ask-ai 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b9aac9c44ecfb8fd4a68b1329f7dc48e37b09910011b7a1d1ad085f2afd48407
-  data.tar.gz: c3b58592fce1e23ae7184a3b6d37d58451b8da82992c90dd503c4705a1fb94ff
+  metadata.gz: f43705932c791d2b495230b8ac87567a23dafddd7c76fbbdc9cd0d88d960aac7
+  data.tar.gz: c8376477e6a4261f04ced1c99ccc07d433551e1613c317b51ab1b21b53e50b58
 SHA512:
-  metadata.gz: f075fc93f55659c6f8d9331f5e508dcc6ac9f77e52cf4a8e2df2d25e3e727a8bef5f8f253588a4de06b8c67f7f2704792ed62f7c7e98d4337b5df0b843f34652
-  data.tar.gz: 4d74731c56a046231dc11bd889c37c42e591dfcaebef7e1aaba11d6a0dd64c4942e01474e67f526b2a061a1681d806befa9934f60c070d2b0ead4fa45bc75d8e
+  metadata.gz: 26841ece9e0ee1cb05d1df0d6cbf1b20db94d18daf74be4decf82599bf3614f5a6823a2ae3398a26d9a4df480c9eedb51285934bd674d4f3b2eac5355b617da1
+  data.tar.gz: 0ecfbc09c3944f62ef1cfdc3ddbef975d7b2cf3ea69913bff416ba25f6df063436bdbf7359bf40ca9b80fa4ce0d4dfc74d54d79b487f1f8c9b22365a2027c70e

data/config/config.yml

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

data/lib/config.rb

@@ -0,0 +1,65 @@
+require_relative './files.rb'
+
+module Config
+  include Files
+  def self.load_key()
+    config = YAML.load_file(Files.config_path)
+    config['OPENAI_API_KEY']
+  end
+
+  def self.save_key(api_key)
+    config = YAML.load_file(Files.config_path)
+    config['OPENAI_API_KEY'] = api_key
+    File.open(Files.config_path, 'w') { |f| YAML.dump(config, f) }
+  end
+
+  def self.load_temperature
+    config = YAML.load_file(Files.config_path)
+    config['TEMPERATURE']
+  end
+
+  def self.save_temperature(temperature)
+    config = YAML.load_file(Files.config_path)
+    config['TEMPERATURE'] = temperature.to_f
+    File.open(Files.config_path, 'w') { |f| YAML.dump(config, f) }
+  end
+
+  def self.load_context_length
+    config = YAML.load_file(Files.config_path)
+    config['CONTEXT_LENGTH']
+  end
+
+  def self.save_context_length(context_length)
+    config = YAML.load_file(Files.config_path)
+    config['CONTEXT_LENGTH'] = context_length.to_i
+    File.open(Files.config_path, 'w') { |f| YAML.dump(config, f) }
+  end
+
+  def self.set_config(value)
+    puts 'value',value
+    if value.include?('key')
+      stript_value = value.sub(/^key/, '').strip
+      if stript_value.empty?
+        puts 'No API key given'
+        exit
+      end
+      save_key(stript_value)
+    elsif value.include?('temp')
+      stript_value = value.sub(/^temp/, '').strip
+      if stript_value.empty?
+        puts 'No temperature given'
+        exit
+      end
+      save_temperature(stript_value)
+    elsif value.include?('context')
+      stript_value = value.sub(/^context/, '').strip
+      if stript_value.empty?
+        puts 'No context length given'
+        exit
+      end
+      save_context_length(stript_value)
+    else
+      puts 'Invalid config value'
+    end
+  end
+end

data/lib/context.rb

@@ -1,7 +1,7 @@
 require_relative './files.rb'
-
+require_relative './config.rb'
 class Context
-  include Files
+  include Files, Config
   def self.load_context()
     if File.exist?(Files.context_path)
       conversation = File.readlines(Files.context_path).map { |line| JSON.parse(line) }
@@ -32,7 +32,8 @@ class Context
       File.open(Files.context_path, "w") {}
     end
     File.readlines(Files.context_path).map { |line| tmp_arr.push(JSON.parse(line)) }
-    if tmp_arr.length > 9
+    length = Config.load_context_length().nil? ? 10 : Config.load_context_length()
+    if tmp_arr.length > length
       tmp_arr.shift()
     end
     File.truncate(Files.context_path, 0)
@@ -46,14 +47,18 @@ class Context
   end
 
   def self.save_context_file(file_path)
+    puts "File path: #{file_path}"
     unless file_path.nil?
       file_in = File.open(file_path, 'r')
       file_out = File.open(Files.context_file_path, 'w')
       char_count = 0
       file_in.each do |line|
+        puts "Line: #{line}"
         char_count += line.length
         file_out.write(line)
       end
+      file_in.close
+      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.")

data/lib/help.rb

@@ -45,4 +45,41 @@ class Help
 
     Logging.log("There are two types of options, flags and arguments.")
   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.")
+    when '-t'
+      Logging.log("Ex: -t /home/name/sound_file.m4a")
+      Logging.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.")
+    when '-f'
+      Logging.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.")
+    else
+      Logging.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.")
+  end
 end

data/lib/main.rb

@@ -10,15 +10,16 @@ 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
-  include Files
+  include Logging, Files, Config
+
   def self.run()
     config = load_env()
 
     ## This does not work. Need to fix this.
-    if config == false
+    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.')
@@ -28,7 +29,7 @@ class Main
           exit
         elsif input == "y" || input == "yes" || input == "Y" || input == "Yes"
           set_key()
-          exit
+          break
         else
           Logging.log('Invalid input (y)es or press enter to exit.')
         end
@@ -92,14 +93,7 @@ class Main
           Logging.log(Prompt.whisper_translate(input))
         when "-i", "--interactive"
           Logging.log("Interactive mode...")
-          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' to change config.")
-          Logging.log("Type '-w' to whisper transcribe.")
-          Logging.log("Type '-t' to whisper translate.")
-
+          Help.interactive_desc()
           while input = Readline.readline("\n> ", true) do
             case input
               when "exit", "quit"
@@ -107,9 +101,12 @@ class Main
               when "clear"
                 Logging.log("Clearing context...")
                 Context.delete_context()
-              when "help"
+              when 'help'
+                Help.interactive_desc()
+              when /^help/
+                strip_input = input.sub(/^help/, "").strip
                 #TODO: This should be a specific help for interactive. 
-                Help.display_help()
+                Help.interactive_help(strip_input)
               when "show"
                 Logging.log("\n")
                 Logging.log(Context.load_context())
@@ -119,8 +116,26 @@ class Main
               when /^-t/
                 stript_input = input.sub(/^-t/, "").strip
                 Logging.log(Prompt.whisper_translate(stript_input, interactive: true))
-              when "config"
-                set_key(api_key: nil)
+              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()
@@ -168,9 +183,11 @@ class Main
       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

data/lib/prompt.rb

@@ -1,10 +1,15 @@
 require "openai"
 require_relative './files.rb'
+require_relative './config.rb'
 
 class Prompt
   include Files
+  include Config
   ## Streams the response, VERY NICE
-  def self.stream_prompt(input, conversation = '', temp = 0.7)
+  def self.stream_prompt(input, conversation = '', temp = Config.load_temperature())
+    if temp.nil?
+      temp = 0.7
+    end
     if conversation.length == 0
       conversation += input
     else

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ask-ai
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors:
 - Oskar Franck
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-10-03 00:00:00.000000000 Z
+date: 2023-10-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-openai
@@ -36,13 +36,13 @@ files:
 - config/config.yml
 - files/context.jsonl
 - files/context_file.txt
+- lib/config.rb
 - lib/context.rb
 - lib/files.rb
 - lib/handle_args.rb
 - lib/help.rb
 - lib/logging.rb
 - lib/main.rb
-- lib/paths.rb
 - lib/prompt.rb
 homepage: https://github.com/OskarFranck/terminal_chat
 licenses:
@@ -63,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.

data/lib/paths.rb

@@ -1,26 +0,0 @@
-
-module Files
-  def self.root
-    File.expand_path("./../", __dir__)
-  end
-
-  def self.context
-    File.expand_path("./../files/context.jsonl", __dir__)
-  end
-
-  def self.files
-    FILE_PATH
-  end
-
-  def self.config
-    CONFIG_PATH
-  end
-
-  def self.context_file
-    CONTEXT_FILE_PATH
-  end
-end
-CONTEXT_PATH = File.expand_path("./../files/context.jsonl", __dir__)
-FILE_PATH = File.expand_path("./../files/", __dir__)
-CONFIG_PATH = File.expand_path("./../config/config.yml", __dir__)
-CONTEXT_FILE_PATH = File.expand_path("./../files/context_file.txt", __dir__)