mirrors2/opencatd-open
1
0 Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fetch model & add apitype

Browse Source
This commit is contained in:
Sakurasan committed 2025-04-19 01:21:28 +08:00
1 parent f8e539c9b4
commit ca305f4199
5 files changed
+128 -5
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

internal/controller/proxy/proxy.go
+2 -2
View File
@@ -186,7 +186,7 @@ func (p *Proxy) SelectApiKey(model string) error {
akpikeys, err := p.apiKeyDao.FindApiKeysBySupportModel(p.db, model)
if err != nil || len(akpikeys) == 0 {
if strings.HasPrefix(model, "gpt") || strings.HasPrefix(model, "o1") || strings.HasPrefix(model, "o3") {
if strings.HasPrefix(model, "gpt") || strings.HasPrefix(model, "o1") || strings.HasPrefix(model, "o3") || strings.HasPrefix(model, "o4") {
keys, err := p.apiKeyDao.FindKeys(map[string]any{"apitype = ?": "openai"})
if err != nil {
return err
@@ -227,7 +227,7 @@ func (p *Proxy) SelectApiKey(model string) error {
func (p *Proxy) updateSupportModel() {
keys, err := p.apiKeyDao.FindKeys(map[string]interface{}{"apitype in ?": "openai,azure,claude"})
keys, err := p.apiKeyDao.FindKeys(map[string]interface{}{"apitype in ?": []string{"openai", "azure", "claude"}})
if err != nil {
return
}