fix: ToolChoice数据类型解析异常

Author: bincooo

internal/agent/com.go

@@ -28,6 +28,7 @@ TOOL_CALL:
 你是一个智能机器人，你拥有专注于拆解多个任务的能力。有时候，你可以依赖工具的运行结果，来更准确的回答用户。
 
 请你根据用户请求，拆解出3个以内的子任务。在完成拆解过程中，USER代表用户的输入，TOOL_RESPONSE代表工具运行结果。ASSISTANT 代表你的输出，task为子任务字符串描述。
+浏览上面的上下文，避免出现与最新用户请求无关的子任务。
 
 你的每次输出都必须以0,1开头，代表是否需要拆解任务：
 0: 无拆解任务。

internal/plugin/toolcall.go

@@ -95,6 +95,31 @@ func CompleteToolCalls(ctx *gin.Context, completion pkg.ChatCompletion, callback
 		}
 	}
 
+	// toolChoice自推荐toolId处理
+	if completion.ToolChoice != "" && completion.ToolChoice != "auto" {
+		var (
+			keyv       pkg.Keyv[interface{}]
+			toolChoice pkg.Keyv[interface{}]
+			ok         = false
+		)
+		toolChoice, ok = completion.ToolChoice.(map[string]interface{})
+		if !ok || !toolChoice.Is("type", "function") {
+			goto label
+		}
+
+		keyv = toolChoice.GetKeyv("function")
+		if !keyv.Has("name") {
+			goto label
+		}
+
+		if toolId := toolIdWithTools(keyv.GetString("name"), completion.Tools); toolId != "-1" {
+			completion.Messages = append(completion.Messages, pkg.Keyv[interface{}]{
+				"role": "user", "content": "continue。 工具推荐： toolId = " + toolId,
+			})
+		}
+	label:
+	}
+
 	message, err := buildTemplate(ctx, completion, agent.ToolCall)
 	if err != nil {
 		return false, err
@@ -503,14 +528,16 @@ func toolIdWithTools(name string, tools []pkg.Keyv[interface{}]) (value string)
 
 	for _, t := range tools {
 		fn := t.GetKeyv("function")
+		if fn.Has("id") && value == fn.GetString("id") {
+			return
+		}
+
 		if fn.Has("name") {
 			if fn.Has("id") && value == fn.GetString("name") {
 				value = fn.GetString("id")
 				return
 			}
-		}
-
-		if fn.Has("id") && value == fn.GetString("id") {
+			value = fn.GetString("name")
 			return
 		}
 	}
@@ -543,6 +570,11 @@ func nameWithTools(name string, tools []pkg.Keyv[interface{}]) (value string) {
 }
 
 func toolIsEnabled(ctx *gin.Context) bool {
+	completion := common.GetGinCompletion(ctx)
+	if completion.ToolChoice != "" && completion.ToolChoice != "auto" {
+		return false
+	}
+
 	t := common.GetGinToolValue(ctx)
 	return t.Is("tasks", true)
 }

pkg/model.go

@@ -15,7 +15,7 @@ type ChatCompletion struct {
 	TopK          int                 `json:"topK"`
 	TopP          float32             `json:"topP"`
 	Stream        bool                `json:"stream"`
-	ToolChoice    string              `json:"tool_choice"`
+	ToolChoice    interface{}         `json:"tool_choice"`
 }
 
 type ChatGeneration struct {