discourse-ai/lib/completions/dialects/open_ai_tools.rb

# frozen_string_literal: true

module DiscourseAi
  module Completions
    module Dialects
      class OpenAiTools
        def initialize(tools)
          @raw_tools = tools
        end

        def translated_tools
          raw_tools.map do |t|
            tool = t.dup

            tool[:parameters] = t[:parameters]
              .to_a
              .reduce({ type: "object", properties: {}, required: [] }) do |memo, p|
                name = p[:name]
                memo[:required] << name if p[:required]

                except = %i[name required item_type]
                except << :enum if p[:enum].blank?

                memo[:properties][name] = p.except(*except)

                memo[:properties][name][:items] = { type: p[:item_type] } if p[:item_type]
                memo
              end

            { type: "function", function: tool }
          end
        end

        def instructions
          "" # Noop. Tools are listed separate.
        end

        def from_raw_tool_call(raw_message)
          call_details = JSON.parse(raw_message[:content], symbolize_names: true)
          call_details[:arguments] = call_details[:arguments].to_json
          call_details[:name] = raw_message[:name]

          {
            role: "assistant",
            content: nil,
            tool_calls: [{ type: "function", function: call_details, id: raw_message[:id] }],
          }
        end

        def from_raw_tool(raw_message)
          {
            role: "tool",
            tool_call_id: raw_message[:id],
            content: raw_message[:content],
            name: raw_message[:name],
          }
        end

        private

        attr_reader :raw_tools
      end
    end
  end
end