070d4c3459
后端： 1. 排程工具从 tools/ 根目录拆分为 tools/schedule 独立子包 - 12 个排程工具文件等价迁入 tools/schedule/，tools/ 根目录仅保留 registry.go 作为统一注册入口 - 所有依赖方（conv / model / node / prompt / service）import 统一切到 schedule 子包 2. Web 搜索工具链落地（tools/web 子包） - 新增 web_search（结构化检索）与 web_fetch（正文抓取）两个读工具，支持博查 API / mock 降级 - 启动流程按配置选择 provider，未识别类型自动降级为 mock，不阻断主流程 - 执行提示补齐 web 工具使用约束与返回值示例 - config.example.yaml 补齐 websearch 配置段 前端：无 仓库：无
126 lines
3.0 KiB
Go
126 lines
3.0 KiB
Go
package schedule
|
||
|
||
import "fmt"
|
||
|
||
// ==================== 参数解析辅助 ====================
|
||
// 这些函数专门用于从 LLM 输出的 map[string]any 中提取工具参数。
|
||
// JSON 反序列化后数字默认为 float64，字符串为 string，需要类型断言。
|
||
|
||
// argsInt 从 map 中提取 int 值。支持 float64（JSON 反序列化的默认类型）。
|
||
func ArgsInt(args map[string]any, key string) (int, bool) {
|
||
v, ok := args[key]
|
||
if !ok {
|
||
return 0, false
|
||
}
|
||
switch n := v.(type) {
|
||
case float64:
|
||
return int(n), true
|
||
case int:
|
||
return n, true
|
||
}
|
||
return 0, false
|
||
}
|
||
|
||
// argsString 从 map 中提取 string 值。
|
||
func ArgsString(args map[string]any, key string) (string, bool) {
|
||
v, ok := args[key]
|
||
if !ok {
|
||
return "", false
|
||
}
|
||
s, ok := v.(string)
|
||
return s, ok
|
||
}
|
||
|
||
// argsIntPtr 从 map 中提取可选 int 值，不存在返回 nil。
|
||
func ArgsIntPtr(args map[string]any, key string) *int {
|
||
v, ok := ArgsInt(args, key)
|
||
if !ok {
|
||
return nil
|
||
}
|
||
return &v
|
||
}
|
||
|
||
// argsStringPtr 从 map 中提取可选 string 值，不存在返回 nil。
|
||
func ArgsStringPtr(args map[string]any, key string) *string {
|
||
v, ok := ArgsString(args, key)
|
||
if !ok {
|
||
return nil
|
||
}
|
||
return &v
|
||
}
|
||
|
||
// argsIntSlice 从 map 中提取 int 数组，支持 []any / []int / []float64。
|
||
func ArgsIntSlice(args map[string]any, key string) ([]int, bool) {
|
||
v, ok := args[key]
|
||
if !ok {
|
||
return nil, false
|
||
}
|
||
switch arr := v.(type) {
|
||
case []int:
|
||
if len(arr) == 0 {
|
||
return []int{}, true
|
||
}
|
||
result := make([]int, len(arr))
|
||
copy(result, arr)
|
||
return result, true
|
||
case []float64:
|
||
result := make([]int, 0, len(arr))
|
||
for _, item := range arr {
|
||
result = append(result, int(item))
|
||
}
|
||
return result, true
|
||
case []any:
|
||
result := make([]int, 0, len(arr))
|
||
for _, item := range arr {
|
||
switch n := item.(type) {
|
||
case float64:
|
||
result = append(result, int(n))
|
||
case int:
|
||
result = append(result, n)
|
||
default:
|
||
return nil, false
|
||
}
|
||
}
|
||
return result, true
|
||
default:
|
||
return nil, false
|
||
}
|
||
}
|
||
|
||
// argsMoveList 从 map 中提取 batch_move 的 moves 数组。
|
||
func ArgsMoveList(args map[string]any) ([]MoveRequest, error) {
|
||
v, ok := args["moves"]
|
||
if !ok {
|
||
return nil, fmt.Errorf("缺少 moves 参数")
|
||
}
|
||
arr, ok := v.([]any)
|
||
if !ok {
|
||
return nil, fmt.Errorf("moves 参数必须是数组")
|
||
}
|
||
moves := make([]MoveRequest, 0, len(arr))
|
||
for i, item := range arr {
|
||
m, ok := item.(map[string]any)
|
||
if !ok {
|
||
return nil, fmt.Errorf("moves[%d] 不是有效对象", i)
|
||
}
|
||
taskID, ok := ArgsInt(m, "task_id")
|
||
if !ok {
|
||
return nil, fmt.Errorf("moves[%d].task_id 缺失或无效", i)
|
||
}
|
||
newDay, ok := ArgsInt(m, "new_day")
|
||
if !ok {
|
||
return nil, fmt.Errorf("moves[%d].new_day 缺失或无效", i)
|
||
}
|
||
newSlotStart, ok := ArgsInt(m, "new_slot_start")
|
||
if !ok {
|
||
return nil, fmt.Errorf("moves[%d].new_slot_start 缺失或无效", i)
|
||
}
|
||
moves = append(moves, MoveRequest{
|
||
TaskID: taskID,
|
||
NewDay: newDay,
|
||
NewSlotStart: newSlotStart,
|
||
})
|
||
}
|
||
return moves, nil
|
||
}
|