tweak: adjust variant logic for muse (#36079)

This commit is contained in:
Aiden Cline 2026-07-09 09:07:02 -05:00 committed by GitHub
parent 518772c2ba
commit 2e43d41af0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 29 additions and 1 deletions

View file

@ -207,6 +207,13 @@ function custom(dep: CustomDep): Record<string, CustomLoader> {
},
options: { headerTimeout: OPENAI_HEADER_TIMEOUT_DEFAULT },
}),
meta: () =>
Effect.succeed({
autoload: false,
async getModel(sdk: any, modelID: string, _options?: Record<string, any>) {
return sdk.responses(modelID)
},
}),
xai: () =>
Effect.succeed({
autoload: false,

View file

@ -884,6 +884,18 @@ export function variants(model: Provider.Model): Record<string, Record<string, a
)
case "@ai-sdk/amazon-bedrock/mantle":
case "@ai-sdk/openai": {
if (model.providerID === "meta") {
return Object.fromEntries(
OPENAI_EFFORTS.map((effort) => [
effort,
{
reasoningEffort: effort,
reasoningSummary: "auto",
include: INCLUDE_ENCRYPTED_REASONING,
},
]),
)
}
// https://v5.ai-sdk.dev/providers/ai-sdk-providers/openai
const efforts = openaiReasoningEfforts(model.api.id, model.release_date)
return Object.fromEntries(
@ -1128,11 +1140,20 @@ export function options(input: {
if (
input.providerOptions?.setCacheKey !== false &&
(input.model.providerID === "openai" || input.model.api.npm === "@ai-sdk/xai" || input.providerOptions?.setCacheKey)
(input.model.providerID === "openai" ||
input.model.api.npm === "@ai-sdk/openai" ||
input.model.api.npm === "@ai-sdk/xai" ||
input.providerOptions?.setCacheKey)
) {
result["promptCacheKey"] = input.sessionID
}
if (input.model.providerID === "meta" && input.model.api.npm === "@ai-sdk/openai") {
result["reasoningEffort"] = "high"
result["reasoningSummary"] = "auto"
result["include"] = INCLUDE_ENCRYPTED_REASONING
}
if (input.model.api.npm === "@ai-sdk/google" || input.model.api.npm === "@ai-sdk/google-vertex") {
if (input.model.capabilities.reasoning) {
result["thinkingConfig"] = {