Merge pull request #5 from icebaker/ib-tools

Adding support for Spec 1.0.0: Tools (Functions)

2 files changed, 115 insertions, 0 deletions

diff --git a/logic/providers/openai.rb b/logic/providers/openai.rb
new file mode 100644
index 0000000..00e7a43
--- /dev/null
+++ b/logic/providers/openai.rb
@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+require 'json'
+
+module NanoBot
+  module Logic
+    module OpenAI
+      def self.prepare_tools(cartridge, tools)
+        applies = []
+        tools.each do |tool|
+          cartridge.each do |candidate|
+            next unless candidate[:type] == 'function' &&
+                        tool[:type] == candidate[:type] &&
+                        tool[:function][:name] == candidate[:name]
+
+            source = {}
+
+            source[:fennel] = candidate[:fennel] if candidate[:fennel]
+            source[:lua] = candidate[:lua] if candidate[:lua]
+
+            applies << {
+              name: tool[:function][:name],
+              type: candidate[:type],
+              parameters: JSON.parse(tool[:function][:arguments]),
+              source:
+            }
+          end
+        end
+
+        applies
+      end
+
+      def self.adapt_tool(cartridge)
+        raise 'unsupported tool' if cartridge[:type] != 'function'
+
+        adapted = {
+          type: 'function',
+          function: {
+            name: cartridge[:name], description: cartridge[:description],
+            parameters: { type: 'object', properties: {} }
+          }
+        }
+
+        properties = adapted[:function][:parameters][:properties]
+
+        cartridge[:parameters].each do |parameter|
+          key = parameter[:name].to_sym
+          properties[key] = {}
+          properties[key][:type] = parameter[:type] || 'string'
+          properties[key][:description] = parameter[:description] if parameter[:description]
+        end
+
+        adapted
+      end
+    end
+  end
+end
diff --git a/logic/providers/openai/tools.rb b/logic/providers/openai/tools.rb
new file mode 100644
index 0000000..1b2882a
--- /dev/null
+++ b/logic/providers/openai/tools.rb
@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+
+require 'json'
+
+require_relative '../../helpers/hash'
+
+module NanoBot
+  module Logic
+    module OpenAI
+      module Tools
+        def self.prepare(cartridge, tools)
+          applies = []
+
+          tools = Marshal.load(Marshal.dump(tools))
+
+          tools.each do |tool|
+            tool = Helpers::Hash.symbolize_keys(tool)
+
+            cartridge.each do |candidate|
+              next unless tool[:function][:name] == candidate[:name]
+
+              source = {}
+
+              source[:clojure] = candidate[:clojure] if candidate[:clojure]
+              source[:fennel] = candidate[:fennel] if candidate[:fennel]
+              source[:lua] = candidate[:lua] if candidate[:lua]
+
+              applies << {
+                id: tool[:id],
+                name: tool[:function][:name],
+                type: 'function',
+                parameters: JSON.parse(tool[:function][:arguments]),
+                source:
+              }
+            end
+          end
+
+          raise 'missing tool' if applies.size != tools.size
+
+          applies
+        end
+
+        def self.adapt(cartridge)
+          output = {
+            type: 'function',
+            function: {
+              name: cartridge[:name], description: cartridge[:description]
+            }
+          }
+
+          output[:function][:parameters] = (cartridge[:parameters] || { type: 'object', properties: {} })
+
+          output
+        end
+      end
+    end
+  end
+end