feat: support third part models of baidu (#1046)

百度千帆平台上的第三方大模型调用
songquanpeng · Mar 3, 2024 · 4fb22ad · 4fb22ad
1 parent 95cfb8e
commit 4fb22ad
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/relay/channel/baidu/adaptor.go b/relay/channel/baidu/adaptor.go
@@ -36,6 +36,8 @@ func (a *Adaptor) GetRequestURL(meta *util.RelayMeta) (string, error) {
 		fullRequestURL = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/bloomz_7b1"
 	case "Embedding-V1":
 		fullRequestURL = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/embeddings/embedding-v1"
+	default:
+               fullRequestURL = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/" + meta.ActualModelName
 	}
 	var accessToken string
 	var err error