v1.2 真实模型接入 + Thinking 死循环修复

- 新增 OpenAIProvider / ClaudeProvider，接入 Deepseek 真实 API - 修复 Deepseek thinking 模式 reasoning_content 回传问题 - 修复 Thinking 阶段过渡指令每轮重复插入导致的死循环 - 新增 maxTurns 保护、dumpMessages/dumpTools 调试输出
2026-05-14 20:35:44 +08:00
parent f75d0c49ae
commit daa89531f5
9 changed files with 491 additions and 47 deletions
@@ -0,0 +1,144 @@
+// internal/provider/claude.go
+package provider
+
+import (
+    "context"
+    "encoding/json"
+    "fmt"
+    // "os"
+
+    "github.com/anthropics/anthropic-sdk-go"
+    "github.com/anthropics/anthropic-sdk-go/option"
+    "go-tiny-claw/internal/schema"
+)
+
+type ClaudeProvider struct {
+    client anthropic.Client
+    model  string
+}
+
+func DeepseekClaudeProvider(model string) *ClaudeProvider {
+	apiKey := "sk-1f44696abe644bd684f09cc43f12c557"
+    if apiKey == "" {
+        panic("请设置 ZHIPU_API_KEY 环境变量")
+    }
+    baseURL := "https://api.deepseek.com/anthropic"
+    return &ClaudeProvider{
+        client: anthropic.NewClient(option.WithAPIKey(apiKey), option.WithBaseURL(baseURL)),
+        model:  model,
+    }
+}
+
+func (p *ClaudeProvider) Generate(ctx context.Context, msgs []schema.Message, availableTools []schema.ToolDefinition) (*schema.Message, error) {
+    var anthropicMsgs []anthropic.MessageParam
+    var systemPrompt string
+
+    // 1. 消息翻译
+    for _, msg := range msgs {
+        switch msg.Role {
+        case schema.RoleSystem:
+            systemPrompt = msg.Content
+        case schema.RoleUser:
+            if msg.ToolCallID != "" {
+                anthropicMsgs = append(anthropicMsgs, anthropic.NewUserMessage(
+                    anthropic.NewToolResultBlock(msg.ToolCallID, msg.Content, false),
+                ))
+            } else {
+                anthropicMsgs = append(anthropicMsgs, anthropic.NewUserMessage(
+                    anthropic.NewTextBlock(msg.Content),
+                ))
+            }
+        case schema.RoleAssistant:
+            var blocks []anthropic.ContentBlockParamUnion
+            if msg.Content != "" {
+                blocks = append(blocks, anthropic.NewTextBlock(msg.Content))
+            }
+
+            // 将历史工具调用转回 Claude 特有的 ToolUseBlockParam
+            for _, tc := range msg.ToolCalls {
+                var inputMap map[string]interface{}
+                _ = json.Unmarshal(tc.Arguments, &inputMap)
+                blocks = append(blocks, anthropic.ContentBlockParamUnion{
+                    OfToolUse: &anthropic.ToolUseBlockParam{
+                        ID:    tc.ID,
+                        Name:  tc.Name,
+                        Input: inputMap,
+                    },
+                })
+            }
+            if len(blocks) > 0 {
+                anthropicMsgs = append(anthropicMsgs, anthropic.NewAssistantMessage(blocks...))
+            }
+        }
+    }
+
+    // 2. 工具 Schema 翻译
+    var anthropicTools []anthropic.ToolUnionParam
+    for _, toolDef := range availableTools {
+        // ToolInputSchemaParam 是结构体，需要通过 Properties 字段精准填充
+        var properties map[string]any
+        var required []string
+
+        if m, ok := toolDef.InputSchema.(map[string]interface{}); ok {
+            if p, ok := m["properties"].(map[string]interface{}); ok {
+                properties = p
+            }
+            if r, ok := m["required"].([]string); ok {
+                required = r
+            }
+        }
+
+        tp := anthropic.ToolParam{
+            Name:        toolDef.Name,
+            Description: anthropic.String(toolDef.Description),
+            InputSchema: anthropic.ToolInputSchemaParam{
+                Properties: properties,
+                Required:   required,
+            },
+        }
+        anthropicTools = append(anthropicTools, anthropic.ToolUnionParam{OfTool: &tp})
+    }
+
+    // 3. 构建请求并发送
+    params := anthropic.MessageNewParams{
+        Model:     anthropic.Model(p.model),
+        MaxTokens: 4096,
+        Messages:  anthropicMsgs,
+    }
+
+    if systemPrompt != "" {
+        params.System = []anthropic.TextBlockParam{
+            {Text: systemPrompt},
+        }
+    }
+
+    if len(anthropicTools) > 0 {
+        params.Tools = anthropicTools
+    }
+
+    resp, err := p.client.Messages.New(ctx, params)
+    if err != nil {
+        return nil, fmt.Errorf("Claude/Zhipu API 请求失败: %w", err)
+    }
+
+    // 4. 反向解析
+    resultMsg := &schema.Message{
+        Role: schema.RoleAssistant,
+    }
+
+    for _, block := range resp.Content {
+        switch block.Type {
+        case "text":
+            resultMsg.Content += block.Text
+        case "tool_use":
+            argsBytes, _ := json.Marshal(block.Input)
+            resultMsg.ToolCalls = append(resultMsg.ToolCalls, schema.ToolCall{
+                ID:        block.ID,
+                Name:      block.Name,
+                Arguments: argsBytes,
+            })
+        }
+    }
+
+    return resultMsg, nil
+}