From 1c80119fde1d775d972333389bcdf8c30d40f8ec Mon Sep 17 00:00:00 2001 From: igophper Date: Sat, 12 Aug 2023 16:30:13 +0800 Subject: [PATCH] fix:judge baidu embeddings input type --- controller/relay-baidu.go | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/controller/relay-baidu.go b/controller/relay-baidu.go index 664bbd11..f887fa62 100644 --- a/controller/relay-baidu.go +++ b/controller/relay-baidu.go @@ -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 }