import 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),
};
}