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

🐛 fix: custom channel error

Browse Source
This commit is contained in:
MartialBE 2024-01-06 16:22:49 +08:00 committed by Buer
parent 47c9630169
commit b3f1b5953c
3 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
providers/azure/base.go
View File

@ -27,6 +27,7 @@ func (f AzureProviderFactory) Create(c *gin.Context) base.ProviderInterface {
// AudioSpeech: "/audio/speech", // AudioSpeech: "/audio/speech",
}, },
IsAzure: true, IsAzure: true,
BalanceAction: false,
}, },
} }
} }

2
providers/openai/base.go
View File

@ -54,7 +54,7 @@ func CreateOpenAIProvider(c *gin.Context, baseURL string) *OpenAIProvider {
Context: c, Context: c,
}, },
IsAzure: false, IsAzure: false,
BalanceAction: false, BalanceAction: true,
} }
} }

3
providers/providers.go
View File

@ -72,8 +72,9 @@ func GetProvider(channel *model.Channel, c *gin.Context) base.ProviderInterface
} }
provider = openai.CreateOpenAIProvider(c, baseURL) provider = openai.CreateOpenAIProvider(c, baseURL)
} } else {
provider = factory.Create(c) provider = factory.Create(c)
}
provider.SetChannel(channel) provider.SetChannel(channel)
return provider return provider