discourse-ai/lib/ai_bot/tools/tool.rb

# frozen_string_literal: true

module DiscourseAi
  module AiBot
    module Tools
      class Tool
        class << self
          def signature
            raise NotImplemented
          end

          def name
            raise NotImplemented
          end

          def accepted_options
            []
          end

          def option(name, type:)
            Option.new(tool: self, name: name, type: type)
          end

          def help
            I18n.t("discourse_ai.ai_bot.command_help.#{signature[:name]}")
          end

          def custom_system_message
            nil
          end
        end

        attr_accessor :custom_raw

        def initialize(parameters, tool_call_id: "", persona_options: {})
          @parameters = parameters
          @tool_call_id = tool_call_id
          @persona_options = persona_options
        end

        attr_reader :parameters, :tool_call_id

        def name
          self.class.name
        end

        def summary
          I18n.t("discourse_ai.ai_bot.command_summary.#{name}")
        end

        def details
          I18n.t("discourse_ai.ai_bot.command_description.#{name}", description_args)
        end

        def help
          I18n.t("discourse_ai.ai_bot.command_help.#{name}")
        end

        def options
          self
            .class
            .accepted_options
            .reduce(HashWithIndifferentAccess.new) do |memo, option|
              val = @persona_options[option.name]
              memo[option.name] = val if val
              memo
            end
        end

        def chain_next_response?
          true
        end

        def standalone?
          false
        end

        protected

        def accepted_options
          []
        end

        def option(name, type:)
          Option.new(tool: self, name: name, type: type)
        end

        def description_args
          {}
        end

        def format_results(rows, column_names = nil, args: nil)
          rows = rows&.map { |row| yield row } if block_given?

          if !column_names
            index = -1
            column_indexes = {}

            rows =
              rows&.map do |data|
                new_row = []
                data.each do |key, value|
                  found_index = column_indexes[key.to_s] ||= (index += 1)
                  new_row[found_index] = value
                end
                new_row
              end
            column_names = column_indexes.keys
          end

          # this is not the most efficient format
          # however this is needed cause GPT 3.5 / 4 was steered using JSON
          result = { column_names: column_names, rows: rows }
          result[:args] = args if args
          result
        end
      end
    end
  end
end