File size: 2,540 Bytes
80ffd2e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 | package parser
import (
"testing"
)
func TestTaskIdAsString(t *testing.T) {
// 测试 taskId 参数应该保持为字符串类型，即使值看起来像数字
text := `<function_calls>
<invoke name="TaskUpdate">
<parameter name="taskId">1</parameter>
<parameter name="status">completed</parameter>
</invoke>
</function_calls>`
result := ParseToolCalls(text)
if len(result.ToolCalls) != 1 {
t.Fatalf("Expected 1 tool call, got %d", len(result.ToolCalls))
}
toolCall := result.ToolCalls[0]
if toolCall.Name != "TaskUpdate" {
t.Errorf("Expected tool name 'TaskUpdate', got '%s'", toolCall.Name)
}
// 检查 taskId 是否为字符串类型
taskId, ok := toolCall.Input["taskId"]
if !ok {
t.Fatal("taskId parameter not found")
}
taskIdStr, ok := taskId.(string)
if !ok {
t.Errorf("taskId should be string type, got %T with value %v", taskId, taskId)
}
if taskIdStr != "1" {
t.Errorf("Expected taskId to be '1', got '%s'", taskIdStr)
}
// 检查 status 参数应该保持为字符串（因为它本来就是字符串）
status, ok := toolCall.Input["status"]
if !ok {
t.Fatal("status parameter not found")
}
statusStr, ok := status.(string)
if !ok {
t.Errorf("status should be string type, got %T", status)
}
if statusStr != "completed" {
t.Errorf("Expected status to be 'completed', got '%s'", statusStr)
}
}
func TestOtherNumericParams(t *testing.T) {
// 测试其他数字参数应该被解析为数字类型
text := `<function_calls>
<invoke name="SomeTool">
<parameter name="count">42</parameter>
<parameter name="enabled">true</parameter>
</invoke>
</function_calls>`
result := ParseToolCalls(text)
if len(result.ToolCalls) != 1 {
t.Fatalf("Expected 1 tool call, got %d", len(result.ToolCalls))
}
toolCall := result.ToolCalls[0]
// count 应该是数字类型
count, ok := toolCall.Input["count"]
if !ok {
t.Fatal("count parameter not found")
}
countFloat, ok := count.(float64)
if !ok {
t.Errorf("count should be float64 type, got %T", count)
}
if countFloat != 42 {
t.Errorf("Expected count to be 42, got %f", countFloat)
}
// enabled 应该是布尔类型
enabled, ok := toolCall.Input["enabled"]
if !ok {
t.Fatal("enabled parameter not found")
}
enabledBool, ok := enabled.(bool)
if !ok {
t.Errorf("enabled should be bool type, got %T", enabled)
}
if !enabledBool {
t.Errorf("Expected enabled to be true, got %v", enabledBool)
}
} |