summaryrefslogtreecommitdiff
path: root/components/providers/tools.rb
blob: 122bc141505250618beb5c822553be72ef6b330b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
# frozen_string_literal: true

require_relative '../embedding'
require_relative '../../logic/cartridge/safety'

require 'concurrent'

module NanoBot
  module Components
    module Providers
      module Tools
        def self.confirming(tool, feedback)
          feedback.call(
            { should_be_stored: false,
              interaction: { who: 'AI', message: nil, meta: {
                tool: { action: 'confirming', id: tool[:id], name: tool[:label], parameters: tool[:parameters] }
              } } }
          )
        end

        def self.apply(cartridge, function_cartridge, tools, feedback, tools_logic)
          prepared_tools = tools_logic.prepare(function_cartridge, tools)

          if Logic::Cartridge::Safety.confirmable?(cartridge)
            prepared_tools.each { |tool| tool[:allowed] = confirming(tool, feedback) }
          else
            prepared_tools.each { |tool| tool[:allowed] = true }
          end

          futures = prepared_tools.map do |tool|
            Concurrent::Promises.future do
              if tool[:allowed]
                process!(tool, feedback, function_cartridge, cartridge)
              else
                tool[:output] =
                  "We asked the user you're chatting with for permission, but the user did not allow you to run this tool or function."
                tool
              end
            end
          end

          results = Concurrent::Promises.zip(*futures).value!

          results.map do |applied_tool|
            {
              who: 'tool',
              message: applied_tool[:output],
              meta: { id: applied_tool[:id], name: applied_tool[:name] }
            }
          end
        end

        def self.process!(tool, feedback, _function_cartridge, cartridge)
          feedback.call(
            { should_be_stored: false,
              interaction: { who: 'AI', message: nil, meta: {
                tool: { action: 'executing', id: tool[:id], name: tool[:label], parameters: tool[:parameters] }
              } } }
          )

          call = {
            parameters: %w[parameters],
            values: [tool[:parameters]],
            safety: { sandboxed: Logic::Cartridge::Safety.sandboxed?(cartridge) }
          }

          if %i[fennel lua clojure].count { |key| !tool[:source][key].nil? } > 1
            raise StandardError, 'conflicting tools'
          end

          if !tool[:source][:fennel].nil?
            call[:source] = tool[:source][:fennel]
            tool[:output] = Components::Embedding.fennel(**call)
          elsif !tool[:source][:clojure].nil?
            call[:source] = tool[:source][:clojure]
            tool[:output] = Components::Embedding.clojure(**call)
          elsif !tool[:source][:lua].nil?
            call[:source] = tool[:source][:lua]
            tool[:output] = Components::Embedding.lua(**call)
          else
            raise 'missing source code'
          end

          feedback.call(
            { should_be_stored: false,
              interaction: { who: 'AI', message: nil, meta: {
                tool: {
                  action: 'responding', id: tool[:id], name: tool[:label],
                  parameters: tool[:parameters], output: tool[:output]
                }
              } } }
          )

          tool
        end
      end
    end
  end
end