From 02f8e7a8579ea5296f8dd43965bf4ced92ffde9d Mon Sep 17 00:00:00 2001
From: MyPrototypeWhat <daoquqiexing@gmail.com>
Date: Thu, 14 Aug 2025 19:31:17 +0800
Subject: [PATCH] feat(aiCore): add enableUrlContext capability and new support
 function

- Enhanced the buildStreamTextParams function to include enableUrlContext in the capabilities object, improving the parameter set for AI interactions.
- Introduced a new isSupportedFlexServiceTier function to streamline model support checks, enhancing code clarity and maintainability.
---
 src/renderer/src/aiCore/transformParameters.ts | 13 +++++++++++--
 src/renderer/src/config/models.ts              |  3 +++
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/src/renderer/src/aiCore/transformParameters.ts b/src/renderer/src/aiCore/transformParameters.ts
index 5a8226e4ee..a16e4a887b 100644
--- a/src/renderer/src/aiCore/transformParameters.ts
+++ b/src/renderer/src/aiCore/transformParameters.ts
@@ -243,7 +243,12 @@ export async function buildStreamTextParams(
 ): Promise<{
   params: StreamTextParams
   modelId: string
-  capabilities: { enableReasoning: boolean; enableWebSearch: boolean; enableGenerateImage: boolean }
+  capabilities: {
+    enableReasoning: boolean
+    enableWebSearch: boolean
+    enableGenerateImage: boolean
+    enableUrlContext: boolean
+  }
 }> {
   const { mcpTools } = options
 
@@ -300,7 +305,11 @@ export async function buildStreamTextParams(
     params.system = assistant.prompt
   }
 
-  return { params, modelId: model.id, capabilities: { enableReasoning, enableWebSearch, enableGenerateImage } }
+  return {
+    params,
+    modelId: model.id,
+    capabilities: { enableReasoning, enableWebSearch, enableGenerateImage, enableUrlContext }
+  }
 }
 
 /**
diff --git a/src/renderer/src/config/models.ts b/src/renderer/src/config/models.ts
index 1947a5d36d..eef1044fb3 100644
--- a/src/renderer/src/config/models.ts
+++ b/src/renderer/src/config/models.ts
@@ -2508,6 +2508,9 @@ export function isSupportFlexServiceTierModel(model: Model): boolean {
     (modelId.includes('o3') && !modelId.includes('o3-mini')) || modelId.includes('o4-mini') || modelId.includes('gpt-5')
   )
 }
+export function isSupportedFlexServiceTier(model: Model): boolean {
+  return isSupportFlexServiceTierModel(model)
+}
 
 export function isSupportVerbosityModel(model: Model): boolean {
   const modelId = getLowerBaseModelName(model.id)