on main help sign in
fix: gemini tool calling
commit792e2b164b2370ebe966d51634277603fd02b8fb authorLukáš Loukota <lukasloukota94@gmail.com> dateSun 27 Apr 2025, 15:44 (2 years ago) parent5859dcd
1 file changed+4−5
internal/llm/provider/gemini.go+4−5
internal/llm/provider/gemini.go+4−5
@@ -132,7 +132,8 @@ func (g *geminiClient) convertMessages(messages []message.Message) []*genai.Cont
132132}
133133
134134func (g *geminiClient) convertTools(tools []tools.BaseTool) []*genai.Tool {
135− geminiTools := make([]*genai.Tool, 0, len(tools))
135+ geminiTool := &genai.Tool{}
136+ geminiTool.FunctionDeclarations = make([]*genai.FunctionDeclaration, 0, len(tools))
136137
137138 for _, tool := range tools {
138139 info := tool.Info()
@@ -146,12 +147,10 @@ func (g *geminiClient) convertTools(tools []tools.BaseTool) []*genai.Tool {
146147 },
147148 }
148149
149− geminiTools = append(geminiTools, &genai.Tool{
150− FunctionDeclarations: []*genai.FunctionDeclaration{declaration},
151− })
150+ geminiTool.FunctionDeclarations = append(geminiTool.FunctionDeclarations, declaration)
152151 }
153152
154− return geminiTools
153+ return []*genai.Tool{geminiTool}
155154}
156155
157156func (g *geminiClient) finishReason(reason genai.FinishReason) message.FinishReason {