From 8b5c6feb5e138a4ef6b7745300d99c73c50c2c1e Mon Sep 17 00:00:00 2001
From: Sakurasan <26715255+Sakurasan@users.noreply.github.com>
Date: Wed, 1 Jan 2025 23:12:05 +0800
Subject: [PATCH] add azure o1

---
 store/cache.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/store/cache.go b/store/cache.go
index 37aab6c..5f8376b 100644
--- a/store/cache.go
+++ b/store/cache.go
@@ -82,7 +82,7 @@ func SelectKeyCacheByModel(model string) (Key, error) {
 	}
 	items := KeysCache.Items()
 	for _, item := range items {
-		if strings.Contains(model, "realtime") {
+		if strings.Contains(model, "realtime") || strings.HasPrefix(model, "o1-") {
 			if item.Object.(Key).ApiType == "openai" {
 				keys = append(keys, item.Object.(Key))
 			}
@@ -101,7 +101,7 @@ func SelectKeyCacheByModel(model string) (Key, error) {
 				keys = append(keys, item.Object.(Key))
 			}
 		}
-		if strings.HasPrefix(model, "o1-") || strings.HasPrefix(model, "chatgpt-") {
+		if strings.HasPrefix(model, "chatgpt-") {
 			if item.Object.(Key).ApiType == "openai" {
 				keys = append(keys, item.Object.(Key))
 			}