From 20c3cda4a710f6d05a3c415d7b394ee9b075c205 Mon Sep 17 00:00:00 2001
From: Sakurasan <26715255+Sakurasan@users.noreply.github.com>
Date: Thu, 17 Apr 2025 21:02:00 +0800
Subject: [PATCH] fix bugs

---
 internal/controller/proxy/proxy.go | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/internal/controller/proxy/proxy.go b/internal/controller/proxy/proxy.go
index 96db855..fb0c36d 100644
--- a/internal/controller/proxy/proxy.go
+++ b/internal/controller/proxy/proxy.go
@@ -187,7 +187,7 @@ func (p *Proxy) SelectApiKey(model string) error {
 
 	if err != nil || len(akpikeys) == 0 {
 		if strings.HasPrefix(model, "gpt") || strings.HasPrefix(model, "o1") || strings.HasPrefix(model, "o3") {
-			keys, err := p.apiKeyDao.FindKeys(map[string]any{"type = ?": "openai"})
+			keys, err := p.apiKeyDao.FindKeys(map[string]any{"apitype = ?": "openai"})
 			if err != nil {
 				return err
 			}
@@ -195,7 +195,7 @@ func (p *Proxy) SelectApiKey(model string) error {
 		}
 
 		if strings.HasPrefix(model, "gemini") {
-			keys, err := p.apiKeyDao.FindKeys(map[string]any{"type = ?": "gemini"})
+			keys, err := p.apiKeyDao.FindKeys(map[string]any{"apitype = ?": "gemini"})
 			if err != nil {
 				return err
 			}
@@ -203,7 +203,7 @@ func (p *Proxy) SelectApiKey(model string) error {
 		}
 
 		if strings.HasPrefix(model, "claude") {
-			keys, err := p.apiKeyDao.FindKeys(map[string]any{"type = ?": "claude"})
+			keys, err := p.apiKeyDao.FindKeys(map[string]any{"apitype = ?": "claude"})
 			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{}{"type in ?": "openai,azure,claude"})
+	keys, err := p.apiKeyDao.FindKeys(map[string]interface{}{"apitype in ?": "openai,azure,claude"})
 	if err != nil {
 		return
 	}