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)