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

fix:judge baidu embeddings input type

Browse Source
This commit is contained in:
igophper 2023-08-12 16:30:13 +08:00
parent 7e2bca7e9c
commit 1c80119fde
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
controller/relay-baidu.go
View File

@ -140,8 +140,12 @@ func embeddingRequestOpenAI2Baidu(request GeneralOpenAIRequest) *BaiduEmbeddingR
switch request.Input.(type) {
case string:
baiduEmbeddingRequest.Input = []string{request.Input.(string)}
case []string:
baiduEmbeddingRequest.Input = request.Input.([]string)
case []any:
for _, item := range request.Input.([]any) {
if str, isStr := item.(string); isStr {
baiduEmbeddingRequest.Input = append(baiduEmbeddingRequest.Input, str)
}
}
}
return &baiduEmbeddingRequest
}