From 9024d489388fa4ee32f87981a06ad5c90231e990 Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Wed, 5 Feb 2025 16:14:35 +0800
Subject: [PATCH] feat: add support for o1 models and update provider
 configurations

---
 src/renderer/src/config/models.ts            | 3 ++-
 src/renderer/src/config/providers.ts         | 2 +-
 src/renderer/src/providers/OpenAIProvider.ts | 4 ++--
 3 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/renderer/src/config/models.ts b/src/renderer/src/config/models.ts
index 69d74aae..7d9f54b8 100644
--- a/src/renderer/src/config/models.ts
+++ b/src/renderer/src/config/models.ts
@@ -144,7 +144,8 @@ const visionAllowedModels = [
   'pixtral',
   'gpt-4(?:-[\\w-]+)',
   'gpt-4o(?:-[\\w-]+)?',
-  'chatgpt-4o(?:-[\\w-]+)?'
+  'chatgpt-4o(?:-[\\w-]+)?',
+  'o1(?:-[\\w-]+)?'
 ]
 
 const visionExcludedModels = ['gpt-4-\\d+-preview', 'gpt-4-turbo-preview', 'gpt-4-32k', 'gpt-4-\\d+']
diff --git a/src/renderer/src/config/providers.ts b/src/renderer/src/config/providers.ts
index 5e313b4e..37719919 100644
--- a/src/renderer/src/config/providers.ts
+++ b/src/renderer/src/config/providers.ts
@@ -358,7 +358,7 @@ export const PROVIDER_CONFIG = {
   },
   aihubmix: {
     api: {
-      url: 'https://aihubmix.com?aff=SJyh'
+      url: 'https://aihubmix.com'
     },
     websites: {
       official: 'https://aihubmix.com?aff=SJyh',
diff --git a/src/renderer/src/providers/OpenAIProvider.ts b/src/renderer/src/providers/OpenAIProvider.ts
index 1b863f29..77330f1f 100644
--- a/src/renderer/src/providers/OpenAIProvider.ts
+++ b/src/renderer/src/providers/OpenAIProvider.ts
@@ -165,7 +165,7 @@ export default class OpenAIProvider extends BaseProvider {
     const isOpenAIo1 = model.id.startsWith('o1')
 
     const isSupportStreamOutput = () => {
-      if (this.provider.id === 'github' && isOpenAIo1) {
+      if (isOpenAIo1) {
         return false
       }
       return streamOutput
@@ -251,7 +251,7 @@ export default class OpenAIProvider extends BaseProvider {
       if (!onResponse) {
         return false
       }
-      if (this.provider.id === 'github' && isOpenAIo1) {
+      if (isOpenAIo1) {
         return false
       }
       return true