wushenghua/sub2api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test(gemini): check function declaration assertions

Browse Source
This commit is contained in:
chinnsenn
2026-07-27 13:19:00 +09:00
parent 3e08106116
commit cc84cd8b4c
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/internal/service/gemini_messages_compat_service_test.go
+6 -2
View File
@@ -228,8 +228,12 @@ func TestGeminiForwardAsChatCompletions_FunctionNamedWebSearchStaysClientSide(t
functionDecls, ok := functionTool["functionDeclarations"].([]any)
require.True(t, ok)
require.Len(t, functionDecls, 2)
require.Equal(t, "web_search", functionDecls[0].(map[string]any)["name"])
require.Equal(t, "read_file", functionDecls[1].(map[string]any)["name"])
webSearchDecl, ok := functionDecls[0].(map[string]any)
require.True(t, ok)
readFileDecl, ok := functionDecls[1].(map[string]any)
require.True(t, ok)
require.Equal(t, "web_search", webSearchDecl["name"])
require.Equal(t, "read_file", readFileDecl["name"])
require.NotContains(t, functionTool, "googleSearch")
require.NotContains(t, functionTool, "google_search")
}