diff options
| author | icebaker <icebaker@proton.me> | 2023-11-18 19:07:10 -0300 |
|---|---|---|
| committer | icebaker <icebaker@proton.me> | 2023-11-18 19:07:10 -0300 |
| commit | 8ae78b954350755a47a13133668dba93bac15f37 (patch) | |
| tree | 9cdc3bb770d778bd8d00675fdbc1f27a6e27e37c /logic/providers/openai.rb | |
| parent | ab22d1bbe37093912cb7418b3c945153a15f4255 (diff) | |
adding support for tools
Diffstat (limited to 'logic/providers/openai.rb')
| -rw-r--r-- | logic/providers/openai.rb | 57 |
1 files changed, 57 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 |