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

🐛 fix: function call error

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

2
providers/baidu/chat.go
View File

@ -150,7 +150,7 @@ func convertFromChatOpenai(request *types.ChatCompletionRequest) *BaiduChatReque
if message.Role == types.ChatMessageRoleSystem {
baiduChatRequest.System = message.StringContent()
continue
} else if message.Role == types.ChatMessageRoleFunction {
} else if message.Role == types.ChatMessageRoleFunction || message.Role == types.ChatMessageRoleTool {
baiduChatRequest.Messages = append(baiduChatRequest.Messages, BaiduMessage{
Role: types.ChatMessageRoleAssistant,
FunctionCall: &types.ChatCompletionToolCallsFunction{

2
providers/gemini/type.go
View File

@ -151,7 +151,7 @@ func OpenAIToGeminiChatContent(openaiContents []types.ChatCompletionMessage) ([]
},
},
}
} else if openaiContent.Role == types.ChatMessageRoleTool {
} else if openaiContent.Role == types.ChatMessageRoleFunction || openaiContent.Role == types.ChatMessageRoleTool {
content = GeminiChatContent{
Role: "function",
Parts: []GeminiPart{

2
providers/minimax/chat.go
View File

@ -145,7 +145,7 @@ func convertFromChatOpenai(request *types.ChatCompletionRequest) *MiniMaxChatReq
}
// 如果role为function， 则需要在前面一条记录添加function_call，如果没有消息，则添加一个message
if message.Role == types.ChatMessageRoleFunction {
if message.Role == types.ChatMessageRoleFunction || message.Role == types.ChatMessageRoleTool {
if len(messges) == 0 {
messges = append(messges, MiniMaxChatMessage{
SenderType: "USER",

2
providers/xunfei/chat.go
View File

@ -75,7 +75,7 @@ func (p *XunfeiProvider) getChatRequest(request *types.ChatCompletionRequest) (*
func (p *XunfeiProvider) convertFromChatOpenai(request *types.ChatCompletionRequest) *XunfeiChatRequest {
messages := make([]XunfeiMessage, 0, len(request.Messages))
for _, message := range request.Messages {
if message.Role == types.ChatMessageRoleFunction {
if message.Role == types.ChatMessageRoleFunction || message.Role == types.ChatMessageRoleTool {
messages = append(messages, XunfeiMessage{
Role: types.ChatMessageRoleUser,
Content: "这是函数调用返回的内容，请回答之前的问题：\n" + message.StringContent(),