Explorar o código

fix: update adaptor support mode (#371)

zijiren hai 3 meses
pai
achega
448bfa753e

+ 7 - 0
core/relay/adaptor/deepseek/adaptor.go

@@ -3,6 +3,7 @@ package deepseek
 import (
 	"github.com/labring/aiproxy/core/relay/adaptor"
 	"github.com/labring/aiproxy/core/relay/adaptor/openai"
+	"github.com/labring/aiproxy/core/relay/mode"
 )
 
 var _ adaptor.Adaptor = (*Adaptor)(nil)
@@ -17,6 +18,12 @@ func (a *Adaptor) DefaultBaseURL() string {
 	return baseURL
 }
 
+func (a *Adaptor) SupportMode(m mode.Mode) bool {
+	return m == mode.ChatCompletions ||
+		m == mode.Completions ||
+		m == mode.Anthropic
+}
+
 func (a *Adaptor) Metadata() adaptor.Metadata {
 	return adaptor.Metadata{
 		Models: ModelList,

+ 6 - 0
core/relay/adaptor/doubao/main.go

@@ -77,6 +77,12 @@ func (a *Adaptor) DefaultBaseURL() string {
 	return baseURL
 }
 
+func (a *Adaptor) SupportMode(m mode.Mode) bool {
+	return m == mode.ChatCompletions ||
+		m == mode.Anthropic ||
+		m == mode.Embeddings
+}
+
 func (a *Adaptor) Metadata() adaptor.Metadata {
 	return adaptor.Metadata{
 		Features: []string{

+ 8 - 0
core/relay/adaptor/qianfan/adaptor.go

@@ -8,6 +8,7 @@ import (
 	"github.com/labring/aiproxy/core/relay/adaptor"
 	"github.com/labring/aiproxy/core/relay/adaptor/openai"
 	"github.com/labring/aiproxy/core/relay/meta"
+	"github.com/labring/aiproxy/core/relay/mode"
 )
 
 type Adaptor struct {
@@ -20,6 +21,13 @@ func (a *Adaptor) DefaultBaseURL() string {
 	return baseURL
 }
 
+func (a *Adaptor) SupportMode(m mode.Mode) bool {
+	return m == mode.ChatCompletions ||
+		m == mode.Completions ||
+		m == mode.Anthropic ||
+		m == mode.Embeddings
+}
+
 func (a *Adaptor) DoResponse(
 	meta *meta.Meta,
 	store adaptor.Store,