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

momomobinx · web-flow · commit 4f1f052211e1 · 2024-03-03T23:50:28.000+08:00
百度千帆平台上的第三方大模型调用
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

Original file line number	Diff line number	Diff line change
`@@ -36,6 +36,8 @@ func (a Adaptor) GetRequestURL(meta util.RelayMeta) (string, error) {`
`36`	`36`	`fullRequestURL = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/bloomz_7b1"`
`37`	`37`	`case "Embedding-V1":`
`38`	`38`	`fullRequestURL = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/embeddings/embedding-v1"`
	`39`	`+ default:`
	`40`	`+ fullRequestURL = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/" + meta.ActualModelName`
`39`	`41`	`}`
`40`	`42`	`var accessToken string`
`41`	`43`	`var err error`