Add qwq-32b-preview model to Blackbox provider

XInTheDark · Dec 19, 2024 · 9dde572 · 9dde572
1 parent 58a29ee
commit 9dde572
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -90,11 +90,12 @@ updating manually also allows you to fetch and view the latest changes to the so
 | DeepInfra                    | DeepSeek-V2.5                           | ▶️       | ![Active][active-badge]     | Fast           | 7.5/10                                                                                      |
 | DeepInfra                    | Qwen2.5-72B                             | ▶️       | ![Active][active-badge]     | Medium         | 7.5/10                                                                                      |
 | DeepInfra                    | Qwen2.5-Coder-32B                       | ▶️       | ![Active][active-badge]     | Fast           | 7/10                                                                                        |
-| DeepInfra                    | QwQ-32B-Preview                         | ▶️       | ![Active][active-badge]     | Fast           | 7.5/10                                                                                      |
+| DeepInfra                    | QwQ-32B-Preview                         | ▶️       | ![Active][active-badge]     | Very fast      | 7.5/10                                                                                      |
 | Blackbox                     | custom model                            | ▶️       | ![Active][active-badge]     | Fast           | 7.5/10, very fast generation with built-in web search ability, but is optimized for coding. |
 | Blackbox                     | llama-3.1-405b                          | ▶️       | ![Active][active-badge]     | Fast           | 8.5/10                                                                                      |
 | Blackbox                     | llama-3.1-70b                           | ▶️       | ![Active][active-badge]     | Very fast      | 8/10                                                                                        |
 | Blackbox                     | gemini-1.5-flash                        | ▶️       | ![Active][active-badge]     | Extremely fast | 7.5/10                                                                                      |
+| Blackbox                     | qwq-32b-preview                         | ▶️       | ![Active][active-badge]     | Extremely fast | 6.5/10                                                                                      |
 | Blackbox                     | gpt-4o                                  | ▶️       | ![Active][active-badge]     | Very fast      | 7.5/10                                                                                      |
 | Blackbox                     | claude-3.5-sonnet                       | ▶️       | ![Active][active-badge]     | Fast           | 8.5/10                                                                                      |
 | Blackbox                     | gemini-pro                              | ▶️       | ![Active][active-badge]     | Fast           | 8/10                                                                                        |

diff --git a/package.json b/package.json
@@ -325,6 +325,10 @@
           "title": "Blackbox (gemini-1.5-flash)",
           "value": "BlackboxGemini15Flash"
         },
+        {
+          "title": "Blackbox (qwq-32b-preview)",
+          "value": "BlackboxQwQ32BPreview"
+        },
         {
           "title": "Blackbox (gpt-4o)",
           "value": "BlackboxGPT4o"

diff --git a/src/api/Providers/blackbox.js b/src/api/Providers/blackbox.js
@@ -32,6 +32,10 @@ const uuid4 = function () {
   return randomUUID();
 };
 
+const agentModeConfig = {
+  "qwq-32b-preview": { mode: true, id: "Qwen/QwQ-32B-Preview", name: "Qwen-QwQ-32B-Preview" },
+};
+
 const trendingAgentModeConfig = {
   blackbox: {},
   "llama-3.1-405b": { mode: true, id: "llama-3.1-405b" },
@@ -57,6 +61,9 @@ const paramOverrides = {
   "gemini-pro": {
     maxTokens: 8192,
   },
+  "qwq-32b-preview": {
+    maxTokens: 50000,
+  },
 };
 
 const defaultValidatedToken = Buffer.from("MDBmMzdiMzQtYTE2Ni00ZWZiLWJjZTUtMTMxMmQ4N2YyZjk0", "base64").toString(
@@ -97,7 +104,7 @@ export const BlackboxProvider = {
       userId: random_user_id,
       previewToken: null,
       codeModelMode: true,
-      agentMode: {},
+      agentMode: agentModeConfig[options.model] || {},
       trendingAgentMode: trendingAgentModeConfig[options.model] || {},
       userSelectedModel: userSelectedModelConfig[options.model] || undefined,
       isMicMode: false,

diff --git a/src/api/providers.js b/src/api/providers.js
@@ -51,6 +51,7 @@ export const providers_info = {
   BlackboxLlama31_405B: { provider: BlackboxProvider, model: "llama-3.1-405b", stream: true },
   BlackboxLlama31_70B: { provider: BlackboxProvider, model: "llama-3.1-70b", stream: true },
   BlackboxGemini15Flash: { provider: BlackboxProvider, model: "gemini-1.5-flash", stream: true },
+  BlackboxQwQ32BPreview: { provider: BlackboxProvider, model: "qwq-32b-preview", stream: true },
   BlackboxGPT4o: { provider: BlackboxProvider, model: "gpt-4o", stream: true },
   BlackboxClaude35Sonnet: { provider: BlackboxProvider, model: "claude-3.5-sonnet", stream: true },
   BlackboxGeminiPro: { provider: BlackboxProvider, model: "gemini-pro", stream: true },