eve/dist/src/internal/model-auth/transport.js

Filesystem-first framework for durable backend AI agents that run anywhere.

import{resolveModelApiKey}from"./api-key.js";import{VERCEL_TEAM_HEADER}from"./vercel.js";import{developmentModelBrokerAvailable,readDevelopmentModelCredential}from"./development-broker-client.js";import{resolveGatewayModelCredential}from"./gateway-credential.js";import{createGateway}from"ai";import{MODEL_CONNECTION_ENV,MODEL_CONNECTION_ENV as MODEL_CONNECTION_ENV$1}from"#shared/model-helper.js";import{isEveDevEnvironment}from"#internal/application/dev-environment.js";function createDirectModelFetch(e){return async(t,n)=>{let r=new Headers(n?.headers),i=(await readDevelopmentModelCredential(e,void 0,n?.signal))?.token??await resolveModelApiKey(e);return r.set(e===`openai`?`authorization`:`x-api-key`,e===`openai`?`Bearer ${i}`:i),fetch(t,{...n,headers:r})}}function createLocalGateway(){if(!isEveDevEnvironment())return;let e=process.env[MODEL_CONNECTION_ENV$1];if(!developmentModelBrokerAvailable()&&e!==`vercel`&&e!==`vercel-cli`&&e!==`ai-gateway-key`&&e!==`ai-gateway-project`)return;let resolve=async(e,t)=>await readDevelopmentModelCredential(`gateway`,e,t)??await resolveGatewayModelCredential(e);return createGateway({apiKey:`eve-local-credential`,fetch:async(e,t)=>{let n=new Headers(t?.headers),authenticate=e=>{n.set(`authorization`,`Bearer ${e.token}`),n.delete(VERCEL_TEAM_HEADER),e.teamId&&n.set(VERCEL_TEAM_HEADER,e.teamId)},r=await resolve(void 0,t?.signal);authenticate(r);let i=await fetch(e,{...t,headers:n});return i.status!==401||r.kind!==`oauth`||t?.body!==void 0&&t.body!==null&&typeof t.body!=`string`?i:(await i.body?.cancel(),authenticate(await resolve(r.token,t?.signal)),fetch(e,{...t,headers:n}))}})}function localGatewayModel(e){return createLocalGateway()?.languageModel(e)}function localGatewayEvaluationModel(e){return createLocalGateway()?.evaluationModel(e)}export{MODEL_CONNECTION_ENV,createDirectModelFetch,localGatewayEvaluationModel,localGatewayModel,resolveModelApiKey};