mirror of
https://gh.wpcy.net/https://github.com/discourse/discourse.git
synced 2026-06-19 07:03:42 +08:00
e27f493995
This is an extensive set of changes to the way we handle LLM thinking blocks - We now treat "thinking" as a unified entity meaning that tool calls and thinking blocks all live in a single details block - We reworked Thinking object so it stores provider specific data, that allows thinking blocks to work differently for anthropic / open ai while keeping a consistent UI - "Show tool details" is now "Show thinking" which is far more intuitive - The Open AI responses API was implemented more completely
69 lines
1.7 KiB
Ruby
Vendored
69 lines
1.7 KiB
Ruby
Vendored
# frozen_string_literal: true
|
|
|
|
module DiscourseAi
|
|
module Completions
|
|
module Dialects
|
|
class ClaudeTools
|
|
def initialize(tools)
|
|
@raw_tools = tools
|
|
end
|
|
|
|
def translated_tools
|
|
raw_tools.map do |t|
|
|
{ name: t.name, description: t.description, input_schema: t.parameters_json_schema }
|
|
end
|
|
end
|
|
|
|
def instructions
|
|
""
|
|
end
|
|
|
|
def from_raw_tool_call(raw_message)
|
|
call_details = JSON.parse(raw_message[:content], symbolize_names: true)
|
|
result = []
|
|
|
|
anthropic = anthropic_reasoning(raw_message)
|
|
|
|
if anthropic.present?
|
|
if raw_message[:thinking] && anthropic[:signature]
|
|
result << {
|
|
type: "thinking",
|
|
thinking: raw_message[:thinking],
|
|
signature: anthropic[:signature],
|
|
}
|
|
end
|
|
|
|
if anthropic[:redacted_signature]
|
|
result << { type: "redacted_thinking", data: anthropic[:redacted_signature] }
|
|
end
|
|
end
|
|
|
|
tool_call_id = raw_message[:id]
|
|
|
|
result << {
|
|
type: "tool_use",
|
|
id: tool_call_id,
|
|
name: raw_message[:name],
|
|
input: call_details[:arguments],
|
|
}
|
|
|
|
result
|
|
end
|
|
|
|
def from_raw_tool(raw_message)
|
|
[{ type: "tool_result", tool_use_id: raw_message[:id], content: raw_message[:content] }]
|
|
end
|
|
|
|
private
|
|
|
|
def anthropic_reasoning(message)
|
|
info = message[:thinking_provider_info]
|
|
return if info.blank?
|
|
info[:anthropic] || info["anthropic"]
|
|
end
|
|
|
|
attr_reader :raw_tools
|
|
end
|
|
end
|
|
end
|
|
end
|
