Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🐛 fix: gemini function call error

Browse Source
This commit is contained in:
MartialBE 2024-05-18 17:22:26 +08:00
parent 6ab116c0eb
commit 6cd0f932c9
No known key found for this signature in database
GPG Key ID: 27C0267EC84B0A5C
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
providers/gemini/type.go
View File

@ -139,18 +139,19 @@ func OpenAIToGeminiChatContent(openaiContents []types.ChatCompletionMessage) ([]
Parts: make([]GeminiPart, 0), Parts: make([]GeminiPart, 0),
} }
content.Role = ConvertRole(openaiContent.Role) content.Role = ConvertRole(openaiContent.Role)
if openaiContent.Role == types.ChatMessageRoleFunction { if openaiContent.ToolCalls != nil {
contents = append(contents, GeminiChatContent{ content = GeminiChatContent{
Role: "model", Role: "model",
Parts: []GeminiPart{ Parts: []GeminiPart{
{ {
FunctionCall: &GeminiFunctionCall{ FunctionCall: &GeminiFunctionCall{
Name: *openaiContent.Name, Name: openaiContent.ToolCalls[0].Function.Name,
Args: map[string]interface{}{}, Args: map[string]interface{}{},
}, },
}, },
}, },
}) }
} else if openaiContent.Role == types.ChatMessageRoleTool {
content = GeminiChatContent{ content = GeminiChatContent{
Role: "function", Role: "function",
Parts: []GeminiPart{ Parts: []GeminiPart{