71 lines
1.9 KiB
TypeScript
71 lines
1.9 KiB
TypeScript
import { getApiProvider } from "./api-registry.ts";
|
|
import { getEnvApiKey } from "./env-api-keys.ts";
|
|
import type {
|
|
Api,
|
|
AssistantMessage,
|
|
AssistantMessageEventStream,
|
|
Context,
|
|
Model,
|
|
ProviderStreamOptions,
|
|
SimpleStreamOptions,
|
|
StreamOptions,
|
|
} from "./types.ts";
|
|
|
|
function hasExplicitApiKey(apiKey: string | undefined): apiKey is string {
|
|
return typeof apiKey === "string" && apiKey.trim().length > 0;
|
|
}
|
|
|
|
function withEnvApiKey<TOptions extends StreamOptions>(
|
|
model: Model<Api>,
|
|
options: TOptions | undefined,
|
|
): TOptions | undefined {
|
|
if (hasExplicitApiKey(options?.apiKey)) return options;
|
|
const apiKey = getEnvApiKey(model.provider, options?.env);
|
|
if (!apiKey) return options;
|
|
return { ...options, apiKey } as TOptions;
|
|
}
|
|
|
|
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, withEnvApiKey(model, 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, withEnvApiKey(model, 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();
|
|
}
|