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.
This commit is contained in:
MyPrototypeWhat 2025-08-14 19:31:17 +08:00
parent 6c093f72d8
commit 02f8e7a857
2 changed files with 14 additions and 2 deletions

View File

@ -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 }
}
}
/**

View File

@ -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)