@blocklet/pages-kit-agents 0.6.57 → 0.6.59

package/lib/esm/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/esm/utils/agent-utils.js

@@ -1,12 +1,8 @@
-import { AnthropicChatModel } from '@aigne/anthropic';
-import { DeepSeekChatModel } from '@aigne/deepseek';
+import { AIGNEHubChatModel } from '@aigne/aigne-hub';
 import { FSMemory } from '@aigne/fs-memory';
-import { GeminiChatModel } from '@aigne/gemini';
-import { OpenAIChatModel } from '@aigne/openai';
-import { XAIChatModel } from '@aigne/xai';
-import assert from 'node:assert';
+import axios from 'axios';
 import path from 'node:path';
-const { OPENAI_API_KEY, GEMINI_API_KEY, CLAUDE_API_KEY, XAI_API_KEY, DEEPSEEK_API_KEY } = process.env;
+import { joinURL } from 'ufo';
 // === 内置 prompt 内容，避免运行时读取文件 ===
 const recorderInstructions = `你是一个智能的站点生成记忆管理器，负责从多轮对话和用户反馈中提取对后续站点生成有用的知识和偏好，并以结构化方式存储。
 
@@ -99,54 +95,44 @@ const retrieverInstructions = `你是一个智能的站点生成记忆检索器
 {{search}}
 </search-query>
 `;
-export function getModel(modelName = 'openai') {
-    if (modelName?.includes('gpt')) {
-        assert(OPENAI_API_KEY, 'Please set the OPENAI_API_KEY environment variable');
-        const openai = new 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.get(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 joinURL(appUrl, aigneHubMountPoint.mountPoint);
     }
-    if (modelName?.includes('gemini')) {
-        assert(GEMINI_API_KEY, 'Please set the GEMINI_API_KEY environment variable');
-        const gemini = new GeminiChatModel({
-            apiKey: GEMINI_API_KEY,
-            model: modelName,
-        });
-        return gemini;
+    catch (error) {
+        throw new Error('Invalid url');
     }
-    if (modelName?.includes('claude')) {
-        assert(CLAUDE_API_KEY, 'Please set the CLAUDE_API_KEY environment variable');
-        const model = new AnthropicChatModel({
-            apiKey: CLAUDE_API_KEY,
-            model: modelName,
-            modelOptions: {
-                temperature: 0.7,
-            },
-        });
-        return model;
+};
+export const getAigneConfig = async () => {
+    let apiKey = {};
+    const credential = process.env.BLOCKLET_AIGNE_API_CREDENTIAL;
+    if (credential && typeof credential === 'string') {
+        apiKey = JSON.parse(credential);
     }
-    if (modelName?.includes('grok')) {
-        assert(XAI_API_KEY, 'Please set the XAI_API_KEY environment variable');
-        const grok = new XAIChatModel({
-            apiKey: XAI_API_KEY,
-            model: modelName,
-        });
-        return grok;
-    }
-    if (modelName?.includes('deepseek')) {
-        assert(DEEPSEEK_API_KEY, 'Please set the DEEPSEEK_API_KEY environment variable');
-        const deepseek = new DeepSeekChatModel({
-            apiKey: DEEPSEEK_API_KEY,
-            model: modelName,
-        });
-        return deepseek;
-    }
-    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,
+    };
+};
+export async function getModel(modelName) {
+    const config = await getAigneConfig();
+    const model = new AIGNEHubChatModel({
+        model: modelName ?? config.model,
+        apiKey: config.apiKey,
+        url: config.url,
+    });
+    return model;
 }
 export function getMemory({ projectId = 'debug', ...rest }) {
     // === FSMemory 配置 ===