xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-17 13:43:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

🐛 fix: function call error (#190)

Browse Source
This commit is contained in:
MartialBE
2024-05-19 13:11:46 +08:00
parent 68e5cf1183
commit c04dfc735f
6 changed files with 54 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
providers/gemini/type.go
View File

@@ -133,25 +133,34 @@ func (g *GeminiChatResponse) GetResponseText() string {
func OpenAIToGeminiChatContent(openaiContents []types.ChatCompletionMessage) ([]GeminiChatContent, *types.OpenAIErrorWithStatusCode) {
contents := make([]GeminiChatContent, 0)
useToolName := ""
for _, openaiContent := range openaiContents {
content := GeminiChatContent{
Role: ConvertRole(openaiContent.Role),
Parts: make([]GeminiPart, 0),
}
content.Role = ConvertRole(openaiContent.Role)
if openaiContent.ToolCalls != nil {
if openaiContent.ToolCalls != nil || openaiContent.FunctionCall != nil {
if openaiContent.ToolCalls != nil {
useToolName = openaiContent.ToolCalls[0].Function.Name
} else {
useToolName = openaiContent.FunctionCall.Name
}
content = GeminiChatContent{
Role: "model",
Parts: []GeminiPart{
{
FunctionCall: &GeminiFunctionCall{
Name: openaiContent.ToolCalls[0].Function.Name,
Name: useToolName,
Args: map[string]interface{}{},
},
},
},
}
} else if openaiContent.Role == types.ChatMessageRoleFunction || openaiContent.Role == types.ChatMessageRoleTool {
if openaiContent.Name == nil {
openaiContent.Name = &useToolName
}
content = GeminiChatContent{
Role: "function",
Parts: []GeminiPart{