| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930 |
- import { describe, expect, test } from "bun:test"
- import { ProviderTransform } from "../../src/provider/transform"
- const OUTPUT_TOKEN_MAX = 32000
- describe("ProviderTransform.options - setCacheKey", () => {
- const sessionID = "test-session-123"
- const mockModel = {
- id: "anthropic/claude-3-5-sonnet",
- providerID: "anthropic",
- api: {
- id: "claude-3-5-sonnet-20241022",
- url: "https://api.anthropic.com",
- npm: "@ai-sdk/anthropic",
- },
- name: "Claude 3.5 Sonnet",
- capabilities: {
- temperature: true,
- reasoning: false,
- attachment: true,
- toolcall: true,
- input: { text: true, audio: false, image: true, video: false, pdf: true },
- output: { text: true, audio: false, image: false, video: false, pdf: false },
- interleaved: false,
- },
- cost: {
- input: 0.003,
- output: 0.015,
- cache: { read: 0.0003, write: 0.00375 },
- },
- limit: {
- context: 200000,
- output: 8192,
- },
- status: "active",
- options: {},
- headers: {},
- } as any
- test("should set promptCacheKey when providerOptions.setCacheKey is true", () => {
- const result = ProviderTransform.options({
- model: mockModel,
- sessionID,
- providerOptions: { setCacheKey: true },
- })
- expect(result.promptCacheKey).toBe(sessionID)
- })
- test("should not set promptCacheKey when providerOptions.setCacheKey is false", () => {
- const result = ProviderTransform.options({
- model: mockModel,
- sessionID,
- providerOptions: { setCacheKey: false },
- })
- expect(result.promptCacheKey).toBeUndefined()
- })
- test("should not set promptCacheKey when providerOptions is undefined", () => {
- const result = ProviderTransform.options({
- model: mockModel,
- sessionID,
- providerOptions: undefined,
- })
- expect(result.promptCacheKey).toBeUndefined()
- })
- test("should not set promptCacheKey when providerOptions does not have setCacheKey", () => {
- const result = ProviderTransform.options({ model: mockModel, sessionID, providerOptions: {} })
- expect(result.promptCacheKey).toBeUndefined()
- })
- test("should set promptCacheKey for openai provider regardless of setCacheKey", () => {
- const openaiModel = {
- ...mockModel,
- providerID: "openai",
- api: {
- id: "gpt-4",
- url: "https://api.openai.com",
- npm: "@ai-sdk/openai",
- },
- }
- const result = ProviderTransform.options({ model: openaiModel, sessionID, providerOptions: {} })
- expect(result.promptCacheKey).toBe(sessionID)
- })
- test("should set store=false for openai provider", () => {
- const openaiModel = {
- ...mockModel,
- providerID: "openai",
- api: {
- id: "gpt-4",
- url: "https://api.openai.com",
- npm: "@ai-sdk/openai",
- },
- }
- const result = ProviderTransform.options({
- model: openaiModel,
- sessionID,
- providerOptions: {},
- })
- expect(result.store).toBe(false)
- })
- })
- describe("ProviderTransform.options - gpt-5 textVerbosity", () => {
- const sessionID = "test-session-123"
- const createGpt5Model = (apiId: string) =>
- ({
- id: `openai/${apiId}`,
- providerID: "openai",
- api: {
- id: apiId,
- url: "https://api.openai.com",
- npm: "@ai-sdk/openai",
- },
- name: apiId,
- capabilities: {
- temperature: true,
- reasoning: true,
- attachment: true,
- toolcall: true,
- input: { text: true, audio: false, image: true, video: false, pdf: false },
- output: { text: true, audio: false, image: false, video: false, pdf: false },
- interleaved: false,
- },
- cost: { input: 0.03, output: 0.06, cache: { read: 0.001, write: 0.002 } },
- limit: { context: 128000, output: 4096 },
- status: "active",
- options: {},
- headers: {},
- }) as any
- test("gpt-5.2 should have textVerbosity set to low", () => {
- const model = createGpt5Model("gpt-5.2")
- const result = ProviderTransform.options({ model, sessionID, providerOptions: {} })
- expect(result.textVerbosity).toBe("low")
- })
- test("gpt-5.1 should have textVerbosity set to low", () => {
- const model = createGpt5Model("gpt-5.1")
- const result = ProviderTransform.options({ model, sessionID, providerOptions: {} })
- expect(result.textVerbosity).toBe("low")
- })
- test("gpt-5.2-chat-latest should NOT have textVerbosity set (only supports medium)", () => {
- const model = createGpt5Model("gpt-5.2-chat-latest")
- const result = ProviderTransform.options({ model, sessionID, providerOptions: {} })
- expect(result.textVerbosity).toBeUndefined()
- })
- test("gpt-5.1-chat-latest should NOT have textVerbosity set (only supports medium)", () => {
- const model = createGpt5Model("gpt-5.1-chat-latest")
- const result = ProviderTransform.options({ model, sessionID, providerOptions: {} })
- expect(result.textVerbosity).toBeUndefined()
- })
- test("gpt-5.2-chat should NOT have textVerbosity set", () => {
- const model = createGpt5Model("gpt-5.2-chat")
- const result = ProviderTransform.options({ model, sessionID, providerOptions: {} })
- expect(result.textVerbosity).toBeUndefined()
- })
- test("gpt-5-chat should NOT have textVerbosity set", () => {
- const model = createGpt5Model("gpt-5-chat")
- const result = ProviderTransform.options({ model, sessionID, providerOptions: {} })
- expect(result.textVerbosity).toBeUndefined()
- })
- test("gpt-5.2-codex should NOT have textVerbosity set (codex models excluded)", () => {
- const model = createGpt5Model("gpt-5.2-codex")
- const result = ProviderTransform.options({ model, sessionID, providerOptions: {} })
- expect(result.textVerbosity).toBeUndefined()
- })
- })
- describe("ProviderTransform.maxOutputTokens", () => {
- test("returns 32k when modelLimit > 32k", () => {
- const modelLimit = 100000
- const result = ProviderTransform.maxOutputTokens("@ai-sdk/openai", {}, modelLimit, OUTPUT_TOKEN_MAX)
- expect(result).toBe(OUTPUT_TOKEN_MAX)
- })
- test("returns modelLimit when modelLimit < 32k", () => {
- const modelLimit = 16000
- const result = ProviderTransform.maxOutputTokens("@ai-sdk/openai", {}, modelLimit, OUTPUT_TOKEN_MAX)
- expect(result).toBe(16000)
- })
- describe("azure", () => {
- test("returns 32k when modelLimit > 32k", () => {
- const modelLimit = 100000
- const result = ProviderTransform.maxOutputTokens("@ai-sdk/azure", {}, modelLimit, OUTPUT_TOKEN_MAX)
- expect(result).toBe(OUTPUT_TOKEN_MAX)
- })
- test("returns modelLimit when modelLimit < 32k", () => {
- const modelLimit = 16000
- const result = ProviderTransform.maxOutputTokens("@ai-sdk/azure", {}, modelLimit, OUTPUT_TOKEN_MAX)
- expect(result).toBe(16000)
- })
- })
- describe("bedrock", () => {
- test("returns 32k when modelLimit > 32k", () => {
- const modelLimit = 100000
- const result = ProviderTransform.maxOutputTokens("@ai-sdk/amazon-bedrock", {}, modelLimit, OUTPUT_TOKEN_MAX)
- expect(result).toBe(OUTPUT_TOKEN_MAX)
- })
- test("returns modelLimit when modelLimit < 32k", () => {
- const modelLimit = 16000
- const result = ProviderTransform.maxOutputTokens("@ai-sdk/amazon-bedrock", {}, modelLimit, OUTPUT_TOKEN_MAX)
- expect(result).toBe(16000)
- })
- })
- describe("anthropic without thinking options", () => {
- test("returns 32k when modelLimit > 32k", () => {
- const modelLimit = 100000
- const result = ProviderTransform.maxOutputTokens("@ai-sdk/anthropic", {}, modelLimit, OUTPUT_TOKEN_MAX)
- expect(result).toBe(OUTPUT_TOKEN_MAX)
- })
- test("returns modelLimit when modelLimit < 32k", () => {
- const modelLimit = 16000
- const result = ProviderTransform.maxOutputTokens("@ai-sdk/anthropic", {}, modelLimit, OUTPUT_TOKEN_MAX)
- expect(result).toBe(16000)
- })
- })
- describe("anthropic with thinking options", () => {
- test("returns 32k when budgetTokens + 32k <= modelLimit", () => {
- const modelLimit = 100000
- const options = {
- thinking: {
- type: "enabled",
- budgetTokens: 10000,
- },
- }
- const result = ProviderTransform.maxOutputTokens("@ai-sdk/anthropic", options, modelLimit, OUTPUT_TOKEN_MAX)
- expect(result).toBe(OUTPUT_TOKEN_MAX)
- })
- test("returns modelLimit - budgetTokens when budgetTokens + 32k > modelLimit", () => {
- const modelLimit = 50000
- const options = {
- thinking: {
- type: "enabled",
- budgetTokens: 30000,
- },
- }
- const result = ProviderTransform.maxOutputTokens("@ai-sdk/anthropic", options, modelLimit, OUTPUT_TOKEN_MAX)
- expect(result).toBe(20000)
- })
- test("returns 32k when thinking type is not enabled", () => {
- const modelLimit = 100000
- const options = {
- thinking: {
- type: "disabled",
- budgetTokens: 10000,
- },
- }
- const result = ProviderTransform.maxOutputTokens("@ai-sdk/anthropic", options, modelLimit, OUTPUT_TOKEN_MAX)
- expect(result).toBe(OUTPUT_TOKEN_MAX)
- })
- })
- describe("openai-compatible with thinking options (snake_case)", () => {
- test("returns 32k when budget_tokens + 32k <= modelLimit", () => {
- const modelLimit = 100000
- const options = {
- thinking: {
- type: "enabled",
- budget_tokens: 10000,
- },
- }
- const result = ProviderTransform.maxOutputTokens(
- "@ai-sdk/openai-compatible",
- options,
- modelLimit,
- OUTPUT_TOKEN_MAX,
- )
- expect(result).toBe(OUTPUT_TOKEN_MAX)
- })
- test("returns modelLimit - budget_tokens when budget_tokens + 32k > modelLimit", () => {
- const modelLimit = 50000
- const options = {
- thinking: {
- type: "enabled",
- budget_tokens: 30000,
- },
- }
- const result = ProviderTransform.maxOutputTokens(
- "@ai-sdk/openai-compatible",
- options,
- modelLimit,
- OUTPUT_TOKEN_MAX,
- )
- expect(result).toBe(20000)
- })
- test("returns 32k when thinking type is not enabled", () => {
- const modelLimit = 100000
- const options = {
- thinking: {
- type: "disabled",
- budget_tokens: 10000,
- },
- }
- const result = ProviderTransform.maxOutputTokens(
- "@ai-sdk/openai-compatible",
- options,
- modelLimit,
- OUTPUT_TOKEN_MAX,
- )
- expect(result).toBe(OUTPUT_TOKEN_MAX)
- })
- test("returns 32k when budget_tokens is 0", () => {
- const modelLimit = 100000
- const options = {
- thinking: {
- type: "enabled",
- budget_tokens: 0,
- },
- }
- const result = ProviderTransform.maxOutputTokens(
- "@ai-sdk/openai-compatible",
- options,
- modelLimit,
- OUTPUT_TOKEN_MAX,
- )
- expect(result).toBe(OUTPUT_TOKEN_MAX)
- })
- })
- })
- describe("ProviderTransform.schema - gemini array items", () => {
- test("adds missing items for array properties", () => {
- const geminiModel = {
- providerID: "google",
- api: {
- id: "gemini-3-pro",
- },
- } as any
- const schema = {
- type: "object",
- properties: {
- nodes: { type: "array" },
- edges: { type: "array", items: { type: "string" } },
- },
- } as any
- const result = ProviderTransform.schema(geminiModel, schema) as any
- expect(result.properties.nodes.items).toBeDefined()
- expect(result.properties.edges.items.type).toBe("string")
- })
- })
- describe("ProviderTransform.message - DeepSeek reasoning content", () => {
- test("DeepSeek with tool calls includes reasoning_content in providerOptions", () => {
- const msgs = [
- {
- role: "assistant",
- content: [
- { type: "reasoning", text: "Let me think about this..." },
- {
- type: "tool-call",
- toolCallId: "test",
- toolName: "bash",
- input: { command: "echo hello" },
- },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(
- msgs,
- {
- id: "deepseek/deepseek-chat",
- providerID: "deepseek",
- api: {
- id: "deepseek-chat",
- url: "https://api.deepseek.com",
- npm: "@ai-sdk/openai-compatible",
- },
- name: "DeepSeek Chat",
- capabilities: {
- temperature: true,
- reasoning: true,
- attachment: false,
- toolcall: true,
- input: { text: true, audio: false, image: false, video: false, pdf: false },
- output: { text: true, audio: false, image: false, video: false, pdf: false },
- interleaved: {
- field: "reasoning_content",
- },
- },
- cost: {
- input: 0.001,
- output: 0.002,
- cache: { read: 0.0001, write: 0.0002 },
- },
- limit: {
- context: 128000,
- output: 8192,
- },
- status: "active",
- options: {},
- headers: {},
- release_date: "2023-04-01",
- },
- {},
- )
- expect(result).toHaveLength(1)
- expect(result[0].content).toEqual([
- {
- type: "tool-call",
- toolCallId: "test",
- toolName: "bash",
- input: { command: "echo hello" },
- },
- ])
- expect(result[0].providerOptions?.openaiCompatible?.reasoning_content).toBe("Let me think about this...")
- })
- test("Non-DeepSeek providers leave reasoning content unchanged", () => {
- const msgs = [
- {
- role: "assistant",
- content: [
- { type: "reasoning", text: "Should not be processed" },
- { type: "text", text: "Answer" },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(
- msgs,
- {
- id: "openai/gpt-4",
- providerID: "openai",
- api: {
- id: "gpt-4",
- url: "https://api.openai.com",
- npm: "@ai-sdk/openai",
- },
- name: "GPT-4",
- capabilities: {
- temperature: true,
- reasoning: false,
- attachment: true,
- toolcall: true,
- input: { text: true, audio: false, image: true, video: false, pdf: false },
- output: { text: true, audio: false, image: false, video: false, pdf: false },
- interleaved: false,
- },
- cost: {
- input: 0.03,
- output: 0.06,
- cache: { read: 0.001, write: 0.002 },
- },
- limit: {
- context: 128000,
- output: 4096,
- },
- status: "active",
- options: {},
- headers: {},
- release_date: "2023-04-01",
- },
- {},
- )
- expect(result[0].content).toEqual([
- { type: "reasoning", text: "Should not be processed" },
- { type: "text", text: "Answer" },
- ])
- expect(result[0].providerOptions?.openaiCompatible?.reasoning_content).toBeUndefined()
- })
- })
- describe("ProviderTransform.message - empty image handling", () => {
- const mockModel = {
- id: "anthropic/claude-3-5-sonnet",
- providerID: "anthropic",
- api: {
- id: "claude-3-5-sonnet-20241022",
- url: "https://api.anthropic.com",
- npm: "@ai-sdk/anthropic",
- },
- name: "Claude 3.5 Sonnet",
- capabilities: {
- temperature: true,
- reasoning: false,
- attachment: true,
- toolcall: true,
- input: { text: true, audio: false, image: true, video: false, pdf: true },
- output: { text: true, audio: false, image: false, video: false, pdf: false },
- interleaved: false,
- },
- cost: {
- input: 0.003,
- output: 0.015,
- cache: { read: 0.0003, write: 0.00375 },
- },
- limit: {
- context: 200000,
- output: 8192,
- },
- status: "active",
- options: {},
- headers: {},
- } as any
- test("should replace empty base64 image with error text", () => {
- const msgs = [
- {
- role: "user",
- content: [
- { type: "text", text: "What is in this image?" },
- { type: "image", image: "data:image/png;base64," },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, mockModel, {})
- expect(result).toHaveLength(1)
- expect(result[0].content).toHaveLength(2)
- expect(result[0].content[0]).toEqual({ type: "text", text: "What is in this image?" })
- expect(result[0].content[1]).toEqual({
- type: "text",
- text: "ERROR: Image file is empty or corrupted. Please provide a valid image.",
- })
- })
- test("should keep valid base64 images unchanged", () => {
- const validBase64 =
- "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNk+M9QDwADhgGAWjR9awAAAABJRU5ErkJggg=="
- const msgs = [
- {
- role: "user",
- content: [
- { type: "text", text: "What is in this image?" },
- { type: "image", image: `data:image/png;base64,${validBase64}` },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, mockModel, {})
- expect(result).toHaveLength(1)
- expect(result[0].content).toHaveLength(2)
- expect(result[0].content[0]).toEqual({ type: "text", text: "What is in this image?" })
- expect(result[0].content[1]).toEqual({ type: "image", image: `data:image/png;base64,${validBase64}` })
- })
- test("should handle mixed valid and empty images", () => {
- const validBase64 =
- "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNk+M9QDwADhgGAWjR9awAAAABJRU5ErkJggg=="
- const msgs = [
- {
- role: "user",
- content: [
- { type: "text", text: "Compare these images" },
- { type: "image", image: `data:image/png;base64,${validBase64}` },
- { type: "image", image: "data:image/jpeg;base64," },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, mockModel, {})
- expect(result).toHaveLength(1)
- expect(result[0].content).toHaveLength(3)
- expect(result[0].content[0]).toEqual({ type: "text", text: "Compare these images" })
- expect(result[0].content[1]).toEqual({ type: "image", image: `data:image/png;base64,${validBase64}` })
- expect(result[0].content[2]).toEqual({
- type: "text",
- text: "ERROR: Image file is empty or corrupted. Please provide a valid image.",
- })
- })
- })
- describe("ProviderTransform.message - anthropic empty content filtering", () => {
- const anthropicModel = {
- id: "anthropic/claude-3-5-sonnet",
- providerID: "anthropic",
- api: {
- id: "claude-3-5-sonnet-20241022",
- url: "https://api.anthropic.com",
- npm: "@ai-sdk/anthropic",
- },
- name: "Claude 3.5 Sonnet",
- capabilities: {
- temperature: true,
- reasoning: false,
- attachment: true,
- toolcall: true,
- input: { text: true, audio: false, image: true, video: false, pdf: true },
- output: { text: true, audio: false, image: false, video: false, pdf: false },
- interleaved: false,
- },
- cost: {
- input: 0.003,
- output: 0.015,
- cache: { read: 0.0003, write: 0.00375 },
- },
- limit: {
- context: 200000,
- output: 8192,
- },
- status: "active",
- options: {},
- headers: {},
- } as any
- test("filters out messages with empty string content", () => {
- const msgs = [
- { role: "user", content: "Hello" },
- { role: "assistant", content: "" },
- { role: "user", content: "World" },
- ] as any[]
- const result = ProviderTransform.message(msgs, anthropicModel, {})
- expect(result).toHaveLength(2)
- expect(result[0].content).toBe("Hello")
- expect(result[1].content).toBe("World")
- })
- test("filters out empty text parts from array content", () => {
- const msgs = [
- {
- role: "assistant",
- content: [
- { type: "text", text: "" },
- { type: "text", text: "Hello" },
- { type: "text", text: "" },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, anthropicModel, {})
- expect(result).toHaveLength(1)
- expect(result[0].content).toHaveLength(1)
- expect(result[0].content[0]).toEqual({ type: "text", text: "Hello" })
- })
- test("filters out empty reasoning parts from array content", () => {
- const msgs = [
- {
- role: "assistant",
- content: [
- { type: "reasoning", text: "" },
- { type: "text", text: "Answer" },
- { type: "reasoning", text: "" },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, anthropicModel, {})
- expect(result).toHaveLength(1)
- expect(result[0].content).toHaveLength(1)
- expect(result[0].content[0]).toEqual({ type: "text", text: "Answer" })
- })
- test("removes entire message when all parts are empty", () => {
- const msgs = [
- { role: "user", content: "Hello" },
- {
- role: "assistant",
- content: [
- { type: "text", text: "" },
- { type: "reasoning", text: "" },
- ],
- },
- { role: "user", content: "World" },
- ] as any[]
- const result = ProviderTransform.message(msgs, anthropicModel, {})
- expect(result).toHaveLength(2)
- expect(result[0].content).toBe("Hello")
- expect(result[1].content).toBe("World")
- })
- test("keeps non-text/reasoning parts even if text parts are empty", () => {
- const msgs = [
- {
- role: "assistant",
- content: [
- { type: "text", text: "" },
- { type: "tool-call", toolCallId: "123", toolName: "bash", input: { command: "ls" } },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, anthropicModel, {})
- expect(result).toHaveLength(1)
- expect(result[0].content).toHaveLength(1)
- expect(result[0].content[0]).toEqual({
- type: "tool-call",
- toolCallId: "123",
- toolName: "bash",
- input: { command: "ls" },
- })
- })
- test("keeps messages with valid text alongside empty parts", () => {
- const msgs = [
- {
- role: "assistant",
- content: [
- { type: "reasoning", text: "Thinking..." },
- { type: "text", text: "" },
- { type: "text", text: "Result" },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, anthropicModel, {})
- expect(result).toHaveLength(1)
- expect(result[0].content).toHaveLength(2)
- expect(result[0].content[0]).toEqual({ type: "reasoning", text: "Thinking..." })
- expect(result[0].content[1]).toEqual({ type: "text", text: "Result" })
- })
- test("does not filter for non-anthropic providers", () => {
- const openaiModel = {
- ...anthropicModel,
- providerID: "openai",
- api: {
- id: "gpt-4",
- url: "https://api.openai.com",
- npm: "@ai-sdk/openai",
- },
- }
- const msgs = [
- { role: "assistant", content: "" },
- {
- role: "assistant",
- content: [{ type: "text", text: "" }],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, openaiModel, {})
- expect(result).toHaveLength(2)
- expect(result[0].content).toBe("")
- expect(result[1].content).toHaveLength(1)
- })
- })
- describe("ProviderTransform.message - strip openai metadata when store=false", () => {
- const openaiModel = {
- id: "openai/gpt-5",
- providerID: "openai",
- api: {
- id: "gpt-5",
- url: "https://api.openai.com",
- npm: "@ai-sdk/openai",
- },
- name: "GPT-5",
- capabilities: {
- temperature: true,
- reasoning: true,
- attachment: true,
- toolcall: true,
- input: { text: true, audio: false, image: true, video: false, pdf: false },
- output: { text: true, audio: false, image: false, video: false, pdf: false },
- interleaved: false,
- },
- cost: { input: 0.03, output: 0.06, cache: { read: 0.001, write: 0.002 } },
- limit: { context: 128000, output: 4096 },
- status: "active",
- options: {},
- headers: {},
- } as any
- test("preserves itemId and reasoningEncryptedContent when store=false", () => {
- const msgs = [
- {
- role: "assistant",
- content: [
- {
- type: "reasoning",
- text: "thinking...",
- providerOptions: {
- openai: {
- itemId: "rs_123",
- reasoningEncryptedContent: "encrypted",
- },
- },
- },
- {
- type: "text",
- text: "Hello",
- providerOptions: {
- openai: {
- itemId: "msg_456",
- },
- },
- },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, openaiModel, { store: false }) as any[]
- expect(result).toHaveLength(1)
- expect(result[0].content[0].providerOptions?.openai?.itemId).toBe("rs_123")
- expect(result[0].content[1].providerOptions?.openai?.itemId).toBe("msg_456")
- })
- test("preserves itemId and reasoningEncryptedContent when store=false even when not openai", () => {
- const zenModel = {
- ...openaiModel,
- providerID: "zen",
- }
- const msgs = [
- {
- role: "assistant",
- content: [
- {
- type: "reasoning",
- text: "thinking...",
- providerOptions: {
- openai: {
- itemId: "rs_123",
- reasoningEncryptedContent: "encrypted",
- },
- },
- },
- {
- type: "text",
- text: "Hello",
- providerOptions: {
- openai: {
- itemId: "msg_456",
- },
- },
- },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, zenModel, { store: false }) as any[]
- expect(result).toHaveLength(1)
- expect(result[0].content[0].providerOptions?.openai?.itemId).toBe("rs_123")
- expect(result[0].content[1].providerOptions?.openai?.itemId).toBe("msg_456")
- })
- test("preserves other openai options including itemId", () => {
- const msgs = [
- {
- role: "assistant",
- content: [
- {
- type: "text",
- text: "Hello",
- providerOptions: {
- openai: {
- itemId: "msg_123",
- otherOption: "value",
- },
- },
- },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, openaiModel, { store: false }) as any[]
- expect(result[0].content[0].providerOptions?.openai?.itemId).toBe("msg_123")
- expect(result[0].content[0].providerOptions?.openai?.otherOption).toBe("value")
- })
- test("preserves metadata for openai package when store is true", () => {
- const msgs = [
- {
- role: "assistant",
- content: [
- {
- type: "text",
- text: "Hello",
- providerOptions: {
- openai: {
- itemId: "msg_123",
- },
- },
- },
- ],
- },
- ] as any[]
- // openai package preserves itemId regardless of store value
- const result = ProviderTransform.message(msgs, openaiModel, { store: true }) as any[]
- expect(result[0].content[0].providerOptions?.openai?.itemId).toBe("msg_123")
- })
- test("preserves metadata for non-openai packages when store is false", () => {
- const anthropicModel = {
- ...openaiModel,
- providerID: "anthropic",
- api: {
- id: "claude-3",
- url: "https://api.anthropic.com",
- npm: "@ai-sdk/anthropic",
- },
- }
- const msgs = [
- {
- role: "assistant",
- content: [
- {
- type: "text",
- text: "Hello",
- providerOptions: {
- openai: {
- itemId: "msg_123",
- },
- },
- },
- ],
- },
- ] as any[]
- // store=false preserves metadata for non-openai packages
- const result = ProviderTransform.message(msgs, anthropicModel, { store: false }) as any[]
- expect(result[0].content[0].providerOptions?.openai?.itemId).toBe("msg_123")
- })
- test("preserves metadata using providerID key when store is false", () => {
- const opencodeModel = {
- ...openaiModel,
- providerID: "opencode",
- api: {
- id: "opencode-test",
- url: "https://api.opencode.ai",
- npm: "@ai-sdk/openai-compatible",
- },
- }
- const msgs = [
- {
- role: "assistant",
- content: [
- {
- type: "text",
- text: "Hello",
- providerOptions: {
- opencode: {
- itemId: "msg_123",
- otherOption: "value",
- },
- },
- },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, opencodeModel, { store: false }) as any[]
- expect(result[0].content[0].providerOptions?.opencode?.itemId).toBe("msg_123")
- expect(result[0].content[0].providerOptions?.opencode?.otherOption).toBe("value")
- })
- test("preserves itemId across all providerOptions keys", () => {
- const opencodeModel = {
- ...openaiModel,
- providerID: "opencode",
- api: {
- id: "opencode-test",
- url: "https://api.opencode.ai",
- npm: "@ai-sdk/openai-compatible",
- },
- }
- const msgs = [
- {
- role: "assistant",
- providerOptions: {
- openai: { itemId: "msg_root" },
- opencode: { itemId: "msg_opencode" },
- extra: { itemId: "msg_extra" },
- },
- content: [
- {
- type: "text",
- text: "Hello",
- providerOptions: {
- openai: { itemId: "msg_openai_part" },
- opencode: { itemId: "msg_opencode_part" },
- extra: { itemId: "msg_extra_part" },
- },
- },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, opencodeModel, { store: false }) as any[]
- expect(result[0].providerOptions?.openai?.itemId).toBe("msg_root")
- expect(result[0].providerOptions?.opencode?.itemId).toBe("msg_opencode")
- expect(result[0].providerOptions?.extra?.itemId).toBe("msg_extra")
- expect(result[0].content[0].providerOptions?.openai?.itemId).toBe("msg_openai_part")
- expect(result[0].content[0].providerOptions?.opencode?.itemId).toBe("msg_opencode_part")
- expect(result[0].content[0].providerOptions?.extra?.itemId).toBe("msg_extra_part")
- })
- test("does not strip metadata for non-openai packages when store is not false", () => {
- const anthropicModel = {
- ...openaiModel,
- providerID: "anthropic",
- api: {
- id: "claude-3",
- url: "https://api.anthropic.com",
- npm: "@ai-sdk/anthropic",
- },
- }
- const msgs = [
- {
- role: "assistant",
- content: [
- {
- type: "text",
- text: "Hello",
- providerOptions: {
- openai: {
- itemId: "msg_123",
- },
- },
- },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, anthropicModel, {}) as any[]
- expect(result[0].content[0].providerOptions?.openai?.itemId).toBe("msg_123")
- })
- })
- describe("ProviderTransform.message - providerOptions key remapping", () => {
- const createModel = (providerID: string, npm: string) =>
- ({
- id: `${providerID}/test-model`,
- providerID,
- api: {
- id: "test-model",
- url: "https://api.test.com",
- npm,
- },
- name: "Test Model",
- capabilities: {
- temperature: true,
- reasoning: false,
- attachment: true,
- toolcall: true,
- input: { text: true, audio: false, image: true, video: false, pdf: true },
- output: { text: true, audio: false, image: false, video: false, pdf: false },
- interleaved: false,
- },
- cost: { input: 0.001, output: 0.002, cache: { read: 0.0001, write: 0.0002 } },
- limit: { context: 128000, output: 8192 },
- status: "active",
- options: {},
- headers: {},
- }) as any
- test("azure keeps 'azure' key and does not remap to 'openai'", () => {
- const model = createModel("azure", "@ai-sdk/azure")
- const msgs = [
- {
- role: "user",
- content: "Hello",
- providerOptions: {
- azure: { someOption: "value" },
- },
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, model, {})
- expect(result[0].providerOptions?.azure).toEqual({ someOption: "value" })
- expect(result[0].providerOptions?.openai).toBeUndefined()
- })
- test("openai with github-copilot npm remaps providerID to 'openai'", () => {
- const model = createModel("github-copilot", "@ai-sdk/github-copilot")
- const msgs = [
- {
- role: "user",
- content: "Hello",
- providerOptions: {
- "github-copilot": { someOption: "value" },
- },
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, model, {})
- expect(result[0].providerOptions?.openai).toEqual({ someOption: "value" })
- expect(result[0].providerOptions?.["github-copilot"]).toBeUndefined()
- })
- test("bedrock remaps providerID to 'bedrock' key", () => {
- const model = createModel("my-bedrock", "@ai-sdk/amazon-bedrock")
- const msgs = [
- {
- role: "user",
- content: "Hello",
- providerOptions: {
- "my-bedrock": { someOption: "value" },
- },
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, model, {})
- expect(result[0].providerOptions?.bedrock).toEqual({ someOption: "value" })
- expect(result[0].providerOptions?.["my-bedrock"]).toBeUndefined()
- })
- })
- describe("ProviderTransform.message - claude w/bedrock custom inference profile", () => {
- test("adds cachePoint", () => {
- const model = {
- id: "amazon-bedrock/custom-claude-sonnet-4.5",
- providerID: "amazon-bedrock",
- api: {
- id: "arn:aws:bedrock:xxx:yyy:application-inference-profile/zzz",
- url: "https://api.test.com",
- npm: "@ai-sdk/amazon-bedrock",
- },
- name: "Custom inference profile",
- capabilities: {},
- options: {},
- headers: {},
- } as any
- const msgs = [
- {
- role: "user",
- content: "Hello",
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, model, {})
- expect(result[0].providerOptions?.bedrock).toEqual(
- expect.objectContaining({
- cachePoint: {
- type: "ephemeral",
- },
- }),
- )
- })
- })
- describe("ProviderTransform.variants", () => {
- const createMockModel = (overrides: Partial<any> = {}): any => ({
- id: "test/test-model",
- providerID: "test",
- api: {
- id: "test-model",
- url: "https://api.test.com",
- npm: "@ai-sdk/openai",
- },
- name: "Test Model",
- capabilities: {
- temperature: true,
- reasoning: true,
- attachment: true,
- toolcall: true,
- input: { text: true, audio: false, image: true, video: false, pdf: false },
- output: { text: true, audio: false, image: false, video: false, pdf: false },
- interleaved: false,
- },
- cost: {
- input: 0.001,
- output: 0.002,
- cache: { read: 0.0001, write: 0.0002 },
- },
- limit: {
- context: 200_000,
- output: 64_000,
- },
- status: "active",
- options: {},
- headers: {},
- release_date: "2024-01-01",
- ...overrides,
- })
- test("returns empty object when model has no reasoning capabilities", () => {
- const model = createMockModel({
- capabilities: { reasoning: false },
- })
- const result = ProviderTransform.variants(model)
- expect(result).toEqual({})
- })
- test("deepseek returns empty object", () => {
- const model = createMockModel({
- id: "deepseek/deepseek-chat",
- providerID: "deepseek",
- api: {
- id: "deepseek-chat",
- url: "https://api.deepseek.com",
- npm: "@ai-sdk/openai-compatible",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(result).toEqual({})
- })
- test("minimax returns empty object", () => {
- const model = createMockModel({
- id: "minimax/minimax-model",
- providerID: "minimax",
- api: {
- id: "minimax-model",
- url: "https://api.minimax.com",
- npm: "@ai-sdk/openai-compatible",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(result).toEqual({})
- })
- test("glm returns empty object", () => {
- const model = createMockModel({
- id: "glm/glm-4",
- providerID: "glm",
- api: {
- id: "glm-4",
- url: "https://api.glm.com",
- npm: "@ai-sdk/openai-compatible",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(result).toEqual({})
- })
- test("mistral returns empty object", () => {
- const model = createMockModel({
- id: "mistral/mistral-large",
- providerID: "mistral",
- api: {
- id: "mistral-large-latest",
- url: "https://api.mistral.com",
- npm: "@ai-sdk/mistral",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(result).toEqual({})
- })
- describe("@openrouter/ai-sdk-provider", () => {
- test("returns empty object for non-qualifying models", () => {
- const model = createMockModel({
- id: "openrouter/test-model",
- providerID: "openrouter",
- api: {
- id: "test-model",
- url: "https://openrouter.ai",
- npm: "@openrouter/ai-sdk-provider",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(result).toEqual({})
- })
- test("gpt models return OPENAI_EFFORTS with reasoning", () => {
- const model = createMockModel({
- id: "openrouter/gpt-4",
- providerID: "openrouter",
- api: {
- id: "gpt-4",
- url: "https://openrouter.ai",
- npm: "@openrouter/ai-sdk-provider",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["none", "minimal", "low", "medium", "high", "xhigh"])
- expect(result.low).toEqual({ reasoning: { effort: "low" } })
- expect(result.high).toEqual({ reasoning: { effort: "high" } })
- })
- test("gemini-3 returns OPENAI_EFFORTS with reasoning", () => {
- const model = createMockModel({
- id: "openrouter/gemini-3-5-pro",
- providerID: "openrouter",
- api: {
- id: "gemini-3-5-pro",
- url: "https://openrouter.ai",
- npm: "@openrouter/ai-sdk-provider",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["none", "minimal", "low", "medium", "high", "xhigh"])
- })
- test("grok-4 returns empty object", () => {
- const model = createMockModel({
- id: "openrouter/grok-4",
- providerID: "openrouter",
- api: {
- id: "grok-4",
- url: "https://openrouter.ai",
- npm: "@openrouter/ai-sdk-provider",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(result).toEqual({})
- })
- test("grok-3-mini returns low and high with reasoning", () => {
- const model = createMockModel({
- id: "openrouter/grok-3-mini",
- providerID: "openrouter",
- api: {
- id: "grok-3-mini",
- url: "https://openrouter.ai",
- npm: "@openrouter/ai-sdk-provider",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "high"])
- expect(result.low).toEqual({ reasoning: { effort: "low" } })
- expect(result.high).toEqual({ reasoning: { effort: "high" } })
- })
- })
- describe("@ai-sdk/gateway", () => {
- test("returns OPENAI_EFFORTS with reasoningEffort", () => {
- const model = createMockModel({
- id: "gateway/gateway-model",
- providerID: "gateway",
- api: {
- id: "gateway-model",
- url: "https://gateway.ai",
- npm: "@ai-sdk/gateway",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["none", "minimal", "low", "medium", "high", "xhigh"])
- expect(result.low).toEqual({ reasoningEffort: "low" })
- expect(result.high).toEqual({ reasoningEffort: "high" })
- })
- })
- describe("@ai-sdk/github-copilot", () => {
- test("standard models return low, medium, high", () => {
- const model = createMockModel({
- id: "gpt-4.5",
- providerID: "github-copilot",
- api: {
- id: "gpt-4.5",
- url: "https://api.githubcopilot.com",
- npm: "@ai-sdk/github-copilot",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high"])
- expect(result.low).toEqual({
- reasoningEffort: "low",
- reasoningSummary: "auto",
- include: ["reasoning.encrypted_content"],
- })
- })
- test("gpt-5.1-codex-max includes xhigh", () => {
- const model = createMockModel({
- id: "gpt-5.1-codex-max",
- providerID: "github-copilot",
- api: {
- id: "gpt-5.1-codex-max",
- url: "https://api.githubcopilot.com",
- npm: "@ai-sdk/github-copilot",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high", "xhigh"])
- })
- test("gpt-5.1-codex-mini does not include xhigh", () => {
- const model = createMockModel({
- id: "gpt-5.1-codex-mini",
- providerID: "github-copilot",
- api: {
- id: "gpt-5.1-codex-mini",
- url: "https://api.githubcopilot.com",
- npm: "@ai-sdk/github-copilot",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high"])
- })
- test("gpt-5.1-codex does not include xhigh", () => {
- const model = createMockModel({
- id: "gpt-5.1-codex",
- providerID: "github-copilot",
- api: {
- id: "gpt-5.1-codex",
- url: "https://api.githubcopilot.com",
- npm: "@ai-sdk/github-copilot",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high"])
- })
- test("gpt-5.2 includes xhigh", () => {
- const model = createMockModel({
- id: "gpt-5.2",
- providerID: "github-copilot",
- api: {
- id: "gpt-5.2",
- url: "https://api.githubcopilot.com",
- npm: "@ai-sdk/github-copilot",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high", "xhigh"])
- expect(result.xhigh).toEqual({
- reasoningEffort: "xhigh",
- reasoningSummary: "auto",
- include: ["reasoning.encrypted_content"],
- })
- })
- test("gpt-5.2-codex includes xhigh", () => {
- const model = createMockModel({
- id: "gpt-5.2-codex",
- providerID: "github-copilot",
- api: {
- id: "gpt-5.2-codex",
- url: "https://api.githubcopilot.com",
- npm: "@ai-sdk/github-copilot",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high", "xhigh"])
- })
- })
- describe("@ai-sdk/cerebras", () => {
- test("returns WIDELY_SUPPORTED_EFFORTS with reasoningEffort", () => {
- const model = createMockModel({
- id: "cerebras/llama-4",
- providerID: "cerebras",
- api: {
- id: "llama-4-sc",
- url: "https://api.cerebras.ai",
- npm: "@ai-sdk/cerebras",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high"])
- expect(result.low).toEqual({ reasoningEffort: "low" })
- expect(result.high).toEqual({ reasoningEffort: "high" })
- })
- })
- describe("@ai-sdk/togetherai", () => {
- test("returns WIDELY_SUPPORTED_EFFORTS with reasoningEffort", () => {
- const model = createMockModel({
- id: "togetherai/llama-4",
- providerID: "togetherai",
- api: {
- id: "llama-4-sc",
- url: "https://api.togetherai.com",
- npm: "@ai-sdk/togetherai",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high"])
- expect(result.low).toEqual({ reasoningEffort: "low" })
- expect(result.high).toEqual({ reasoningEffort: "high" })
- })
- })
- describe("@ai-sdk/xai", () => {
- test("grok-3 returns empty object", () => {
- const model = createMockModel({
- id: "xai/grok-3",
- providerID: "xai",
- api: {
- id: "grok-3",
- url: "https://api.x.ai",
- npm: "@ai-sdk/xai",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(result).toEqual({})
- })
- test("grok-3-mini returns low and high with reasoningEffort", () => {
- const model = createMockModel({
- id: "xai/grok-3-mini",
- providerID: "xai",
- api: {
- id: "grok-3-mini",
- url: "https://api.x.ai",
- npm: "@ai-sdk/xai",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "high"])
- expect(result.low).toEqual({ reasoningEffort: "low" })
- expect(result.high).toEqual({ reasoningEffort: "high" })
- })
- })
- describe("@ai-sdk/deepinfra", () => {
- test("returns WIDELY_SUPPORTED_EFFORTS with reasoningEffort", () => {
- const model = createMockModel({
- id: "deepinfra/llama-4",
- providerID: "deepinfra",
- api: {
- id: "llama-4-sc",
- url: "https://api.deepinfra.com",
- npm: "@ai-sdk/deepinfra",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high"])
- expect(result.low).toEqual({ reasoningEffort: "low" })
- expect(result.high).toEqual({ reasoningEffort: "high" })
- })
- })
- describe("@ai-sdk/openai-compatible", () => {
- test("returns WIDELY_SUPPORTED_EFFORTS with reasoningEffort", () => {
- const model = createMockModel({
- id: "custom-provider/custom-model",
- providerID: "custom-provider",
- api: {
- id: "custom-model",
- url: "https://api.custom.com",
- npm: "@ai-sdk/openai-compatible",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high"])
- expect(result.low).toEqual({ reasoningEffort: "low" })
- expect(result.high).toEqual({ reasoningEffort: "high" })
- })
- test("Claude via LiteLLM returns thinking with snake_case budget_tokens", () => {
- const model = createMockModel({
- id: "anthropic/claude-sonnet-4-5",
- providerID: "anthropic",
- api: {
- id: "claude-sonnet-4-5-20250929",
- url: "http://localhost:4000",
- npm: "@ai-sdk/openai-compatible",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["high", "max"])
- expect(result.high).toEqual({
- thinking: {
- type: "enabled",
- budget_tokens: 16000,
- },
- })
- expect(result.max).toEqual({
- thinking: {
- type: "enabled",
- budget_tokens: 31999,
- },
- })
- })
- test("Claude model (by model.id) via openai-compatible uses snake_case", () => {
- const model = createMockModel({
- id: "litellm/claude-3-opus",
- providerID: "litellm",
- api: {
- id: "claude-3-opus-20240229",
- url: "http://localhost:4000",
- npm: "@ai-sdk/openai-compatible",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["high", "max"])
- expect(result.high).toEqual({
- thinking: {
- type: "enabled",
- budget_tokens: 16000,
- },
- })
- })
- test("Anthropic model (by model.api.id) via openai-compatible uses snake_case", () => {
- const model = createMockModel({
- id: "custom/my-model",
- providerID: "custom",
- api: {
- id: "anthropic.claude-sonnet",
- url: "http://localhost:4000",
- npm: "@ai-sdk/openai-compatible",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["high", "max"])
- expect(result.high.thinking.budget_tokens).toBe(16000)
- })
- })
- describe("@ai-sdk/azure", () => {
- test("o1-mini returns empty object", () => {
- const model = createMockModel({
- id: "o1-mini",
- providerID: "azure",
- api: {
- id: "o1-mini",
- url: "https://azure.com",
- npm: "@ai-sdk/azure",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(result).toEqual({})
- })
- test("standard azure models return custom efforts with reasoningSummary", () => {
- const model = createMockModel({
- id: "o1",
- providerID: "azure",
- api: {
- id: "o1",
- url: "https://azure.com",
- npm: "@ai-sdk/azure",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high"])
- expect(result.low).toEqual({
- reasoningEffort: "low",
- reasoningSummary: "auto",
- include: ["reasoning.encrypted_content"],
- })
- })
- test("gpt-5 adds minimal effort", () => {
- const model = createMockModel({
- id: "gpt-5",
- providerID: "azure",
- api: {
- id: "gpt-5",
- url: "https://azure.com",
- npm: "@ai-sdk/azure",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["minimal", "low", "medium", "high"])
- })
- })
- describe("@ai-sdk/openai", () => {
- test("gpt-5-pro returns empty object", () => {
- const model = createMockModel({
- id: "gpt-5-pro",
- providerID: "openai",
- api: {
- id: "gpt-5-pro",
- url: "https://api.openai.com",
- npm: "@ai-sdk/openai",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(result).toEqual({})
- })
- test("standard openai models return custom efforts with reasoningSummary", () => {
- const model = createMockModel({
- id: "gpt-5",
- providerID: "openai",
- api: {
- id: "gpt-5",
- url: "https://api.openai.com",
- npm: "@ai-sdk/openai",
- },
- release_date: "2024-06-01",
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["minimal", "low", "medium", "high"])
- expect(result.low).toEqual({
- reasoningEffort: "low",
- reasoningSummary: "auto",
- include: ["reasoning.encrypted_content"],
- })
- })
- test("models after 2025-11-13 include 'none' effort", () => {
- const model = createMockModel({
- id: "gpt-5-nano",
- providerID: "openai",
- api: {
- id: "gpt-5-nano",
- url: "https://api.openai.com",
- npm: "@ai-sdk/openai",
- },
- release_date: "2025-11-14",
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["none", "minimal", "low", "medium", "high"])
- })
- test("models after 2025-12-04 include 'xhigh' effort", () => {
- const model = createMockModel({
- id: "openai/gpt-5-chat",
- providerID: "openai",
- api: {
- id: "gpt-5-chat",
- url: "https://api.openai.com",
- npm: "@ai-sdk/openai",
- },
- release_date: "2025-12-05",
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["none", "minimal", "low", "medium", "high", "xhigh"])
- })
- })
- describe("@ai-sdk/anthropic", () => {
- test("returns high and max with thinking config", () => {
- const model = createMockModel({
- id: "anthropic/claude-4",
- providerID: "anthropic",
- api: {
- id: "claude-4",
- url: "https://api.anthropic.com",
- npm: "@ai-sdk/anthropic",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["high", "max"])
- expect(result.high).toEqual({
- thinking: {
- type: "enabled",
- budgetTokens: 16000,
- },
- })
- expect(result.max).toEqual({
- thinking: {
- type: "enabled",
- budgetTokens: 31999,
- },
- })
- })
- })
- describe("@ai-sdk/amazon-bedrock", () => {
- test("returns WIDELY_SUPPORTED_EFFORTS with reasoningConfig", () => {
- const model = createMockModel({
- id: "bedrock/llama-4",
- providerID: "bedrock",
- api: {
- id: "llama-4-sc",
- url: "https://bedrock.amazonaws.com",
- npm: "@ai-sdk/amazon-bedrock",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high"])
- expect(result.low).toEqual({
- reasoningConfig: {
- type: "enabled",
- maxReasoningEffort: "low",
- },
- })
- })
- })
- describe("@ai-sdk/google", () => {
- test("gemini-2.5 returns high and max with thinkingConfig and thinkingBudget", () => {
- const model = createMockModel({
- id: "google/gemini-2.5-pro",
- providerID: "google",
- api: {
- id: "gemini-2.5-pro",
- url: "https://generativelanguage.googleapis.com",
- npm: "@ai-sdk/google",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["high", "max"])
- expect(result.high).toEqual({
- thinkingConfig: {
- includeThoughts: true,
- thinkingBudget: 16000,
- },
- })
- expect(result.max).toEqual({
- thinkingConfig: {
- includeThoughts: true,
- thinkingBudget: 24576,
- },
- })
- })
- test("other gemini models return low and high with thinkingLevel", () => {
- const model = createMockModel({
- id: "google/gemini-2.0-pro",
- providerID: "google",
- api: {
- id: "gemini-2.0-pro",
- url: "https://generativelanguage.googleapis.com",
- npm: "@ai-sdk/google",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "high"])
- expect(result.low).toEqual({
- includeThoughts: true,
- thinkingLevel: "low",
- })
- expect(result.high).toEqual({
- includeThoughts: true,
- thinkingLevel: "high",
- })
- })
- })
- describe("@ai-sdk/google-vertex", () => {
- test("gemini-2.5 returns high and max with thinkingConfig and thinkingBudget", () => {
- const model = createMockModel({
- id: "google-vertex/gemini-2.5-pro",
- providerID: "google-vertex",
- api: {
- id: "gemini-2.5-pro",
- url: "https://vertexai.googleapis.com",
- npm: "@ai-sdk/google-vertex",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["high", "max"])
- })
- test("other vertex models return low and high with thinkingLevel", () => {
- const model = createMockModel({
- id: "google-vertex/gemini-2.0-pro",
- providerID: "google-vertex",
- api: {
- id: "gemini-2.0-pro",
- url: "https://vertexai.googleapis.com",
- npm: "@ai-sdk/google-vertex",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "high"])
- })
- })
- describe("@ai-sdk/cohere", () => {
- test("returns empty object", () => {
- const model = createMockModel({
- id: "cohere/command-r",
- providerID: "cohere",
- api: {
- id: "command-r",
- url: "https://api.cohere.com",
- npm: "@ai-sdk/cohere",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(result).toEqual({})
- })
- })
- describe("@ai-sdk/groq", () => {
- test("returns none and WIDELY_SUPPORTED_EFFORTS with thinkingLevel", () => {
- const model = createMockModel({
- id: "groq/llama-4",
- providerID: "groq",
- api: {
- id: "llama-4-sc",
- url: "https://api.groq.com",
- npm: "@ai-sdk/groq",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["none", "low", "medium", "high"])
- expect(result.none).toEqual({
- includeThoughts: true,
- thinkingLevel: "none",
- })
- expect(result.low).toEqual({
- includeThoughts: true,
- thinkingLevel: "low",
- })
- })
- })
- describe("@ai-sdk/perplexity", () => {
- test("returns empty object", () => {
- const model = createMockModel({
- id: "perplexity/sonar-plus",
- providerID: "perplexity",
- api: {
- id: "sonar-plus",
- url: "https://api.perplexity.ai",
- npm: "@ai-sdk/perplexity",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(result).toEqual({})
- })
- })
- })
|