@blocklet/pages-kit-agents 0.6.58 → 0.6.60

package/lib/cjs/utils/agent-utils.d.ts

@@ -1,9 +1,12 @@
-import { AnthropicChatModel } from '@aigne/anthropic';
-import { DeepSeekChatModel } from '@aigne/deepseek';
+import { AIGNEHubChatModel } from '@aigne/aigne-hub';
 import { FSMemory, FSMemoryOptions } from '@aigne/fs-memory';
-import { OpenAIChatModel } from '@aigne/openai';
-import { XAIChatModel } from '@aigne/xai';
-export declare function getModel(modelName?: string): OpenAIChatModel | AnthropicChatModel | XAIChatModel | DeepSeekChatModel | undefined;
+export declare const getAigneConfig: () => Promise<{
+    apiKey: any;
+    url: string;
+    provider: string | undefined;
+    model: string | undefined;
+}>;
+export declare function getModel(modelName?: string): Promise<AIGNEHubChatModel>;
 export declare function getMemory({ projectId, ...rest }: Pick<FSMemoryOptions, 'autoUpdate'> & {
     projectId?: string;
 }): FSMemory;

package/lib/cjs/utils/agent-utils.js

@@ -3,17 +3,14 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.getAigneConfig = void 0;
 exports.getModel = getModel;
 exports.getMemory = getMemory;
-const anthropic_1 = require("@aigne/anthropic");
-const deepseek_1 = require("@aigne/deepseek");
+const aigne_hub_1 = require("@aigne/aigne-hub");
 const fs_memory_1 = require("@aigne/fs-memory");
-const gemini_1 = require("@aigne/gemini");
-const openai_1 = require("@aigne/openai");
-const xai_1 = require("@aigne/xai");
-const node_assert_1 = __importDefault(require("node:assert"));
+const axios_1 = __importDefault(require("axios"));
 const node_path_1 = __importDefault(require("node:path"));
-const { OPENAI_API_KEY, GEMINI_API_KEY, CLAUDE_API_KEY, XAI_API_KEY, DEEPSEEK_API_KEY } = process.env;
+const ufo_1 = require("ufo");
 // === 内置 prompt 内容，避免运行时读取文件 ===
 const recorderInstructions = `你是一个智能的站点生成记忆管理器，负责从多轮对话和用户反馈中提取对后续站点生成有用的知识和偏好，并以结构化方式存储。
 
@@ -106,54 +103,45 @@ const retrieverInstructions = `你是一个智能的站点生成记忆检索器
 {{search}}
 </search-query>
 `;
-function getModel(modelName = 'openai') {
-    if (modelName?.includes('gpt')) {
-        (0, node_assert_1.default)(OPENAI_API_KEY, 'Please set the OPENAI_API_KEY environment variable');
-        const openai = new openai_1.OpenAIChatModel({
-            apiKey: OPENAI_API_KEY,
-            model: modelName,
-            modelOptions: {
-                temperature: 0.8,
-            },
-        });
-        return openai;
+const getAigneHubModelApi = async (url) => {
+    try {
+        const urlObj = new URL(url);
+        const appUrl = urlObj.origin;
+        const { data: blockletJson } = await axios_1.default.get((0, ufo_1.joinURL)(appUrl, '__blocklet__.js?type=json'));
+        const { componentMountPoints = [] } = blockletJson || {};
+        const aigneHubMountPoint = componentMountPoints.find((item) => item.name === 'ai-kit');
+        if (!aigneHubMountPoint) {
+            throw new Error("The current application doesn't have the AIGNE Hub component installed");
+        }
+        return (0, ufo_1.joinURL)(appUrl, aigneHubMountPoint.mountPoint);
     }
-    if (modelName?.includes('gemini')) {
-        (0, node_assert_1.default)(GEMINI_API_KEY, 'Please set the GEMINI_API_KEY environment variable');
-        const gemini = new gemini_1.GeminiChatModel({
-            apiKey: GEMINI_API_KEY,
-            model: modelName,
-        });
-        return gemini;
+    catch (error) {
+        throw new Error('Invalid url');
     }
-    if (modelName?.includes('claude')) {
-        (0, node_assert_1.default)(CLAUDE_API_KEY, 'Please set the CLAUDE_API_KEY environment variable');
-        const model = new anthropic_1.AnthropicChatModel({
-            apiKey: CLAUDE_API_KEY,
-            model: modelName,
-            modelOptions: {
-                temperature: 0.7,
-            },
-        });
-        return model;
-    }
-    if (modelName?.includes('grok')) {
-        (0, node_assert_1.default)(XAI_API_KEY, 'Please set the XAI_API_KEY environment variable');
-        const grok = new xai_1.XAIChatModel({
-            apiKey: XAI_API_KEY,
-            model: modelName,
-        });
-        return grok;
-    }
-    if (modelName?.includes('deepseek')) {
-        (0, node_assert_1.default)(DEEPSEEK_API_KEY, 'Please set the DEEPSEEK_API_KEY environment variable');
-        const deepseek = new deepseek_1.DeepSeekChatModel({
-            apiKey: DEEPSEEK_API_KEY,
-            model: modelName,
-        });
-        return deepseek;
+};
+const getAigneConfig = async () => {
+    let apiKey = {};
+    const credential = process.env.BLOCKLET_AIGNE_API_CREDENTIAL;
+    if (credential && typeof credential === 'string') {
+        apiKey = JSON.parse(credential);
     }
-    return undefined;
+    const baseUrl = await getAigneHubModelApi(process.env.BLOCKLET_AIGNE_API_URL || '');
+    return {
+        apiKey: apiKey.apiKey,
+        url: baseUrl,
+        provider: process.env.BLOCKLET_AIGNE_API_PROVIDER,
+        model: process.env.BLOCKLET_AIGNE_API_MODEL,
+    };
+};
+exports.getAigneConfig = getAigneConfig;
+async function getModel(modelName) {
+    const config = await (0, exports.getAigneConfig)();
+    const model = new aigne_hub_1.AIGNEHubChatModel({
+        model: modelName ?? config.model,
+        apiKey: config.apiKey,
+        url: config.url,
+    });
+    return model;
 }
 function getMemory({ projectId = 'debug', ...rest }) {
     // === FSMemory 配置 ===

package/lib/esm/agents/multi-agent-page-writer.js

@@ -47,7 +47,7 @@ export default async function generatePageContentMultiAgent(input) {
             throw new Error('Client disconnected');
         }
         let structurePlanMap = {};
-        const model = getModel(modelName);
+        const model = await getModel(modelName);
         // 3-4 使用反思工作流生成页面结构
         const planningEngine = new AIGNE({
             model,
@@ -175,7 +175,7 @@ export async function generateSingleSectionContent({ context, question, locale,
             generatedSections: JSON.stringify(existingSections),
         };
         const sectionContentAgent = await createSectionContentAgent(sectionContentInput);
-        const model = getModel(modelName);
+        const model = await getModel(modelName);
         const engine = new AIGNE({
             model,
         });

package/lib/esm/agents/template-designer.js

@@ -3,13 +3,6 @@ import { AIAgent } from '@aigne/core';
 // import assert from 'node:assert';
 import { z } from 'zod';
 import { getMemory } from '../utils';
-// const { OPENAI_API_KEY } = process.env;
-// assert(OPENAI_API_KEY, 'Please set the OPENAI_API_KEY environment variable');
-// const memory = new DefaultMemory({
-//   storage: {
-//     url: 'file:./dbs/memory.db', // Path to store memory data, such as 'file:./memory.db'
-//   },
-// });
 const agentInstructions = `
 你是一个专业的页面设计师，名叫小设。你的任务是根据整理好的用户需求，设计一个页面结构。
 

package/lib/esm/agents/template-generator.js

@@ -6,13 +6,6 @@ import path from 'node:path';
 import { z } from 'zod';
 import { getMemory } from '../utils/agent-utils.js';
 import { ReflectionAgent } from '../workflow-agents/reflection-agent.js';
-// const { OPENAI_API_KEY } = process.env;
-// assert(OPENAI_API_KEY, 'Please set the OPENAI_API_KEY environment variable');
-// const memory = new DefaultMemory({
-//   storage: {
-//     url: 'file:./dbs/memory.db', // Path to store memory data, such as 'file:./memory.db'
-//   },
-// });
 const pagesKitTemplateRule = fs.readFileSync(path.join(__dirname, '../prompts/pages-kit-template-rule.md'), 'utf-8');
 const contextRule = `
 ## 技术规范参考