@undefineds.co/xpod 0.2.0 → 0.2.2-preview.0

package/dist/cli/lib/model-manager.js

@@ -0,0 +1,251 @@
+"use strict";
+/**
+ * Get available AI models based on configured providers.
+ *
+ * Queries the user's Pod for configured AI providers and returns
+ * the list of available models from each provider.
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getAvailableModels = getAvailableModels;
+exports.getModelById = getModelById;
+exports.listAvailableModels = listAvailableModels;
+exports.addCustomModel = addCustomModel;
+const drizzle_solid_1 = require("@undefineds.co/drizzle-solid");
+const tables_1 = require("../../credential/schema/tables");
+const provider_1 = require("../../ai/schema/provider");
+const model_1 = require("../../ai/schema/model");
+const types_1 = require("../../credential/schema/types");
+const pi_optional_1 = require("./pi-optional");
+/**
+ * Get all available models from configured providers.
+ *
+ * Returns models from:
+ * 1. Providers with active API key credentials
+ * 2. Providers with active OAuth credentials
+ * 3. Platform default models (from xpod API)
+ */
+async function getAvailableModels(session, xpodUrl) {
+    const models = [];
+    try {
+        const db = (0, drizzle_solid_1.drizzle)({
+            fetch: session.fetch,
+            info: session.info,
+        });
+        // 1. Get all active credentials
+        const allApiKeyCredentials = await db.select().from(tables_1.ApiKeyCredential);
+        const apiKeyCredentials = allApiKeyCredentials.filter((c) => c.service === types_1.ServiceType.AI && c.status === types_1.CredentialStatus.ACTIVE);
+        const allOAuthCredentials = await db.select().from(tables_1.OAuthCredential);
+        const oauthCredentials = allOAuthCredentials.filter((c) => c.service === types_1.ServiceType.AI && c.status === types_1.CredentialStatus.ACTIVE);
+        const allCredentials = [...apiKeyCredentials, ...oauthCredentials];
+        // 2. Get all providers
+        const allProviders = await db.select().from(provider_1.Provider);
+        const providerByUri = new Map(allProviders.map(p => [p['@id'], p]));
+        // 3. Collect provider IDs from credentials
+        const configuredProviderIds = new Set();
+        for (const cred of allCredentials) {
+            if (!cred.provider)
+                continue;
+            const provider = providerByUri.get(cred.provider);
+            if (provider && provider.id) {
+                configuredProviderIds.add(provider.id);
+            }
+        }
+        // 4. Get models from each configured provider
+        for (const providerId of configuredProviderIds) {
+            try {
+                // Only get models for known providers
+                const knownProviders = ['openai', 'anthropic', 'openrouter', 'ollama', 'google', 'codebuddy'];
+                if (!knownProviders.includes(providerId)) {
+                    console.warn(`Skipping unknown provider: ${providerId}`);
+                    continue;
+                }
+                const { getModels } = await (0, pi_optional_1.loadPiAi)();
+                const providerModels = getModels(providerId);
+                for (const model of providerModels) {
+                    models.push({
+                        id: model.id,
+                        name: model.name || model.id,
+                        provider: providerId,
+                        providerName: model.provider || providerId,
+                        contextWindow: model.contextWindow,
+                        maxTokens: model.maxTokens,
+                    });
+                }
+            }
+            catch (error) {
+                console.warn(`Failed to get models for provider ${providerId}:`, error);
+            }
+        }
+        // 4.5. Get custom models from Pod
+        try {
+            const allModels = await db.select().from(model_1.Model);
+            for (const model of allModels) {
+                // Avoid duplicates
+                if (!models.some(m => m.id === model.id)) {
+                    const provider = providerByUri.get(model.isProvidedBy);
+                    models.push({
+                        id: model.id,
+                        name: model.displayName || model.id,
+                        provider: provider?.id || 'custom',
+                        providerName: provider?.displayName || 'Custom',
+                        contextWindow: model.contextLength,
+                        maxTokens: model.maxOutputTokens,
+                    });
+                }
+            }
+        }
+        catch (error) {
+            console.warn('Failed to get custom models:', error);
+        }
+        // 5. Always include platform default models (from xpod API)
+        try {
+            const baseUrl = xpodUrl.endsWith('/') ? xpodUrl : `${xpodUrl}/`;
+            const modelsUrl = `${baseUrl}v1/models`;
+            const response = await session.fetch(modelsUrl, {
+                headers: {
+                    'Accept': 'application/json',
+                },
+            });
+            if (response.ok) {
+                const data = await response.json();
+                for (const model of data.data) {
+                    // Avoid duplicates
+                    if (!models.some(m => m.id === model.id)) {
+                        models.push({
+                            id: model.id,
+                            name: model.name || model.id,
+                            provider: model.provider || 'platform',
+                            providerName: model.owned_by || 'Platform Default',
+                            contextWindow: model.context_window,
+                            maxTokens: model.max_tokens,
+                        });
+                    }
+                }
+            }
+            else {
+                console.warn(`Failed to fetch platform models: ${response.status}`);
+            }
+        }
+        catch (error) {
+            console.warn('Failed to get platform default models:', error);
+        }
+    }
+    catch (error) {
+        console.error('Failed to get available models:', error);
+        // Return empty array on error
+    }
+    return models;
+}
+/**
+ * Get a specific model by ID.
+ */
+async function getModelById(session, xpodUrl, modelId) {
+    const models = await getAvailableModels(session, xpodUrl);
+    return models.find(m => m.id === modelId) || null;
+}
+/**
+ * List available models (for CLI display).
+ */
+async function listAvailableModels(session, xpodUrl) {
+    const models = await getAvailableModels(session, xpodUrl);
+    console.log('\n📋 Available Models:\n');
+    // Group by provider
+    const byProvider = new Map();
+    for (const model of models) {
+        const list = byProvider.get(model.provider) || [];
+        list.push(model);
+        byProvider.set(model.provider, list);
+    }
+    for (const [, providerModels] of byProvider) {
+        console.log(`\n${providerModels[0].providerName}:`);
+        for (const model of providerModels) {
+            const ctx = model.contextWindow ? ` (${model.contextWindow} tokens)` : '';
+            console.log(`  ${model.id}${ctx}`);
+        }
+    }
+    console.log('');
+}
+/**
+ * Add a custom model to the user's Pod.
+ */
+async function addCustomModel(session, xpodUrl, modelId) {
+    try {
+        const db = (0, drizzle_solid_1.drizzle)({
+            fetch: session.fetch,
+            info: session.info,
+        });
+        // Check if model already exists
+        const allModels = await db.select().from(model_1.Model);
+        if (allModels.some(m => m.id === modelId)) {
+            console.log(`✓ Model ${modelId} already exists`);
+            return;
+        }
+        // Prompt for provider
+        const { promptText } = await Promise.resolve().then(() => __importStar(require('./prompt.js')));
+        console.log('\n📝 Add custom model\n');
+        const displayName = await promptText(`Display name (default: ${modelId}): `);
+        const providerInput = await promptText('Provider ID (e.g., openai, anthropic, custom): ');
+        const providerId = providerInput || 'custom';
+        // Get or create provider
+        const allProviders = await db.select().from(provider_1.Provider);
+        let provider = allProviders.find(p => p.id === providerId);
+        if (!provider) {
+            console.log(`\n⚠️  Provider ${providerId} not found. Creating new provider...`);
+            const providerName = await promptText(`Provider display name (default: ${providerId}): `);
+            const baseUrl = await promptText('Base URL (optional): ');
+            await db.insert(provider_1.Provider).values({
+                id: providerId,
+                displayName: providerName || providerId,
+                baseUrl: baseUrl || undefined,
+            });
+            // Reload provider
+            const updatedProviders = await db.select().from(provider_1.Provider);
+            provider = updatedProviders.find(p => p.id === providerId);
+        }
+        if (!provider) {
+            throw new Error('Failed to create provider');
+        }
+        // Create model
+        const podUrl = xpodUrl.endsWith('/') ? xpodUrl : `${xpodUrl}/`;
+        const providerUri = `${podUrl}settings/ai/providers.ttl#${providerId}`;
+        await db.insert(model_1.Model).values({
+            id: modelId,
+            displayName: displayName || modelId,
+            isProvidedBy: providerUri,
+            status: 'active',
+            createdAt: new Date().toISOString(),
+            updatedAt: new Date().toISOString(),
+        });
+        console.log(`\n✅ Model ${modelId} added successfully!`);
+        console.log(`   Provider: ${provider.displayName || providerId}`);
+        console.log(`   Display name: ${displayName || modelId}\n`);
+    }
+    catch (error) {
+        console.error('Failed to add custom model:', error);
+        throw error;
+    }
+}
+//# sourceMappingURL=model-manager.js.map

package/dist/cli/lib/model-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-manager.js","sourceRoot":"","sources":["../../../src/cli/lib/model-manager.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;;;;;;;;;;;;;;;;;;;;;;;AA4BH,gDA0HC;AAKD,oCAGC;AAKD,kDAsBC;AAKD,wCAkEC;AA9PD,gEAAuD;AACvD,2DAAmF;AACnF,uDAAoD;AACpD,iDAA8C;AAC9C,yDAA8E;AAE9E,+CAAyC;AAYzC;;;;;;;GAOG;AACI,KAAK,UAAU,kBAAkB,CAAC,OAAgB,EAAE,OAAe;IACxE,MAAM,MAAM,GAAqB,EAAE,CAAC;IAEpC,IAAI,CAAC;QACH,MAAM,EAAE,GAAQ,IAAA,uBAAO,EAAC;YACtB,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,IAAI,EAAE,OAAO,CAAC,IAAI;SACZ,CAAC,CAAC;QAEV,gCAAgC;QAChC,MAAM,oBAAoB,GAAG,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,yBAAgB,CAAU,CAAC;QAC/E,MAAM,iBAAiB,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAC/D,CAAC,CAAC,OAAO,KAAK,mBAAW,CAAC,EAAE,IAAI,CAAC,CAAC,MAAM,KAAK,wBAAgB,CAAC,MAAM,CACrE,CAAC;QAEF,MAAM,mBAAmB,GAAG,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,wBAAe,CAAU,CAAC;QAC7E,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAC7D,CAAC,CAAC,OAAO,KAAK,mBAAW,CAAC,EAAE,IAAI,CAAC,CAAC,MAAM,KAAK,wBAAgB,CAAC,MAAM,CACrE,CAAC;QAEF,MAAM,cAAc,GAAG,CAAC,GAAG,iBAAiB,EAAE,GAAG,gBAAgB,CAAC,CAAC;QAEnE,uBAAuB;QACvB,MAAM,YAAY,GAAG,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,mBAAQ,CAAU,CAAC;QAC/D,MAAM,aAAa,GAAG,IAAI,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAEpE,2CAA2C;QAC3C,MAAM,qBAAqB,GAAG,IAAI,GAAG,EAAU,CAAC;QAChD,KAAK,MAAM,IAAI,IAAI,cAAc,EAAE,CAAC;YAClC,IAAI,CAAC,IAAI,CAAC,QAAQ;gBAAE,SAAS;YAC7B,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAClD,IAAI,QAAQ,IAAI,QAAQ,CAAC,EAAE,EAAE,CAAC;gBAC5B,qBAAqB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAY,CAAC,CAAC;YACnD,CAAC;QACH,CAAC;QAED,8CAA8C;QAC9C,KAAK,MAAM,UAAU,IAAI,qBAAqB,EAAE,CAAC;YAC/C,IAAI,CAAC;gBACH,sCAAsC;gBACtC,MAAM,cAAc,GAAG,CAAC,QAAQ,EAAE,WAAW,EAAE,YAAY,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC;gBAC9F,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;oBACzC,OAAO,CAAC,IAAI,CAAC,8BAA8B,UAAU,EAAE,CAAC,CAAC;oBACzD,SAAS;gBACX,CAAC;gBAED,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,IAAA,sBAAQ,GAAE,CAAC;gBACvC,MAAM,cAAc,GAAG,SAAS,CAAC,UAAU,CAAC,CAAC;gBAC7C,KAAK,MAAM,KAAK,IAAI,cAAc,EAAE,CAAC;oBACnC,MAAM,CAAC,IAAI,CAAC;wBACV,EAAE,EAAE,KAAK,CAAC,EAAE;wBACZ,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,EAAE;wBAC5B,QAAQ,EAAE,UAAU;wBACpB,YAAY,EAAE,KAAK,CAAC,QAAQ,IAAI,UAAU;wBAC1C,aAAa,EAAE,KAAK,CAAC,aAAa;wBAClC,SAAS,EAAE,KAAK,CAAC,SAAS;qBAC3B,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,IAAI,CAAC,qCAAqC,UAAU,GAAG,EAAE,KAAK,CAAC,CAAC;YAC1E,CAAC;QACH,CAAC;QAED,kCAAkC;QAClC,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,aAAK,CAAU,CAAC;YACzD,KAAK,MAAM,KAAK,IAAI,SAAS,EAAE,CAAC;gBAC9B,mBAAmB;gBACnB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,KAAK,CAAC,EAAE,CAAC,EAAE,CAAC;oBACzC,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;oBACvD,MAAM,CAAC,IAAI,CAAC;wBACV,EAAE,EAAE,KAAK,CAAC,EAAE;wBACZ,IAAI,EAAE,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,EAAE;wBACnC,QAAQ,EAAE,QAAQ,EAAE,EAAY,IAAI,QAAQ;wBAC5C,YAAY,EAAE,QAAQ,EAAE,WAAqB,IAAI,QAAQ;wBACzD,aAAa,EAAE,KAAK,CAAC,aAAa;wBAClC,SAAS,EAAE,KAAK,CAAC,eAAe;qBACjC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,IAAI,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;QACtD,CAAC;QAED,4DAA4D;QAC5D,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,GAAG,CAAC;YAChE,MAAM,SAAS,GAAG,GAAG,OAAO,WAAW,CAAC;YAExC,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,SAAS,EAAE;gBAC9C,OAAO,EAAE;oBACP,QAAQ,EAAE,kBAAkB;iBAC7B;aACF,CAAC,CAAC;YAEH,IAAI,QAAQ,CAAC,EAAE,EAAE,CAAC;gBAChB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAqC,CAAC;gBACtE,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;oBAC9B,mBAAmB;oBACnB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,KAAK,CAAC,EAAE,CAAC,EAAE,CAAC;wBACzC,MAAM,CAAC,IAAI,CAAC;4BACV,EAAE,EAAE,KAAK,CAAC,EAAE;4BACZ,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,EAAE;4BAC5B,QAAQ,EAAE,KAAK,CAAC,QAAQ,IAAI,UAAU;4BACtC,YAAY,EAAE,KAAK,CAAC,QAAQ,IAAI,kBAAkB;4BAClD,aAAa,EAAE,KAAK,CAAC,cAAc;4BACnC,SAAS,EAAE,KAAK,CAAC,UAAU;yBAC5B,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,IAAI,CAAC,oCAAoC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;YACtE,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,IAAI,CAAC,wCAAwC,EAAE,KAAK,CAAC,CAAC;QAChE,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;QACxD,8BAA8B;IAChC,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACI,KAAK,UAAU,YAAY,CAAC,OAAgB,EAAE,OAAe,EAAE,OAAe;IACnF,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAC1D,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,OAAO,CAAC,IAAI,IAAI,CAAC;AACpD,CAAC;AAED;;GAEG;AACI,KAAK,UAAU,mBAAmB,CAAC,OAAgB,EAAE,OAAe;IACzE,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAE1D,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;IAExC,oBAAoB;IACpB,MAAM,UAAU,GAAG,IAAI,GAAG,EAA4B,CAAC;IACvD,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,MAAM,IAAI,GAAG,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAClD,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACjB,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;IACvC,CAAC;IAED,KAAK,MAAM,CAAC,EAAE,cAAc,CAAC,IAAI,UAAU,EAAE,CAAC;QAC5C,OAAO,CAAC,GAAG,CAAC,KAAK,cAAc,CAAC,CAAC,CAAC,CAAC,YAAY,GAAG,CAAC,CAAC;QACpD,KAAK,MAAM,KAAK,IAAI,cAAc,EAAE,CAAC;YACnC,MAAM,GAAG,GAAG,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC,KAAK,KAAK,CAAC,aAAa,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;YAC1E,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,EAAE,GAAG,GAAG,EAAE,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;AAClB,CAAC;AAED;;GAEG;AACI,KAAK,UAAU,cAAc,CAAC,OAAgB,EAAE,OAAe,EAAE,OAAe;IACrF,IAAI,CAAC;QACH,MAAM,EAAE,GAAQ,IAAA,uBAAO,EAAC;YACtB,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,IAAI,EAAE,OAAO,CAAC,IAAI;SACZ,CAAC,CAAC;QAEV,gCAAgC;QAChC,MAAM,SAAS,GAAG,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,aAAK,CAAU,CAAC;QACzD,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,OAAO,CAAC,EAAE,CAAC;YAC1C,OAAO,CAAC,GAAG,CAAC,WAAW,OAAO,iBAAiB,CAAC,CAAC;YACjD,OAAO;QACT,CAAC;QAED,sBAAsB;QACtB,MAAM,EAAE,UAAU,EAAE,GAAG,wDAAa,aAAa,GAAC,CAAC;QACnD,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;QAEvC,MAAM,WAAW,GAAG,MAAM,UAAU,CAAC,0BAA0B,OAAO,KAAK,CAAC,CAAC;QAC7E,MAAM,aAAa,GAAG,MAAM,UAAU,CAAC,iDAAiD,CAAC,CAAC;QAC1F,MAAM,UAAU,GAAG,aAAa,IAAI,QAAQ,CAAC;QAE7C,yBAAyB;QACzB,MAAM,YAAY,GAAG,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,mBAAQ,CAAU,CAAC;QAC/D,IAAI,QAAQ,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,UAAU,CAAC,CAAC;QAE3D,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,OAAO,CAAC,GAAG,CAAC,kBAAkB,UAAU,sCAAsC,CAAC,CAAC;YAChF,MAAM,YAAY,GAAG,MAAM,UAAU,CAAC,mCAAmC,UAAU,KAAK,CAAC,CAAC;YAC1F,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,uBAAuB,CAAC,CAAC;YAE1D,MAAM,EAAE,CAAC,MAAM,CAAC,mBAAQ,CAAC,CAAC,MAAM,CAAC;gBAC/B,EAAE,EAAE,UAAU;gBACd,WAAW,EAAE,YAAY,IAAI,UAAU;gBACvC,OAAO,EAAE,OAAO,IAAI,SAAS;aAC9B,CAAC,CAAC;YAEH,kBAAkB;YAClB,MAAM,gBAAgB,GAAG,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,mBAAQ,CAAU,CAAC;YACnE,QAAQ,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,UAAU,CAAC,CAAC;QAC7D,CAAC;QAED,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC/C,CAAC;QAED,eAAe;QACf,MAAM,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,GAAG,CAAC;QAC/D,MAAM,WAAW,GAAG,GAAG,MAAM,6BAA6B,UAAU,EAAE,CAAC;QAEvE,MAAM,EAAE,CAAC,MAAM,CAAC,aAAK,CAAC,CAAC,MAAM,CAAC;YAC5B,EAAE,EAAE,OAAO;YACX,WAAW,EAAE,WAAW,IAAI,OAAO;YACnC,YAAY,EAAE,WAAW;YACzB,MAAM,EAAE,QAAQ;YAChB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC,CAAC;QAEH,OAAO,CAAC,GAAG,CAAC,aAAa,OAAO,sBAAsB,CAAC,CAAC;QACxD,OAAO,CAAC,GAAG,CAAC,gBAAgB,QAAQ,CAAC,WAAW,IAAI,UAAU,EAAE,CAAC,CAAC;QAClE,OAAO,CAAC,GAAG,CAAC,oBAAoB,WAAW,IAAI,OAAO,IAAI,CAAC,CAAC;IAC9D,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;QACpD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC","sourcesContent":["/**\n * Get available AI models based on configured providers.\n *\n * Queries the user's Pod for configured AI providers and returns\n * the list of available models from each provider.\n */\n\nimport { drizzle } from '@undefineds.co/drizzle-solid';\nimport { ApiKeyCredential, OAuthCredential } from '../../credential/schema/tables';\nimport { Provider } from '../../ai/schema/provider';\nimport { Model } from '../../ai/schema/model';\nimport { ServiceType, CredentialStatus } from '../../credential/schema/types';\nimport type { Session } from '@inrupt/solid-client-authn-node';\nimport { loadPiAi } from './pi-optional';\n\nexport interface AvailableModel {\n  id: string;\n  name: string;\n  provider: string;\n  providerName: string;\n  description?: string;\n  contextWindow?: number;\n  maxTokens?: number;\n}\n\n/**\n * Get all available models from configured providers.\n *\n * Returns models from:\n * 1. Providers with active API key credentials\n * 2. Providers with active OAuth credentials\n * 3. Platform default models (from xpod API)\n */\nexport async function getAvailableModels(session: Session, xpodUrl: string): Promise<AvailableModel[]> {\n  const models: AvailableModel[] = [];\n\n  try {\n    const db: any = drizzle({\n      fetch: session.fetch,\n      info: session.info,\n    } as any);\n\n    // 1. Get all active credentials\n    const allApiKeyCredentials = await db.select().from(ApiKeyCredential) as any[];\n    const apiKeyCredentials = allApiKeyCredentials.filter((c: any) =>\n      c.service === ServiceType.AI && c.status === CredentialStatus.ACTIVE,\n    );\n\n    const allOAuthCredentials = await db.select().from(OAuthCredential) as any[];\n    const oauthCredentials = allOAuthCredentials.filter((c: any) =>\n      c.service === ServiceType.AI && c.status === CredentialStatus.ACTIVE,\n    );\n\n    const allCredentials = [...apiKeyCredentials, ...oauthCredentials];\n\n    // 2. Get all providers\n    const allProviders = await db.select().from(Provider) as any[];\n    const providerByUri = new Map(allProviders.map(p => [p['@id'], p]));\n\n    // 3. Collect provider IDs from credentials\n    const configuredProviderIds = new Set<string>();\n    for (const cred of allCredentials) {\n      if (!cred.provider) continue;\n      const provider = providerByUri.get(cred.provider);\n      if (provider && provider.id) {\n        configuredProviderIds.add(provider.id as string);\n      }\n    }\n\n    // 4. Get models from each configured provider\n    for (const providerId of configuredProviderIds) {\n      try {\n        // Only get models for known providers\n        const knownProviders = ['openai', 'anthropic', 'openrouter', 'ollama', 'google', 'codebuddy'];\n        if (!knownProviders.includes(providerId)) {\n          console.warn(`Skipping unknown provider: ${providerId}`);\n          continue;\n        }\n\n        const { getModels } = await loadPiAi();\n        const providerModels = getModels(providerId);\n        for (const model of providerModels) {\n          models.push({\n            id: model.id,\n            name: model.name || model.id,\n            provider: providerId,\n            providerName: model.provider || providerId,\n            contextWindow: model.contextWindow,\n            maxTokens: model.maxTokens,\n          });\n        }\n      } catch (error) {\n        console.warn(`Failed to get models for provider ${providerId}:`, error);\n      }\n    }\n\n    // 4.5. Get custom models from Pod\n    try {\n      const allModels = await db.select().from(Model) as any[];\n      for (const model of allModels) {\n        // Avoid duplicates\n        if (!models.some(m => m.id === model.id)) {\n          const provider = providerByUri.get(model.isProvidedBy);\n          models.push({\n            id: model.id,\n            name: model.displayName || model.id,\n            provider: provider?.id as string || 'custom',\n            providerName: provider?.displayName as string || 'Custom',\n            contextWindow: model.contextLength,\n            maxTokens: model.maxOutputTokens,\n          });\n        }\n      }\n    } catch (error) {\n      console.warn('Failed to get custom models:', error);\n    }\n\n    // 5. Always include platform default models (from xpod API)\n    try {\n      const baseUrl = xpodUrl.endsWith('/') ? xpodUrl : `${xpodUrl}/`;\n      const modelsUrl = `${baseUrl}v1/models`;\n\n      const response = await session.fetch(modelsUrl, {\n        headers: {\n          'Accept': 'application/json',\n        },\n      });\n\n      if (response.ok) {\n        const data = await response.json() as { object: string; data: any[] };\n        for (const model of data.data) {\n          // Avoid duplicates\n          if (!models.some(m => m.id === model.id)) {\n            models.push({\n              id: model.id,\n              name: model.name || model.id,\n              provider: model.provider || 'platform',\n              providerName: model.owned_by || 'Platform Default',\n              contextWindow: model.context_window,\n              maxTokens: model.max_tokens,\n            });\n          }\n        }\n      } else {\n        console.warn(`Failed to fetch platform models: ${response.status}`);\n      }\n    } catch (error) {\n      console.warn('Failed to get platform default models:', error);\n    }\n  } catch (error) {\n    console.error('Failed to get available models:', error);\n    // Return empty array on error\n  }\n\n  return models;\n}\n\n/**\n * Get a specific model by ID.\n */\nexport async function getModelById(session: Session, xpodUrl: string, modelId: string): Promise<AvailableModel | null> {\n  const models = await getAvailableModels(session, xpodUrl);\n  return models.find(m => m.id === modelId) || null;\n}\n\n/**\n * List available models (for CLI display).\n */\nexport async function listAvailableModels(session: Session, xpodUrl: string): Promise<void> {\n  const models = await getAvailableModels(session, xpodUrl);\n\n  console.log('\\n📋 Available Models:\\n');\n\n  // Group by provider\n  const byProvider = new Map<string, AvailableModel[]>();\n  for (const model of models) {\n    const list = byProvider.get(model.provider) || [];\n    list.push(model);\n    byProvider.set(model.provider, list);\n  }\n\n  for (const [, providerModels] of byProvider) {\n    console.log(`\\n${providerModels[0].providerName}:`);\n    for (const model of providerModels) {\n      const ctx = model.contextWindow ? ` (${model.contextWindow} tokens)` : '';\n      console.log(`  ${model.id}${ctx}`);\n    }\n  }\n\n  console.log('');\n}\n\n/**\n * Add a custom model to the user's Pod.\n */\nexport async function addCustomModel(session: Session, xpodUrl: string, modelId: string): Promise<void> {\n  try {\n    const db: any = drizzle({\n      fetch: session.fetch,\n      info: session.info,\n    } as any);\n\n    // Check if model already exists\n    const allModels = await db.select().from(Model) as any[];\n    if (allModels.some(m => m.id === modelId)) {\n      console.log(`✓ Model ${modelId} already exists`);\n      return;\n    }\n\n    // Prompt for provider\n    const { promptText } = await import('./prompt.js');\n    console.log('\\n📝 Add custom model\\n');\n\n    const displayName = await promptText(`Display name (default: ${modelId}): `);\n    const providerInput = await promptText('Provider ID (e.g., openai, anthropic, custom): ');\n    const providerId = providerInput || 'custom';\n\n    // Get or create provider\n    const allProviders = await db.select().from(Provider) as any[];\n    let provider = allProviders.find(p => p.id === providerId);\n\n    if (!provider) {\n      console.log(`\\n⚠️  Provider ${providerId} not found. Creating new provider...`);\n      const providerName = await promptText(`Provider display name (default: ${providerId}): `);\n      const baseUrl = await promptText('Base URL (optional): ');\n\n      await db.insert(Provider).values({\n        id: providerId,\n        displayName: providerName || providerId,\n        baseUrl: baseUrl || undefined,\n      });\n\n      // Reload provider\n      const updatedProviders = await db.select().from(Provider) as any[];\n      provider = updatedProviders.find(p => p.id === providerId);\n    }\n\n    if (!provider) {\n      throw new Error('Failed to create provider');\n    }\n\n    // Create model\n    const podUrl = xpodUrl.endsWith('/') ? xpodUrl : `${xpodUrl}/`;\n    const providerUri = `${podUrl}settings/ai/providers.ttl#${providerId}`;\n\n    await db.insert(Model).values({\n      id: modelId,\n      displayName: displayName || modelId,\n      isProvidedBy: providerUri,\n      status: 'active',\n      createdAt: new Date().toISOString(),\n      updatedAt: new Date().toISOString(),\n    });\n\n    console.log(`\\n✅ Model ${modelId} added successfully!`);\n    console.log(`   Provider: ${provider.displayName || providerId}`);\n    console.log(`   Display name: ${displayName || modelId}\\n`);\n  } catch (error) {\n    console.error('Failed to add custom model:', error);\n    throw error;\n  }\n}\n"]}

package/dist/cli/lib/oauth-credential-manager.d.ts

@@ -0,0 +1,37 @@
+/**
+ * OAuth Credential Manager
+ *
+ * 管理 AI Provider 的 OAuth credentials：
+ * - 存储到 Pod
+ * - 从 Pod 读取
+ * - Token 刷新
+ */
+import type { Session } from '@inrupt/solid-client-authn-node';
+import { type OAuthCredentials } from './pi-optional';
+/**
+ * 保存 OAuth credentials 到 Pod
+ *
+ * @param session - Solid session
+ * @param providerId - Provider ID (e.g., 'codebuddy')
+ * @param providerUri - Provider 的完整 URI (e.g., 'http://pod/settings/ai/providers.ttl#codebuddy')
+ * @param credentials - OAuth credentials from provider.login()
+ * @param label - 可选的标签
+ */
+export declare function saveOAuthCredential(session: Session, providerId: string, providerUri: string, credentials: OAuthCredentials, label?: string): Promise<string>;
+/**
+ * 刷新 OAuth token
+ *
+ * @param session - Solid session
+ * @param credentialId - Credential ID
+ * @param providerId - Provider ID (用于查找 OAuth provider)
+ */
+export declare function refreshOAuthToken(session: Session, credentialId: string, providerId: string): Promise<void>;
+/**
+ * 检查并自动刷新过期的 OAuth token
+ *
+ * @param session - Solid session
+ * @param credentialId - Credential ID
+ * @param providerId - Provider ID
+ * @returns 是否成功刷新
+ */
+export declare function ensureOAuthTokenValid(session: Session, credentialId: string, providerId: string): Promise<boolean>;

package/dist/cli/lib/oauth-credential-manager.js

@@ -0,0 +1,129 @@
+"use strict";
+/**
+ * OAuth Credential Manager
+ *
+ * 管理 AI Provider 的 OAuth credentials：
+ * - 存储到 Pod
+ * - 从 Pod 读取
+ * - Token 刷新
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.saveOAuthCredential = saveOAuthCredential;
+exports.refreshOAuthToken = refreshOAuthToken;
+exports.ensureOAuthTokenValid = ensureOAuthTokenValid;
+const drizzle_solid_1 = require("@undefineds.co/drizzle-solid");
+const tables_1 = require("../../credential/schema/tables");
+const types_1 = require("../../credential/schema/types");
+const pi_optional_1 = require("./pi-optional");
+/**
+ * 保存 OAuth credentials 到 Pod
+ *
+ * @param session - Solid session
+ * @param providerId - Provider ID (e.g., 'codebuddy')
+ * @param providerUri - Provider 的完整 URI (e.g., 'http://pod/settings/ai/providers.ttl#codebuddy')
+ * @param credentials - OAuth credentials from provider.login()
+ * @param label - 可选的标签
+ */
+async function saveOAuthCredential(session, providerId, providerUri, credentials, label) {
+    const db = (0, drizzle_solid_1.drizzle)({
+        fetch: session.fetch,
+        info: session.info,
+    });
+    // 生成 credential ID
+    const credentialId = `oauth-${providerId}-${Date.now()}`;
+    // 插入 OAuthCredential
+    await db.insert(tables_1.OAuthCredential).values({
+        id: credentialId,
+        provider: providerUri,
+        service: types_1.ServiceType.AI,
+        status: types_1.CredentialStatus.ACTIVE,
+        label: label || `${providerId} OAuth`,
+        oauthRefreshToken: credentials.refresh,
+        oauthAccessToken: credentials.access,
+        oauthExpiresAt: new Date(credentials.expires),
+    });
+    console.log(`✓ OAuth credential saved: ${credentialId}`);
+    return credentialId;
+}
+/**
+ * 刷新 OAuth token
+ *
+ * @param session - Solid session
+ * @param credentialId - Credential ID
+ * @param providerId - Provider ID (用于查找 OAuth provider)
+ */
+async function refreshOAuthToken(session, credentialId, providerId) {
+    const db = (0, drizzle_solid_1.drizzle)({
+        fetch: session.fetch,
+        info: session.info,
+    });
+    // 1. 读取现有 credential
+    const allCredentials = await db.select().from(tables_1.OAuthCredential);
+    const credential = allCredentials.find((c) => c.id === credentialId);
+    if (!credential) {
+        throw new Error(`OAuth credential not found: ${credentialId}`);
+    }
+    // 2. 获取 OAuth provider
+    const { getOAuthProvider } = await (0, pi_optional_1.loadPiAiOAuthUtils)();
+    const provider = getOAuthProvider(providerId);
+    if (!provider) {
+        throw new Error(`OAuth provider not found: ${providerId}`);
+    }
+    // 3. 刷新 token
+    const oldCredentials = {
+        refresh: credential.oauthRefreshToken || '',
+        access: credential.oauthAccessToken || '',
+        expires: credential.oauthExpiresAt ? new Date(credential.oauthExpiresAt).getTime() : 0,
+    };
+    const newCredentials = await provider.refreshToken(oldCredentials);
+    // 4. 更新 Pod 中的 credential
+    await db.update(tables_1.OAuthCredential)
+        .set({
+        oauthRefreshToken: newCredentials.refresh,
+        oauthAccessToken: newCredentials.access,
+        oauthExpiresAt: new Date(newCredentials.expires),
+    })
+        .where({ id: credentialId });
+    console.log(`✓ OAuth token refreshed: ${credentialId}`);
+}
+/**
+ * 检查并自动刷新过期的 OAuth token
+ *
+ * @param session - Solid session
+ * @param credentialId - Credential ID
+ * @param providerId - Provider ID
+ * @returns 是否成功刷新
+ */
+async function ensureOAuthTokenValid(session, credentialId, providerId) {
+    const db = (0, drizzle_solid_1.drizzle)({
+        fetch: session.fetch,
+        info: session.info,
+    });
+    // 读取 credential
+    const allCredentials = await db.select().from(tables_1.OAuthCredential);
+    const credential = allCredentials.find((c) => c.id === credentialId);
+    if (!credential) {
+        return false;
+    }
+    // 检查是否过期（提前 5 分钟刷新）
+    const expiresAt = credential.oauthExpiresAt ? new Date(credential.oauthExpiresAt) : null;
+    if (!expiresAt) {
+        return false;
+    }
+    const now = new Date();
+    const fiveMinutesLater = new Date(now.getTime() + 5 * 60 * 1000);
+    if (expiresAt > fiveMinutesLater) {
+        // Token 还有效
+        return true;
+    }
+    // Token 即将过期或已过期，尝试刷新
+    try {
+        await refreshOAuthToken(session, credentialId, providerId);
+        return true;
+    }
+    catch (error) {
+        console.error(`Failed to refresh OAuth token: ${error}`);
+        return false;
+    }
+}
+//# sourceMappingURL=oauth-credential-manager.js.map

package/dist/cli/lib/oauth-credential-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"oauth-credential-manager.js","sourceRoot":"","sources":["../../../src/cli/lib/oauth-credential-manager.ts"],"names":[],"mappings":";AAAA;;;;;;;GAOG;;AAiBH,kDA6BC;AASD,8CA4CC;AAUD,sDAwCC;AAnJD,gEAAuD;AACvD,2DAAiE;AACjE,yDAA8E;AAE9E,+CAA0E;AAE1E;;;;;;;;GAQG;AACI,KAAK,UAAU,mBAAmB,CACvC,OAAgB,EAChB,UAAkB,EAClB,WAAmB,EACnB,WAA6B,EAC7B,KAAc;IAEd,MAAM,EAAE,GAAQ,IAAA,uBAAO,EAAC;QACtB,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,IAAI,EAAE,OAAO,CAAC,IAAI;KACZ,CAAC,CAAC;IAEV,mBAAmB;IACnB,MAAM,YAAY,GAAG,SAAS,UAAU,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;IAEzD,qBAAqB;IACrB,MAAM,EAAE,CAAC,MAAM,CAAC,wBAAe,CAAC,CAAC,MAAM,CAAC;QACtC,EAAE,EAAE,YAAY;QAChB,QAAQ,EAAE,WAAW;QACrB,OAAO,EAAE,mBAAW,CAAC,EAAE;QACvB,MAAM,EAAE,wBAAgB,CAAC,MAAM;QAC/B,KAAK,EAAE,KAAK,IAAI,GAAG,UAAU,QAAQ;QACrC,iBAAiB,EAAE,WAAW,CAAC,OAAO;QACtC,gBAAgB,EAAE,WAAW,CAAC,MAAM;QACpC,cAAc,EAAE,IAAI,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC;KAC9C,CAAC,CAAC;IAEH,OAAO,CAAC,GAAG,CAAC,6BAA6B,YAAY,EAAE,CAAC,CAAC;IACzD,OAAO,YAAY,CAAC;AACtB,CAAC;AAED;;;;;;GAMG;AACI,KAAK,UAAU,iBAAiB,CACrC,OAAgB,EAChB,YAAoB,EACpB,UAAkB;IAElB,MAAM,EAAE,GAAQ,IAAA,uBAAO,EAAC;QACtB,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,IAAI,EAAE,OAAO,CAAC,IAAI;KACZ,CAAC,CAAC;IAEV,qBAAqB;IACrB,MAAM,cAAc,GAAG,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,wBAAe,CAAC,CAAC;IAC/D,MAAM,UAAU,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,YAAY,CAAC,CAAC;IAE1E,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,MAAM,IAAI,KAAK,CAAC,+BAA+B,YAAY,EAAE,CAAC,CAAC;IACjE,CAAC;IAED,uBAAuB;IACvB,MAAM,EAAE,gBAAgB,EAAE,GAAG,MAAM,IAAA,gCAAkB,GAAE,CAAC;IACxD,MAAM,QAAQ,GAAG,gBAAgB,CAAC,UAAU,CAAC,CAAC;IAC9C,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CAAC,6BAA6B,UAAU,EAAE,CAAC,CAAC;IAC7D,CAAC;IAED,cAAc;IACd,MAAM,cAAc,GAAqB;QACvC,OAAO,EAAE,UAAU,CAAC,iBAAiB,IAAI,EAAE;QAC3C,MAAM,EAAE,UAAU,CAAC,gBAAgB,IAAI,EAAE;QACzC,OAAO,EAAE,UAAU,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;KACvF,CAAC;IAEF,MAAM,cAAc,GAAG,MAAM,QAAQ,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC;IAEnE,0BAA0B;IAC1B,MAAM,EAAE,CAAC,MAAM,CAAC,wBAAe,CAAC;SAC7B,GAAG,CAAC;QACH,iBAAiB,EAAE,cAAc,CAAC,OAAO;QACzC,gBAAgB,EAAE,cAAc,CAAC,MAAM;QACvC,cAAc,EAAE,IAAI,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC;KACjD,CAAC;SACD,KAAK,CAAC,EAAE,EAAE,EAAE,YAAY,EAAS,CAAC,CAAC;IAEtC,OAAO,CAAC,GAAG,CAAC,4BAA4B,YAAY,EAAE,CAAC,CAAC;AAC1D,CAAC;AAED;;;;;;;GAOG;AACI,KAAK,UAAU,qBAAqB,CACzC,OAAgB,EAChB,YAAoB,EACpB,UAAkB;IAElB,MAAM,EAAE,GAAQ,IAAA,uBAAO,EAAC;QACtB,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,IAAI,EAAE,OAAO,CAAC,IAAI;KACZ,CAAC,CAAC;IAEV,gBAAgB;IAChB,MAAM,cAAc,GAAG,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,wBAAe,CAAC,CAAC;IAC/D,MAAM,UAAU,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,YAAY,CAAC,CAAC;IAE1E,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,oBAAoB;IACpB,MAAM,SAAS,GAAG,UAAU,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IACzF,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;IACvB,MAAM,gBAAgB,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;IAEjE,IAAI,SAAS,GAAG,gBAAgB,EAAE,CAAC;QACjC,YAAY;QACZ,OAAO,IAAI,CAAC;IACd,CAAC;IAED,sBAAsB;IACtB,IAAI,CAAC;QACH,MAAM,iBAAiB,CAAC,OAAO,EAAE,YAAY,EAAE,UAAU,CAAC,CAAC;QAC3D,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,kCAAkC,KAAK,EAAE,CAAC,CAAC;QACzD,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC","sourcesContent":["/**\n * OAuth Credential Manager\n *\n * 管理 AI Provider 的 OAuth credentials：\n * - 存储到 Pod\n * - 从 Pod 读取\n * - Token 刷新\n */\n\nimport { drizzle } from '@undefineds.co/drizzle-solid';\nimport { OAuthCredential } from '../../credential/schema/tables';\nimport { ServiceType, CredentialStatus } from '../../credential/schema/types';\nimport type { Session } from '@inrupt/solid-client-authn-node';\nimport { loadPiAiOAuthUtils, type OAuthCredentials } from './pi-optional';\n\n/**\n * 保存 OAuth credentials 到 Pod\n *\n * @param session - Solid session\n * @param providerId - Provider ID (e.g., 'codebuddy')\n * @param providerUri - Provider 的完整 URI (e.g., 'http://pod/settings/ai/providers.ttl#codebuddy')\n * @param credentials - OAuth credentials from provider.login()\n * @param label - 可选的标签\n */\nexport async function saveOAuthCredential(\n  session: Session,\n  providerId: string,\n  providerUri: string,\n  credentials: OAuthCredentials,\n  label?: string,\n): Promise<string> {\n  const db: any = drizzle({\n    fetch: session.fetch,\n    info: session.info,\n  } as any);\n\n  // 生成 credential ID\n  const credentialId = `oauth-${providerId}-${Date.now()}`;\n\n  // 插入 OAuthCredential\n  await db.insert(OAuthCredential).values({\n    id: credentialId,\n    provider: providerUri,\n    service: ServiceType.AI,\n    status: CredentialStatus.ACTIVE,\n    label: label || `${providerId} OAuth`,\n    oauthRefreshToken: credentials.refresh,\n    oauthAccessToken: credentials.access,\n    oauthExpiresAt: new Date(credentials.expires),\n  });\n\n  console.log(`✓ OAuth credential saved: ${credentialId}`);\n  return credentialId;\n}\n\n/**\n * 刷新 OAuth token\n *\n * @param session - Solid session\n * @param credentialId - Credential ID\n * @param providerId - Provider ID (用于查找 OAuth provider)\n */\nexport async function refreshOAuthToken(\n  session: Session,\n  credentialId: string,\n  providerId: string,\n): Promise<void> {\n  const db: any = drizzle({\n    fetch: session.fetch,\n    info: session.info,\n  } as any);\n\n  // 1. 读取现有 credential\n  const allCredentials = await db.select().from(OAuthCredential);\n  const credential = allCredentials.find((c: any) => c.id === credentialId);\n\n  if (!credential) {\n    throw new Error(`OAuth credential not found: ${credentialId}`);\n  }\n\n  // 2. 获取 OAuth provider\n  const { getOAuthProvider } = await loadPiAiOAuthUtils();\n  const provider = getOAuthProvider(providerId);\n  if (!provider) {\n    throw new Error(`OAuth provider not found: ${providerId}`);\n  }\n\n  // 3. 刷新 token\n  const oldCredentials: OAuthCredentials = {\n    refresh: credential.oauthRefreshToken || '',\n    access: credential.oauthAccessToken || '',\n    expires: credential.oauthExpiresAt ? new Date(credential.oauthExpiresAt).getTime() : 0,\n  };\n\n  const newCredentials = await provider.refreshToken(oldCredentials);\n\n  // 4. 更新 Pod 中的 credential\n  await db.update(OAuthCredential)\n    .set({\n      oauthRefreshToken: newCredentials.refresh,\n      oauthAccessToken: newCredentials.access,\n      oauthExpiresAt: new Date(newCredentials.expires),\n    })\n    .where({ id: credentialId } as any);\n\n  console.log(`✓ OAuth token refreshed: ${credentialId}`);\n}\n\n/**\n * 检查并自动刷新过期的 OAuth token\n *\n * @param session - Solid session\n * @param credentialId - Credential ID\n * @param providerId - Provider ID\n * @returns 是否成功刷新\n */\nexport async function ensureOAuthTokenValid(\n  session: Session,\n  credentialId: string,\n  providerId: string,\n): Promise<boolean> {\n  const db: any = drizzle({\n    fetch: session.fetch,\n    info: session.info,\n  } as any);\n\n  // 读取 credential\n  const allCredentials = await db.select().from(OAuthCredential);\n  const credential = allCredentials.find((c: any) => c.id === credentialId);\n\n  if (!credential) {\n    return false;\n  }\n\n  // 检查是否过期（提前 5 分钟刷新）\n  const expiresAt = credential.oauthExpiresAt ? new Date(credential.oauthExpiresAt) : null;\n  if (!expiresAt) {\n    return false;\n  }\n\n  const now = new Date();\n  const fiveMinutesLater = new Date(now.getTime() + 5 * 60 * 1000);\n\n  if (expiresAt > fiveMinutesLater) {\n    // Token 还有效\n    return true;\n  }\n\n  // Token 即将过期或已过期，尝试刷新\n  try {\n    await refreshOAuthToken(session, credentialId, providerId);\n    return true;\n  } catch (error) {\n    console.error(`Failed to refresh OAuth token: ${error}`);\n    return false;\n  }\n}\n"]}

package/dist/cli/lib/oauth-providers/codebuddy.d.ts

@@ -0,0 +1,25 @@
+/**
+ * CodeBuddy OAuth Provider
+ *
+ * 通过 @tencent-ai/agent-sdk 的 authenticate() 实现 OAuth 登录。
+ * 登录后 token 存入 Pod 的 Credential 表，跨环境复用。
+ */
+import type { OAuthCredentials, OAuthProviderInterface } from '../pi-optional';
+/**
+ * CodeBuddy OAuth 登录
+ *
+ * 调用 @tencent-ai/agent-sdk 的 authenticate()，
+ * 打开浏览器让用户登录腾讯账号，获取 token。
+ */
+export declare function loginCodeBuddy(onAuthUrl: (url: string) => void): Promise<OAuthCredentials>;
+/**
+ * CodeBuddy token 刷新
+ *
+ * CodeBuddy SDK 内部管理 token 生命周期，
+ * 如果 token 过期，需要重新登录。
+ */
+export declare function refreshCodeBuddyToken(credentials: OAuthCredentials): Promise<OAuthCredentials>;
+/**
+ * CodeBuddy OAuth Provider 实现
+ */
+export declare const codeBuddyOAuthProvider: OAuthProviderInterface;

package/dist/cli/lib/oauth-providers/codebuddy.js

@@ -0,0 +1,69 @@
+"use strict";
+/**
+ * CodeBuddy OAuth Provider
+ *
+ * 通过 @tencent-ai/agent-sdk 的 authenticate() 实现 OAuth 登录。
+ * 登录后 token 存入 Pod 的 Credential 表，跨环境复用。
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.codeBuddyOAuthProvider = void 0;
+exports.loginCodeBuddy = loginCodeBuddy;
+exports.refreshCodeBuddyToken = refreshCodeBuddyToken;
+/**
+ * CodeBuddy OAuth 登录
+ *
+ * 调用 @tencent-ai/agent-sdk 的 authenticate()，
+ * 打开浏览器让用户登录腾讯账号，获取 token。
+ */
+async function loginCodeBuddy(onAuthUrl) {
+    // 动态导入，避免未安装时报错
+    const mod = await (new Function('specifier', 'return import(specifier)'))('@tencent-ai/agent-sdk');
+    const authenticate = mod.unstable_v2_authenticate;
+    if (typeof authenticate !== 'function') {
+        throw new Error('@tencent-ai/agent-sdk 不可用或版本过低，需要 unstable_v2_authenticate');
+    }
+    const result = await authenticate({
+        environment: 'external',
+        onAuthUrl: (authState) => {
+            onAuthUrl(authState.authUrl);
+        },
+    });
+    return {
+        refresh: '',
+        access: result.userinfo.token,
+        expires: Date.now() + 24 * 3600_000,
+        userId: result.userinfo.userId,
+        userName: result.userinfo.userName,
+    };
+}
+/**
+ * CodeBuddy token 刷新
+ *
+ * CodeBuddy SDK 内部管理 token 生命周期，
+ * 如果 token 过期，需要重新登录。
+ */
+async function refreshCodeBuddyToken(credentials) {
+    if (credentials.expires > Date.now()) {
+        return credentials;
+    }
+    throw new Error('CodeBuddy token expired, please login again');
+}
+/**
+ * CodeBuddy OAuth Provider 实现
+ */
+exports.codeBuddyOAuthProvider = {
+    id: 'codebuddy',
+    name: 'CodeBuddy',
+    async login(callbacks) {
+        return loginCodeBuddy((url) => {
+            callbacks.onAuth({ url, instructions: '请在浏览器中登录腾讯账号' });
+        });
+    },
+    async refreshToken(credentials) {
+        return refreshCodeBuddyToken(credentials);
+    },
+    getApiKey(credentials) {
+        return credentials.access;
+    },
+};
+//# sourceMappingURL=codebuddy.js.map

package/dist/cli/lib/oauth-providers/codebuddy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"codebuddy.js","sourceRoot":"","sources":["../../../../src/cli/lib/oauth-providers/codebuddy.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AA0BH,wCA8BC;AAQD,sDAOC;AAnDD;;;;;GAKG;AACI,KAAK,UAAU,cAAc,CAClC,SAAgC;IAEhC,gBAAgB;IAChB,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,QAAQ,CAC7B,WAAW,EACX,0BAA0B,CAC3B,CAAC,CAAC,uBAAuB,CAAwB,CAAC;IAEnD,MAAM,YAAY,GAAG,GAAG,CAAC,wBAAwB,CAAC;IAClD,IAAI,OAAO,YAAY,KAAK,UAAU,EAAE,CAAC;QACvC,MAAM,IAAI,KAAK,CACb,4DAA4D,CAC7D,CAAC;IACJ,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;QAChC,WAAW,EAAE,UAAU;QACvB,SAAS,EAAE,CAAC,SAA6B,EAAE,EAAE;YAC3C,SAAS,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC;KACF,CAAoC,CAAC;IAEtC,OAAO;QACL,OAAO,EAAE,EAAE;QACX,MAAM,EAAE,MAAM,CAAC,QAAQ,CAAC,KAAK;QAC7B,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,QAAQ;QACnC,MAAM,EAAE,MAAM,CAAC,QAAQ,CAAC,MAAM;QAC9B,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,QAAQ;KACnC,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACI,KAAK,UAAU,qBAAqB,CACzC,WAA6B;IAE7B,IAAI,WAAW,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;QACrC,OAAO,WAAW,CAAC;IACrB,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;AACjE,CAAC;AAED;;GAEG;AACU,QAAA,sBAAsB,GAA2B;IAC5D,EAAE,EAAE,WAAW;IACf,IAAI,EAAE,WAAW;IAEjB,KAAK,CAAC,KAAK,CAAC,SAA8B;QACxC,OAAO,cAAc,CAAC,CAAC,GAAG,EAAE,EAAE;YAC5B,SAAS,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,YAAY,EAAE,cAAc,EAAE,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,WAA6B;QAC9C,OAAO,qBAAqB,CAAC,WAAW,CAAC,CAAC;IAC5C,CAAC;IAED,SAAS,CAAC,WAA6B;QACrC,OAAO,WAAW,CAAC,MAAM,CAAC;IAC5B,CAAC;CACF,CAAC","sourcesContent":["/**\n * CodeBuddy OAuth Provider\n *\n * 通过 @tencent-ai/agent-sdk 的 authenticate() 实现 OAuth 登录。\n * 登录后 token 存入 Pod 的 Credential 表，跨环境复用。\n */\n\nimport type {\n  OAuthCredentials,\n  OAuthLoginCallbacks,\n  OAuthProviderInterface,\n} from '../pi-optional';\n\ninterface CodeBuddyAuthState {\n  state: string;\n  authUrl: string;\n}\n\ninterface CodeBuddyUserinfo {\n  userId: string;\n  userName: string;\n  userNickname: string;\n  token: string;\n}\n\n/**\n * CodeBuddy OAuth 登录\n *\n * 调用 @tencent-ai/agent-sdk 的 authenticate()，\n * 打开浏览器让用户登录腾讯账号，获取 token。\n */\nexport async function loginCodeBuddy(\n  onAuthUrl: (url: string) => void,\n): Promise<OAuthCredentials> {\n  // 动态导入，避免未安装时报错\n  const mod = await (new Function(\n    'specifier',\n    'return import(specifier)',\n  ))('@tencent-ai/agent-sdk') as Record<string, any>;\n\n  const authenticate = mod.unstable_v2_authenticate;\n  if (typeof authenticate !== 'function') {\n    throw new Error(\n      '@tencent-ai/agent-sdk 不可用或版本过低，需要 unstable_v2_authenticate',\n    );\n  }\n\n  const result = await authenticate({\n    environment: 'external',\n    onAuthUrl: (authState: CodeBuddyAuthState) => {\n      onAuthUrl(authState.authUrl);\n    },\n  }) as { userinfo: CodeBuddyUserinfo };\n\n  return {\n    refresh: '',\n    access: result.userinfo.token,\n    expires: Date.now() + 24 * 3600_000,\n    userId: result.userinfo.userId,\n    userName: result.userinfo.userName,\n  };\n}\n\n/**\n * CodeBuddy token 刷新\n *\n * CodeBuddy SDK 内部管理 token 生命周期，\n * 如果 token 过期，需要重新登录。\n */\nexport async function refreshCodeBuddyToken(\n  credentials: OAuthCredentials,\n): Promise<OAuthCredentials> {\n  if (credentials.expires > Date.now()) {\n    return credentials;\n  }\n  throw new Error('CodeBuddy token expired, please login again');\n}\n\n/**\n * CodeBuddy OAuth Provider 实现\n */\nexport const codeBuddyOAuthProvider: OAuthProviderInterface = {\n  id: 'codebuddy',\n  name: 'CodeBuddy',\n\n  async login(callbacks: OAuthLoginCallbacks): Promise<OAuthCredentials> {\n    return loginCodeBuddy((url) => {\n      callbacks.onAuth({ url, instructions: '请在浏览器中登录腾讯账号' });\n    });\n  },\n\n  async refreshToken(credentials: OAuthCredentials): Promise<OAuthCredentials> {\n    return refreshCodeBuddyToken(credentials);\n  },\n\n  getApiKey(credentials: OAuthCredentials): string {\n    return credentials.access;\n  },\n};\n"]}

package/dist/cli/lib/oauth-providers/index.d.ts

@@ -0,0 +1,12 @@
+/**
+ * OAuth Providers Registry
+ *
+ * 注册所有 AI Provider 的 OAuth 实现。
+ * Solid OIDC 不在这里——它是"统一钥匙"，存本地。
+ * 这里的 provider token 存 Pod，跨环境复用。
+ */
+/**
+ * 注册所有自定义 OAuth providers 到 pi-mono 框架
+ */
+export declare function registerCustomOAuthProviders(): Promise<void>;
+export { codeBuddyOAuthProvider } from './codebuddy';

package/dist/cli/lib/oauth-providers/index.js

@@ -0,0 +1,23 @@
+"use strict";
+/**
+ * OAuth Providers Registry
+ *
+ * 注册所有 AI Provider 的 OAuth 实现。
+ * Solid OIDC 不在这里——它是"统一钥匙"，存本地。
+ * 这里的 provider token 存 Pod，跨环境复用。
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.codeBuddyOAuthProvider = void 0;
+exports.registerCustomOAuthProviders = registerCustomOAuthProviders;
+const pi_optional_1 = require("../pi-optional");
+const codebuddy_1 = require("./codebuddy");
+/**
+ * 注册所有自定义 OAuth providers 到 pi-mono 框架
+ */
+async function registerCustomOAuthProviders() {
+    const { registerOAuthProvider } = await (0, pi_optional_1.loadPiAiOAuthUtils)();
+    registerOAuthProvider(codebuddy_1.codeBuddyOAuthProvider);
+}
+var codebuddy_2 = require("./codebuddy");
+Object.defineProperty(exports, "codeBuddyOAuthProvider", { enumerable: true, get: function () { return codebuddy_2.codeBuddyOAuthProvider; } });
+//# sourceMappingURL=index.js.map

package/dist/cli/lib/oauth-providers/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/cli/lib/oauth-providers/index.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;;AAQH,oEAGC;AATD,gDAAoD;AACpD,2CAAqD;AAErD;;GAEG;AACI,KAAK,UAAU,4BAA4B;IAChD,MAAM,EAAE,qBAAqB,EAAE,GAAG,MAAM,IAAA,gCAAkB,GAAE,CAAC;IAC7D,qBAAqB,CAAC,kCAAsB,CAAC,CAAC;AAChD,CAAC;AAED,yCAAqD;AAA5C,mHAAA,sBAAsB,OAAA","sourcesContent":["/**\n * OAuth Providers Registry\n *\n * 注册所有 AI Provider 的 OAuth 实现。\n * Solid OIDC 不在这里——它是\"统一钥匙\"，存本地。\n * 这里的 provider token 存 Pod，跨环境复用。\n */\n\nimport { loadPiAiOAuthUtils } from '../pi-optional';\nimport { codeBuddyOAuthProvider } from './codebuddy';\n\n/**\n * 注册所有自定义 OAuth providers 到 pi-mono 框架\n */\nexport async function registerCustomOAuthProviders(): Promise<void> {\n  const { registerOAuthProvider } = await loadPiAiOAuthUtils();\n  registerOAuthProvider(codeBuddyOAuthProvider);\n}\n\nexport { codeBuddyOAuthProvider } from './codebuddy';\n"]}