ask-ai 0.0.5 → 1.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f43705932c791d2b495230b8ac87567a23dafddd7c76fbbdc9cd0d88d960aac7
-  data.tar.gz: c8376477e6a4261f04ced1c99ccc07d433551e1613c317b51ab1b21b53e50b58
+  metadata.gz: e73628fad8025e8de63e07c4256f42426eb9651fb62a418f8f73c6b7b4db432d
+  data.tar.gz: 4575f5535ddedd005e846cf598c3f464717130b2ce065b7c29e2e174cadebd30
 SHA512:
-  metadata.gz: 26841ece9e0ee1cb05d1df0d6cbf1b20db94d18daf74be4decf82599bf3614f5a6823a2ae3398a26d9a4df480c9eedb51285934bd674d4f3b2eac5355b617da1
-  data.tar.gz: 0ecfbc09c3944f62ef1cfdc3ddbef975d7b2cf3ea69913bff416ba25f6df063436bdbf7359bf40ca9b80fa4ce0d4dfc74d54d79b487f1f8c9b22365a2027c70e
+  metadata.gz: a2fccb29fba0174e10556377b2422fb6d3de99eef58820af2289ef77dd0379029c5d989d745eb388ef0901c153de2ce22163a9fb8b11d1f133d84e0d3bdb6b62
+  data.tar.gz: 45bc0ea0ef5e78a465b196fe262a2b1b88cdeb2bd9b15af86300d72ac2cf220c4b1ebeb860b41782e02bad51dc87d34e32a7522c1ff3129bc1902995c0dfcebf

data/config/config.yml

@@ -1,3 +0,0 @@
----
-TEMPERATURE: 0.7
-CONTEXT_LENGTH: 12

data/lib/config.rb

@@ -1,65 +1,110 @@
 require_relative './files.rb'
 
 module Config
-  include Files
-  def self.load_key()
-    config = YAML.load_file(Files.config_path)
+  extend Files
+  def load_key()
+    config = YAML.load_file(config_path)
     config['OPENAI_API_KEY']
   end
 
-  def self.save_key(api_key)
-    config = YAML.load_file(Files.config_path)
+  def save_key(api_key)
+    config = YAML.load_file(config_path)
+    if (config == false || config.nil?)
+      config = {}
+    end
     config['OPENAI_API_KEY'] = api_key
-    File.open(Files.config_path, 'w') { |f| YAML.dump(config, f) }
+    File.open(config_path, 'w') { |f| YAML.dump(config, f) }
   end
 
-  def self.load_temperature
-    config = YAML.load_file(Files.config_path)
-    config['TEMPERATURE']
+  def load_temperature
+    config = YAML.load_file(config_path)
+    unless (config == false || config.nil?)
+      config['TEMPERATURE']
+    end
   end
 
-  def self.save_temperature(temperature)
-    config = YAML.load_file(Files.config_path)
+  def save_temperature(temperature)
+    config = YAML.load_file(config_path)
+    if config == false
+      config = {}
+    end
     config['TEMPERATURE'] = temperature.to_f
-    File.open(Files.config_path, 'w') { |f| YAML.dump(config, f) }
+    File.open(config_path, 'w') { |f| YAML.dump(config, f) }
   end
 
-  def self.load_context_length
-    config = YAML.load_file(Files.config_path)
-    config['CONTEXT_LENGTH']
+  def load_context_length
+    config = YAML.load_file(config_path)
+    unless (config == false || config.nil?)
+      config['CONTEXT_LENGTH']
+    end
   end
 
-  def self.save_context_length(context_length)
-    config = YAML.load_file(Files.config_path)
+  def save_context_length(context_length)
+    config = YAML.load_file(config_path)
+    if config == false
+      config = {}
+    end
     config['CONTEXT_LENGTH'] = context_length.to_i
-    File.open(Files.config_path, 'w') { |f| YAML.dump(config, f) }
+    File.open(config_path, 'w') { |f| YAML.dump(config, f) }
   end
 
-  def self.set_config(value)
-    puts 'value',value
+  def set_config(value)
     if value.include?('key')
       stript_value = value.sub(/^key/, '').strip
       if stript_value.empty?
-        puts 'No API key given'
-        exit
+        return 'No API key given'
       end
       save_key(stript_value)
+      return 'API key saved'
     elsif value.include?('temp')
-      stript_value = value.sub(/^temp/, '').strip
-      if stript_value.empty?
-        puts 'No temperature given'
-        exit
+      stript_value = value.sub(/^temp/, '').strip.to_f
+      if stript_value.to_f > 1.0 || stript_value.to_f < 0.1
+        return 'Temperature must be between 0.1 and 1.0'
       end
       save_temperature(stript_value)
+      return 'Temperature saved'
     elsif value.include?('context')
-      stript_value = value.sub(/^context/, '').strip
-      if stript_value.empty?
-        puts 'No context length given'
-        exit
+      stript_value = value.sub(/^context/, '').strip.to_i
+      if stript_value.to_i > 100 || stript_value.to_i < 1
+        return 'Context length must be between 1 and 100'
       end
       save_context_length(stript_value)
+      return 'Context length saved'
     else
-      puts 'Invalid config value'
+      return 'Invalid config value'
     end
   end
+
+  def set_key(api_key: nil)
+    if api_key.nil?
+      log("Setting API key...")
+      log("Enter API key: (or press enter to exit)")
+
+      while input = Readline.readline("> ", true) do
+        if input.empty?
+          log("Exiting.")
+          exit
+        else
+          api_key = input.strip
+          break
+        end
+      end
+      log("Saving API key...")
+    end
+
+    FileUtils.mkdir_p(File.dirname(config_path))
+    File.open(config_path, "w") do |f|
+      f.write(YAML.dump({ "OPENAI_API_KEY" => api_key }))
+    end
+    log("API key saved.")
+    log("")
+  end
+
+  def load_env()
+    #Config.load_key()
+    YAML.load(File.read(config_path))
+
+  rescue Errno::ENOENT
+    log("No config.yml found.")
+  end
 end

data/lib/context.rb

@@ -1,10 +1,11 @@
 require_relative './files.rb'
 require_relative './config.rb'
+require_relative './file_format_error.rb'
 class Context
-  include Files, Config
-  def self.load_context()
-    if File.exist?(Files.context_path)
-      conversation = File.readlines(Files.context_path).map { |line| JSON.parse(line) }
+  extend Files, Config, Logging
+  def self.load_context(file_with_context: false)
+    if File.exist?(context_path)
+      conversation = File.readlines(context_path).map { |line| JSON.parse(line) }
     else
       conversation = []
     end
@@ -21,39 +22,50 @@ class Context
       end
     end
 
+    if file_with_context
+      return load_context_file() + context_as_string
+    end
+
     return context_as_string
   end
 
   ## Here we save the context to a file.
   ## Max 10 previous Q / A to save tokens.
   def self.save_context(context)
+    return if context.nil?
     tmp_arr = []
-    unless File.exist?(Files.context_path)
-      File.open(Files.context_path, "w") {}
+    unless File.exist?(context_path)
+      File.open(context_path, "w") {}
     end
-    File.readlines(Files.context_path).map { |line| tmp_arr.push(JSON.parse(line)) }
-    length = Config.load_context_length().nil? ? 10 : Config.load_context_length()
+    File.readlines(context_path).map { |line| tmp_arr.push(JSON.parse(line)) }
+    length = load_context_length().nil? ? 10 : load_context_length()
     if tmp_arr.length > length
       tmp_arr.shift()
     end
-    File.truncate(Files.context_path, 0)
-    tmp_arr.each { |line| File.open(Files.context_path, "a") { |file| file.write("#{line.to_json}\n") } }
-    File.open(Files.context_path, "a") { |file| file.write("#{context.to_json}\n") }
+    File.truncate(context_path, 0)
+    tmp_arr.each { |line| File.open(context_path, "a") { |file| file.write("#{line.to_json}\n") } }
+    File.open(context_path, "a") { |file| file.write("#{context.to_json}\n") }
   end
 
   def self.delete_context()
-    Logging.log("Deleting previous context.")
-    File.truncate(Files.context_path, 0)
+    log("Deleting previous context.")
+    File.truncate(context_path, 0)
   end
 
   def self.save_context_file(file_path)
-    puts "File path: #{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(Files.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
@@ -61,17 +73,19 @@ class Context
       file_out.close
 
       if char_count > 10000
-        Logging.log("Warning: The file you are trying to feed to the API is #{char_count} characters long. This consumes a lot of tokens.")
+        log("Warning: The file you are trying to feed to the API is #{char_count} characters long. This consumes a lot of tokens.")
       end
     else
-      Logging.log("No file path given.")
+      log("No file path given.")
     end
   rescue Errno::ENOENT
-    Logging.log("No file at '#{file_path}' found.")
+    log("No file at '#{file_path}' found.")
+  rescue FileFormatError => e
+    log(e.message)
   end
 
   def self.load_context_file()
-    file = File.open(Files.context_file_path, 'r')
+    file = File.open(context_file_path, 'r')
     file_as_string = ""
     file.each do |line|
       file_as_string += line
@@ -79,9 +93,74 @@ class Context
 
     return file_as_string
   rescue Errno::ENOENT
-    Logging.log("No file at '#{Files.context_file_path}' found.")
-    Logging.log("Load a file with 'aa -lf <file_path>'")
+    log("No file at '#{context_file_path}' found.")
+    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/files.rb

@@ -1,21 +1,21 @@
 module Files
-  def self.root
+  def root
     File.expand_path("./../", __dir__)
   end
 
-  def self.file_path
+  def file_path
     File.expand_path("./../files/", __dir__)
   end
 
-  def self.context_path
+  def context_path
     File.expand_path("./../files/context.jsonl", __dir__)
   end
 
-  def self.config_path
+  def config_path
     File.expand_path("./../config/config.yml", __dir__)
   end
 
-  def self.context_file_path
+  def context_file_path
     File.expand_path("./../files/context_file.txt", __dir__)
   end
 end

data/lib/help.rb

@@ -1,85 +1,50 @@
 require 'rubygems'
 
-class Help
-  def self.display_help()
-    Logging.log("Usage: aa [options] [input]")
-    Logging.log("  -lf, --loadfile <path>: Load file into context")
-    Logging.log("  -f, --file: Read from context file")
-    Logging.log("  -c, --conversation: Append to conversation (max 10 Questions / Answers pairs saved)")
-    Logging.log("  -d, --delete: Delete conversation")
-    Logging.log("  -i, --interactive: Interactive mode, always a conversation. Clear context with 'clear' (exit with 'exit' or 'quit')")
-    Logging.log("  -w, --whisper <path>: Transcribe audio file")
-    Logging.log("  -t, --translate <path>: Translate audio file")
-    Logging.log("\n  Options:")
-    Logging.log("    --config: Edit config file")
-    Logging.log("    -v, --version: Display version")
-    Logging.log("    -h, --help: Display this help message")
-
-  end
- 
-  def self.display_api_key()
-    Logging.log("You need to set your API key in the file: ./config/config.yml")
-    Logging.log("Create the file if it doesn't exist.")
-    Logging.log("Add the following line to the file:")
-    Logging.log("  OPENAI_API_KEY: <your API key>")
-    Logging.log("You can get your API key from: https://openai.com/")
-  end
-
-  ## This don't work yet. Need to rework the way we handle args.
-  def self.display_help_file()
-    ## TODO: How to work with files
-  end
 
-  ## This don't work yet. Need to rework the way we handle args.
-  def self.display_help_conversation()
-    ## TODO: How to work with conversation
-  end
+class Help
+  extend Logging
 
   def self.display_version()
-    spec = Gem::Specification::load("ask-ai.gemspec")
-    Logging.log("Version: #{spec.version}")
-  end
-
-  def self.display_usage()
-    Logging.log("Usage: ./main.rb [options] [input]")
-
-    Logging.log("There are two types of options, flags and arguments.")
+    ## Without using gemspec
+    spec = Gem::Specification::find_by_name("ask-ai")
+    log("Version: ask-ai-#{spec.version}")
   end
 
   def self.interactive_help(command)
     case command
     when '-w'
-      Logging.log("Ex: -w /home/name/sound_file.m4a")
-      Logging.log("Will transcribe the audio file.")
+      log("Ex: -w /home/name/sound_file.m4a")
+      log("Will transcribe the audio file.")
     when '-t'
-      Logging.log("Ex: -t /home/name/sound_file.m4a")
-      Logging.log("Will translate the audio file to English.")
+      log("Ex: -t /home/name/sound_file.m4a")
+      log("Will translate the audio file to English.")
     when '-lf'
-      Logging.log("Ex: -lf /home/name/some_text_file.txt'")
-      Logging.log("Will load the file into context.")
-      Logging.log("The file should a [txt, CSV]. More formats coming soon.")
+      log("Ex: -lf /home/name/some_text_file.txt'")
+      log("Will load the file into context.")
+      log("The file should a [txt, CSV]. More formats coming soon.")
     when '-f'
-      Logging.log("Ex: -f Can you describe the file i provided?")
+      log("Ex: -f Can you describe the file i provided?")
     when 'config'
-      Logging.log("Ex: config key <your API key>")
-      Logging.log("Ex: config temp <0.0 - 1.0>")
-      Logging.log("Ex: config context <0 - 100>")
-      Logging.log("Beaware that the more context you use, the more expensive it will be.")
+      log("Ex: config key <your API key>")
+      log("Ex: config temp <0.0 - 1.0>")
+      log("Ex: config context <0 - 100>")
+      log("Beaware that the more context you use, the more expensive it will be.")
     else
-      Logging.log("No help for: #{command}")
+      log("No help for: #{command}")
     end
 
   end
 
   def self.interactive_desc()
-    Logging.log("Type 'exit' or 'quit' to exit.")
-    Logging.log("Type 'clear' to clear context.")
-    Logging.log("Type 'show' to show context.")
-    Logging.log("Type 'help' to show help.")
-    Logging.log("Type 'config [key, temp, context]' to change config.")
-    Logging.log("Type '-w <filepath>' to whisper transcribe.")
-    Logging.log("Type '-t' <filepath> to whisper translate.")
-    Logging.log("Type '-lf' <filepath> to load file.")
-    Logging.log("Type '-f' to use loaded file as context.")
+    log("Type 'exit' or 'quit' to exit.")
+    log("Type 'clear' to clear context.")
+    log("Type 'show' to show context.")
+    log("Type 'help' to show help.")
+    log("Type 'config [key, temp, context]' to change config.")
+    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/logging.rb

@@ -1,7 +1,7 @@
 require 'logger'
 
 module Logging
-  def self.log(msg)
+  def log(msg)
     logger = Logger.new(STDOUT)
     logger.formatter = proc { |_severity, _datetime, _progname, msg| "#{msg}\n" }
     logger.info(msg)

data/lib/main.rb

@@ -6,192 +6,98 @@ require 'readline'
 require 'io/console'
 require_relative './logging.rb'
 require_relative './prompt.rb'
-require_relative './handle_args.rb'
 require_relative './help.rb'
 require_relative './context.rb'
 require_relative './files.rb'
 require_relative './config.rb'
 
 class Main
-  include Logging, Files, Config
+  extend Logging, Files, Config
+  LIST = [
+    "exit", "quit", "version", "clear", "help", "show",
+    "-w", "-t", "-lf", "-f", "-df", "config", "temp", "context"
+  ].sort
 
   def self.run()
-    config = load_env()
 
-    ## This does not work. Need to fix this.
-    if (config == false || config.nil?)
-      Logging.log("No API key found.")
-      Help.display_api_key()
-      Logging.log('If you want to set your API key now, type (y)es and press enter.')
-      while input = Readline.readline("> ", true) do
-        if input.empty?
-          Logging.log("Exiting.")
-          exit
-        elsif input == "y" || input == "yes" || input == "Y" || input == "Yes"
-          set_key()
-          break
-        else
-          Logging.log('Invalid input (y)es or press enter to exit.')
-        end
-      end
-    end
+    ## When using Readline, TAB will auto-complete
+    ## But it will only auto-complete from the LIST
+    ## Need to handle directories and files when -lf, -w, -t are used
+    #comp = proc { |s| LIST.grep(/^#{Regexp.escape(s)}/) }
+    #Readline.completion_append_character = ""
+    #Readline.completion_proc = comp
 
-    context = Context.load_context()
-    options_and_input = HandleArgs.handle_args()
-    options = options_and_input.select { |k, v| k.start_with?("option_") }
-    input = options_and_input["input"]
+    ## 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
 
-    halt_options = ["-h", "--help", "-v", "--version", "--key"]
-
-    ## Hack... Need to fix this.
-    if options.empty?
-      options = { "option_0" => "simple" }
-    end
+    # list_auto_complete()
 
-    options.each do |k, v|
-      if halt_options.include?(v)
-        ## Options that halt the program.
-        case v
-        when "-h", "--help"
-          Help.display_help()
-          exit
-        when "-v", "--version"
-          Help.display_version()
-          exit
-        
-        when "--key"
-          set_key(api_key: nil)
-        else
-          Help.display_help()
-          exit
+    Help.interactive_desc()
+    while input = Readline.readline("\n> ", true) do
+      case input
+      when "exit", "quit"
+        break
+      when "version"
+        Help.display_version()
+      when "clear"
+        log("Clearing context...")
+        Context.delete_context()
+      when 'help'
+        Help.interactive_desc()
+      when /^help/
+        strip_input = input.sub(/^help/, "").strip
+        Help.interactive_help(strip_input)
+      when "show"
+        log("\n")
+        log(Context.load_context())
+      when /^-w/ 
+        stript_input = input.sub(/^-w/, "").strip
+        log(Prompt.whisper_transcribe(stript_input, interactive: true))
+      when /^-t/
+        stript_input = input.sub(/^-t/, "").strip
+        log(Prompt.whisper_translate(stript_input, interactive: true))
+      when /^-lf/
+        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()
+        if file_as_string.empty?
+          log("No file loaded.")
+          next
         end
-      else
-        ## Options that don't halt the program.
-        case v
-        when "-f", "--file"
-          file_as_string = Context.load_context_file()
-          if file_as_string.empty?
-            exit
-          end
-          Prompt.stream_prompt(input, file_as_string)
-        when "-lf", "--loadfile"
-          Logging.log("Loading File #{input}")
-          Context.save_context_file(input)
-        when "-d", "--delete"
-          if input.nil?
-            Context.delete_context()
-          else
-            Context.delete_context()
-            Context.save_context(Prompt.stream_prompt(input, context))
-          end
-        when "-c", "--conversation"
-          Logging.log(input)
-          Context.save_context(Prompt.stream_prompt(input, context))
-        when "-w", "--whisper"
-          Logging.log(Prompt.whisper_transcribe(input))
-        when "-t", "--translate"
-          Logging.log(Prompt.whisper_translate(input))
-        when "-i", "--interactive"
-          Logging.log("Interactive mode...")
-          Help.interactive_desc()
-          while input = Readline.readline("\n> ", true) do
-            case input
-              when "exit", "quit"
-                break
-              when "clear"
-                Logging.log("Clearing context...")
-                Context.delete_context()
-              when 'help'
-                Help.interactive_desc()
-              when /^help/
-                strip_input = input.sub(/^help/, "").strip
-                #TODO: This should be a specific help for interactive. 
-                Help.interactive_help(strip_input)
-              when "show"
-                Logging.log("\n")
-                Logging.log(Context.load_context())
-              when /^-w/ 
-                stript_input = input.sub(/^-w/, "").strip
-                Logging.log(Prompt.whisper_transcribe(stript_input, interactive: true))
-              when /^-t/
-                stript_input = input.sub(/^-t/, "").strip
-                Logging.log(Prompt.whisper_translate(stript_input, interactive: true))
-              when /^-lf/
-                stript_input = input.sub(/^-lf/, "").strip
-                Logging.log("Loading File #{stript_input}")
-                Context.save_context_file(stript_input)
-              when /^-f/
-                stript_input = input.sub(/^-f/, "").strip
-                file_as_string = Context.load_context_file()
-                if file_as_string.empty?
-                  Logging.log("No file loaded.")
-                  next
-                end
-                Logging.log("")
-                Prompt.stream_prompt(stript_input, file_as_string)
-                Logging.log("")
-              when ""
-                Logging.log("No input given.")
-              when /^config/
-                strip_input = input.sub(/^config/, "").strip
-                Config.set_config(strip_input)
-                #set_key(api_key: nil)
-              else
-                #options_and_input = HandleArgs.handle_args()
-                context = Context.load_context()
-                Logging.log("")
-                Context.save_context(Prompt.stream_prompt(input, context))
-                Logging.log("")
-            end
-          end
-          Logging.log("Exiting...")
-        when "simple"
-          if !input.nil?
-            Prompt.stream_prompt(input)
-          else
-            Logging.log("No input given.")
-            Help.display_help()
-          end
-        else
-          Help.display_help()
-        end
-      end
-    end
-  end
-
-  private
-
-  def self.set_key(api_key: nil)
-    if api_key.nil?
-      Logging.log("Setting API key...")
-      Logging.log("Enter API key: (or press enter to exit)")
-
-      while input = Readline.readline("> ", true) do
-        if input.empty?
-          Logging.log("Exiting.")
-          exit
-        else
-          api_key = input.strip
-          break
+        context = Context.load_context(file_with_context: true)
+        log("")
+        Context.save_context(Prompt.stream_prompt(stript_input, context))
+        log("")
+      when ""
+        log("No input given.")
+      when /^config/
+        strip_input = input.sub(/^config/, "").strip
+        res = set_config(strip_input)
+        if res.is_a?(String)
+          log(res)
         end
+      else
+        context = Context.load_context()
+        log("")
+        Context.save_context(Prompt.stream_prompt(input, context))
+        log("")
       end
-      Logging.log("Saving API key...")
     end
-
-    FileUtils.mkdir_p(File.dirname(Files.config_path))
-    File.open(Files.config_path, "w") do |f|
-      f.write(YAML.dump({ "OPENAI_API_KEY" => api_key }))
-    end
-    Logging.log("API key saved.")
-    Logging.log("")
-  end
-
-  def self.load_env()
-    #Config.load_key()
-    YAML.load(File.read(Files.config_path))
-
-  rescue Errno::ENOENT
-    Logging.log("No config.yml found.")
   end
 end
 

data/lib/prompt.rb

@@ -3,10 +3,9 @@ require_relative './files.rb'
 require_relative './config.rb'
 
 class Prompt
-  include Files
-  include Config
+  extend Files, Config, Logging
   ## Streams the response, VERY NICE
-  def self.stream_prompt(input, conversation = '', temp = Config.load_temperature())
+  def self.stream_prompt(input, conversation = '', temp = load_temperature())
     if temp.nil?
       temp = 0.7
     end
@@ -16,23 +15,25 @@ class Prompt
       conversation += "\n My question: #{input}"
     end
     response = ''
-    client.chat(
-      parameters: {
-        model: "gpt-3.5-turbo",
-        messages: [{ role: "user", content: conversation}],
-        temperature: temp, ## Should be a parameter
-        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")
-        end
+    unless client.nil?
+      client.chat(
+        parameters: {
+          model: "gpt-3.5-turbo",
+          messages: [{ role: "user", content: conversation}],
+          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")
+          end
+        }
+      )
+      context = {
+        "input" => input,
+        "response" => response,
       }
-    )
-    context = {
-      "input" => input,
-      "response" => response,
-    }
 
-    return context
+      return context
+    end
   end
 
   ## Not implemented only scaffolding
@@ -47,7 +48,7 @@ class Prompt
 
   def self.whisper_translate(file_path, interactive = false)
     if (file_path.nil? || !file_path.end_with?(*['.mp3', '.wav', '.m4a', '.webm', '.mpeg', '.mpga']))
-      Logging.log("No file given or wrong file type")
+      log("No file given or wrong file type")
       unless interactive
         exit
       end
@@ -59,27 +60,29 @@ class Prompt
           exit
         end
       else
-        response = client.audio.translate(
-        parameters: {
-            model: "whisper-1",
-            file: File.open(file_path, "rb"),
-        })
-        if (response["text"].nil? || response["text"].empty?)
-          Logging.log("No text found")
-          unless interactive
-            exit
+        unless client.nil?
+          response = client.audio.translate(
+          parameters: {
+              model: "whisper-1",
+              file: File.open(file_path, "rb"),
+          })
+          if (response["text"].nil? || response["text"].empty?)
+            log("No text found")
+            unless interactive
+              exit
+            end
           end
+          return response["text"]
         end
-        return response["text"]
       end
     end
   rescue Errno::ENOENT => e
-    Logging.log(e)
+    log(e)
   end
 
   def self.whisper_transcribe(file_path, interactive = false)
     if (file_path.nil? || !file_path.end_with?(*['.mp3', '.wav', '.m4a', '.webm', '.mpeg', '.mpga']))
-      Logging.log("No file given")
+      log("No file given")
       unless interactive
         exit
       end
@@ -91,31 +94,40 @@ class Prompt
           exit
         end
       else
-        response = client.audio.transcribe(
-        parameters: {
-            model: "whisper-1",
-            file: File.open(file_path, "rb"),
-        })
-        if (response["text"].nil? || response["text"].empty?)
-          Logging.log("No text found")
-          unless interactive
-            exit
+        unless client.nil?
+          response = client.audio.transcribe(
+          parameters: {
+              model: "whisper-1",
+              file: File.open(file_path, "rb"),
+          })
+          if (response["text"].nil? || response["text"].empty?)
+            log("No text found")
+            unless interactive
+              exit
+            end
           end
+          return response["text"]
         end
-        return response["text"]
       end
     end
   rescue Errno::ENOENT => e
     #Logging.log("File not found")
-    Logging.log(e)
+    log(e)
   end
 
   private
 
   def self.client()
-    conf = YAML.load(File.read(Files.config_path))
-    key = conf["OPENAI_API_KEY"]
+    conf = YAML.load(File.read(config_path))
+    unless (conf == false || conf.nil?)
+      key = conf["OPENAI_API_KEY"]
+    end
 
-    OpenAI::Client.new(access_token: key)
+    begin
+      OpenAI::Client.new(access_token: key)
+    rescue OpenAI::ConfigurationError => e
+      log("OpenAI API key not found, run 'config key' to set it")
+      return nil
+    end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ask-ai
 version: !ruby/object:Gem::Version
-  version: 0.0.5
+  version: 1.0.1
 platform: ruby
 authors:
 - Oskar Franck
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-10-04 00:00:00.000000000 Z
+date: 2023-10-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-openai
@@ -38,8 +38,8 @@ files:
 - files/context_file.txt
 - lib/config.rb
 - lib/context.rb
+- lib/file_format_error.rb
 - lib/files.rb
-- lib/handle_args.rb
 - lib/help.rb
 - lib/logging.rb
 - lib/main.rb

data/lib/handle_args.rb

@@ -1,61 +0,0 @@
-class HandleArgs
-  def self.permitted_options()
-    ## TODO: Add more options.
-    ## -t --translate
-    ## -temp --temperature (need to handle value after input)
-
-    [
-      '-h',
-      '--help',
-      '-v',
-      '--version',
-      '-f',
-      '--file',
-      '-lf',
-      '--loadfile',
-      '-d',
-      '--delete',
-      '-c',
-      '--conversation',
-      '--finetune',
-      '-w', '--whisper',
-      '-t', '--translate',
-      '-i', '--interactive',
-      '--key',
-  ]
-end
-
-  def self.handle_args()
-    p_args = HandleArgs.permitted_options()
-    ## This is the hash that will be returned.
-    args_hash = {}
-
-    ## This handles args then called as ruby script.
-    ARGV.each_with_index do |arg, i|
-      #Logging.log("Arg: #{arg}")
-
-      if arg.start_with?('-')
-        ## This is an option.
-        if p_args.include?(arg)
-          ## This is a permitted / available option.
-          #Logging.log("Option: #{arg}")
-          args_hash["option_#{i}"] = arg
-        else
-          ## This is an unknown option.
-          ## TODO: Handle unknown option. display help? discard?
-          Logging.log("Unknown option: #{arg}")
-          args_hash["option_#{i}"] = arg
-        end
-      else
-        ## This is input.
-        ## This if statement append all 'input' args to one string.
-        if args_hash["input"].nil?
-          args_hash["input"] = arg
-        else
-          args_hash["input"] = "#{args_hash['input']} #{arg}"
-        end
-      end
-    end
-    return args_hash
-  end
-end