Something went wrong. Try again.
fork of prime agent with changes i want but are not upstreamable probqbly (and some fixes)
Something went wrong. Try again.
1.5 kB · 59 lines
TypeScript
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960import "./providers/register-builtins.js";
import { getApiProvider } from "./api-registry.js";import type { Api, AssistantMessage, AssistantMessageEventStream, Context, Model, ProviderStreamOptions, SimpleStreamOptions, StreamOptions,} from "./types.js";
export { getEnvApiKey } from "./env-api-keys.js";
function resolveApiProvider(api: Api) { const provider = getApiProvider(api); if (!provider) { throw new Error(`No API provider registered for api: ${api}`); } return provider;}
export function stream<TApi extends Api>( model: Model<TApi>, context: Context, options?: ProviderStreamOptions,): AssistantMessageEventStream { const provider = resolveApiProvider(model.api); return provider.stream(model, context, options as StreamOptions);}
export async function complete<TApi extends Api>( model: Model<TApi>, context: Context, options?: ProviderStreamOptions,): Promise<AssistantMessage> { const s = stream(model, context, options); return s.result();}
export function streamSimple<TApi extends Api>( model: Model<TApi>, context: Context, options?: SimpleStreamOptions,): AssistantMessageEventStream { const provider = resolveApiProvider(model.api); return provider.streamSimple(model, context, options);}
export async function completeSimple<TApi extends Api>( model: Model<TApi>, context: Context, options?: SimpleStreamOptions,): Promise<AssistantMessage> { const s = streamSimple(model, context, options); return s.result();}