ai-world-sdk 1.2.7 → 1.3.2

package/dist/__tests__/example.test.js

@@ -40,6 +40,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 const dotenv = __importStar(require("dotenv"));
 const index_1 = require("../index");
 const fs_1 = require("fs");
+const ai_1 = require("ai");
 // Load environment variables from .env file
 dotenv.config();
 index_1.sdkConfig.setBaseUrl("http://localhost:8000");
@@ -1187,6 +1188,32 @@ describe("Langchain SDK Tests", () => {
             console.log("响应内容:", content);
         }
     }, 30000);
+    // ========== OpenRouter 测试 ==========
+    test("ChatOpenAI - OpenRouter 基础测试", async () => {
+        const provider = (0, index_1.createProvider)("openrouter", "openai", "test");
+        const result = await (0, ai_1.generateText)({
+            model: provider.languageModel('openai/gpt-4o-mini'),
+            prompt: '你好',
+        });
+        console.log("result:", result);
+    }, 30000);
+    test("ChatOpenAI - OpenRouter 流式调用", async () => {
+        const provider = (0, index_1.createProvider)("openrouter", "openai", "test");
+        const stream = (0, ai_1.streamText)({
+            model: provider.languageModel('openai/gpt-4o-mini'),
+            prompt: '你好',
+        });
+        let fullText = "";
+        for await (const part of stream.fullStream) {
+            switch (part.type) {
+                case 'text-delta':
+                    fullText += part.text;
+                    break;
+            }
+        }
+        expect(fullText.length).toBeGreaterThan(0);
+        console.log("fullText:", fullText);
+    }, 30000);
     // ========== OpenAI 视频生成测试 ==========
     test("OpenAIVideoGenerationClient - 创建视频生成任务（文本）", async () => {
         const openaiVideoClient = new index_1.OpenAIVideoGenerationClient({});

package/dist/__tests__/vscode-login.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/__tests__/vscode-login.test.js

@@ -0,0 +1,447 @@
+"use strict";
+/* eslint-disable @typescript-eslint/no-var-requires */
+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 () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+const http = __importStar(require("http"));
+// ── Mock vscode module ──────────────────────────────────────────────────────
+const mockStatusBarItem = {
+    text: "",
+    tooltip: "",
+    command: "",
+    backgroundColor: undefined,
+    show: jest.fn(),
+    hide: jest.fn(),
+    dispose: jest.fn(),
+};
+const mockEventEmitter = {
+    event: jest.fn(),
+    fire: jest.fn(),
+    dispose: jest.fn(),
+};
+const mockPanel = {
+    webview: {
+        html: "",
+        onDidReceiveMessage: jest.fn(() => ({ dispose: jest.fn() })),
+        postMessage: jest.fn(),
+    },
+    reveal: jest.fn(),
+    dispose: jest.fn(),
+    onDidDispose: jest.fn(() => ({ dispose: jest.fn() })),
+};
+jest.mock("vscode", () => ({
+    EventEmitter: jest.fn(() => mockEventEmitter),
+    StatusBarAlignment: { Left: 1, Right: 2 },
+    ThemeColor: jest.fn((id) => ({ id })),
+    ViewColumn: { One: 1 },
+    Uri: {
+        parse: jest.fn((s) => ({ toString: () => s, fsPath: s })),
+        file: jest.fn((s) => ({ fsPath: s, toString: () => `file://${s}` })),
+    },
+    window: {
+        createStatusBarItem: jest.fn(() => mockStatusBarItem),
+        createWebviewPanel: jest.fn(() => mockPanel),
+        showInformationMessage: jest.fn(),
+        showWarningMessage: jest.fn(),
+        showErrorMessage: jest.fn(),
+    },
+    workspace: { workspaceFolders: undefined },
+    commands: {
+        registerCommand: jest.fn((_cmd, _cb) => ({ dispose: jest.fn() })),
+    },
+    env: {
+        openExternal: jest.fn().mockResolvedValue(true),
+    },
+}), { virtual: true });
+const vscode_login_1 = require("../vscode-login");
+// ── Helpers ──────────────────────────────────────────────────────────────────
+let nextPort = 19200;
+function getTestPort() { return nextPort++; }
+/** 创建带可用 SecretStorage mock 的 context */
+function createMockContext(extName = "test-extension") {
+    const store = new Map();
+    return {
+        subscriptions: [],
+        secrets: {
+            get: jest.fn((key) => Promise.resolve(store.get(key))),
+            store: jest.fn((key, value) => { store.set(key, value); return Promise.resolve(); }),
+            delete: jest.fn((key) => { store.delete(key); return Promise.resolve(); }),
+            _store: store,
+        },
+        extension: { packageJSON: { name: extName } },
+        extensionUri: { fsPath: "/mock" },
+        globalStorageUri: { fsPath: "/mock/.global" },
+    };
+}
+// ── Tests ────────────────────────────────────────────────────────────────────
+beforeEach(() => {
+    jest.clearAllMocks();
+    mockStatusBarItem.text = "";
+    mockStatusBarItem.tooltip = "";
+});
+describe("AIWorldExtension - 构造与基本属性", () => {
+    it("使用默认值创建实例", () => {
+        const ctx = createMockContext();
+        const ext = new vscode_login_1.AIWorldExtension(ctx);
+        expect(ext.baseUrl).toBe("https://aiworld.local:8000");
+        expect(ext.isAuthenticated).toBe(false);
+        expect(ext.token).toBeUndefined();
+        expect(ext.user).toBeUndefined();
+        ext.dispose();
+    });
+    it("pluginId 默认取 context.extension.packageJSON.name", () => {
+        const ctx = createMockContext("my-cool-plugin");
+        const ext = new vscode_login_1.AIWorldExtension(ctx);
+        expect(ext.pluginId).toBe("my-cool-plugin");
+        ext.dispose();
+    });
+    it("options.pluginId 覆盖默认值", () => {
+        const ctx = createMockContext("default-name");
+        const ext = new vscode_login_1.AIWorldExtension(ctx, { pluginId: "override-id" });
+        expect(ext.pluginId).toBe("override-id");
+        ext.dispose();
+    });
+    it("使用自定义选项", () => {
+        const ctx = createMockContext();
+        const ext = new vscode_login_1.AIWorldExtension(ctx, {
+            baseUrl: "http://localhost:9999",
+            pluginId: "test-ext",
+            debug: true,
+        });
+        expect(ext.baseUrl).toBe("http://localhost:9999");
+        expect(ext.pluginId).toBe("test-ext");
+        ext.dispose();
+    });
+    it("注册了 3 个命令", () => {
+        const vsc = require("vscode");
+        const ctx = createMockContext();
+        const ext = new vscode_login_1.AIWorldExtension(ctx);
+        const calls = vsc.commands.registerCommand.mock.calls.map((c) => c[0]);
+        expect(calls).toContain("aiWorld.login");
+        expect(calls).toContain("aiWorld.logout");
+        expect(calls).toContain("aiWorld.showLoginPanel");
+        ext.dispose();
+    });
+});
+describe("AIWorldExtension - SecretStorage 缓存", () => {
+    let authServer = null;
+    afterEach((done) => {
+        if (authServer?.listening) {
+            authServer.close(() => { authServer = null; done(); });
+        }
+        else {
+            authServer = null;
+            done();
+        }
+    });
+    it("_bootstrap 从 SecretStorage 读取缓存 token", async () => {
+        authServer = http.createServer((req, res) => {
+            if (req.url === "/api/auth/me" && req.headers.authorization === "Bearer secret-tok") {
+                res.writeHead(200, { "Content-Type": "application/json" });
+                res.end(JSON.stringify({ id: 1, full_name: "SecretUser" }));
+            }
+            else {
+                res.writeHead(401);
+                res.end();
+            }
+        });
+        await new Promise((r) => authServer.listen(0, r));
+        const port = authServer.address().port;
+        const ctx = createMockContext();
+        ctx.secrets._store.set("ai-world:token", "secret-tok");
+        ctx.secrets._store.set("ai-world:baseUrl", `http://localhost:${port}`);
+        const ext = new vscode_login_1.AIWorldExtension(ctx, { baseUrl: `http://localhost:${port}` });
+        await ext._bootstrap(false);
+        expect(ext.isAuthenticated).toBe(true);
+        expect(ext.token).toBe("secret-tok");
+        expect(ext.user?.full_name).toBe("SecretUser");
+        ext.dispose();
+    });
+    it("login 成功后写入 SecretStorage", async () => {
+        authServer = http.createServer((req, res) => {
+            if (req.url === "/api/auth/me" && req.headers.authorization === "Bearer new-tok") {
+                res.writeHead(200, { "Content-Type": "application/json" });
+                res.end(JSON.stringify({ id: 2, full_name: "NewUser" }));
+            }
+            else {
+                res.writeHead(401);
+                res.end();
+            }
+        });
+        await new Promise((r) => authServer.listen(0, r));
+        const authPort = authServer.address().port;
+        const cbPort = getTestPort();
+        const vsc = require("vscode");
+        vsc.env.openExternal.mockImplementation(async () => {
+            await new Promise((r) => setTimeout(r, 100));
+            await new Promise((resolve, reject) => {
+                const req = http.get(`http://localhost:${cbPort}/callback?token=new-tok`, (res) => {
+                    res.resume();
+                    res.on("end", resolve);
+                });
+                req.on("error", reject);
+            });
+            return true;
+        });
+        const ctx = createMockContext();
+        const ext = new vscode_login_1.AIWorldExtension(ctx, {
+            baseUrl: `http://localhost:${authPort}`,
+            callbackPort: cbPort,
+        });
+        await ext.login();
+        expect(ctx.secrets._store.get("ai-world:token")).toBe("new-tok");
+        expect(JSON.parse(ctx.secrets._store.get("ai-world:user")).full_name).toBe("NewUser");
+        expect(ctx.secrets._store.get("ai-world:baseUrl")).toBe(`http://localhost:${authPort}`);
+        ext.dispose();
+    });
+    it("logout 清除 SecretStorage", async () => {
+        const ctx = createMockContext();
+        ctx.secrets._store.set("ai-world:token", "old");
+        ctx.secrets._store.set("ai-world:user", '{"id":1}');
+        ctx.secrets._store.set("ai-world:baseUrl", "http://x");
+        const ext = new vscode_login_1.AIWorldExtension(ctx);
+        ext._token = "old";
+        await ext.logout();
+        expect(ctx.secrets._store.has("ai-world:token")).toBe(false);
+        expect(ctx.secrets._store.has("ai-world:user")).toBe(false);
+        expect(ctx.secrets._store.has("ai-world:baseUrl")).toBe(false);
+        ext.dispose();
+    });
+    it("缓存 token 过期时自动清除 SecretStorage", async () => {
+        authServer = http.createServer((_req, res) => { res.writeHead(401); res.end(); });
+        await new Promise((r) => authServer.listen(0, r));
+        const port = authServer.address().port;
+        const ctx = createMockContext();
+        ctx.secrets._store.set("ai-world:token", "expired");
+        ctx.secrets._store.set("ai-world:baseUrl", `http://localhost:${port}`);
+        const ext = new vscode_login_1.AIWorldExtension(ctx, { baseUrl: `http://localhost:${port}` });
+        await ext._bootstrap(false);
+        expect(ext.isAuthenticated).toBe(false);
+        expect(ctx.secrets._store.has("ai-world:token")).toBe(false);
+        ext.dispose();
+    });
+});
+describe("AIWorldExtension - _bootstrap 提示", () => {
+    it("autoLogin=true 时弹出提示而不直接打开浏览器", async () => {
+        const vsc = require("vscode");
+        vsc.window.showWarningMessage.mockResolvedValue("稍后");
+        vsc.env.openExternal.mockClear();
+        const ctx = createMockContext();
+        const ext = new vscode_login_1.AIWorldExtension(ctx);
+        await ext._bootstrap(true);
+        expect(vsc.window.showWarningMessage).toHaveBeenCalledWith("AI World: 未登录，部分功能不可用", "登录", "稍后");
+        expect(vsc.env.openExternal).not.toHaveBeenCalled();
+        ext.dispose();
+    });
+    it("弹窗点击登录后才触发浏览器打开", async () => {
+        let authServer = http.createServer((_req, res) => { res.writeHead(401); res.end(); });
+        await new Promise((r) => authServer.listen(0, r));
+        const authPort = authServer.address().port;
+        const cbPort = getTestPort();
+        const vsc = require("vscode");
+        vsc.env.openExternal.mockImplementation(async () => {
+            await new Promise((r) => setTimeout(r, 50));
+            await new Promise((resolve, reject) => {
+                const req = http.get(`http://localhost:${cbPort}/callback?token=prompt-tok`, (res) => {
+                    res.resume();
+                    res.on("end", resolve);
+                });
+                req.on("error", reject);
+            });
+            return true;
+        });
+        vsc.window.showWarningMessage.mockResolvedValue("登录");
+        const ctx = createMockContext();
+        const ext = new vscode_login_1.AIWorldExtension(ctx, {
+            baseUrl: `http://localhost:${authPort}`,
+            callbackPort: cbPort,
+        });
+        await ext._bootstrap(true);
+        await new Promise((r) => setTimeout(r, 500));
+        expect(vsc.env.openExternal).toHaveBeenCalled();
+        expect(ext.isAuthenticated).toBe(true);
+        expect(ext.token).toBe("prompt-tok");
+        ext.dispose();
+        authServer.close();
+    });
+});
+describe("AIWorldExtension - login / logout", () => {
+    let authServer = null;
+    afterEach((done) => {
+        if (authServer?.listening) {
+            authServer.close(() => { authServer = null; done(); });
+        }
+        else {
+            authServer = null;
+            done();
+        }
+    });
+    it("login 超时时抛出错误", async () => {
+        authServer = http.createServer((_req, res) => { res.writeHead(401); res.end(); });
+        await new Promise((r) => authServer.listen(0, r));
+        const authPort = authServer.address().port;
+        const cbPort = getTestPort();
+        const vsc = require("vscode");
+        vsc.env.openExternal.mockResolvedValue(true);
+        const ctx = createMockContext();
+        const ext = new vscode_login_1.AIWorldExtension(ctx, {
+            baseUrl: `http://localhost:${authPort}`,
+            callbackPort: cbPort,
+            timeout: 500,
+        });
+        await expect(ext.login()).rejects.toThrow("登录超时");
+        ext.dispose();
+    });
+});
+describe("AIWorldExtension - showLoginPanel", () => {
+    it("创建 webview 面板", () => {
+        const vsc = require("vscode");
+        const ctx = createMockContext();
+        const ext = new vscode_login_1.AIWorldExtension(ctx);
+        ext.showLoginPanel();
+        expect(vsc.window.createWebviewPanel).toHaveBeenCalledWith("aiWorldLogin", "AI World 登录", 1, expect.objectContaining({ enableScripts: true }));
+        ext.dispose();
+    });
+});
+describe("AIWorldExtension - _getLoginPanelHtml", () => {
+    it("未登录时包含登录按钮", () => {
+        const ctx = createMockContext();
+        const ext = new vscode_login_1.AIWorldExtension(ctx);
+        const html = ext._getLoginPanelHtml();
+        expect(html).toContain("通过飞书登录");
+        expect(html).toContain("未登录");
+        expect(html).not.toContain("退出登录");
+        ext.dispose();
+    });
+    it("已登录时显示用户信息和退出按钮", () => {
+        const ctx = createMockContext("my-ext");
+        const ext = new vscode_login_1.AIWorldExtension(ctx);
+        ext._token = "x";
+        ext._user = { id: 1, full_name: "张三" };
+        const html = ext._getLoginPanelHtml();
+        expect(html).toContain("已登录");
+        expect(html).toContain("张三");
+        expect(html).toContain("my-ext");
+        expect(html).toContain("退出登录");
+        ext.dispose();
+    });
+    it("包含 acquireVsCodeApi 和 postMessage", () => {
+        const ctx = createMockContext();
+        const ext = new vscode_login_1.AIWorldExtension(ctx);
+        const html = ext._getLoginPanelHtml();
+        expect(html).toContain("acquireVsCodeApi");
+        expect(html).toContain("postMessage");
+        ext.dispose();
+    });
+});
+describe("initAIWorld / getAIWorld", () => {
+    let authServer = null;
+    afterEach((done) => {
+        if (authServer?.listening) {
+            authServer.close(() => { authServer = null; done(); });
+        }
+        else {
+            authServer = null;
+            done();
+        }
+    });
+    it("initAIWorld 返回实例", async () => {
+        const ctx = createMockContext();
+        const ext = await (0, vscode_login_1.initAIWorld)(ctx, { autoLogin: false });
+        expect(ext).toBeInstanceOf(vscode_login_1.AIWorldExtension);
+        expect(ext.isAuthenticated).toBe(false);
+        ext.dispose();
+    });
+    it("getAIWorld 返回同一实例", async () => {
+        const ctx = createMockContext();
+        const ext = await (0, vscode_login_1.initAIWorld)(ctx, { autoLogin: false });
+        expect((0, vscode_login_1.getAIWorld)()).toBe(ext);
+        ext.dispose();
+    });
+    it("未初始化时 getAIWorld 抛出错误", () => {
+        try {
+            (0, vscode_login_1.getAIWorld)().dispose();
+        }
+        catch { /* noop */ }
+        expect(() => (0, vscode_login_1.getAIWorld)()).toThrow("请先调用 initAIWorld");
+    });
+    it("initAIWorld 带缓存 token 自动恢复", async () => {
+        authServer = http.createServer((req, res) => {
+            if (req.url === "/api/auth/me" && req.headers.authorization === "Bearer cached") {
+                res.writeHead(200, { "Content-Type": "application/json" });
+                res.end(JSON.stringify({ id: 3, full_name: "Cached" }));
+            }
+            else {
+                res.writeHead(401);
+                res.end();
+            }
+        });
+        await new Promise((r) => authServer.listen(0, r));
+        const port = authServer.address().port;
+        const ctx = createMockContext();
+        ctx.secrets._store.set("ai-world:token", "cached");
+        ctx.secrets._store.set("ai-world:baseUrl", `http://localhost:${port}`);
+        const ext = await (0, vscode_login_1.initAIWorld)(ctx, {
+            baseUrl: `http://localhost:${port}`,
+            autoLogin: false,
+        });
+        expect(ext.isAuthenticated).toBe(true);
+        expect(ext.user?.full_name).toBe("Cached");
+        ext.dispose();
+    });
+    it("pluginId 自动取 package.json name", async () => {
+        const ctx = createMockContext("auto-name-plugin");
+        const ext = await (0, vscode_login_1.initAIWorld)(ctx, { autoLogin: false });
+        expect(ext.pluginId).toBe("auto-name-plugin");
+        ext.dispose();
+    });
+});
+describe("Status Bar", () => {
+    it("未登录时显示警告样式", () => {
+        const ctx = createMockContext();
+        const ext = new vscode_login_1.AIWorldExtension(ctx);
+        expect(mockStatusBarItem.text).toContain("未登录");
+        expect(mockStatusBarItem.show).toHaveBeenCalled();
+        ext.dispose();
+    });
+    it("已登录时显示用户名", () => {
+        const ctx = createMockContext();
+        const ext = new vscode_login_1.AIWorldExtension(ctx);
+        ext._token = "t";
+        ext._user = { id: 1, full_name: "StatusUser" };
+        ext._updateStatusBar();
+        expect(mockStatusBarItem.text).toContain("StatusUser");
+        ext.dispose();
+    });
+});

package/dist/agent-skills.d.ts

@@ -0,0 +1,96 @@
+/**
+ * Agent Skill Client
+ * Agent Skills 管理客户端
+ * 通过后端 /api/agent-skills 上传、下载、删除、列出 Skills
+ */
+export interface AgentSkillClientConfig {
+    baseUrl?: string;
+    token?: string;
+    headers?: Record<string, string>;
+}
+export interface AgentSkillInfo {
+    skill_name: string;
+    name: string;
+    description: string;
+    user_id?: number;
+    created_at?: string;
+    updated_at?: string;
+}
+export interface AgentSkillListResponse {
+    items: AgentSkillInfo[];
+    total: number;
+    page: number;
+    page_size: number;
+}
+export interface ListSkillsOptions {
+    name?: string;
+    page?: number;
+    pageSize?: number;
+}
+/**
+ * Agent Skill Client
+ * Agent Skills 管理客户端类
+ *
+ * @example
+ * ```typescript
+ * import { AgentSkillClient } from 'ai-world-sdk';
+ *
+ * const skills = new AgentSkillClient();
+ *
+ * // 列出所有 skills
+ * const list = await skills.list({ page: 1, pageSize: 20 });
+ *
+ * // 获取 skill 详情
+ * const detail = await skills.get('web-search');
+ *
+ * // 上传 skill（需要 File 或 Blob）
+ * const zipFile = new File([zipContent], 'web-search.zip');
+ * await skills.upload('web-search', zipFile);
+ *
+ * // 下载 skill
+ * const blob = await skills.download('web-search');
+ *
+ * // 删除 skill
+ * await skills.delete('web-search');
+ * ```
+ */
+export declare class AgentSkillClient {
+    private baseUrl;
+    private headers;
+    constructor(config?: AgentSkillClientConfig);
+    private handleErrorResponse;
+    /**
+     * List agent skills
+     * 列出 Agent Skills
+     */
+    list(options?: ListSkillsOptions): Promise<AgentSkillListResponse>;
+    /**
+     * Get skill detail
+     * 获取 Skill 详情
+     */
+    get(skillName: string): Promise<AgentSkillInfo>;
+    /**
+     * Upload a skill (zip file)
+     * 上传 Skill
+     * @param skillName - Skill 名称
+     * @param file - zip 文件
+     * @param description - 自定义显示描述（不填则使用 SKILL.md 中的描述）
+     */
+    upload(skillName: string, file: File | Blob, description?: string): Promise<{
+        message: string;
+        skill_name: string;
+        is_update: boolean;
+    }>;
+    /**
+     * Download a skill as zip blob
+     * 下载 Skill（返回 Blob）
+     */
+    download(skillName: string): Promise<Blob>;
+    /**
+     * Delete a skill
+     * 删除 Skill
+     */
+    delete(skillName: string): Promise<{
+        message: string;
+    }>;
+}