Add Anthropic tool calling support
This commit is contained in:
154
internal/providers/anthropic/convert.go
Normal file
154
internal/providers/anthropic/convert.go
Normal file
@@ -0,0 +1,154 @@
|
||||
package anthropic
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
|
||||
"github.com/ajac-zero/latticelm/internal/api"
|
||||
"github.com/anthropics/anthropic-sdk-go"
|
||||
)
|
||||
|
||||
// parseTools converts Open Responses tools to Anthropic format
|
||||
func parseTools(req *api.ResponseRequest) ([]anthropic.ToolUnionParam, error) {
|
||||
if req.Tools == nil || len(req.Tools) == 0 {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
var toolDefs []map[string]interface{}
|
||||
if err := json.Unmarshal(req.Tools, &toolDefs); err != nil {
|
||||
return nil, fmt.Errorf("unmarshal tools: %w", err)
|
||||
}
|
||||
|
||||
var tools []anthropic.ToolUnionParam
|
||||
for _, td := range toolDefs {
|
||||
// Extract: name, description, parameters
|
||||
// Note: Anthropic uses "input_schema" instead of "parameters"
|
||||
name, _ := td["name"].(string)
|
||||
desc, _ := td["description"].(string)
|
||||
params, _ := td["parameters"].(map[string]interface{})
|
||||
|
||||
inputSchema := anthropic.ToolInputSchemaParam{
|
||||
Type: "object",
|
||||
Properties: params["properties"],
|
||||
}
|
||||
|
||||
// Add required fields if present
|
||||
if required, ok := params["required"].([]interface{}); ok {
|
||||
requiredStrs := make([]string, 0, len(required))
|
||||
for _, r := range required {
|
||||
if str, ok := r.(string); ok {
|
||||
requiredStrs = append(requiredStrs, str)
|
||||
}
|
||||
}
|
||||
inputSchema.Required = requiredStrs
|
||||
}
|
||||
|
||||
// Create the tool using ToolUnionParamOfTool
|
||||
tool := anthropic.ToolUnionParamOfTool(inputSchema, name)
|
||||
|
||||
if desc != "" {
|
||||
tool.OfTool.Description = anthropic.String(desc)
|
||||
}
|
||||
|
||||
tools = append(tools, tool)
|
||||
}
|
||||
|
||||
return tools, nil
|
||||
}
|
||||
|
||||
// parseToolChoice converts Open Responses tool_choice to Anthropic format
|
||||
func parseToolChoice(req *api.ResponseRequest) (anthropic.ToolChoiceUnionParam, error) {
|
||||
var result anthropic.ToolChoiceUnionParam
|
||||
|
||||
if req.ToolChoice == nil || len(req.ToolChoice) == 0 {
|
||||
return result, nil
|
||||
}
|
||||
|
||||
var choice interface{}
|
||||
if err := json.Unmarshal(req.ToolChoice, &choice); err != nil {
|
||||
return result, fmt.Errorf("unmarshal tool_choice: %w", err)
|
||||
}
|
||||
|
||||
// Handle string values: "auto", "any", "required"
|
||||
if str, ok := choice.(string); ok {
|
||||
switch str {
|
||||
case "auto":
|
||||
result.OfAuto = &anthropic.ToolChoiceAutoParam{
|
||||
Type: "auto",
|
||||
}
|
||||
case "any", "required":
|
||||
result.OfAny = &anthropic.ToolChoiceAnyParam{
|
||||
Type: "any",
|
||||
}
|
||||
case "none":
|
||||
result.OfNone = &anthropic.ToolChoiceNoneParam{
|
||||
Type: "none",
|
||||
}
|
||||
default:
|
||||
return result, fmt.Errorf("unknown tool_choice string: %s", str)
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
|
||||
// Handle specific tool selection: {"type": "tool", "function": {"name": "..."}}
|
||||
if obj, ok := choice.(map[string]interface{}); ok {
|
||||
// Check for OpenAI format: {"type": "function", "function": {"name": "..."}}
|
||||
if funcObj, ok := obj["function"].(map[string]interface{}); ok {
|
||||
if name, ok := funcObj["name"].(string); ok {
|
||||
result.OfTool = &anthropic.ToolChoiceToolParam{
|
||||
Type: "tool",
|
||||
Name: name,
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
}
|
||||
|
||||
// Check for direct name field
|
||||
if name, ok := obj["name"].(string); ok {
|
||||
result.OfTool = &anthropic.ToolChoiceToolParam{
|
||||
Type: "tool",
|
||||
Name: name,
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
}
|
||||
|
||||
return result, fmt.Errorf("invalid tool_choice format")
|
||||
}
|
||||
|
||||
// extractToolCalls converts Anthropic content blocks to api.ToolCall
|
||||
func extractToolCalls(content []anthropic.ContentBlockUnion) []api.ToolCall {
|
||||
var toolCalls []api.ToolCall
|
||||
|
||||
for _, block := range content {
|
||||
// Check if this is a tool_use block
|
||||
if block.Type == "tool_use" {
|
||||
// Cast to ToolUseBlock to access the fields
|
||||
toolUse := block.AsToolUse()
|
||||
|
||||
// Marshal the input to JSON string for Arguments
|
||||
argsJSON, _ := json.Marshal(toolUse.Input)
|
||||
|
||||
toolCalls = append(toolCalls, api.ToolCall{
|
||||
ID: toolUse.ID,
|
||||
Name: toolUse.Name,
|
||||
Arguments: string(argsJSON),
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
return toolCalls
|
||||
}
|
||||
|
||||
// extractToolCallDelta extracts tool call delta from streaming content block delta
|
||||
func extractToolCallDelta(delta anthropic.RawContentBlockDeltaUnion, index int) *api.ToolCallDelta {
|
||||
// Check if this is an input_json_delta (streaming tool arguments)
|
||||
if delta.Type == "input_json_delta" {
|
||||
return &api.ToolCallDelta{
|
||||
Index: index,
|
||||
Arguments: delta.PartialJSON,
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user