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.1 kB · 34 lines
TypeScript
at main
1234567891011121314151617181920212223242526272829303132333435import type { Api, Model } from "@earendil-works/pi-ai";import type { ModelRegistry } from "./model-registry.js";import { PRIME_INFERENCE_DEFAULT_MODEL_ID } from "./model-resolver.js";import { PRIME_INFERENCE_PROVIDER_ID } from "./prime-inference-auth.js";
type ProviderLoginResult = | { status: "success"; providerId: string; kind?: "provider" | "service" } | { status: "cancelled" | "failed" };
export interface PrimeInferencePostLoginModelAction { openModelPicker: boolean; fallbackModel?: Model<Api>;}
export function resolvePrimeInferencePostLoginModelAction( authResult: ProviderLoginResult, currentModel: Model<Api> | undefined, modelRegistry: Pick<ModelRegistry, "find">,): PrimeInferencePostLoginModelAction { if ( authResult.status !== "success" || authResult.kind === "service" || authResult.providerId !== PRIME_INFERENCE_PROVIDER_ID ) { return { openModelPicker: false }; }
return { openModelPicker: true, fallbackModel: currentModel ? undefined : modelRegistry.find(PRIME_INFERENCE_PROVIDER_ID, PRIME_INFERENCE_DEFAULT_MODEL_ID), };}