Module: Raif::Concerns::Llms::OpenAiResponses::ToolFormatting

Extended by:
ActiveSupport::Concern
Included in:
Llms::OpenAiResponses
Defined in:
app/models/raif/concerns/llms/open_ai_responses/tool_formatting.rb

Instance Method Summary collapse

Instance Method Details

#build_tools_parameter(model_completion) ⇒ Object 



6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
 
# File 'app/models/raif/concerns/llms/open_ai_responses/tool_formatting.rb', line 6

def build_tools_parameter(model_completion)
  model_completion.available_model_tools_map.map do |_tool_name, tool|
    if tool.provider_managed?
      format_provider_managed_tool(tool)
    else
      # It's a developer-managed tool
      validate_json_schema!(tool.tool_arguments_schema)

      {
        type: "function",
        name: tool.tool_name,
        description: tool.tool_description,
        parameters: tool.tool_arguments_schema
      }
    end
  end
end

#format_provider_managed_tool(tool) ⇒ Object 



24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
 
# File 'app/models/raif/concerns/llms/open_ai_responses/tool_formatting.rb', line 24

def format_provider_managed_tool(tool)
  validate_provider_managed_tool_support!(tool)

  case tool.name
  when "Raif::ModelTools::ProviderManaged::WebSearch"
    { type: "web_search_preview" }
  when "Raif::ModelTools::ProviderManaged::CodeExecution"
    {
      type: "code_interpreter",
      container: { "type": "auto" }
    }
  when "Raif::ModelTools::ProviderManaged::ImageGeneration"
    { type: "image_generation" }
  else
    raise Raif::Errors::UnsupportedFeatureError,
      "Invalid provider-managed tool: #{tool.name} for #{key}"
  end
end