bug: always prefer tool's given name over the referenced name

gptscript-ai · Sep 6, 2024 · 8128bbc · 8128bbc
1 parent d97d564
commit 8128bbc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/pkg/tests/testdata/TestCase2/call1.golden b/pkg/tests/testdata/TestCase2/call1.golden
@@ -4,7 +4,7 @@
     {
       "function": {
         "toolID": "testdata/TestCase2/test.gpt:bob",
-        "name": "Bob",
+        "name": "bob",
         "description": "I'm Bob, a friendly guy.",
         "parameters": {
           "properties": {

diff --git a/pkg/types/tool.go b/pkg/types/tool.go
@@ -698,7 +698,10 @@ func toolRefsToCompletionTools(completionTools []ToolReference, prg Program) (re
 	for _, subToolRef := range completionTools {
 		subTool := prg.ToolSet[subToolRef.ToolID]
 
-		subToolName := subToolRef.Reference
+		subToolName := subTool.Name
+		if subToolName == "" {
+			subToolName = subToolRef.Reference
+		}
 		if subToolRef.Named != "" {
 			subToolName = subToolRef.Named
 		}