| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917391839193920392139223923392439253926392739283929393039313932393339343935393639373938393939403941394239433944394539463947394839493950395139523953395439553956395739583959396039613962396339643965396639673968396939703971397239733974397539763977397839793980398139823983398439853986398739883989399039913992399339943995399639973998399940004001400240034004400540064007400840094010401140124013401440154016401740184019402040214022402340244025402640274028402940304031403240334034403540364037403840394040404140424043404440454046404740484049405040514052405340544055405640574058405940604061406240634064406540664067406840694070407140724073407440754076407740784079408040814082408340844085408640874088408940904091409240934094409540964097409840994100410141024103410441054106410741084109411041114112411341144115411641174118411941204121412241234124412541264127412841294130413141324133413441354136413741384139414041414142414341444145414641474148414941504151415241534154415541564157415841594160416141624163416441654166416741684169417041714172417341744175417641774178417941804181418241834184418541864187418841894190419141924193419441954196419741984199420042014202420342044205420642074208420942104211421242134214421542164217421842194220422142224223422442254226422742284229423042314232423342344235423642374238423942404241424242434244424542464247424842494250425142524253425442554256425742584259426042614262426342644265426642674268426942704271427242734274427542764277427842794280428142824283428442854286428742884289429042914292429342944295429642974298429943004301430243034304430543064307430843094310431143124313431443154316431743184319432043214322432343244325432643274328432943304331433243334334433543364337433843394340434143424343434443454346434743484349435043514352435343544355435643574358435943604361436243634364436543664367436843694370437143724373437443754376437743784379438043814382438343844385438643874388438943904391439243934394439543964397439843994400440144024403440444054406440744084409441044114412441344144415441644174418441944204421442244234424442544264427442844294430443144324433443444354436443744384439444044414442444344444445444644474448444944504451445244534454445544564457445844594460446144624463446444654466446744684469447044714472447344744475447644774478447944804481448244834484448544864487448844894490449144924493449444954496449744984499450045014502450345044505450645074508450945104511451245134514451545164517451845194520452145224523452445254526452745284529453045314532453345344535453645374538453945404541454245434544454545464547454845494550455145524553455445554556455745584559456045614562456345644565456645674568456945704571457245734574457545764577457845794580458145824583458445854586458745884589459045914592459345944595459645974598459946004601460246034604460546064607460846094610461146124613461446154616461746184619462046214622462346244625462646274628462946304631463246334634463546364637463846394640464146424643464446454646464746484649465046514652465346544655465646574658465946604661466246634664466546664667466846694670467146724673467446754676467746784679468046814682468346844685468646874688468946904691469246934694469546964697469846994700470147024703470447054706470747084709471047114712471347144715471647174718471947204721472247234724472547264727472847294730473147324733473447354736473747384739474047414742474347444745474647474748474947504751475247534754475547564757475847594760476147624763476447654766476747684769477047714772477347744775477647774778477947804781478247834784478547864787478847894790479147924793479447954796479747984799480048014802480348044805480648074808480948104811481248134814481548164817481848194820482148224823482448254826482748284829483048314832483348344835483648374838483948404841484248434844484548464847484848494850485148524853485448554856485748584859486048614862486348644865486648674868486948704871487248734874487548764877487848794880488148824883488448854886488748884889489048914892489348944895489648974898489949004901490249034904490549064907490849094910491149124913491449154916491749184919492049214922492349244925492649274928492949304931493249334934493549364937493849394940494149424943494449454946494749484949495049514952495349544955495649574958495949604961496249634964496549664967496849694970497149724973497449754976497749784979498049814982498349844985498649874988498949904991499249934994499549964997499849995000500150025003500450055006500750085009501050115012501350145015501650175018501950205021502250235024502550265027502850295030503150325033503450355036503750385039504050415042504350445045504650475048504950505051505250535054505550565057505850595060506150625063506450655066506750685069507050715072507350745075507650775078507950805081508250835084508550865087508850895090509150925093509450955096509750985099510051015102510351045105510651075108510951105111511251135114511551165117511851195120512151225123512451255126512751285129513051315132513351345135513651375138513951405141514251435144514551465147514851495150515151525153515451555156515751585159516051615162516351645165516651675168516951705171517251735174517551765177517851795180518151825183518451855186518751885189519051915192519351945195519651975198519952005201520252035204520552065207520852095210521152125213521452155216521752185219522052215222522352245225522652275228522952305231523252335234523552365237523852395240524152425243524452455246524752485249525052515252525352545255525652575258525952605261526252635264526552665267526852695270527152725273527452755276527752785279528052815282528352845285528652875288528952905291529252935294529552965297529852995300530153025303530453055306530753085309531053115312531353145315531653175318531953205321532253235324532553265327532853295330533153325333533453355336533753385339534053415342534353445345534653475348534953505351535253535354535553565357535853595360536153625363536453655366536753685369537053715372537353745375537653775378537953805381538253835384538553865387538853895390539153925393539453955396539753985399540054015402540354045405540654075408540954105411541254135414541554165417541854195420542154225423542454255426542754285429 |
- import { describe, expect, test } from "bun:test"
- import { Effect } from "effect"
- import { ProviderTransform } from "@/provider/transform"
- import { LLMRequestPrep } from "@/session/llm/request"
- import { ProviderV2 } from "@opencode-ai/core/provider"
- import { ModelV2 } from "@opencode-ai/core/model"
- import { ModelsDev } from "@opencode-ai/core/models-dev"
- import { jsonSchema } from "ai"
- 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 by default", () => {
- 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 promptCacheKey for the OpenAI SDK regardless of provider ID", () => {
- const result = ProviderTransform.options({
- model: {
- ...mockModel,
- providerID: "custom-openai",
- api: { id: "gpt-5", url: "https://example.com", npm: "@ai-sdk/openai" },
- },
- sessionID,
- providerOptions: {},
- })
- expect(result.promptCacheKey).toBe(sessionID)
- })
- test("should not set promptCacheKey for the OpenAI-compatible SDK by provider name", () => {
- const result = ProviderTransform.options({
- model: {
- ...mockModel,
- providerID: "openai",
- api: { id: "gpt-5", url: "https://example.com", npm: "@ai-sdk/openai-compatible" },
- },
- sessionID,
- providerOptions: {},
- })
- expect(result.promptCacheKey).toBeUndefined()
- })
- test("should not set promptCacheKey for openai when explicitly disabled", () => {
- 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: { setCacheKey: false },
- })
- expect(result.promptCacheKey).toBeUndefined()
- })
- test("should set promptCacheKey for the xAI SDK by default regardless of provider ID", () => {
- const xaiModel = {
- ...mockModel,
- providerID: "custom-xai",
- api: {
- id: "grok-4",
- url: "https://api.x.ai",
- npm: "@ai-sdk/xai",
- },
- }
- const result = ProviderTransform.options({ model: xaiModel, sessionID, providerOptions: {} })
- expect(result.promptCacheKey).toBe(sessionID)
- })
- test("should not set promptCacheKey for the xAI SDK when explicitly disabled", () => {
- const xaiModel = {
- ...mockModel,
- providerID: "xai",
- api: {
- id: "grok-4",
- url: "https://api.x.ai",
- npm: "@ai-sdk/xai",
- },
- }
- const result = ProviderTransform.options({
- model: xaiModel,
- sessionID,
- providerOptions: { setCacheKey: false },
- })
- expect(result.promptCacheKey).toBeUndefined()
- })
- 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)
- })
- test("should set store=false for xAI provider by default", () => {
- const xaiModel = {
- ...mockModel,
- providerID: "xai",
- api: {
- id: "grok-4",
- url: "https://api.x.ai",
- npm: "@ai-sdk/xai",
- },
- }
- const result = ProviderTransform.options({
- model: xaiModel,
- sessionID,
- providerOptions: {},
- })
- expect(result.store).toBe(false)
- expect(result.promptCacheKey).toBe(sessionID)
- })
- test("should set store=false for xAI SDK regardless of provider ID", () => {
- const xaiModel = {
- ...mockModel,
- providerID: "custom-xai",
- api: {
- id: "grok-4",
- url: "https://api.x.ai",
- npm: "@ai-sdk/xai",
- },
- }
- const result = ProviderTransform.options({
- model: xaiModel,
- sessionID,
- providerOptions: {},
- })
- expect(result.store).toBe(false)
- })
- test("should set store=false for azure provider by default", () => {
- const azureModel = {
- ...mockModel,
- providerID: "azure",
- api: {
- id: "gpt-4",
- url: "https://azure.com",
- npm: "@ai-sdk/azure",
- },
- }
- const result = ProviderTransform.options({
- model: azureModel,
- sessionID,
- providerOptions: {},
- })
- expect(result.store).toBe(false)
- expect(result.promptCacheKey).toBe(sessionID)
- })
- test("should disable the Azure cache key without disabling store=false", () => {
- const result = ProviderTransform.options({
- model: {
- ...mockModel,
- providerID: "azure",
- api: { id: "gpt-5", url: "https://azure.com", npm: "@ai-sdk/azure" },
- },
- sessionID,
- providerOptions: { setCacheKey: false },
- })
- expect(result.store).toBe(false)
- expect(result.promptCacheKey).toBeUndefined()
- })
- test("should keep the Azure cache key for gpt-5.5 early return", () => {
- const result = ProviderTransform.options({
- model: {
- ...mockModel,
- providerID: "azure",
- api: { id: "gpt-5.5", url: "https://azure.com", npm: "@ai-sdk/azure" },
- },
- sessionID,
- providerOptions: {},
- })
- expect(result.store).toBe(false)
- expect(result.reasoningSummary).toBe("auto")
- expect(result.promptCacheKey).toBe(sessionID)
- })
- for (const npm of ["@ai-sdk/deepinfra", "@ai-sdk/cerebras"]) {
- test(`should set the snake-case cache key for ${npm}`, () => {
- const result = ProviderTransform.options({
- model: { ...mockModel, providerID: "custom", api: { ...mockModel.api, npm } },
- sessionID,
- providerOptions: {},
- })
- expect(result.prompt_cache_key).toBe(sessionID)
- expect(result.promptCacheKey).toBeUndefined()
- })
- }
- test("should set promptCacheKey for the Mistral SDK", () => {
- const result = ProviderTransform.options({
- model: { ...mockModel, providerID: "custom", api: { ...mockModel.api, npm: "@ai-sdk/mistral" } },
- sessionID,
- providerOptions: {},
- })
- expect(result.promptCacheKey).toBe(sessionID)
- })
- test("should not send an undocumented OpenRouter prompt_cache_key", () => {
- const result = ProviderTransform.options({
- model: {
- ...mockModel,
- providerID: "openrouter",
- api: { ...mockModel.api, npm: "@openrouter/ai-sdk-provider" },
- },
- sessionID,
- providerOptions: {},
- })
- expect(result.prompt_cache_key).toBeUndefined()
- })
- })
- describe("ProviderTransform.options - zai/zhipuai thinking", () => {
- const sessionID = "test-session-123"
- const createModel = (providerID: string) =>
- ({
- id: `${providerID}/glm-4.6`,
- providerID,
- api: {
- id: "glm-4.6",
- url: "https://open.bigmodel.cn/api/paas/v4",
- npm: "@ai-sdk/openai-compatible",
- },
- name: "GLM 4.6",
- capabilities: {
- temperature: true,
- reasoning: true,
- 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
- for (const providerID of ["zai-coding-plan", "zai", "zhipuai-coding-plan", "zhipuai"]) {
- test(`${providerID} should set thinking cfg`, () => {
- const result = ProviderTransform.options({
- model: createModel(providerID),
- sessionID,
- providerOptions: {},
- })
- expect(result.thinking).toEqual({
- type: "enabled",
- clear_thinking: false,
- })
- })
- }
- })
- describe("ProviderTransform.options - minimax m3 thinking", () => {
- const createModel = (npm: string) =>
- ({
- id: "minimax/minimax-m3",
- providerID: "minimax",
- api: {
- id: "minimax-m3",
- url: "https://api.minimax.com",
- npm,
- },
- capabilities: { reasoning: true },
- limit: { output: 64_000 },
- }) as any
- test("explicitly enables adaptive thinking with the anthropic SDK", () => {
- expect(
- ProviderTransform.options({
- model: createModel("@ai-sdk/anthropic"),
- sessionID: "test-session-123",
- }).thinking,
- ).toEqual({ type: "adaptive" })
- })
- test("uses the native default with the openai-compatible SDK", () => {
- expect(
- ProviderTransform.options({
- model: createModel("@ai-sdk/openai-compatible"),
- sessionID: "test-session-123",
- }).thinking,
- ).toBeUndefined()
- })
- })
- describe("ProviderTransform.options - google thinkingConfig gating", () => {
- const sessionID = "test-session-123"
- const createGoogleModel = (reasoning: boolean, npm: "@ai-sdk/google" | "@ai-sdk/google-vertex") =>
- ({
- id: `${npm === "@ai-sdk/google" ? "google" : "google-vertex"}/gemini-2.0-flash`,
- providerID: npm === "@ai-sdk/google" ? "google" : "google-vertex",
- api: {
- id: "gemini-2.0-flash",
- url: npm === "@ai-sdk/google" ? "https://generativelanguage.googleapis.com" : "https://vertexai.googleapis.com",
- npm,
- },
- name: "Gemini 2.0 Flash",
- capabilities: {
- temperature: true,
- reasoning,
- 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: 1_000_000,
- output: 8192,
- },
- status: "active",
- options: {},
- headers: {},
- }) as any
- test("does not set thinkingConfig for google models without reasoning capability", () => {
- const result = ProviderTransform.options({
- model: createGoogleModel(false, "@ai-sdk/google"),
- sessionID,
- providerOptions: {},
- })
- expect(result.thinkingConfig).toBeUndefined()
- })
- test("sets thinkingConfig for google models with reasoning capability", () => {
- const result = ProviderTransform.options({
- model: createGoogleModel(true, "@ai-sdk/google"),
- sessionID,
- providerOptions: {},
- })
- expect(result.thinkingConfig).toEqual({
- includeThoughts: true,
- })
- })
- test("does not set thinkingConfig for vertex models without reasoning capability", () => {
- const result = ProviderTransform.options({
- model: createGoogleModel(false, "@ai-sdk/google-vertex"),
- sessionID,
- providerOptions: {},
- })
- expect(result.thinkingConfig).toBeUndefined()
- })
- })
- 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")
- expect(result.include).toEqual(["reasoning.encrypted_content"])
- })
- test("Bedrock Mantle gpt-5.5 uses OpenAI Responses defaults", () => {
- const model = {
- ...createGpt5Model("openai.gpt-5.5"),
- id: "amazon-bedrock/openai.gpt-5.5",
- providerID: "amazon-bedrock",
- api: {
- id: "openai.gpt-5.5",
- url: "https://bedrock-mantle.us-east-2.api.aws/openai/v1",
- npm: "@ai-sdk/amazon-bedrock/mantle",
- },
- }
- const result = ProviderTransform.options({ model, sessionID, providerOptions: {} })
- expect(result.store).toBe(false)
- expect(result.reasoningEffort).toBe("medium")
- expect(result.reasoningSummary).toBe("auto")
- expect(result.include).toEqual(["reasoning.encrypted_content"])
- expect(result.textVerbosity).toBe("low")
- })
- test("openai-compatible gpt-5 models omit Responses-only reasoningSummary", () => {
- const model = {
- ...createGpt5Model("gpt-5.4"),
- id: "cortecs/gpt-5.4",
- providerID: "cortecs",
- api: {
- id: "gpt-5.4",
- url: "https://api.cortecs.ai/v1",
- npm: "@ai-sdk/openai-compatible",
- },
- }
- const result = ProviderTransform.options({ model, sessionID, providerOptions: {} })
- expect(result.reasoningEffort).toBe("medium")
- expect(result.reasoningSummary).toBeUndefined()
- expect(result.include).toBeUndefined()
- })
- test("azure chat completions omit Responses-only reasoning options after variants merge", async () => {
- const model = {
- ...createGpt5Model("gpt-5.4"),
- id: "azure/gpt-5.4",
- providerID: "azure",
- api: {
- id: "gpt-5.4",
- url: "https://azure.com",
- npm: "@ai-sdk/azure",
- },
- variants: {
- high: {
- reasoningEffort: "high",
- reasoningSummary: "auto",
- include: ["reasoning.encrypted_content"],
- },
- },
- }
- const result = await Effect.runPromise(
- LLMRequestPrep.prepare({
- user: {
- id: "msg_user-test",
- sessionID,
- role: "user",
- time: { created: Date.now() },
- agent: "test",
- model: { providerID: "azure", modelID: "gpt-5.4", variant: "high" },
- } as any,
- sessionID,
- model,
- agent: {
- name: "test",
- mode: "primary",
- options: {},
- permission: [],
- } as any,
- system: [],
- messages: [{ role: "user", content: "Hello" }],
- tools: {
- lookup: {
- description: "Look up a value",
- inputSchema: jsonSchema({ type: "object", properties: {} }),
- },
- },
- provider: { id: "azure", options: { useCompletionUrls: true } } as any,
- auth: undefined,
- plugin: {
- trigger: (_name: string, _input: unknown, output: unknown) => Effect.succeed(output),
- list: () => Effect.succeed([]),
- init: () => Effect.void,
- } as any,
- flags: { outputTokenMax: 32_000, client: "test" } as any,
- isWorkflow: false,
- }),
- )
- expect(result.params.options.reasoningEffort).toBe("high")
- expect(result.params.options.reasoningSummary).toBeUndefined()
- expect(result.params.options.include).toBeUndefined()
- expect(result.tools.lookup.strict).toBe(false)
- })
- 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.options - gpt-5 reasoningEffort", () => {
- const sessionID = "test-session-123"
- const createModel = (apiId: string) =>
- ({
- id: `azure/${apiId}`,
- providerID: "azure",
- api: {
- id: apiId,
- url: "https://azure.com",
- npm: "@ai-sdk/azure",
- },
- 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-chat should NOT set reasoningEffort", () => {
- const result = ProviderTransform.options({
- model: createModel("gpt-5-chat"),
- sessionID,
- providerOptions: {},
- })
- expect(result.reasoningEffort).toBeUndefined()
- })
- test("gpt-5.5 should NOT set reasoningEffort", () => {
- const result = ProviderTransform.options({
- model: createModel("gpt-5.5"),
- sessionID,
- providerOptions: {},
- })
- expect(result.reasoningEffort).toBeUndefined()
- })
- })
- describe("ProviderTransform.options - gateway", () => {
- const sessionID = "test-session-123"
- const createModel = (id: string) =>
- ({
- id,
- providerID: "vercel",
- api: {
- id,
- url: "https://ai-gateway.vercel.sh/v3/ai",
- npm: "@ai-sdk/gateway",
- },
- name: id,
- capabilities: {
- temperature: true,
- reasoning: true,
- 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: 200_000,
- output: 8192,
- },
- status: "active",
- options: {},
- headers: {},
- release_date: "2024-01-01",
- }) as any
- test("puts gateway defaults under gateway key", () => {
- const model = createModel("anthropic/claude-sonnet-4")
- const result = ProviderTransform.options({ model, sessionID, providerOptions: {} })
- expect(result).toEqual({
- gateway: {
- caching: "auto",
- },
- })
- })
- })
- describe("ProviderTransform.providerOptions", () => {
- const createModel = (overrides: Partial<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,
- }) as any
- test("uses sdk key for non-gateway models", () => {
- const model = createModel({
- providerID: "my-bedrock",
- api: {
- id: "anthropic.claude-sonnet-4",
- url: "https://bedrock.aws",
- npm: "@ai-sdk/amazon-bedrock",
- },
- })
- expect(ProviderTransform.providerOptions(model, { cachePoint: { type: "default" } })).toEqual({
- bedrock: { cachePoint: { type: "default" } },
- })
- })
- test("forces reasoning for custom OpenAI package models with explicit effort", () => {
- const model = createModel({
- providerID: "meta",
- api: {
- id: "muse-spark",
- url: "https://api.ai.meta.com/v1",
- npm: "@ai-sdk/openai",
- },
- })
- expect(ProviderTransform.providerOptions(model, { reasoningEffort: "xhigh", reasoningSummary: "auto" })).toEqual({
- openai: { forceReasoning: true, reasoningEffort: "xhigh", reasoningSummary: "auto" },
- })
- })
- test("forces reasoning for OpenAI package models marked reasoning-capable", () => {
- expect(ProviderTransform.providerOptions(createModel(), { store: false })).toEqual({
- openai: { forceReasoning: true, store: false },
- })
- })
- test("uses canonical sdk key for custom xAI models", () => {
- const model = createModel({
- providerID: "my-xai",
- api: { id: "grok-4", url: "https://api.x.ai", npm: "@ai-sdk/xai" },
- })
- expect(ProviderTransform.providerOptions(model, { promptCacheKey: "session" })).toEqual({
- xai: { promptCacheKey: "session" },
- })
- })
- test("forces reasoning for explicit effort even when model is not marked reasoning-capable", () => {
- const model = createModel({
- 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,
- },
- })
- expect(ProviderTransform.providerOptions(model, { reasoningEffort: "xhigh" })).toEqual({
- openai: { forceReasoning: true, reasoningEffort: "xhigh" },
- })
- })
- test("forces reasoning for Azure OpenAI models with explicit effort", () => {
- const model = createModel({
- providerID: "azure",
- api: {
- id: "custom-gpt-5-deployment",
- url: "https://azure.openai.example.com/openai/v1",
- npm: "@ai-sdk/azure",
- },
- })
- expect(ProviderTransform.providerOptions(model, { reasoningEffort: "xhigh" })).toEqual({
- openai: { forceReasoning: true, reasoningEffort: "xhigh" },
- azure: { forceReasoning: true, reasoningEffort: "xhigh" },
- })
- })
- test("forces reasoning for Bedrock Mantle OpenAI models with explicit effort", () => {
- const model = createModel({
- providerID: "amazon-bedrock",
- api: {
- id: "openai.gpt-5-custom",
- url: "https://bedrock-mantle.us-east-2.api.aws/openai/v1",
- npm: "@ai-sdk/amazon-bedrock/mantle",
- },
- })
- expect(ProviderTransform.providerOptions(model, { reasoningEffort: "xhigh" })).toEqual({
- openai: { forceReasoning: true, reasoningEffort: "xhigh" },
- })
- })
- test("overrides forceReasoning false when reasoning should be forced", () => {
- expect(
- ProviderTransform.providerOptions(createModel(), { forceReasoning: false, reasoningEffort: "xhigh" }),
- ).toEqual({
- openai: { forceReasoning: true, reasoningEffort: "xhigh" },
- })
- })
- test("uses gateway model provider slug for gateway models", () => {
- const model = createModel({
- providerID: "vercel",
- api: {
- id: "anthropic/claude-sonnet-4",
- url: "https://ai-gateway.vercel.sh/v3/ai",
- npm: "@ai-sdk/gateway",
- },
- })
- expect(ProviderTransform.providerOptions(model, { thinking: { type: "enabled", budgetTokens: 12_000 } })).toEqual({
- anthropic: { thinking: { type: "enabled", budgetTokens: 12_000 } },
- })
- })
- test("falls back to gateway key when gateway api id is unscoped", () => {
- const model = createModel({
- id: "anthropic/claude-sonnet-4",
- providerID: "vercel",
- api: {
- id: "claude-sonnet-4",
- url: "https://ai-gateway.vercel.sh/v3/ai",
- npm: "@ai-sdk/gateway",
- },
- })
- expect(ProviderTransform.providerOptions(model, { thinking: { type: "enabled", budgetTokens: 12_000 } })).toEqual({
- gateway: { thinking: { type: "enabled", budgetTokens: 12_000 } },
- })
- })
- test("splits gateway routing options from provider-specific options", () => {
- const model = createModel({
- providerID: "vercel",
- api: {
- id: "anthropic/claude-sonnet-4",
- url: "https://ai-gateway.vercel.sh/v3/ai",
- npm: "@ai-sdk/gateway",
- },
- })
- expect(
- ProviderTransform.providerOptions(model, {
- gateway: { order: ["vertex", "anthropic"] },
- thinking: { type: "enabled", budgetTokens: 12_000 },
- }),
- ).toEqual({
- gateway: { order: ["vertex", "anthropic"] },
- anthropic: { thinking: { type: "enabled", budgetTokens: 12_000 } },
- } as any)
- })
- test("falls back to gateway key when model id has no provider slug", () => {
- const model = createModel({
- id: "claude-sonnet-4",
- providerID: "vercel",
- api: {
- id: "claude-sonnet-4",
- url: "https://ai-gateway.vercel.sh/v3/ai",
- npm: "@ai-sdk/gateway",
- },
- })
- expect(ProviderTransform.providerOptions(model, { reasoningEffort: "high" })).toEqual({
- gateway: { reasoningEffort: "high" },
- })
- })
- test("maps amazon slug to bedrock for provider options", () => {
- const model = createModel({
- providerID: "vercel",
- api: {
- id: "amazon/nova-2-lite",
- url: "https://ai-gateway.vercel.sh/v3/ai",
- npm: "@ai-sdk/gateway",
- },
- })
- expect(ProviderTransform.providerOptions(model, { reasoningConfig: { type: "enabled" } })).toEqual({
- bedrock: { reasoningConfig: { type: "enabled" } },
- })
- })
- test("maps Bedrock Mantle provider options to OpenAI namespace", () => {
- const model = createModel({
- providerID: "amazon-bedrock",
- api: {
- id: "openai.gpt-5.5",
- url: "https://bedrock-mantle.us-east-2.api.aws/openai/v1",
- npm: "@ai-sdk/amazon-bedrock/mantle",
- },
- })
- expect(ProviderTransform.providerOptions(model, { reasoningEffort: "medium" })).toEqual({
- openai: { forceReasoning: true, reasoningEffort: "medium" },
- })
- })
- test("uses groq slug for groq models", () => {
- const model = createModel({
- providerID: "vercel",
- api: {
- id: "groq/llama-3.3-70b-versatile",
- url: "https://ai-gateway.vercel.sh/v3/ai",
- npm: "@ai-sdk/gateway",
- },
- })
- expect(ProviderTransform.providerOptions(model, { reasoningFormat: "parsed" })).toEqual({
- groq: { reasoningFormat: "parsed" },
- })
- })
- })
- 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.schema - gemini nested array items", () => {
- const geminiModel = {
- providerID: "google",
- api: {
- id: "gemini-3-pro",
- },
- } as any
- test("adds type to 2D array with empty inner items", () => {
- const schema = {
- type: "object",
- properties: {
- values: {
- type: "array",
- items: {
- type: "array",
- items: {}, // Empty items object
- },
- },
- },
- } as any
- const result = ProviderTransform.schema(geminiModel, schema) as any
- // Inner items should have a default type
- expect(result.properties.values.items.items.type).toBe("string")
- })
- test("adds items and type to 2D array with missing inner items", () => {
- const schema = {
- type: "object",
- properties: {
- data: {
- type: "array",
- items: { type: "array" }, // No items at all
- },
- },
- } as any
- const result = ProviderTransform.schema(geminiModel, schema) as any
- expect(result.properties.data.items.items).toBeDefined()
- expect(result.properties.data.items.items.type).toBe("string")
- })
- test("handles deeply nested arrays (3D)", () => {
- const schema = {
- type: "object",
- properties: {
- matrix: {
- type: "array",
- items: {
- type: "array",
- items: {
- type: "array",
- // No items
- },
- },
- },
- },
- } as any
- const result = ProviderTransform.schema(geminiModel, schema) as any
- expect(result.properties.matrix.items.items.items).toBeDefined()
- expect(result.properties.matrix.items.items.items.type).toBe("string")
- })
- test("preserves existing item types in nested arrays", () => {
- const schema = {
- type: "object",
- properties: {
- numbers: {
- type: "array",
- items: {
- type: "array",
- items: { type: "number" }, // Has explicit type
- },
- },
- },
- } as any
- const result = ProviderTransform.schema(geminiModel, schema) as any
- // Should preserve the explicit type
- expect(result.properties.numbers.items.items.type).toBe("number")
- })
- test("handles mixed nested structures with objects and arrays", () => {
- const schema = {
- type: "object",
- properties: {
- spreadsheetData: {
- type: "object",
- properties: {
- rows: {
- type: "array",
- items: {
- type: "array",
- items: {}, // Empty items
- },
- },
- },
- },
- },
- } as any
- const result = ProviderTransform.schema(geminiModel, schema) as any
- expect(result.properties.spreadsheetData.properties.rows.items.items.type).toBe("string")
- })
- })
- describe("ProviderTransform.schema - gemini type arrays", () => {
- // Mirrors @ai-sdk/google's convertJSONSchemaToOpenAPISchema: JSON Schema type
- // arrays (e.g. `["number","string"]`, common in MCP tool schemas) become an
- // `anyOf` of single-type schemas, with `null` lifted into `nullable`. Plain
- // @ai-sdk/google rewrites these, but OpenAI-compatible transports such as
- // GitHub Copilot (proxying to Gemini) forward them verbatim and the backend
- // rejects the array form.
- const geminiModel = {
- providerID: "google",
- api: {
- id: "gemini-3-pro",
- },
- } as any
- test("splits a multi-type array into anyOf and drops the type array", () => {
- const schema = {
- type: "object",
- properties: {
- status: { type: ["number", "string"], description: "status filter" },
- },
- } as any
- const result = ProviderTransform.schema(geminiModel, schema) as any
- expect(result.properties.status.type).toBeUndefined()
- expect(result.properties.status.anyOf).toEqual([{ type: "number" }, { type: "string" }])
- expect(result.properties.status.nullable).toBeUndefined()
- // Sibling keywords stay alongside the generated anyOf.
- expect(result.properties.status.description).toBe("status filter")
- })
- test("lifts null into nullable for a nullable type array", () => {
- const schema = {
- type: "object",
- properties: {
- maybe: { type: ["string", "null"], description: "nullable string" },
- },
- } as any
- const result = ProviderTransform.schema(geminiModel, schema) as any
- expect(result.properties.maybe.type).toBeUndefined()
- expect(result.properties.maybe.anyOf).toEqual([{ type: "string" }])
- expect(result.properties.maybe.nullable).toBe(true)
- })
- test("collapses an all-null type array to type null", () => {
- const schema = {
- type: "object",
- properties: {
- nothing: { type: ["null"] },
- },
- } as any
- const result = ProviderTransform.schema(geminiModel, schema) as any
- expect(result.properties.nothing.type).toBe("null")
- expect(result.properties.nothing.anyOf).toBeUndefined()
- })
- test("rewrites type arrays for gemini served through github-copilot", () => {
- const copilotGeminiModel = {
- providerID: "github-copilot",
- api: {
- id: "gemini-3.5-flash",
- npm: "@ai-sdk/github-copilot",
- },
- } as any
- const schema = {
- type: "object",
- properties: {
- hook_id: { type: "number", description: "ID of the webhook" },
- status: { type: ["number", "string"], description: "Filter by response status code" },
- },
- required: ["hook_id"],
- additionalProperties: false,
- } as any
- const result = ProviderTransform.schema(copilotGeminiModel, schema) as any
- expect(result.properties.status.anyOf).toEqual([{ type: "number" }, { type: "string" }])
- expect(result.properties.status.type).toBeUndefined()
- expect(result.properties.hook_id.type).toBe("number")
- })
- })
- describe("ProviderTransform.schema - gemini combiner nodes", () => {
- const geminiModel = {
- providerID: "google",
- api: {
- id: "gemini-3-pro",
- },
- } as any
- const walk = (node: any, cb: (node: any, path: (string | number)[]) => void, path: (string | number)[] = []) => {
- if (node === null || typeof node !== "object") {
- return
- }
- if (Array.isArray(node)) {
- node.forEach((item, i) => walk(item, cb, [...path, i]))
- return
- }
- cb(node, path)
- Object.entries(node).forEach(([key, value]) => walk(value, cb, [...path, key]))
- }
- test("keeps edits.items.anyOf without adding type", () => {
- const schema = {
- type: "object",
- properties: {
- edits: {
- type: "array",
- items: {
- anyOf: [
- {
- type: "object",
- properties: {
- old_string: { type: "string" },
- new_string: { type: "string" },
- },
- required: ["old_string", "new_string"],
- },
- {
- type: "object",
- properties: {
- old_string: { type: "string" },
- new_string: { type: "string" },
- replace_all: { type: "boolean" },
- },
- required: ["old_string", "new_string"],
- },
- ],
- },
- },
- },
- required: ["edits"],
- } as any
- const result = ProviderTransform.schema(geminiModel, schema) as any
- expect(Array.isArray(result.properties.edits.items.anyOf)).toBe(true)
- expect(result.properties.edits.items.type).toBeUndefined()
- })
- test("does not add sibling keys to combiner nodes during sanitize", () => {
- const schema = {
- type: "object",
- properties: {
- edits: {
- type: "array",
- items: {
- anyOf: [{ type: "string" }, { type: "number" }],
- },
- },
- value: {
- oneOf: [{ type: "string" }, { type: "boolean" }],
- },
- meta: {
- allOf: [
- {
- type: "object",
- properties: { a: { type: "string" } },
- },
- {
- type: "object",
- properties: { b: { type: "string" } },
- },
- ],
- },
- },
- } as any
- const input = JSON.parse(JSON.stringify(schema))
- const result = ProviderTransform.schema(geminiModel, schema) as any
- walk(result, (node, path) => {
- const hasCombiner = Array.isArray(node.anyOf) || Array.isArray(node.oneOf) || Array.isArray(node.allOf)
- if (!hasCombiner) {
- return
- }
- const before = path.reduce((acc: any, key) => acc?.[key], input)
- const added = Object.keys(node).filter((key) => !(key in before))
- expect(added).toEqual([])
- })
- })
- })
- describe("ProviderTransform.schema - gemini non-object properties removal", () => {
- const geminiModel = {
- providerID: "google",
- api: {
- id: "gemini-3-pro",
- },
- } as any
- test("removes properties from non-object types", () => {
- const schema = {
- type: "object",
- properties: {
- data: {
- type: "string",
- properties: { invalid: { type: "string" } },
- },
- },
- } as any
- const result = ProviderTransform.schema(geminiModel, schema) as any
- expect(result.properties.data.type).toBe("string")
- expect(result.properties.data.properties).toBeUndefined()
- })
- test("removes required from non-object types", () => {
- const schema = {
- type: "object",
- properties: {
- data: {
- type: "array",
- items: { type: "string" },
- required: ["invalid"],
- },
- },
- } as any
- const result = ProviderTransform.schema(geminiModel, schema) as any
- expect(result.properties.data.type).toBe("array")
- expect(result.properties.data.required).toBeUndefined()
- })
- test("removes properties and required from nested non-object types", () => {
- const schema = {
- type: "object",
- properties: {
- outer: {
- type: "object",
- properties: {
- inner: {
- type: "number",
- properties: { bad: { type: "string" } },
- required: ["bad"],
- },
- },
- },
- },
- } as any
- const result = ProviderTransform.schema(geminiModel, schema) as any
- expect(result.properties.outer.properties.inner.type).toBe("number")
- expect(result.properties.outer.properties.inner.properties).toBeUndefined()
- expect(result.properties.outer.properties.inner.required).toBeUndefined()
- })
- test("keeps properties and required on object types", () => {
- const schema = {
- type: "object",
- properties: {
- data: {
- type: "object",
- properties: { name: { type: "string" } },
- required: ["name"],
- },
- },
- } as any
- const result = ProviderTransform.schema(geminiModel, schema) as any
- expect(result.properties.data.type).toBe("object")
- expect(result.properties.data.properties).toBeDefined()
- expect(result.properties.data.required).toEqual(["name"])
- })
- test("does not affect non-gemini providers", () => {
- const openaiModel = {
- providerID: "openai",
- api: {
- id: "gpt-4",
- },
- } as any
- const schema = {
- type: "object",
- properties: {
- data: {
- type: "string",
- properties: { invalid: { type: "string" } },
- },
- },
- } as any
- const result = ProviderTransform.schema(openaiModel, schema) as any
- expect(result.properties.data.properties).toBeDefined()
- })
- })
- describe("ProviderTransform.schema - openai supported schema subset", () => {
- const openaiModel = {
- providerID: "openai",
- api: {
- id: "gpt-4.1",
- npm: "@ai-sdk/openai",
- },
- } as any
- test("removes unsupported JSON Schema keywords recursively", () => {
- const result = ProviderTransform.schema(openaiModel, {
- $schema: "https://json-schema.org/draft/2020-12/schema",
- title: "Search",
- type: "object",
- properties: {
- query: {
- type: "string",
- description: "Search query",
- format: "uri",
- pattern: "^https://",
- minLength: 1,
- maxLength: 100,
- default: "https://example.com",
- },
- count: {
- type: "integer",
- minimum: 1,
- maximum: 10,
- multipleOf: 1,
- },
- createdAt: {
- format: "date-time",
- },
- mode: {
- const: "fast",
- },
- tags: {
- type: "array",
- minItems: 1,
- maxItems: 3,
- uniqueItems: true,
- },
- tuple: {
- type: "array",
- items: [
- { type: "number", minimum: 0 },
- { type: "string", pattern: "^ok$" },
- ],
- },
- metadata: {
- type: "object",
- patternProperties: {
- "^x-": { type: "string" },
- },
- additionalProperties: {
- type: "string",
- pattern: "^safe$",
- },
- },
- },
- patternProperties: {
- "^extra": { type: "string" },
- },
- required: ["query"],
- additionalProperties: false,
- } as any) as any
- expect(result).toEqual({
- type: "object",
- properties: {
- query: {
- type: "string",
- description: "Search query",
- },
- count: {
- type: "integer",
- },
- createdAt: {
- type: "string",
- },
- mode: {
- enum: ["fast"],
- type: "string",
- },
- tags: {
- type: "array",
- items: { type: "string" },
- },
- tuple: {
- type: "array",
- items: [{ type: "number" }, { type: "string" }],
- },
- metadata: {
- type: "object",
- properties: {},
- additionalProperties: {
- type: "string",
- },
- },
- },
- required: ["query"],
- additionalProperties: false,
- })
- })
- test("keeps local references and sanitizes definitions", () => {
- const result = ProviderTransform.schema(openaiModel, {
- type: "object",
- properties: {
- value: {
- $ref: "#/$defs/Value",
- description: "Referenced value",
- examples: ["ignored"],
- },
- },
- $defs: {
- Value: {
- type: "string",
- pattern: "^value$",
- description: "Definition description",
- },
- Unused: {
- type: "number",
- minimum: 0,
- },
- },
- } as any) as any
- expect(result.properties.value).toEqual({
- $ref: "#/$defs/Value",
- description: "Referenced value",
- })
- expect(result.$defs).toEqual({
- Value: {
- type: "string",
- description: "Definition description",
- },
- Unused: {
- type: "number",
- },
- })
- })
- test("does not sanitize non-openai providers", () => {
- const result = ProviderTransform.schema(
- {
- providerID: "anthropic",
- api: {
- id: "claude-sonnet-4",
- npm: "@ai-sdk/anthropic",
- },
- } as any,
- {
- type: "object",
- properties: {
- query: {
- type: "string",
- pattern: "^https://",
- },
- },
- } as any,
- ) as any
- expect(result.properties.query.pattern).toBe("^https://")
- })
- test.each([
- ["opencode", "@ai-sdk/openai"],
- ["custom-openai-compatible", "@ai-sdk/openai"],
- ["azure", "@ai-sdk/azure"],
- ])("sanitizes %s models using %s", (providerID, npm) => {
- expect(
- ProviderTransform.schema(
- {
- providerID,
- api: {
- id: "custom-model",
- npm,
- },
- } as any,
- {
- type: "object",
- properties: {
- query: {
- type: "string",
- pattern: "^https://",
- },
- },
- } as any,
- ),
- ).toEqual({
- type: "object",
- properties: {
- query: {
- type: "string",
- },
- },
- })
- })
- })
- describe("ProviderTransform.schema - moonshot $ref siblings", () => {
- const moonshotModel = {
- providerID: "moonshotai",
- api: {
- id: "kimi-k2",
- },
- } as any
- test("removes sibling descriptions from referenced tool parameter schemas", () => {
- const schema = {
- type: "object",
- properties: {
- deviceType: {
- description: "Optional. The type of device that captured the screenshot, e.g. mobile or desktop.",
- enum: ["DEVICE_TYPE_UNSPECIFIED", "MOBILE", "DESKTOP", "TABLET", "AGNOSTIC"],
- type: "string",
- },
- modelId: {
- description: "Optional. The model to use for generation.",
- enum: ["MODEL_ID_UNSPECIFIED", "GEMINI_3_PRO", "GEMINI_3_FLASH", "GEMINI_3_1_PRO"],
- type: "string",
- },
- projectId: {
- description: "Required. The project ID of screens to generate variants for.",
- type: "string",
- },
- prompt: {
- description: "Required. The input text used to generate the variants.",
- type: "string",
- },
- selectedScreenIds: {
- description: "Required. The screen ids of screen to generate variants for.",
- items: {
- type: "string",
- },
- type: "array",
- },
- variantOptions: {
- $ref: "#/$defs/VariantOptions",
- description:
- "Required. The variant options for generation, including the number of variants, creative range, and aspects to focus on.",
- },
- },
- required: ["projectId", "selectedScreenIds", "prompt", "variantOptions"],
- $defs: {
- VariantOptions: {
- description:
- "Configuration options for design variant generation. This message captures all parameters used to generate variants, allowing the configuration to be stored, replayed, or analyzed.",
- properties: {
- aspects: {
- description: "Optional. Specific aspects to focus on. If empty, all aspects may be varied.",
- items: {
- enum: ["VARIANT_ASPECT_UNSPECIFIED", "LAYOUT", "COLOR_SCHEME", "IMAGES", "TEXT_FONT", "TEXT_CONTENT"],
- type: "string",
- },
- type: "array",
- },
- creativeRange: {
- description: "Optional. Creative range for variations. Default: EXPLORE",
- enum: ["CREATIVE_RANGE_UNSPECIFIED", "REFINE", "EXPLORE", "REIMAGINE"],
- type: "string",
- },
- variantCount: {
- description: "Optional. Number of variants to generate (1-5). Default: 3",
- format: "int32",
- type: "integer",
- },
- },
- type: "object",
- },
- },
- description: "Request message for GenerateVariants.",
- additionalProperties: false,
- } as any
- const result = ProviderTransform.schema(moonshotModel, schema) as any
- expect(result.properties.variantOptions).toEqual({
- $ref: "#/$defs/VariantOptions",
- })
- expect(result.$defs.VariantOptions.description).toBe(schema.$defs.VariantOptions.description)
- })
- test("also runs for kimi models outside the moonshot provider", () => {
- const result = ProviderTransform.schema(
- {
- providerID: "openrouter",
- name: "Kimi K2",
- api: {
- id: "moonshotai/kimi-k2",
- },
- } as any,
- {
- type: "object",
- properties: {
- value: {
- $ref: "#/$defs/Value",
- description: "Moonshot rejects this sibling after ref expansion.",
- },
- },
- $defs: {
- Value: {
- description: "Referenced schema description stays here.",
- type: "object",
- },
- },
- } as any,
- ) as any
- expect(result.properties.value).toEqual({
- $ref: "#/$defs/Value",
- })
- })
- test("converts tuple-style array items to a single item schema", () => {
- const result = ProviderTransform.schema(moonshotModel, {
- type: "object",
- properties: {
- codeSpec: {
- type: "object",
- properties: {
- accessibility: {
- type: "object",
- properties: {
- renderedSize: {
- description: "Rendered size [width, height] in px",
- type: "array",
- items: [{ type: "number" }, { type: "number" }],
- minItems: 2,
- maxItems: 2,
- },
- },
- },
- },
- },
- },
- } as any) as any
- expect(result.properties.codeSpec.properties.accessibility.properties.renderedSize.items).toEqual({
- type: "number",
- })
- })
- })
- describe("ProviderTransform.message - Mistral tool call IDs", () => {
- test.each(["codestral-latest", "pixtral-large-latest", "open-mixtral-8x22b"])(
- "normalizes IDs for custom OpenAI-compatible %s models",
- (id) => {
- const result = ProviderTransform.message(
- [
- {
- role: "assistant",
- content: [
- {
- type: "tool-call",
- toolCallId: "toolu_01CBhTTz95qkd9LJMdC9sf8t",
- toolName: "read",
- input: { filePath: "/tmp/test" },
- },
- ],
- },
- {
- role: "tool",
- content: [
- {
- type: "tool-result",
- toolCallId: "toolu_01CBhTTz95qkd9LJMdC9sf8t",
- toolName: "read",
- output: { type: "text", value: "test" },
- },
- ],
- },
- ] as any,
- {
- id: `custom/${id}`,
- providerID: "custom",
- api: {
- id,
- url: "https://example.com/v1",
- npm: "@ai-sdk/openai-compatible",
- },
- } as any,
- {},
- )
- expect(result).toMatchObject([
- { role: "assistant", content: [{ type: "tool-call", toolCallId: "toolu01CB" }] },
- { role: "tool", content: [{ type: "tool-result", toolCallId: "toolu01CB" }] },
- ])
- },
- )
- })
- 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: ModelV2.ID.make("deepseek/deepseek-chat"),
- providerID: ProviderV2.ID.make("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: ModelV2.ID.make("openai/gpt-4"),
- providerID: ProviderV2.ID.make("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 - surrogate sanitization", () => {
- const model = {
- id: "test/test-model",
- providerID: "test",
- api: {
- id: "test-model",
- url: "https://api.test.com",
- npm: "@ai-sdk/openai-compatible",
- },
- 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: 128000, output: 8192 },
- status: "active",
- options: {},
- headers: {},
- } as any
- test("replaces lone surrogates in model-visible text", () => {
- const lone = "\uD83D"
- const valid = "🚀"
- const sanitized = "�"
- const text = (label: string) => `${label} ${lone} and ${valid}`
- const expected = (label: string) => `${label} ${sanitized} and ${valid}`
- const msgs = [
- { role: "system", content: text("system") },
- { role: "user", content: text("user string") },
- {
- role: "user",
- content: [
- { type: "text", text: text("user text") },
- { type: "image", image: "data:image/png;base64,abcd" },
- ],
- },
- { role: "assistant", content: text("assistant string") },
- {
- role: "assistant",
- content: [
- { type: "text", text: text("assistant text") },
- { type: "reasoning", text: text("assistant reasoning") },
- { type: "tool-call", toolCallId: "call-1", toolName: "Read", input: { filePath: ".opencode/tool/emoji.ts" } },
- {
- type: "tool-result",
- toolCallId: "call-2",
- toolName: "Read",
- output: { type: "text", value: text("assistant tool text") },
- },
- {
- type: "tool-result",
- toolCallId: "call-3",
- toolName: "Read",
- output: { type: "error-text", value: text("assistant tool error") },
- },
- {
- type: "tool-result",
- toolCallId: "call-4",
- toolName: "Read",
- output: { type: "content", value: [{ type: "text", text: text("assistant tool content") }] },
- },
- ],
- },
- {
- role: "tool",
- content: [
- {
- type: "tool-result",
- toolCallId: "call-5",
- toolName: "Read",
- output: { type: "text", value: text("tool text") },
- },
- {
- type: "tool-result",
- toolCallId: "call-6",
- toolName: "Read",
- output: { type: "error-text", value: text("tool error") },
- },
- {
- type: "tool-result",
- toolCallId: "call-7",
- toolName: "Read",
- output: { type: "content", value: [{ type: "text", text: text("tool content") }] },
- },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, model, {}) as any[]
- expect(result[0].content).toBe(expected("system"))
- expect(result[1].content).toBe(expected("user string"))
- expect(result[2].content[0].text).toBe(expected("user text"))
- expect(result[3].content).toBe(expected("assistant string"))
- expect(result[4].content[0].text).toBe(expected("assistant text"))
- expect(result[4].content[1].text).toBe(expected("assistant reasoning"))
- expect(result[4].content[3].output.value).toBe(expected("assistant tool text"))
- expect(result[4].content[4].output.value).toBe(expected("assistant tool error"))
- expect(result[4].content[5].output.value[0].text).toBe(expected("assistant tool content"))
- expect(result[5].content[0].output.value).toBe(expected("tool text"))
- expect(result[5].content[1].output.value).toBe(expected("tool error"))
- expect(result[5].content[2].output.value[0].text).toBe(expected("tool content"))
- expect(result[2].content[1]).toEqual({ type: "image", image: "data:image/png;base64,abcd" })
- })
- })
- 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("filters empty content for bedrock provider", () => {
- const bedrockModel = {
- ...anthropicModel,
- id: "amazon-bedrock/anthropic.claude-opus-4-6",
- providerID: "amazon-bedrock",
- api: {
- id: "anthropic.claude-opus-4-6",
- url: "https://bedrock-runtime.us-east-1.amazonaws.com",
- npm: "@ai-sdk/amazon-bedrock",
- },
- }
- const msgs = [
- { role: "user", content: "Hello" },
- { role: "assistant", content: "" },
- {
- role: "assistant",
- content: [
- { type: "text", text: "" },
- { type: "text", text: "Answer" },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, bedrockModel, {})
- expect(result).toHaveLength(2)
- expect(result[0].content).toBe("Hello")
- expect(result[1].content).toHaveLength(1)
- expect(result[1].content[0]).toEqual({ type: "text", text: "Answer" })
- })
- 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)
- })
- test("leaves valid anthropic assistant tool ordering unchanged", () => {
- const msgs = [
- {
- role: "assistant",
- content: [
- { type: "text", text: "I checked your home directory and looked for PDF files." },
- { type: "tool-call", toolCallId: "toolu_1", toolName: "read", input: { filePath: "/root" } },
- { type: "tool-call", toolCallId: "toolu_2", toolName: "glob", input: { pattern: "**/*.pdf" } },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, anthropicModel, {}) as any[]
- expect(result).toHaveLength(1)
- expect(result[0].content).toMatchObject([
- { type: "text", text: "I checked your home directory and looked for PDF files." },
- { type: "tool-call", toolCallId: "toolu_1", toolName: "read", input: { filePath: "/root" } },
- { type: "tool-call", toolCallId: "toolu_2", toolName: "glob", input: { pattern: "**/*.pdf" } },
- ])
- })
- })
- 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("strips OpenAI itemId and preserves 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).toBeUndefined()
- expect(result[0].content[0].providerOptions?.openai?.reasoningEncryptedContent).toBe("encrypted")
- expect(result[0].content[1].providerOptions?.openai?.itemId).toBeUndefined()
- })
- test("uses the SDK package namespace rather than provider ID", () => {
- 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).toBeUndefined()
- expect(result[0].content[0].providerOptions?.openai?.reasoningEncryptedContent).toBe("encrypted")
- expect(result[0].content[1].providerOptions?.openai?.itemId).toBeUndefined()
- })
- test("preserves other OpenAI options", () => {
- 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).toBeUndefined()
- expect(result[0].content[0].providerOptions?.openai?.otherOption).toBe("value")
- })
- test("strips Azure itemId from the Azure namespace", () => {
- const azureModel = {
- ...openaiModel,
- providerID: "azure",
- api: {
- id: "gpt-5",
- url: "https://example.openai.azure.com",
- npm: "@ai-sdk/azure",
- },
- }
- const msgs = [
- {
- role: "assistant",
- content: [
- {
- type: "text",
- text: "Hello",
- providerOptions: {
- azure: { itemId: "msg_123", otherOption: "value" },
- openai: { itemId: "msg_openai" },
- },
- },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, azureModel, { store: false }) as any[]
- expect(result[0].content[0].providerOptions?.azure?.itemId).toBeUndefined()
- expect(result[0].content[0].providerOptions?.azure?.otherOption).toBe("value")
- expect(result[0].content[0].providerOptions?.openai?.itemId).toBe("msg_openai")
- })
- test("strips Bedrock Mantle itemId from the OpenAI namespace", () => {
- const mantleModel = {
- ...openaiModel,
- providerID: "amazon-bedrock",
- api: {
- id: "openai.gpt-5.5",
- url: "https://bedrock-mantle.us-east-2.api.aws/openai/v1",
- npm: "@ai-sdk/amazon-bedrock/mantle",
- },
- }
- const msgs = [
- {
- role: "assistant",
- providerOptions: { openai: { itemId: "msg_root", otherOption: "root-value" } },
- content: [
- {
- type: "reasoning",
- text: "thinking...",
- providerOptions: {
- openai: { itemId: "rs_123", reasoningEncryptedContent: "encrypted" },
- },
- },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, mantleModel, { store: false }) as any[]
- expect(result[0].providerOptions?.openai?.itemId).toBeUndefined()
- expect(result[0].providerOptions?.openai?.otherOption).toBe("root-value")
- expect(result[0].content[0].providerOptions?.openai?.itemId).toBeUndefined()
- expect(result[0].content[0].providerOptions?.openai?.reasoningEncryptedContent).toBe("encrypted")
- })
- test("strips GitHub Copilot itemId from the copilot namespace, preserving other copilot options", () => {
- const copilotModel = {
- ...openaiModel,
- id: "github-copilot/gpt-5.5",
- providerID: "github-copilot",
- api: {
- id: "gpt-5.5",
- url: "https://api.githubcopilot.com",
- npm: "@ai-sdk/github-copilot",
- },
- }
- const msgs = [
- {
- role: "assistant",
- content: [
- {
- type: "reasoning",
- text: "thinking...",
- providerOptions: {
- copilot: { itemId: "rs_123", reasoningEncryptedContent: "encrypted" },
- },
- },
- {
- // The stale itemId on tool-call parts is what Copilot echoes back as the
- // `function_call` item `id`, which is what the upstream connection rejects.
- type: "tool-call",
- toolCallId: "call_1",
- toolName: "bash",
- input: { command: "ls" },
- providerOptions: {
- copilot: { itemId: "fc_456", reasoningEffort: "medium" },
- },
- },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, copilotModel, { store: false }) as any[]
- expect(result[0].content[0].providerOptions?.copilot?.itemId).toBeUndefined()
- expect(result[0].content[0].providerOptions?.copilot?.reasoningEncryptedContent).toBe("encrypted")
- expect(result[0].content[1].providerOptions?.copilot?.itemId).toBeUndefined()
- expect(result[0].content[1].providerOptions?.copilot?.reasoningEffort).toBe("medium")
- })
- test("leaves a stray openai namespace on a Copilot model untouched, since Copilot's Responses model only reads the copilot namespace", () => {
- const copilotModel = {
- ...openaiModel,
- id: "github-copilot/gpt-5.5",
- providerID: "github-copilot",
- api: {
- id: "gpt-5.5",
- url: "https://api.githubcopilot.com",
- npm: "@ai-sdk/github-copilot",
- },
- }
- const msgs = [
- {
- role: "assistant",
- content: [
- {
- type: "text",
- text: "Hello",
- providerOptions: {
- openai: { itemId: "msg_456" },
- },
- },
- ],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, copilotModel, { store: false }) as any[]
- expect(result[0].content[0].providerOptions?.openai?.itemId).toBe("msg_456")
- })
- 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("azure cognitive services remaps providerID to 'azure' key", () => {
- const model = createModel("azure-cognitive-services", "@ai-sdk/azure")
- const msgs = [
- {
- role: "user",
- content: [
- {
- type: "text",
- text: "Hello",
- providerOptions: {
- "azure-cognitive-services": { part: true },
- },
- },
- ],
- providerOptions: {
- "azure-cognitive-services": { someOption: "value" },
- },
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, model, {}) as any[]
- const part = result[0].content[0] as any
- expect(result[0].providerOptions?.azure).toEqual({ someOption: "value" })
- expect(result[0].providerOptions?.["azure-cognitive-services"]).toBeUndefined()
- expect(part.providerOptions?.azure).toEqual({ part: true })
- expect(part.providerOptions?.["azure-cognitive-services"]).toBeUndefined()
- })
- test("copilot remaps providerID to 'copilot' key", () => {
- const model = createModel("github-copilot", "@ai-sdk/github-copilot")
- const msgs = [
- {
- role: "user",
- content: "Hello",
- providerOptions: {
- copilot: { someOption: "value" },
- },
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, model, {})
- expect(result[0].providerOptions?.copilot).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: "default",
- },
- }),
- )
- })
- })
- describe("ProviderTransform.message - bedrock caching with non-bedrock providerID", () => {
- test("applies cache options at message level when npm package is amazon-bedrock", () => {
- const model = {
- id: "aws/us.anthropic.claude-opus-4-6-v1",
- providerID: "aws",
- api: {
- id: "us.anthropic.claude-opus-4-6-v1",
- url: "https://bedrock-runtime.us-east-1.amazonaws.com",
- npm: "@ai-sdk/amazon-bedrock",
- },
- name: "Claude Opus 4.6",
- capabilities: {},
- options: {},
- headers: {},
- } as any
- const msgs = [
- {
- role: "system",
- content: "You are a helpful assistant",
- },
- {
- role: "user",
- content: [{ type: "text", text: "Hello" }],
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, model, {}) as any[]
- // Cache should be at the message level and not the content-part level
- expect(result[0].providerOptions?.bedrock).toEqual({
- cachePoint: { type: "default" },
- })
- expect(result[0].content).toBe("You are a helpful assistant")
- })
- })
- describe("ProviderTransform.message - cache control on gateway", () => {
- const createModel = (overrides: Partial<any> = {}) =>
- ({
- id: "anthropic/claude-sonnet-4",
- providerID: "vercel",
- api: {
- id: "anthropic/claude-sonnet-4",
- url: "https://ai-gateway.vercel.sh/v3/ai",
- npm: "@ai-sdk/gateway",
- },
- name: "Claude Sonnet 4",
- capabilities: {
- temperature: true,
- reasoning: true,
- 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: 200_000, output: 8192 },
- status: "active",
- options: {},
- headers: {},
- ...overrides,
- }) as any
- test("gateway does not set cache control for anthropic models", () => {
- const model = createModel()
- const msgs = [
- {
- role: "system",
- content: "You are a helpful assistant",
- },
- {
- role: "user",
- content: "Hello",
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, model, {}) as any[]
- expect(result[0].content).toBe("You are a helpful assistant")
- expect(result[0].providerOptions).toBeUndefined()
- })
- test("non-gateway anthropic keeps existing cache control behavior", () => {
- const model = createModel({
- providerID: "anthropic",
- api: {
- id: "claude-sonnet-4",
- url: "https://api.anthropic.com",
- npm: "@ai-sdk/anthropic",
- },
- })
- const msgs = [
- {
- role: "system",
- content: "You are a helpful assistant",
- },
- {
- role: "user",
- content: "Hello",
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, model, {}) as any[]
- expect(result[0].providerOptions).toEqual({
- anthropic: {
- cacheControl: {
- type: "ephemeral",
- },
- },
- openrouter: {
- cacheControl: {
- type: "ephemeral",
- },
- },
- bedrock: {
- cachePoint: {
- type: "default",
- },
- },
- openaiCompatible: {
- cache_control: {
- type: "ephemeral",
- },
- },
- copilot: {
- copilot_cache_control: {
- type: "ephemeral",
- },
- },
- alibaba: {
- cacheControl: {
- type: "ephemeral",
- },
- },
- })
- })
- test("does not add explicit breakpoints when Anthropic automatic caching is enabled", () => {
- const model = createModel({
- providerID: "anthropic",
- api: { id: "claude-sonnet-4", url: "https://api.anthropic.com", npm: "@ai-sdk/anthropic" },
- })
- const msgs = [
- { role: "system", content: "You are a helpful assistant" },
- { role: "user", content: "Hello" },
- ] as any[]
- const result = ProviderTransform.message(msgs, model, { cacheControl: { type: "ephemeral" } }) as any[]
- expect(result.every((message) => message.providerOptions === undefined)).toBe(true)
- })
- test("google-vertex-anthropic applies cache control", () => {
- const model = createModel({
- providerID: "google-vertex-anthropic",
- api: {
- id: "google-vertex-anthropic",
- url: "https://us-central1-aiplatform.googleapis.com",
- npm: "@ai-sdk/google-vertex/anthropic",
- },
- id: "claude-sonnet-4@20250514",
- })
- const msgs = [
- {
- role: "system",
- content: "You are a helpful assistant",
- },
- {
- role: "user",
- content: "Hello",
- },
- ] as any[]
- const result = ProviderTransform.message(msgs, model, {}) as any[]
- expect(result[0].providerOptions).toEqual({
- anthropic: {
- cacheControl: {
- type: "ephemeral",
- },
- },
- openrouter: {
- cacheControl: {
- type: "ephemeral",
- },
- },
- bedrock: {
- cachePoint: {
- type: "default",
- },
- },
- openaiCompatible: {
- cache_control: {
- type: "ephemeral",
- },
- },
- copilot: {
- copilot_cache_control: {
- type: "ephemeral",
- },
- },
- alibaba: {
- cacheControl: {
- type: "ephemeral",
- },
- },
- })
- })
- })
- describe("ProviderTransform.temperature - Cohere North", () => {
- test("defaults north-mini-code models to 1.0", () => {
- expect(ProviderTransform.temperature({ id: "cohere/North-Mini-Code-1-0-latest" } as any)).toBe(1.0)
- })
- })
- describe("ProviderTransform.reasoningVariants", () => {
- const model = (reasoning_options: ModelsDev.Model["reasoning_options"]) => ({ reasoning_options }) as ModelsDev.Model
- const target = (npm: string, id = "test-model") =>
- ({
- id,
- providerID: "test",
- api: { id, npm, url: "" },
- capabilities: { reasoning: true },
- limit: { output: 64_000 },
- }) as any
- test("respects explicitly empty reasoning options", () => {
- expect(ProviderTransform.reasoningVariants(model([]), target("@ai-sdk/openai"))).toEqual({})
- })
- test.each([
- ["@openrouter/ai-sdk-provider", { reasoning: { effort: "high" } }],
- ["@ai-sdk/anthropic", { thinking: { type: "adaptive" }, effort: "high" }, "claude-opus-4-6"],
- [
- "@ai-sdk/google-vertex/anthropic",
- { thinking: { type: "adaptive", display: "summarized" }, effort: "high" },
- "claude-opus-4-7",
- ],
- ["@ai-sdk/google", { thinkingConfig: { includeThoughts: true, thinkingLevel: "high" } }],
- ["@ai-sdk/google-vertex", { thinkingConfig: { includeThoughts: true, thinkingLevel: "high" } }],
- [
- "@ai-sdk/azure",
- {
- reasoningEffort: "high",
- reasoningSummary: "auto",
- include: ["reasoning.encrypted_content"],
- },
- ],
- [
- "@ai-sdk/openai",
- {
- reasoningEffort: "high",
- reasoningSummary: "auto",
- include: ["reasoning.encrypted_content"],
- },
- ],
- [
- "@ai-sdk/amazon-bedrock/mantle",
- {
- reasoningEffort: "high",
- reasoningSummary: "auto",
- include: ["reasoning.encrypted_content"],
- },
- ],
- [
- "@ai-sdk/github-copilot",
- {
- reasoningEffort: "high",
- reasoningSummary: "auto",
- include: ["reasoning.encrypted_content"],
- },
- ],
- ["@ai-sdk/openai-compatible", { reasoningEffort: "high" }],
- ["@ai-sdk/xai", { reasoningEffort: "high" }],
- ["@ai-sdk/mistral", { reasoningEffort: "high" }],
- ["@ai-sdk/groq", { reasoningEffort: "high" }],
- ["@ai-sdk/cerebras", { reasoningEffort: "high" }],
- ["@ai-sdk/deepinfra", { reasoningEffort: "high" }],
- ["@ai-sdk/togetherai", { reasoningEffort: "high" }],
- ["venice-ai-sdk-provider", { reasoningEffort: "high" }],
- ["ai-gateway-provider", { reasoningEffort: "high" }],
- ["@ai-sdk/amazon-bedrock", { reasoningConfig: { type: "enabled", maxReasoningEffort: "high" } }],
- ])("converts effort for %s", (npm, expected, ...args) => {
- const id = args[0] as string | undefined
- expect(ProviderTransform.reasoningVariants(model([{ type: "effort", values: ["high"] }]), target(npm, id))).toEqual(
- { high: expected },
- )
- })
- test("uses bare effort for Claude Opus 4.5", () => {
- expect(
- ProviderTransform.reasoningVariants(
- model([{ type: "effort", values: ["high"] }]),
- target("@ai-sdk/anthropic", "claude-opus-4-5"),
- ),
- ).toEqual({ high: { effort: "high" } })
- })
- test("uses explicit effort metadata for Anthropic-compatible models", () => {
- expect(
- ProviderTransform.reasoningVariants(
- model([{ type: "effort", values: ["high"] }]),
- target("@ai-sdk/anthropic", "claude-sonnet-4"),
- ),
- ).toEqual({ high: { effort: "high" } })
- expect(
- ProviderTransform.reasoningVariants(
- model([{ type: "effort", values: ["max"] }]),
- target("@ai-sdk/anthropic", "k3"),
- ),
- ).toEqual({ max: { effort: "max" } })
- })
- test("maps Kimi effort metadata to adaptive thinking", () => {
- expect(
- ProviderTransform.reasoningVariants(
- model([{ type: "effort", values: ["low", "high", "max"] }]),
- target("@ai-sdk/anthropic", "kimi-k3"),
- ),
- ).toEqual({
- low: { thinking: { type: "adaptive", display: "summarized" }, effort: "low" },
- high: { thinking: { type: "adaptive", display: "summarized" }, effort: "high" },
- max: { thinking: { type: "adaptive", display: "summarized" }, effort: "max" },
- })
- })
- test("uses adaptive reasoning config for Anthropic models on Bedrock", () => {
- expect(
- ProviderTransform.reasoningVariants(
- model([{ type: "effort", values: ["high"] }]),
- target("@ai-sdk/amazon-bedrock", "anthropic.claude-opus-4-7-v1:0"),
- ),
- ).toEqual({
- high: {
- reasoningConfig: {
- type: "adaptive",
- maxReasoningEffort: "high",
- display: "summarized",
- },
- },
- })
- })
- test("does not replace unsupported Anthropic Bedrock effort options with token budgets", () => {
- expect(
- ProviderTransform.reasoningVariants(
- model([{ type: "effort", values: ["high"] }]),
- target("@ai-sdk/amazon-bedrock", "anthropic.claude-sonnet-4-v1:0"),
- ),
- ).toEqual({})
- })
- test.each([
- ["@openrouter/ai-sdk-provider", { reasoning: { max_tokens: 16_000 } }],
- ["@ai-sdk/anthropic", { thinking: { type: "enabled", budgetTokens: 16_000 } }],
- ["@ai-sdk/google-vertex/anthropic", { thinking: { type: "enabled", budgetTokens: 16_000 } }],
- ["@ai-sdk/google", { thinkingConfig: { includeThoughts: true, thinkingBudget: 16_000 } }],
- ["@ai-sdk/google-vertex", { thinkingConfig: { includeThoughts: true, thinkingBudget: 16_000 } }],
- ["@ai-sdk/amazon-bedrock", { reasoningConfig: { type: "enabled", budgetTokens: 16_000 } }],
- ["@ai-sdk/cohere", { thinking: { type: "enabled", tokenBudget: 16_000 } }],
- ["@ai-sdk/alibaba", { enableThinking: true, thinkingBudget: 16_000 }],
- ])("converts token budgets for %s", (npm, high) => {
- const variants = ProviderTransform.reasoningVariants(model([{ type: "budget_tokens", min: 1_024 }]), target(npm))
- expect(variants?.high).toEqual(high)
- expect(Object.keys(variants ?? {})).toEqual(["high", "max"])
- })
- test("maps null effort to none", () => {
- expect(
- ProviderTransform.reasoningVariants(model([{ type: "effort", values: [null] }]), target("@ai-sdk/openai")),
- ).toEqual({
- none: {
- reasoningEffort: "none",
- reasoningSummary: "auto",
- include: ["reasoning.encrypted_content"],
- },
- })
- })
- test.each([
- ["@ai-sdk/alibaba", { none: { enableThinking: false }, high: { enableThinking: true } }],
- [
- "@ai-sdk/cohere",
- {
- none: { thinking: { type: "disabled" } },
- high: { thinking: { type: "enabled" } },
- },
- ],
- ])("converts toggle options for %s", (npm, expected) => {
- expect(ProviderTransform.reasoningVariants(model([{ type: "toggle" }]), target(npm))).toEqual(expected)
- })
- test("combines Cohere toggle and budget options", () => {
- expect(
- ProviderTransform.reasoningVariants(
- model([{ type: "toggle" }, { type: "budget_tokens", min: 1 }]),
- target("@ai-sdk/cohere"),
- ),
- ).toEqual({
- none: { thinking: { type: "disabled" } },
- high: { thinking: { type: "enabled", tokenBudget: 16_000 } },
- max: { thinking: { type: "enabled", tokenBudget: 31_999 } },
- })
- })
- test("generates bounded high and max token budgets", () => {
- expect(
- ProviderTransform.reasoningVariants(
- model([{ type: "budget_tokens", min: 1_024, max: 64_000 }]),
- target("@ai-sdk/anthropic"),
- ),
- ).toEqual({
- high: { thinking: { type: "enabled", budgetTokens: 16_000 } },
- max: { thinking: { type: "enabled", budgetTokens: 31_999 } },
- })
- })
- test("caps token budgets below the model output limit", () => {
- const anthropic = target("@ai-sdk/anthropic")
- anthropic.limit.output = 5_000
- expect(
- ProviderTransform.reasoningVariants(model([{ type: "budget_tokens", min: 1_024, max: 64_000 }]), anthropic),
- ).toEqual({
- high: { thinking: { type: "enabled", budgetTokens: 2_500 } },
- max: { thinking: { type: "enabled", budgetTokens: 4_999 } },
- })
- })
- test("derives high and max budgets when models.dev omits max", () => {
- expect(
- ProviderTransform.reasoningVariants(
- model([{ type: "budget_tokens", min: 1_024 }]),
- target("@ai-sdk/anthropic", "claude-haiku-4-5"),
- ),
- ).toEqual({
- high: { thinking: { type: "enabled", budgetTokens: 16_000 } },
- max: { thinking: { type: "enabled", budgetTokens: 31_999 } },
- })
- })
- test("preserves explicit inclusive budget maxima", () => {
- expect(
- ProviderTransform.reasoningVariants(
- model([{ type: "budget_tokens", min: 1_024, max: 24_576 }]),
- target("@ai-sdk/google", "gemini-2.5-pro"),
- ),
- ).toEqual({
- high: { thinkingConfig: { includeThoughts: true, thinkingBudget: 12_288 } },
- max: { thinkingConfig: { includeThoughts: true, thinkingBudget: 24_576 } },
- })
- })
- test("prefers effort options over token budgets", () => {
- expect(
- ProviderTransform.reasoningVariants(
- model([
- { type: "budget_tokens", min: 1_024, max: 64_000 },
- { type: "effort", values: ["low"] },
- ]),
- target("@ai-sdk/openai"),
- ),
- ).toEqual({
- low: {
- reasoningEffort: "low",
- reasoningSummary: "auto",
- include: ["reasoning.encrypted_content"],
- },
- })
- })
- test("does not replace unsupported effort options with heuristic variants", () => {
- expect(
- ProviderTransform.reasoningVariants(model([{ type: "effort", values: ["high"] }]), target("@ai-sdk/perplexity")),
- ).toEqual({})
- })
- test("leaves unsupported toggle options for heuristic fallback", () => {
- expect(ProviderTransform.reasoningVariants(model([{ type: "toggle" }]), target("@ai-sdk/openai"))).toBeUndefined()
- })
- test("uses model-family options for gateway and GitHub Copilot", () => {
- const effort = model([{ type: "effort", values: ["high"] }])
- expect(ProviderTransform.reasoningVariants(effort, target("@ai-sdk/gateway", "anthropic/claude-sonnet-4"))).toEqual(
- {
- high: { thinking: { type: "adaptive", display: "summarized" }, effort: "high" },
- },
- )
- expect(ProviderTransform.reasoningVariants(effort, target("@ai-sdk/gateway", "google/gemini-3-pro"))).toEqual({
- high: { thinkingConfig: { includeThoughts: true, thinkingLevel: "high" } },
- })
- expect(ProviderTransform.reasoningVariants(effort, target("@ai-sdk/github-copilot", "gemini-3-pro"))).toEqual({})
- })
- test.each(["@ai-sdk/cohere", "@ai-sdk/perplexity", "@ai-sdk/vercel", "@ai-sdk/alibaba", "gitlab-ai-provider"])(
- "does not invent effort controls for %s",
- (npm) => {
- expect(ProviderTransform.reasoningVariants(model([{ type: "effort", values: ["high"] }]), target(npm))).toEqual(
- {},
- )
- },
- )
- })
- 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("minimax m3 using anthropic returns thinking toggles", () => {
- const model = createMockModel({
- id: "minimax/minimax-m3",
- providerID: "minimax",
- api: {
- id: "MiniMax-M3",
- url: "https://api.minimax.com/anthropic/v1",
- npm: "@ai-sdk/anthropic",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(result).toEqual({
- none: { thinking: { type: "disabled" } },
- thinking: { thinking: { type: "adaptive" } },
- })
- })
- test("minimax m3 using openai-compatible returns thinking toggles", () => {
- const model = createMockModel({
- id: "minimax/minimax-m3",
- providerID: "minimax",
- api: {
- id: "minimax-m3",
- url: "https://api.minimax.com/v1",
- npm: "@ai-sdk/openai-compatible",
- },
- })
- expect(ProviderTransform.variants(model)).toEqual({
- none: { thinking: { type: "disabled" } },
- thinking: { thinking: { type: "adaptive" } },
- })
- })
- test.each(["nvidia", "lilac"])("%s minimax m3 returns chat template thinking toggles", (providerID) => {
- const model = createMockModel({
- id: `${providerID}/minimaxai/minimax-m3`,
- providerID,
- api: {
- id: "minimaxai/minimax-m3",
- url: "https://api.example.com/v1",
- npm: "@ai-sdk/openai-compatible",
- },
- })
- expect(ProviderTransform.variants(model)).toEqual({
- none: { chat_template_kwargs: { thinking_mode: "disabled" } },
- thinking: { chat_template_kwargs: { thinking_mode: "enabled" } },
- })
- })
- 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("glm-5.2 returns native effort variants for openai-compatible providers", () => {
- const model = createMockModel({
- id: "zhipuai/glm-5.2",
- providerID: "zhipuai",
- api: {
- id: "glm-5.2",
- url: "https://open.bigmodel.cn/api/paas/v4",
- npm: "@ai-sdk/openai-compatible",
- },
- })
- expect(ProviderTransform.variants(model)).toEqual({
- high: { reasoningEffort: "high" },
- max: { reasoningEffort: "max" },
- })
- })
- test("recognizes GLM-5.2 provider model IDs", () => {
- for (const id of ["accounts/fireworks/models/glm-5p2", "zai-org-glm-5-2", "umans-glm-5.2"]) {
- const model = createMockModel({
- id: `test/${id}`,
- api: {
- id,
- url: "https://api.test.com",
- npm: "@ai-sdk/openai-compatible",
- },
- })
- expect(ProviderTransform.variants(model)).toEqual({
- high: { reasoningEffort: "high" },
- max: { reasoningEffort: "max" },
- })
- }
- })
- test("recognizes GLM-5.2 from the API ID when the configured model ID is an alias", () => {
- const model = createMockModel({
- id: "custom/my-glm",
- api: {
- id: "accounts/fireworks/models/glm-5p2",
- url: "https://api.fireworks.ai/inference/v1",
- npm: "@ai-sdk/openai-compatible",
- },
- })
- expect(ProviderTransform.variants(model)).toEqual({
- high: { reasoningEffort: "high" },
- max: { reasoningEffort: "max" },
- })
- })
- test("glm-5.2 returns openrouter effort variants for openrouter", () => {
- const model = createMockModel({
- id: "openrouter/z-ai/glm-5.2",
- providerID: "openrouter",
- api: {
- id: "z-ai/glm-5.2",
- url: "https://openrouter.ai/api/v1",
- npm: "@openrouter/ai-sdk-provider",
- },
- })
- expect(ProviderTransform.variants(model)).toEqual({
- high: { reasoning: { effort: "high" } },
- xhigh: { reasoning: { effort: "xhigh" } },
- })
- })
- test("glm-5.2 returns effort variants for anthropic-compatible providers", () => {
- const model = createMockModel({
- id: "zai-coding-plan/glm-5.2",
- providerID: "zai-coding-plan",
- api: {
- id: "glm-5.2",
- url: "https://api.z.ai/api/anthropic",
- npm: "@ai-sdk/anthropic",
- },
- })
- expect(ProviderTransform.variants(model)).toEqual({
- high: { effort: "high" },
- max: { effort: "max" },
- })
- })
- test("glm-5.2 falls back to provider defaults for other packages", () => {
- const model = createMockModel({
- id: "test/glm-5.2",
- api: {
- id: "glm-5.2",
- url: "https://api.test.com",
- npm: "@ai-sdk/amazon-bedrock",
- },
- })
- expect(ProviderTransform.variants(model)).toEqual({
- low: { reasoningConfig: { type: "enabled", maxReasoningEffort: "low" } },
- medium: { reasoningConfig: { type: "enabled", maxReasoningEffort: "medium" } },
- high: { reasoningConfig: { type: "enabled", maxReasoningEffort: "high" } },
- })
- })
- test("mistral models with reasoning support return variants", () => {
- const model = createMockModel({
- id: "mistral/mistral-small-latest",
- providerID: "mistral",
- api: {
- id: "mistral-small-latest",
- url: "https://api.mistral.com",
- npm: "@ai-sdk/mistral",
- },
- capabilities: { reasoning: true },
- })
- const result = ProviderTransform.variants(model)
- expect(result).toEqual({
- high: { reasoningEffort: "high" },
- })
- })
- test("mistral-medium-3.5 with reasoning returns variants", () => {
- const model = createMockModel({
- id: "mistral/mistral-medium-3.5",
- providerID: "mistral",
- api: {
- id: "mistral-medium-3.5",
- url: "https://api.mistral.com",
- npm: "@ai-sdk/mistral",
- },
- capabilities: { reasoning: true },
- })
- const result = ProviderTransform.variants(model)
- expect(result).toEqual({
- high: { reasoningEffort: "high" },
- })
- })
- test("mistral without reasoning 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",
- },
- capabilities: { reasoning: false },
- })
- const result = ProviderTransform.variants(model)
- expect(result).toEqual({})
- })
- test("mistral large with reasoning returns empty object (only small supports reasoning)", () => {
- const model = createMockModel({
- id: "mistral/mistral-large",
- providerID: "mistral",
- api: {
- id: "mistral-large-latest",
- url: "https://api.mistral.com",
- npm: "@ai-sdk/mistral",
- },
- capabilities: { reasoning: true },
- })
- const result = ProviderTransform.variants(model)
- expect(result).toEqual({})
- })
- describe("@openrouter/ai-sdk-provider", () => {
- test("returns widely supported efforts for other reasoning 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(Object.keys(result)).toEqual(["low", "medium", "high"])
- expect(result.medium).toEqual({ reasoning: { effort: "medium" } })
- })
- 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" } })
- })
- for (const testCase of [
- { id: "openai/o3-mini", efforts: ["none", "minimal", "low", "medium", "high", "xhigh"] },
- { id: "openai/gpt-5.4", efforts: ["none", "low", "medium", "high", "xhigh"] },
- { id: "openai/gpt-5-pro", efforts: ["high"] },
- { id: "openai/gpt-5.5-pro", efforts: ["medium", "high", "xhigh"] },
- { id: "openai/gpt-5.2-codex", efforts: ["low", "medium", "high", "xhigh"] },
- { id: "openai/gpt-5.3-codex", efforts: ["none", "low", "medium", "high", "xhigh"] },
- { id: "openai/gpt-5.3-codex-max", efforts: ["none", "low", "medium", "high", "xhigh"] },
- { id: "openai/gpt-5-chat-latest", efforts: [] },
- { id: "openai/gpt-5.2-chat-latest", efforts: ["medium"] },
- ]) {
- test(`${testCase.id} returns supported OpenAI reasoning efforts`, () => {
- const result = ProviderTransform.variants(
- createMockModel({
- id: testCase.id,
- providerID: "openrouter",
- api: {
- id: testCase.id,
- url: "https://openrouter.ai",
- npm: "@openrouter/ai-sdk-provider",
- },
- }),
- )
- expect(Object.keys(result)).toEqual(testCase.efforts)
- })
- }
- test("gemini-3 returns widely supported 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(["low", "medium", "high"])
- })
- test("grok-4 uses the provider's standard efforts", () => {
- 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(Object.keys(result)).toEqual(["low", "medium", "high"])
- expect(result.medium).toEqual({ reasoning: { effort: "medium" } })
- })
- 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("configured anthropic aliases route by the API ID", () => {
- const model = createMockModel({
- id: "my-claude",
- providerID: "gateway",
- api: {
- id: "anthropic/claude-sonnet-4-6",
- url: "https://gateway.ai",
- npm: "@ai-sdk/gateway",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high", "max"])
- expect(result.high).toEqual({
- thinking: {
- type: "adaptive",
- },
- effort: "high",
- })
- })
- test("configured google aliases route by the API ID", () => {
- const model = createMockModel({
- id: "my-gemini",
- providerID: "gateway",
- api: {
- id: "google/gemini-2.5-pro",
- url: "https://gateway.ai",
- npm: "@ai-sdk/gateway",
- },
- })
- expect(ProviderTransform.variants(model)).toEqual({
- high: { thinkingConfig: { includeThoughts: true, thinkingBudget: 16_000 } },
- max: { thinkingConfig: { includeThoughts: true, thinkingBudget: 32_768 } },
- })
- })
- test("anthropic sonnet 4.6 models return adaptive thinking options", () => {
- const model = createMockModel({
- id: "anthropic/claude-sonnet-4-6",
- providerID: "gateway",
- api: {
- id: "anthropic/claude-sonnet-4-6",
- url: "https://gateway.ai",
- npm: "@ai-sdk/gateway",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high", "max"])
- expect(result.medium).toEqual({
- thinking: {
- type: "adaptive",
- },
- effort: "medium",
- })
- })
- test("anthropic sonnet 4.6 dot-format models return adaptive thinking options", () => {
- const model = createMockModel({
- id: "anthropic/claude-sonnet-4-6",
- providerID: "gateway",
- api: {
- id: "anthropic/claude-sonnet-4.6",
- url: "https://gateway.ai",
- npm: "@ai-sdk/gateway",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high", "max"])
- expect(result.medium).toEqual({
- thinking: {
- type: "adaptive",
- },
- effort: "medium",
- })
- })
- test("anthropic opus 4.6 dot-format models return adaptive thinking options", () => {
- const model = createMockModel({
- id: "anthropic/claude-opus-4-6",
- providerID: "gateway",
- api: {
- id: "anthropic/claude-opus-4.6",
- url: "https://gateway.ai",
- npm: "@ai-sdk/gateway",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high", "max"])
- expect(result.high).toEqual({
- thinking: {
- type: "adaptive",
- },
- effort: "high",
- })
- })
- test("anthropic opus 4.7 models return adaptive thinking options with xhigh", () => {
- const model = createMockModel({
- id: "anthropic/claude-opus-4-7",
- providerID: "gateway",
- api: {
- id: "anthropic/claude-opus-4-7",
- url: "https://gateway.ai",
- npm: "@ai-sdk/gateway",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high", "xhigh", "max"])
- expect(result.xhigh).toEqual({
- thinking: {
- type: "adaptive",
- display: "summarized",
- },
- effort: "xhigh",
- })
- expect(result.max).toEqual({
- thinking: {
- type: "adaptive",
- display: "summarized",
- },
- effort: "max",
- })
- })
- test("anthropic opus 4.7 dot-format models return adaptive thinking options with xhigh", () => {
- const model = createMockModel({
- id: "anthropic/claude-opus-4-7",
- providerID: "gateway",
- api: {
- id: "anthropic/claude-opus-4.7",
- url: "https://gateway.ai",
- npm: "@ai-sdk/gateway",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high", "xhigh", "max"])
- })
- test("anthropic opus 4.8 forces display summarized for adaptive reasoning", () => {
- const model = createMockModel({
- id: "anthropic/claude-opus-4-8",
- providerID: "gateway",
- api: {
- id: "anthropic/claude-opus-4-8",
- url: "https://gateway.ai",
- npm: "@ai-sdk/gateway",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high", "xhigh", "max"])
- expect(result.high).toEqual({
- thinking: {
- type: "adaptive",
- display: "summarized",
- },
- effort: "high",
- })
- })
- test("anthropic sonnet 5 returns adaptive thinking options with xhigh", () => {
- const model = createMockModel({
- id: "anthropic/claude-sonnet-5",
- providerID: "gateway",
- api: {
- id: "anthropic/claude-sonnet-5",
- url: "https://gateway.ai",
- npm: "@ai-sdk/gateway",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high", "xhigh", "max"])
- expect(result.high).toEqual({
- thinking: {
- type: "adaptive",
- display: "summarized",
- },
- effort: "high",
- })
- })
- test("anthropic opus 4.6 omits display so it keeps the summarized default", () => {
- const model = createMockModel({
- id: "anthropic/claude-opus-4-6",
- providerID: "gateway",
- api: {
- id: "anthropic/claude-opus-4-6",
- url: "https://gateway.ai",
- npm: "@ai-sdk/gateway",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high", "max"])
- expect(result.high).toEqual({
- thinking: {
- type: "adaptive",
- },
- effort: "high",
- })
- })
- test("anthropic models return anthropic thinking options", () => {
- const model = createMockModel({
- id: "anthropic/claude-sonnet-4",
- providerID: "gateway",
- api: {
- id: "anthropic/claude-sonnet-4",
- url: "https://gateway.ai",
- npm: "@ai-sdk/gateway",
- },
- })
- 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,
- },
- })
- })
- 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" })
- })
- for (const testCase of [
- { id: "openai/gpt-5-5", efforts: ["none", "low", "medium", "high", "xhigh"] },
- { id: "openai/gpt-5-pro", efforts: ["high"] },
- { id: "openai/gpt-5-5-pro", efforts: ["medium", "high", "xhigh"] },
- { id: "openai/gpt-5-2-codex", efforts: ["low", "medium", "high", "xhigh"] },
- { id: "openai/gpt-5-3-codex", efforts: ["none", "low", "medium", "high", "xhigh"] },
- { id: "openai/gpt-5-3-codex-max", efforts: ["none", "low", "medium", "high", "xhigh"] },
- { id: "openai/gpt-5-chat-latest", efforts: [] },
- { id: "openai/gpt-5-2-chat-latest", efforts: ["medium"] },
- ]) {
- test(`${testCase.id} returns supported OpenAI reasoning efforts`, () => {
- const result = ProviderTransform.variants(
- createMockModel({
- id: testCase.id,
- providerID: "gateway",
- api: {
- id: testCase.id,
- url: "https://gateway.ai",
- npm: "@ai-sdk/gateway",
- },
- }),
- )
- expect(Object.keys(result)).toEqual(testCase.efforts)
- })
- }
- })
- 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"])
- })
- test("gpt-5.3-codex includes xhigh", () => {
- const model = createMockModel({
- id: "gpt-5.3-codex",
- providerID: "github-copilot",
- api: {
- id: "gpt-5.3-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"])
- })
- test("gpt-5.4 includes xhigh", () => {
- const model = createMockModel({
- id: "gpt-5.4",
- release_date: "2026-03-05",
- providerID: "github-copilot",
- api: {
- id: "gpt-5.4",
- 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-4.5 uses standard reasoning efforts", () => {
- const model = createMockModel({
- id: "xai/grok-4.5",
- providerID: "xai",
- api: {
- id: "grok-4.5",
- url: "https://api.x.ai",
- npm: "@ai-sdk/xai",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high"])
- expect(result.medium).toEqual({ reasoningEffort: "medium" })
- })
- 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("north-mini-code-1-0 returns only none and high", () => {
- const model = createMockModel({
- id: "cohere/north-mini-code-1-0",
- providerID: "cohere",
- api: {
- id: "North-Mini-Code-1-0-latest",
- url: "https://api.cohere.com/compatibility/v1",
- npm: "@ai-sdk/openai-compatible",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(result).toEqual({
- none: { reasoningEffort: "none" },
- high: { reasoningEffort: "high" },
- })
- })
- })
- 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"])
- })
- for (const testCase of [
- { id: "gpt-5-1", efforts: ["none", "low", "medium", "high"] },
- { id: "gpt-5-4", efforts: ["none", "low", "medium", "high", "xhigh"] },
- { id: "gpt-5.4", efforts: ["none", "low", "medium", "high", "xhigh"] },
- { id: "gpt-5-5", efforts: ["none", "low", "medium", "high", "xhigh"] },
- ]) {
- test(`${testCase.id} returns supported Azure reasoning efforts`, () => {
- const result = ProviderTransform.variants(
- createMockModel({
- id: testCase.id,
- providerID: "azure",
- api: {
- id: testCase.id,
- url: "https://azure.com",
- npm: "@ai-sdk/azure",
- },
- }),
- )
- expect(Object.keys(result)).toEqual(testCase.efforts)
- })
- }
- })
- describe("@ai-sdk/openai", () => {
- test("gpt-5-pro returns only high effort", () => {
- 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(Object.keys(result)).toEqual(["high"])
- })
- 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-reasoning",
- providerID: "openai",
- api: {
- id: "gpt-5-reasoning",
- 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"])
- })
- for (const testCase of [
- { id: "o1", releaseDate: "2024-12-17", efforts: ["low", "medium", "high"] },
- { id: "o1-pro", releaseDate: "2025-03-19", efforts: ["low", "medium", "high"] },
- { id: "o3", releaseDate: "2025-04-16", efforts: ["low", "medium", "high"] },
- { id: "o3-mini", releaseDate: "2025-01-31", efforts: ["low", "medium", "high"] },
- { id: "o3-pro", releaseDate: "2025-06-10", efforts: ["low", "medium", "high"] },
- { id: "o4-mini", releaseDate: "2025-04-16", efforts: ["low", "medium", "high"] },
- { id: "o3-deep-research", releaseDate: "2025-06-26", efforts: ["medium"] },
- { id: "o4-mini-deep-research", releaseDate: "2025-06-26", efforts: ["medium"] },
- { id: "gpt-5.1", releaseDate: "2025-11-13", efforts: ["none", "low", "medium", "high"] },
- { id: "gpt-5.4", releaseDate: "2026-03-05", efforts: ["none", "low", "medium", "high", "xhigh"] },
- {
- id: "gpt-5.5",
- modelID: "gpt-5-5",
- releaseDate: "2026-04-23",
- efforts: ["none", "low", "medium", "high", "xhigh"],
- },
- { id: "gpt-5.4-pro", releaseDate: "2026-03-05", efforts: ["medium", "high", "xhigh"] },
- { id: "gpt-5.5-pro", releaseDate: "2026-04-23", efforts: ["medium", "high", "xhigh"] },
- { id: "gpt-5-codex", releaseDate: "2025-09-23", efforts: ["low", "medium", "high"] },
- { id: "gpt-5.1-codex", releaseDate: "2025-11-13", efforts: ["low", "medium", "high"] },
- { id: "gpt-5.1-codex-max", releaseDate: "2025-11-13", efforts: ["low", "medium", "high", "xhigh"] },
- { id: "gpt-5.2-codex", releaseDate: "2025-12-11", efforts: ["low", "medium", "high", "xhigh"] },
- { id: "gpt-5.3-codex", releaseDate: "2026-01-22", efforts: ["none", "low", "medium", "high", "xhigh"] },
- { id: "gpt-5.3-codex-max", releaseDate: "2026-01-22", efforts: ["none", "low", "medium", "high", "xhigh"] },
- { id: "gpt-5-chat-latest", releaseDate: "2025-08-07", efforts: [] },
- { id: "gpt-5.1-chat-latest", releaseDate: "2025-11-13", efforts: ["medium"] },
- { id: "gpt-5.2-chat-latest", releaseDate: "2025-12-11", efforts: ["medium"] },
- ]) {
- test(`${testCase.id} returns supported reasoning efforts`, () => {
- const result = ProviderTransform.variants(
- createMockModel({
- id: testCase.modelID ?? testCase.id,
- providerID: "openai",
- api: {
- id: testCase.id,
- url: "https://api.openai.com",
- npm: "@ai-sdk/openai",
- },
- release_date: testCase.releaseDate,
- }),
- )
- expect(Object.keys(result)).toEqual(testCase.efforts)
- })
- }
- test("gpt-50 (lookalike) does not get gpt-5 family treatment", () => {
- const model = createMockModel({
- id: "gpt-50",
- providerID: "openai",
- api: {
- id: "gpt-50",
- url: "https://api.openai.com",
- npm: "@ai-sdk/openai",
- },
- release_date: "2024-01-01",
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high"])
- })
- })
- describe("@ai-sdk/amazon-bedrock/mantle", () => {
- test("gpt-5.5 returns OpenAI-style reasoning variants", () => {
- const model = createMockModel({
- id: "openai.gpt-5.5",
- providerID: "amazon-bedrock",
- api: {
- id: "openai.gpt-5.5",
- url: "https://bedrock-mantle.us-east-2.api.aws/openai/v1",
- npm: "@ai-sdk/amazon-bedrock/mantle",
- },
- release_date: "2026-04-23",
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["none", "low", "medium", "high", "xhigh"])
- expect(result.medium).toEqual({
- reasoningEffort: "medium",
- reasoningSummary: "auto",
- include: ["reasoning.encrypted_content"],
- })
- })
- })
- describe("@ai-sdk/anthropic", () => {
- for (const testCase of [
- {
- name: "opus 4.5",
- apiIds: ["claude-opus-4-5-20251101", "claude-opus-4.5-20251101"],
- efforts: ["low", "medium", "high"],
- expectedHigh: { effort: "high" },
- },
- {
- name: "sonnet 4.6",
- apiIds: ["claude-sonnet-4-6", "claude-sonnet-4.6"],
- efforts: ["low", "medium", "high", "max"],
- expectedHigh: { thinking: { type: "adaptive" }, effort: "high" },
- },
- {
- name: "opus 4.6",
- apiIds: ["claude-opus-4-6", "claude-opus-4.6"],
- efforts: ["low", "medium", "high", "max"],
- expectedHigh: { thinking: { type: "adaptive" }, effort: "high" },
- },
- {
- name: "opus 4.7",
- apiIds: ["claude-opus-4-7", "claude-opus-4.7"],
- efforts: ["low", "medium", "high", "xhigh", "max"],
- expectedHigh: { thinking: { type: "adaptive", display: "summarized" }, effort: "high" },
- },
- {
- name: "opus 4.8",
- apiIds: ["claude-opus-4-8", "claude-opus-4.8"],
- efforts: ["low", "medium", "high", "xhigh", "max"],
- expectedHigh: { thinking: { type: "adaptive", display: "summarized" }, effort: "high" },
- },
- {
- name: "sonnet 5",
- apiIds: ["claude-sonnet-5", "claude-sonnet-5-20260630"],
- efforts: ["low", "medium", "high", "xhigh", "max"],
- expectedHigh: { thinking: { type: "adaptive", display: "summarized" }, effort: "high" },
- },
- {
- name: "fable 5",
- apiIds: ["claude-fable-5"],
- efforts: ["low", "medium", "high", "xhigh", "max"],
- expectedHigh: { thinking: { type: "adaptive", display: "summarized" }, effort: "high" },
- },
- ]) {
- for (const apiId of testCase.apiIds) {
- test(`${testCase.name} ${apiId} returns supported reasoning efforts`, () => {
- const result = ProviderTransform.variants(
- createMockModel({
- id: `anthropic/${apiId}`,
- providerID: "anthropic",
- api: {
- id: apiId,
- url: "https://api.anthropic.com",
- npm: "@ai-sdk/anthropic",
- },
- }),
- )
- expect(Object.keys(result)).toEqual(testCase.efforts)
- expect(result.high).toEqual(testCase.expectedHigh)
- })
- }
- }
- test("github copilot opus 4.7 returns only medium reasoning effort", () => {
- const model = createMockModel({
- id: "claude-opus-4.7",
- providerID: "github-copilot",
- api: {
- id: "claude-opus-4.7",
- url: "https://api.githubcopilot.com/v1",
- npm: "@ai-sdk/anthropic",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(result).toEqual({
- medium: {
- thinking: {
- type: "adaptive",
- display: "summarized",
- },
- effort: "medium",
- },
- })
- })
- 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/google-vertex/anthropic", () => {
- test("opus 4.8 uses adaptive reasoning for Vertex model IDs", () => {
- const result = ProviderTransform.variants(
- createMockModel({
- id: "google-vertex-anthropic/claude-opus-4-8@default",
- providerID: "google-vertex-anthropic",
- api: {
- id: "claude-opus-4-8@default",
- url: "https://us-central1-aiplatform.googleapis.com",
- npm: "@ai-sdk/google-vertex/anthropic",
- },
- }),
- )
- expect(Object.keys(result)).toEqual(["low", "medium", "high", "xhigh", "max"])
- expect(result.high).toEqual({
- thinking: {
- type: "adaptive",
- display: "summarized",
- },
- effort: "high",
- })
- })
- test("sonnet 5 uses adaptive reasoning for Vertex model IDs", () => {
- const result = ProviderTransform.variants(
- createMockModel({
- id: "google-vertex-anthropic/claude-sonnet-5@default",
- providerID: "google-vertex-anthropic",
- api: {
- id: "claude-sonnet-5@default",
- url: "https://us-central1-aiplatform.googleapis.com",
- npm: "@ai-sdk/google-vertex/anthropic",
- },
- }),
- )
- expect(Object.keys(result)).toEqual(["low", "medium", "high", "xhigh", "max"])
- expect(result.high).toEqual({
- thinking: {
- type: "adaptive",
- display: "summarized",
- },
- effort: "high",
- })
- })
- })
- describe("@ai-sdk/amazon-bedrock", () => {
- test("anthropic sonnet 4.6 returns adaptive reasoning options", () => {
- const model = createMockModel({
- id: "bedrock/anthropic-claude-sonnet-4-6",
- providerID: "bedrock",
- api: {
- id: "anthropic.claude-sonnet-4-6",
- url: "https://bedrock.amazonaws.com",
- npm: "@ai-sdk/amazon-bedrock",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high", "max"])
- expect(result.max).toEqual({
- reasoningConfig: {
- type: "adaptive",
- maxReasoningEffort: "max",
- },
- })
- })
- test("anthropic opus 4.7 returns adaptive reasoning options with xhigh", () => {
- const model = createMockModel({
- id: "bedrock/anthropic-claude-opus-4-7",
- providerID: "bedrock",
- api: {
- id: "anthropic.claude-opus-4-7",
- url: "https://bedrock.amazonaws.com",
- npm: "@ai-sdk/amazon-bedrock",
- },
- })
- const result = ProviderTransform.variants(model)
- expect(Object.keys(result)).toEqual(["low", "medium", "high", "xhigh", "max"])
- expect(result.xhigh).toEqual({
- reasoningConfig: {
- type: "adaptive",
- maxReasoningEffort: "xhigh",
- display: "summarized",
- },
- })
- expect(result.max).toEqual({
- reasoningConfig: {
- type: "adaptive",
- maxReasoningEffort: "max",
- display: "summarized",
- },
- })
- })
- test("anthropic opus 4.8 returns adaptive reasoning options with xhigh", () => {
- const result = ProviderTransform.variants(
- createMockModel({
- id: "bedrock/anthropic-claude-opus-4.8",
- providerID: "bedrock",
- api: {
- id: "anthropic.claude-opus-4.8",
- url: "https://bedrock.amazonaws.com",
- npm: "@ai-sdk/amazon-bedrock",
- },
- }),
- )
- expect(Object.keys(result)).toEqual(["low", "medium", "high", "xhigh", "max"])
- expect(result.high).toEqual({
- reasoningConfig: {
- type: "adaptive",
- maxReasoningEffort: "high",
- display: "summarized",
- },
- })
- })
- test("anthropic sonnet 5 returns adaptive reasoning options with xhigh", () => {
- const result = ProviderTransform.variants(
- createMockModel({
- id: "bedrock/anthropic-claude-sonnet-5",
- providerID: "bedrock",
- api: {
- id: "anthropic.claude-sonnet-5",
- url: "https://bedrock.amazonaws.com",
- npm: "@ai-sdk/amazon-bedrock",
- },
- }),
- )
- expect(Object.keys(result)).toEqual(["low", "medium", "high", "xhigh", "max"])
- expect(result.high).toEqual({
- reasoningConfig: {
- type: "adaptive",
- maxReasoningEffort: "high",
- display: "summarized",
- },
- })
- })
- 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",
- },
- })
- })
- })
- for (const provider of [
- { name: "@ai-sdk/google", providerID: "google", url: "https://generativelanguage.googleapis.com" },
- { name: "@ai-sdk/google-vertex", providerID: "google-vertex", url: "https://vertexai.googleapis.com" },
- ]) {
- describe(provider.name, () => {
- for (const testCase of [
- {
- apiId: "gemini-2.5-pro",
- efforts: ["high", "max"],
- expectedHigh: { thinkingConfig: { includeThoughts: true, thinkingBudget: 16_000 } },
- expectedMax: { thinkingConfig: { includeThoughts: true, thinkingBudget: 32_768 } },
- },
- {
- apiId: "gemini-2.5-flash",
- efforts: ["high", "max"],
- expectedHigh: { thinkingConfig: { includeThoughts: true, thinkingBudget: 16_000 } },
- expectedMax: { thinkingConfig: { includeThoughts: true, thinkingBudget: 24_576 } },
- },
- {
- apiId: "gemini-3-pro-preview",
- efforts: ["low", "medium", "high"],
- expectedHigh: { thinkingConfig: { includeThoughts: true, thinkingLevel: "high" } },
- },
- {
- apiId: "gemini-3.1-pro-preview",
- efforts: ["low", "medium", "high"],
- expectedHigh: { thinkingConfig: { includeThoughts: true, thinkingLevel: "high" } },
- },
- {
- apiId: "gemini-3-flash-preview",
- efforts: ["minimal", "low", "medium", "high"],
- expectedHigh: { thinkingConfig: { includeThoughts: true, thinkingLevel: "high" } },
- },
- {
- apiId: "gemini-3.1-flash-lite",
- efforts: ["minimal", "low", "medium", "high"],
- expectedHigh: { thinkingConfig: { includeThoughts: true, thinkingLevel: "high" } },
- },
- {
- apiId: "gemini-3.1-flash-image-preview",
- efforts: ["minimal", "high"],
- expectedHigh: { thinkingConfig: { includeThoughts: true, thinkingLevel: "high" } },
- },
- {
- apiId: "gemini-3-pro-image-preview",
- efforts: ["high"],
- expectedHigh: { thinkingConfig: { includeThoughts: true, thinkingLevel: "high" } },
- },
- ]) {
- test(`${testCase.apiId} returns supported thinking controls`, () => {
- const result = ProviderTransform.variants(
- createMockModel({
- id: `${provider.providerID}/${testCase.apiId}`,
- providerID: provider.providerID,
- api: {
- id: testCase.apiId,
- url: provider.url,
- npm: provider.name,
- },
- }),
- )
- expect(Object.keys(result)).toEqual(testCase.efforts)
- expect(result.high).toEqual(testCase.expectedHigh)
- if (testCase.expectedMax) expect(result.max).toEqual(testCase.expectedMax)
- })
- }
- })
- }
- 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({
- reasoningEffort: "none",
- })
- expect(result.low).toEqual({
- reasoningEffort: "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({})
- })
- })
- describe("@jerome-benoit/sap-ai-provider-v2", () => {
- const sapModel = (apiId: string, releaseDate = "2024-01-01") =>
- createMockModel({
- id: `sap-ai-core/${apiId}`,
- providerID: "sap-ai-core",
- api: {
- id: apiId,
- url: "https://api.ai.sap",
- npm: "@jerome-benoit/sap-ai-provider-v2",
- },
- release_date: releaseDate,
- })
- for (const testCase of [
- {
- name: "sonnet 4.6",
- apiIds: ["anthropic--claude-sonnet-4-6"],
- efforts: ["low", "medium", "high", "max"],
- thinking: { type: "adaptive" },
- },
- {
- name: "opus 4.6",
- apiIds: ["anthropic--claude-4.6-opus", "anthropic--claude-4-6-opus"],
- efforts: ["low", "medium", "high", "max"],
- thinking: { type: "adaptive" },
- },
- {
- name: "opus 4.7",
- apiIds: ["anthropic--claude-4.7-opus", "anthropic--claude-4-7-opus"],
- efforts: ["low", "medium", "high", "xhigh", "max"],
- thinking: { type: "adaptive", display: "summarized" },
- },
- {
- name: "opus 4.8",
- apiIds: ["anthropic--claude-4.8-opus", "anthropic--claude-4-8-opus"],
- efforts: ["low", "medium", "high", "xhigh", "max"],
- thinking: { type: "adaptive", display: "summarized" },
- },
- {
- name: "sonnet 5",
- apiIds: ["anthropic--claude-sonnet-5", "anthropic--claude-5-sonnet"],
- efforts: ["low", "medium", "high", "xhigh", "max"],
- thinking: { type: "adaptive", display: "summarized" },
- },
- ]) {
- for (const apiId of testCase.apiIds) {
- test(`${testCase.name} ${apiId} returns adaptive thinking variants under modelParams`, () => {
- const result = ProviderTransform.variants(sapModel(apiId))
- expect(Object.keys(result)).toEqual(testCase.efforts)
- for (const effort of testCase.efforts) {
- expect(result[effort]).toEqual({
- modelParams: {
- thinking: testCase.thinking,
- output_config: { effort },
- },
- })
- }
- })
- }
- }
- for (const apiId of ["anthropic--claude-sonnet-4", "anthropic--claude-4.5-opus"]) {
- test(`${apiId} returns budget_tokens variants under modelParams`, () => {
- const result = ProviderTransform.variants(sapModel(apiId))
- expect(Object.keys(result)).toEqual(["high", "max"])
- expect(result.high).toEqual({
- modelParams: { thinking: { type: "enabled", budget_tokens: 16000 } },
- })
- expect(result.max).toEqual({
- modelParams: { thinking: { type: "enabled", budget_tokens: 31999 } },
- })
- })
- }
- for (const testCase of [
- { apiId: "gemini-2.5-pro", maxBudget: 32768 },
- { apiId: "gemini-2.5-flash", maxBudget: 24576 },
- ]) {
- test(`${testCase.apiId} returns thinkingConfig variants under modelParams`, () => {
- const result = ProviderTransform.variants(sapModel(testCase.apiId))
- expect(Object.keys(result)).toEqual(["high", "max"])
- expect(result.high).toEqual({
- modelParams: { thinkingConfig: { includeThoughts: true, thinkingBudget: 16000 } },
- })
- expect(result.max).toEqual({
- modelParams: { thinkingConfig: { includeThoughts: true, thinkingBudget: testCase.maxBudget } },
- })
- })
- }
- for (const testCase of [
- { apiId: "gpt-5", releaseDate: "2025-08-07", efforts: ["minimal", "low", "medium", "high"] },
- { apiId: "gpt-5-mini", releaseDate: "2025-08-07", efforts: ["minimal", "low", "medium", "high"] },
- { apiId: "gpt-5-nano", releaseDate: "2025-08-07", efforts: ["minimal", "low", "medium", "high"] },
- { apiId: "gpt-5.4", releaseDate: "2026-01-15", efforts: ["none", "low", "medium", "high", "xhigh"] },
- { apiId: "azure-openai--o3-mini", releaseDate: "2024-01-01", efforts: ["low", "medium", "high"] },
- ]) {
- test(`${testCase.apiId} returns reasoning_effort variants under modelParams`, () => {
- const result = ProviderTransform.variants(sapModel(testCase.apiId, testCase.releaseDate))
- expect(Object.keys(result)).toEqual(testCase.efforts)
- for (const effort of testCase.efforts) {
- expect(result[effort]).toEqual({ modelParams: { reasoning_effort: effort } })
- }
- })
- }
- for (const apiId of [
- "gemini-3.1-flash-lite",
- "cohere--command-a-reasoning",
- "sonar-deep-research",
- "aws--llama-opus-4.7-fake",
- ]) {
- test(`${apiId} falls through to harmonized reasoning_effort fallback`, () => {
- const result = ProviderTransform.variants(sapModel(apiId))
- expect(Object.keys(result)).toEqual(["low", "medium", "high"])
- for (const effort of ["low", "medium", "high"]) {
- expect(result[effort]).toEqual({ modelParams: { reasoning_effort: effort } })
- }
- })
- }
- })
- describe("ai-gateway-provider (cloudflare-ai-gateway)", () => {
- const cfModel = (apiId: string, releaseDate = "2024-01-01") =>
- createMockModel({
- id: `cloudflare-ai-gateway/${apiId}`,
- providerID: "cloudflare-ai-gateway",
- api: {
- id: apiId,
- url: "https://gateway.ai.cloudflare.com/v1/compat",
- npm: "ai-gateway-provider",
- },
- release_date: releaseDate,
- })
- for (const testCase of [
- { id: "openai/gpt-5.4", efforts: ["none", "low", "medium", "high", "xhigh"] },
- { id: "openai/gpt-5.2-codex", efforts: ["low", "medium", "high", "xhigh"] },
- { id: "openai/gpt-5.3-codex", efforts: ["none", "low", "medium", "high", "xhigh"] },
- { id: "openai/gpt-5-pro", efforts: ["high"] },
- { id: "openai/gpt-5.2-pro", efforts: ["medium", "high", "xhigh"] },
- { id: "openai/gpt-5-chat-latest", efforts: [] },
- { id: "openai/gpt-5.2-chat-latest", efforts: ["medium"] },
- ]) {
- test(`${testCase.id} returns supported reasoning efforts`, () => {
- const result = ProviderTransform.variants(cfModel(testCase.id, "2026-03-05"))
- expect(Object.keys(result)).toEqual(testCase.efforts)
- })
- }
- test("openai gpt-4o (no reasoning) returns empty", () => {
- const model = cfModel("openai/gpt-4o")
- model.capabilities.reasoning = false
- const result = ProviderTransform.variants(model)
- expect(result).toEqual({})
- })
- test("non-openai upstream falls back to widely-supported OAI efforts", () => {
- const result = ProviderTransform.variants(cfModel("anthropic/claude-sonnet-4-6"))
- expect(result).toEqual({
- low: { reasoningEffort: "low" },
- medium: { reasoningEffort: "medium" },
- high: { reasoningEffort: "high" },
- })
- })
- })
- })
- describe("ProviderTransform.smallOptions - gpt-5 chat/search", () => {
- const createModel = (apiId: string) => {
- const model = {
- id: `openai/${apiId}`,
- providerID: "openai",
- api: {
- id: apiId,
- url: "https://api.openai.com",
- npm: "@ai-sdk/openai",
- },
- capabilities: { reasoning: true },
- limit: { output: 64_000 },
- release_date: "2026-01-01",
- } as any
- model.variants = ProviderTransform.variants(model)
- return model
- }
- for (const testCase of [
- { id: "gpt-5-chat-latest", options: { store: false } },
- {
- id: "gpt-5.1-chat-latest",
- options: {
- store: false,
- reasoningEffort: "medium",
- reasoningSummary: "auto",
- include: ["reasoning.encrypted_content"],
- },
- },
- {
- id: "gpt-5.2-chat-latest",
- options: {
- store: false,
- reasoningEffort: "medium",
- reasoningSummary: "auto",
- include: ["reasoning.encrypted_content"],
- },
- },
- {
- id: "gpt-5-search-api",
- options: {
- store: false,
- reasoningEffort: "none",
- reasoningSummary: "auto",
- include: ["reasoning.encrypted_content"],
- },
- },
- ]) {
- test(`${testCase.id} returns only supported small options`, () => {
- expect(ProviderTransform.smallOptions(createModel(testCase.id))).toEqual(testCase.options)
- })
- }
- })
- test("ProviderTransform.smallOptions preserves the weakest OpenRouter reasoning effort", () => {
- expect(
- ProviderTransform.smallOptions({
- providerID: "openrouter",
- api: {
- id: "google/gemini-3.5-flash",
- npm: "@openrouter/ai-sdk-provider",
- },
- variants: {
- low: { reasoning: { effort: "low" } },
- medium: { reasoning: { effort: "medium" } },
- high: { reasoning: { effort: "high" } },
- },
- } as any),
- ).toEqual({ reasoning: { effort: "low" } })
- })
- describe("ProviderTransform.smallOptions - google thinking controls", () => {
- const createGoogleModel = (apiId: string) => {
- const model = {
- id: `google/${apiId}`,
- providerID: "google",
- api: {
- id: apiId,
- url: "https://generativelanguage.googleapis.com",
- npm: "@ai-sdk/google",
- },
- capabilities: { reasoning: true },
- limit: { output: 64_000 },
- } as any
- model.variants = ProviderTransform.variants(model)
- return model
- }
- for (const testCase of [
- { id: "gemini-3-pro-preview", options: { thinkingConfig: { includeThoughts: true, thinkingLevel: "low" } } },
- { id: "gemini-3-flash-preview", options: { thinkingConfig: { includeThoughts: true, thinkingLevel: "minimal" } } },
- {
- id: "gemini-3.1-flash-image-preview",
- options: { thinkingConfig: { includeThoughts: true, thinkingLevel: "minimal" } },
- },
- { id: "gemini-3-pro-image-preview", options: { thinkingConfig: { includeThoughts: true, thinkingLevel: "high" } } },
- { id: "gemini-2.5-pro", options: { thinkingConfig: { includeThoughts: true, thinkingBudget: 16000 } } },
- { id: "gemini-2.5-flash", options: { thinkingConfig: { includeThoughts: true, thinkingBudget: 16000 } } },
- ]) {
- test(`${testCase.id} returns supported small thinking options`, () => {
- expect(ProviderTransform.smallOptions(createGoogleModel(testCase.id))).toEqual(testCase.options)
- })
- }
- test("uses the first configured variant when available", () => {
- expect(
- ProviderTransform.smallOptions({
- ...createGoogleModel("gemini-2.5-pro"),
- variants: {
- high: { thinkingConfig: { includeThoughts: true, thinkingBudget: 16000 } },
- max: { thinkingConfig: { includeThoughts: true, thinkingBudget: 32768 } },
- },
- }),
- ).toEqual({ thinkingConfig: { includeThoughts: true, thinkingBudget: 16000 } })
- })
- test("does not synthesize thinking options when variants are empty", () => {
- expect(ProviderTransform.smallOptions({ ...createGoogleModel("gemini-2.5-pro"), variants: {} })).toEqual({})
- })
- })
- describe("ProviderTransform.providerOptions - ai-gateway-provider", () => {
- const createModel = (overrides: Partial<any> = {}) =>
- ({
- id: "cloudflare-ai-gateway/openai/gpt-5.4",
- providerID: "cloudflare-ai-gateway",
- api: {
- id: "openai/gpt-5.4",
- url: "https://gateway.ai.cloudflare.com/v1/compat",
- npm: "ai-gateway-provider",
- },
- capabilities: {
- temperature: false,
- reasoning: true,
- 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: 1, output: 1, cache: { read: 0, write: 0 } },
- limit: { context: 1_000_000, output: 128_000 },
- status: "active",
- options: {},
- headers: {},
- release_date: "2026-03-05",
- ...overrides,
- }) as any
- test("routes options under openaiCompatible (the key @ai-sdk/openai-compatible reads)", () => {
- // Regression: previously fell back to providerID="cloudflare-ai-gateway",
- // which @ai-sdk/openai-compatible never reads, silently dropping reasoningEffort.
- const result = ProviderTransform.providerOptions(createModel(), { reasoningEffort: "high" })
- expect(result).toEqual({ openaiCompatible: { reasoningEffort: "high" } })
- })
- })
- describe("ProviderTransform.options - kimi family adaptive thinking", () => {
- const createModel = (overrides: Record<string, any> = {}) =>
- ({
- id: "moonshotai/kimi-k2-thinking",
- providerID: "moonshotai",
- api: {
- id: "kimi-k2-thinking",
- url: "https://api.moonshot.ai/anthropic",
- npm: "@ai-sdk/anthropic",
- },
- name: "Kimi K2 Thinking",
- 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: false,
- },
- cost: { input: 0.001, output: 0.002, cache: { read: 0.0001, write: 0.0002 } },
- limit: { context: 262144, output: 262144 },
- status: "active",
- options: {},
- headers: {},
- ...overrides,
- }) as any
- test("uses adaptive thinking with effort instead of budget tokens", () => {
- const result = ProviderTransform.options({ model: createModel(), sessionID: "s1", providerOptions: {} })
- expect(result.thinking).toEqual({ type: "adaptive", display: "summarized" })
- expect(result.effort).toBe("high")
- expect(JSON.stringify(result)).not.toContain("budgetTokens")
- })
- test("uses adaptive thinking through Google Vertex Anthropic", () => {
- const model = createModel()
- model.api.npm = "@ai-sdk/google-vertex/anthropic"
- const result = ProviderTransform.options({ model, sessionID: "s1", providerOptions: {} })
- expect(result.thinking).toEqual({ type: "adaptive", display: "summarized" })
- expect(result.effort).toBe("high")
- })
- test("provides adaptive effort variants without models.dev metadata", () => {
- expect(ProviderTransform.variants(createModel())).toEqual(
- Object.fromEntries(
- ["low", "medium", "high", "xhigh", "max"].map((effort) => [
- effort,
- { thinking: { type: "adaptive", display: "summarized" }, effort },
- ]),
- ),
- )
- const model = createModel()
- model.api.npm = "@ai-sdk/openai-compatible"
- expect(ProviderTransform.variants(model)).toEqual({})
- })
- test("does not enable thinking for kimi models without reasoning capability", () => {
- const model = createModel()
- model.capabilities.reasoning = false
- const result = ProviderTransform.options({ model, sessionID: "s1", providerOptions: {} })
- expect(result.thinking).toBeUndefined()
- })
- test("does not set thinking defaults for non-kimi anthropic models", () => {
- const model = createModel({
- id: "anthropic/claude-sonnet-4-5",
- providerID: "anthropic",
- api: {
- id: "claude-sonnet-4-5",
- url: "https://api.anthropic.com",
- npm: "@ai-sdk/anthropic",
- },
- })
- const result = ProviderTransform.options({ model, sessionID: "s1", providerOptions: {} })
- expect(result.thinking).toBeUndefined()
- expect(result.effort).toBeUndefined()
- })
- test("does not set adaptive thinking for kimi on openai-compatible", () => {
- const model = createModel()
- model.api = {
- id: "kimi-k2-thinking",
- url: "https://api.moonshot.ai/v1",
- npm: "@ai-sdk/openai-compatible",
- }
- const result = ProviderTransform.options({ model, sessionID: "s1", providerOptions: {} })
- expect(result.thinking).toBeUndefined()
- })
- })
|