func_bot 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 68e7cc778676e7b037cff5572cd69c2f61559763fd57b8e31328f487fa839148
-  data.tar.gz: cb9ca4334b4c08384216a03988e7b3343809cf7d0ee9e0ba69706b17a3c91e48
+  metadata.gz: e2642f560765801d53e14efe88c613738759eb83c5ea65bac70696987eb38783
+  data.tar.gz: fc5683415acbaed9f68013aad668dbb180d1059b1496c514bcbcd05d5be6044c
 SHA512:
-  metadata.gz: b22ff1a5ebedc9261dc022bed342d8272b1bd15ed73c749038c4c2711f7785fa9dae7452fe0e8a40337dc46ae43daa2df7bc7ba7b4af28a8fa3f0844aff9efe3
-  data.tar.gz: 45ac6d18634f9615dcbd3807726543bab58409a7c6e20d80c8a0ad039a5f9bbcccf1c2e801c6b1070bce3aaf4632d435538aa6313f468e22dfe889efca24c156
+  metadata.gz: a17a9594c4a6ec3cdfbf31db3531afb2074e314fd092851c76d73fa41f221a3e03c35a183ef2a7a86618e15bbd20cd9ff7d642e5efb5dbb9c55e48495eef1165
+  data.tar.gz: 615b2b10f8c235b67dd7489ea3624a83c3b53a9b0eb57843d816d66dc1672a4c6692cfd615dd7c2238e3c62c5521093dfd32b8186c358fb595b7eedaca201499

data/lib/func_bot/bots/client.rb

@@ -15,7 +15,7 @@ module FuncBot
             model: "gpt-3.5-turbo-0613",
             messages: bot.history.payload,
             temperature: 0.7,
-            functions: FuncBot::Functions::List.call
+            functions: FuncBot::Functions::List.call(bot)
           }
         )
       end

data/lib/func_bot/functions/base_function.rb

@@ -3,15 +3,15 @@
 module FuncBot
   module Functions
     class BaseFunction
-      attr_reader :response
+      attr_reader :bot
 
-      def initialize(response)
-        @response = response
+      def initialize(bot)
+        @bot = bot
       end
 
       def parsed_response
         JSON.parse(
-          response.dig("choices", 0, "message", "function_call", "arguments")
+          bot.response.dig("choices", 0, "message", "function_call", "arguments")
         )
       end
     end

data/lib/func_bot/functions/handler.rb

@@ -17,7 +17,7 @@ module FuncBot
       private
 
       def function_data
-        constantize_function.new(bot.response).execute
+        constantize_function.new(bot).execute
       end
 
       def constantize_function

data/lib/func_bot/functions/list.rb

@@ -3,11 +3,15 @@
 module FuncBot
   module Functions
     class List
-      def self.call
-        if File.exist?(Rails.root.join("app", "lib", "func_bot", "functions", "list.yml"))
-          YAML.load_file(Rails.root.join("app", "lib", "func_bot", "functions", "list.yml"))["functions"]
+      def self.call(bot)
+        if bot.include_functions
+          if File.exist?(Rails.root.join("app", "lib", "func_bot", "functions", "list.yml"))
+            YAML.load_file(Rails.root.join("app", "lib", "func_bot", "functions", "list.yml"))["functions"]
+          else
+            raise "app/lib/func_bot/functions/list.yml file not found. Please create it by running rails func_bot:install."
+          end
         else
-          raise "app/lib/func_bot/functions/list.yml file not found. Please create it by running rails func_bot:install."
+          []
         end
       end
     end

data/lib/func_bot/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module FuncBot
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

data/lib/func_bot.rb

@@ -12,12 +12,13 @@ require_relative "func_bot/functions/list"
 
 module FuncBot
   class Bot
-    attr_accessor :response
+    attr_accessor :response, :include_functions
     attr_reader :client, :history
 
     def initialize
       @history = Bots::History.new
       @client = Bots::Client.new(self)
+      @include_functions = true
     end
 
     def ask(prompt)

data/lib/generators/func_bot/templates/weather_function.rb

@@ -7,6 +7,7 @@ module FuncBot
   module Functions
     class WeatherFunction < BaseFunction
       def execute
+        bot.include_functions = false
         weather_info = {
           location: parsed_response["location"],
           temperature: 98,

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: func_bot
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - lbp