refactor: 重构llm包

backend/internal/llm/tools_test.go

@@ -0,0 +1,113 @@
+package llm
+
+import (
+	"testing"
+)
+
+func TestGetKnowledgeGraphTools_ReturnsTools(t *testing.T) {
+	tools := GetKnowledgeGraphTools()
+	if len(tools) == 0 {
+		t.Fatal("GetKnowledgeGraphTools() returned no tools")
+	}
+	if len(tools) != 6 {
+		t.Fatalf("expected 6 tools, got %d", len(tools))
+	}
+}
+
+func TestGetKnowledgeGraphTools_ToolNames(t *testing.T) {
+	tools := GetKnowledgeGraphTools()
+	expected := map[string]bool{
+		"search_nodes":  true,
+		"get_neighbors": true,
+		"create_node":   true,
+		"create_edge":   true,
+		"delete_node":   true,
+		"delete_edge":   true,
+	}
+
+	for _, tool := range tools {
+		if tool.Function == nil {
+			t.Error("tool has nil Function definition")
+			continue
+		}
+		name := tool.Function.Name
+		if !expected[name] {
+			t.Errorf("unexpected tool name: %s", name)
+		}
+		delete(expected, name)
+	}
+
+	if len(expected) > 0 {
+		t.Errorf("missing tools: %v", expected)
+	}
+}
+
+func TestGetKnowledgeGraphTools_HasDescriptions(t *testing.T) {
+	tools := GetKnowledgeGraphTools()
+	for _, tool := range tools {
+		if tool.Function == nil {
+			continue
+		}
+		if tool.Function.Description == "" {
+			t.Errorf("tool %s has empty description", tool.Function.Name)
+		}
+	}
+}
+
+func TestGetKnowledgeGraphTools_HasParameters(t *testing.T) {
+	tools := GetKnowledgeGraphTools()
+	for _, tool := range tools {
+		if tool.Function == nil {
+			continue
+		}
+		params, ok := tool.Function.Parameters.(map[string]any)
+		if !ok {
+			t.Errorf("tool %s parameters is not map[string]any", tool.Function.Name)
+			continue
+		}
+		if params["type"] != "object" {
+			t.Errorf("tool %s parameters type is not 'object'", tool.Function.Name)
+		}
+		props, ok := params["properties"].(map[string]any)
+		if !ok {
+			t.Errorf("tool %s has no properties map", tool.Function.Name)
+			continue
+		}
+		if len(props) == 0 {
+			t.Errorf("tool %s has empty properties", tool.Function.Name)
+		}
+	}
+}
+
+func TestGetKnowledgeGraphTools_RequiredFields(t *testing.T) {
+	tools := GetKnowledgeGraphTools()
+	for _, tool := range tools {
+		if tool.Function == nil {
+			continue
+		}
+		params := tool.Function.Parameters.(map[string]any)
+		required, ok := params["required"].([]string)
+		if !ok {
+			t.Errorf("tool %s has no required fields", tool.Function.Name)
+			continue
+		}
+		if len(required) == 0 {
+			t.Errorf("tool %s has empty required fields", tool.Function.Name)
+		}
+	}
+}
+
+func TestGetKnowledgeGraphTools_NoDuplicates(t *testing.T) {
+	tools := GetKnowledgeGraphTools()
+	seen := map[string]bool{}
+	for _, tool := range tools {
+		if tool.Function == nil {
+			continue
+		}
+		name := tool.Function.Name
+		if seen[name] {
+			t.Errorf("duplicate tool name: %s", name)
+		}
+		seen[name] = true
+	}
+}