ask-ai 1.0.0 → 1.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3d0033fda00a5a8399a6e9b71c75338b02269df54edb791bd82215e939492d15
-  data.tar.gz: 391d4cc9fe240289358f2489b61224b8d0706254174f9ad4a5ec7a109b7b3a89
+  metadata.gz: e73628fad8025e8de63e07c4256f42426eb9651fb62a418f8f73c6b7b4db432d
+  data.tar.gz: 4575f5535ddedd005e846cf598c3f464717130b2ce065b7c29e2e174cadebd30
 SHA512:
-  metadata.gz: 6f749624e465e35a8ebcae8b7b9b312d1f4ffdbd8e390043a0fc87dae5eaf680931975eb87369eba855043eff52206c2daf47af95fc5136e7fb0134520275a28
-  data.tar.gz: 81d6f56f465808352087d5f53bab44b7a1af2d63292dbba4b5fab7f91b44fd3d78c6a591718d142f6348c6ab78bae8a54d84b2cba73e2164efd32dd3b27d4e1d
+  metadata.gz: a2fccb29fba0174e10556377b2422fb6d3de99eef58820af2289ef77dd0379029c5d989d745eb388ef0901c153de2ce22163a9fb8b11d1f133d84e0d3bdb6b62
+  data.tar.gz: 45bc0ea0ef5e78a465b196fe262a2b1b88cdeb2bd9b15af86300d72ac2cf220c4b1ebeb860b41782e02bad51dc87d34e32a7522c1ff3129bc1902995c0dfcebf

data/lib/config.rb

@@ -9,7 +9,7 @@ module Config
 
   def save_key(api_key)
     config = YAML.load_file(config_path)
-    if config == false
+    if (config == false || config.nil?)
       config = {}
     end
     config['OPENAI_API_KEY'] = api_key
@@ -18,7 +18,7 @@ module Config
 
   def load_temperature
     config = YAML.load_file(config_path)
-    unless config == false
+    unless (config == false || config.nil?)
       config['TEMPERATURE']
     end
   end
@@ -34,7 +34,7 @@ module Config
 
   def load_context_length
     config = YAML.load_file(config_path)
-    unless config == false
+    unless (config == false || config.nil?)
       config['CONTEXT_LENGTH']
     end
   end

data/lib/context.rb

@@ -1,8 +1,9 @@
 require_relative './files.rb'
 require_relative './config.rb'
+require_relative './file_format_error.rb'
 class Context
   extend Files, Config, Logging
-  def self.load_context()
+  def self.load_context(file_with_context: false)
     if File.exist?(context_path)
       conversation = File.readlines(context_path).map { |line| JSON.parse(line) }
     else
@@ -21,6 +22,10 @@ class Context
       end
     end
 
+    if file_with_context
+      return load_context_file() + context_as_string
+    end
+
     return context_as_string
   end
 
@@ -48,12 +53,19 @@ class Context
   end
 
   def self.save_context_file(file_path)
+    ## If the file extenstion is pdf or docx raise an error.
+    ## This is not a complete list of file extensions.
+    if file_path.include?(".pdf") || file_path.include?(".docx")
+      raise FileFormatError, "File type not supported."
+    end
     unless file_path.nil?
+      conter = 0
       file_in = File.open(file_path, 'r')
-      file_out = File.open(context_file_path, 'w')
+      file_out = File.open(context_file_path, 'a')
+      file_out.write("loaded_context_file_path=#{file_path}\n")
       char_count = 0
       file_in.each do |line|
-        puts "Line: #{line}"
+        #puts "Line: #{line}"
         char_count += line.length
         file_out.write(line)
       end
@@ -68,6 +80,8 @@ class Context
     end
   rescue Errno::ENOENT
     log("No file at '#{file_path}' found.")
+  rescue FileFormatError => e
+    log(e.message)
   end
 
   def self.load_context_file()
@@ -83,5 +97,70 @@ class Context
     log("Load a file with 'aa -lf <file_path>'")
     return ""
   end
+  ## This first class pasta method need to be refactored.
+  ## It's a mess.
+  def self.delete_file_context()
+    counter = 1
+    delete_lines = []
+    last_line = 0
+    File.open(context_file_path, "r") { |file|
+      if file.size == 0
+        log("No files loaded.")
+        return
+      end
+      file.readlines.each_with_index { |line, index|
+        if line.include?("loaded_context_file_path=")
+          log("#{counter}: #{line.gsub("loaded_context_file_path=", "")}")
+          delete_lines.push(index)
+          counter += 1
+        end
+        last_line = index
+      }
+    }
+    if counter == 1 + 1
+      log "One file loaded. Enter '1' or 'all' to delete it."
+      log "Enter 'a' to abort."
+    else
+      log("Which file do you want to delete? (1-#{counter - 1}) or 'all'")
+      log("Enter 'a' to abort.")
+    end
+  
+    delete_counter = 0
+    while input = Readline.readline("\nDelete --> ", true) do
+      if input == "a"
+        log("Aborting.")
+        return
+      elsif input == "all"
+        File.truncate(context_file_path, 0)
+        log("Deleted all files.")
+        return
+      elsif input.to_i >= 1 && input.to_i < counter
+        lines_to_save = []
+        File.open(context_file_path, "r") { |file|
+          file.readlines.each_with_index { |line, index|
+            start_line = delete_lines[input.to_i - 1]
+            end_line = delete_lines[input.to_i]
+            if end_line.nil?
+              end_line = last_line + 1
+            end
+            if index < start_line || index > end_line - 1
+              lines_to_save.push(line)
+            end
+          }
+        }
+        File.truncate(context_file_path, 0)
+        File.open(context_file_path, "a") { |file|
+          lines_to_save.each { |line|
+            file.write(line)
+          }
+        }
+        log("Deleting file")
+        return
+      elsif input.to_i <= 0 || input.to_i > counter
+        log("Please enter a number between 1 and #{counter - 1}")
+        log("Enter 'a' to abort.")
+      end
+    end
+  end
 
 end

data/lib/file_format_error.rb

@@ -0,0 +1,7 @@
+class FileFormatError < StandardError
+
+  def initialize(msg="File format not supported")
+    super
+  end
+
+end

data/lib/help.rb

@@ -44,6 +44,7 @@ class Help
     log("Type '-w <filepath>' to whisper transcribe.")
     log("Type '-t' <filepath> to whisper translate.")
     log("Type '-lf' <filepath> to load file.")
+    log "Type '-df' to delete file context."
     log("Type '-f' to use loaded file as context.")
   end
 end

data/lib/main.rb

@@ -15,7 +15,7 @@ class Main
   extend Logging, Files, Config
   LIST = [
     "exit", "quit", "version", "clear", "help", "show",
-    "-w", "-t", "-lf", "-f", "config", "temp", "context"
+    "-w", "-t", "-lf", "-f", "-df", "config", "temp", "context"
   ].sort
 
   def self.run()
@@ -27,6 +27,21 @@ class Main
     #Readline.completion_append_character = ""
     #Readline.completion_proc = comp
 
+    ## Function that when called activates completion from LIST
+    # def self.list_auto_complete()
+    #   Readline.completion_append_character = ""
+    #   Readline.completion_proc = proc do |s|
+    #     if s.start_with?("-w", "-t", "-lf", "-f")
+    #       pattern = s.sub(/^-w/, "") + "*"
+    #       Dir.glob(pattern).grep(/^#{Regexp.escape('')}/)
+    #     else
+    #       LIST.grep(/^#{Regexp.escape(s)}/)
+    #     end
+    #   end
+    # end
+
+    # list_auto_complete()
+
     Help.interactive_desc()
     while input = Readline.readline("\n> ", true) do
       case input
@@ -55,6 +70,8 @@ class Main
         stript_input = input.sub(/^-lf/, "").strip
         log("Loading File #{stript_input}")
         Context.save_context_file(stript_input)
+      when /^-df/
+        Context.delete_file_context()
       when /^-f/
         stript_input = input.sub(/^-f/, "").strip
         file_as_string = Context.load_context_file()
@@ -62,8 +79,9 @@ class Main
           log("No file loaded.")
           next
         end
+        context = Context.load_context(file_with_context: true)
         log("")
-        Prompt.stream_prompt(stript_input, file_as_string)
+        Context.save_context(Prompt.stream_prompt(stript_input, context))
         log("")
       when ""
         log("No input given.")

data/lib/prompt.rb

@@ -20,7 +20,7 @@ class Prompt
         parameters: {
           model: "gpt-3.5-turbo",
           messages: [{ role: "user", content: conversation}],
-          temperature: temp, ## Should be a parameter
+          temperature: temp,
           stream: proc do |chunk, _bytesize|
             response += chunk.dig("choices", 0, "delta", "content") unless chunk.dig("choices", 0, "delta", "content").nil?
             print chunk.dig("choices", 0, "delta", "content")
@@ -119,7 +119,7 @@ class Prompt
 
   def self.client()
     conf = YAML.load(File.read(config_path))
-    unless conf == false
+    unless (conf == false || conf.nil?)
       key = conf["OPENAI_API_KEY"]
     end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ask-ai
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.0.1
 platform: ruby
 authors:
 - Oskar Franck
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-10-08 00:00:00.000000000 Z
+date: 2023-10-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-openai
@@ -38,6 +38,7 @@ files:
 - files/context_file.txt
 - lib/config.rb
 - lib/context.rb
+- lib/file_format_error.rb
 - lib/files.rb
 - lib/help.rb
 - lib/logging.rb
@@ -62,7 +63,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.2.33
+rubygems_version: 3.3.7
 signing_key: 
 specification_version: 4
 summary: A simple CLI for OpenAI's GPT-3 API.