63 lines
1.6 KiB
Ruby
63 lines
1.6 KiB
Ruby
|
# 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]
|
||
|
|
|
||
|
|
memo[:properties][name] = p.except(:name, :required, :item_type)
|
||
|
|
|
||
|
|
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
|