@pasko70/pibo 1.16.5 → 2.0.0

package/dist/agent-runtimes/pi/model-catalog.js

@@ -0,0 +1,87 @@
+import { createAgentSessionServices } from "@earendil-works/pi-coding-agent";
+import { registerMiniMaxProvider } from "../../providers/minimax.js";
+import { registerGlmProvider } from "../../providers/glm.js";
+import { registerQwenTokenPlanProvider } from "../../providers/qwen-token-plan.js";
+import { registerOpenAiGpt56Models } from "../../providers/openai-gpt56.js";
+import { piAuthMethodsForProvider } from "./auth.js";
+export function buildModelCatalogFromRegistry(registry) {
+    const providers = new Map();
+    for (const model of registry.getAll()) {
+        const providerId = model.provider;
+        let provider = providers.get(providerId);
+        if (!provider) {
+            const authConfigured = registry.getProviderAuthStatus?.(providerId).configured ?? false;
+            provider = {
+                id: providerId,
+                label: registry.getProviderDisplayName?.(providerId) ?? providerId,
+                authConfigured,
+                models: [],
+            };
+            providers.set(providerId, provider);
+        }
+        provider.models.push({
+            provider: providerId,
+            id: model.id,
+            label: model.name || model.id,
+            authConfigured: provider.authConfigured,
+            supportsReasoning: model.reasoning || undefined,
+        });
+    }
+    return {
+        providers: [...providers.values()]
+            .sort((left, right) => left.label.localeCompare(right.label) || left.id.localeCompare(right.id))
+            .map((provider) => ({
+            ...provider,
+            models: [...provider.models].sort((left, right) => left.label.localeCompare(right.label) || left.id.localeCompare(right.id)),
+        })),
+    };
+}
+export async function loadModelCatalogWithServices(createServices, cwd = process.cwd(), extensionHook) {
+    try {
+        const services = await createServices({ cwd });
+        extensionHook?.(services.modelRegistry);
+        return buildModelCatalogFromRegistry(services.modelRegistry);
+    }
+    catch {
+        return { providers: [] };
+    }
+}
+export async function loadModelCatalog(cwd = process.cwd()) {
+    return loadModelCatalogWithServices(createAgentSessionServices, cwd, (registry) => {
+        registerOpenAiGpt56Models(registry);
+        registerMiniMaxProvider(registry);
+        registerGlmProvider(registry);
+        registerQwenTokenPlanProvider(registry);
+    });
+}
+export function piAgentRuntimeModelCatalog(runtimeInstanceId, catalog) {
+    return {
+        runtimeInstanceId,
+        models: catalog.providers.flatMap((provider) => provider.models.map((model) => ({
+            id: model.id,
+            provider: model.provider,
+            displayName: model.label,
+            ...(model.supportsReasoning ? { reasoningOptions: ["off", "minimal", "low", "medium", "high", "xhigh", "max"] } : {}),
+            options: {
+                providerDisplayName: provider.label,
+                authConfigured: provider.authConfigured,
+            },
+        }))),
+        ...(catalog.providers.length === 0 ? {
+            diagnostics: [{
+                    severity: "warning",
+                    code: "pi_model_catalog_empty",
+                    message: "The Pi model registry did not return any models.",
+                }],
+        } : {}),
+    };
+}
+export function piAgentRuntimeAuthStatus(catalog) {
+    return catalog.providers.map((provider) => ({
+        id: provider.id,
+        displayName: provider.label,
+        state: provider.authConfigured ? "connected" : "disconnected",
+        configured: provider.authConfigured,
+        methods: [...piAuthMethodsForProvider(provider.id)],
+    }));
+}