hhs
114 lines
2.6 KiB
Go
114 lines
2.6 KiB
Go
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
|
|
}
|
|
}
|