@cicctencent/agent-midway 0.1.1

package/dist/controller/security.controller.js

@@ -0,0 +1,172 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SecurityController = void 0;
+const core_1 = require("@midwayjs/core");
+const midwayjs_base_1 = require("@cicctencent/midwayjs-base");
+const agent_server_1 = require("@cicctencent/agent-server");
+/**
+ * 安全审计 Controller
+ *
+ * 使用 agent-server 的安全服务提供：
+ * - 工具调用审计日志
+ * - 风险评估
+ * - 权限审批
+ * - 安全策略管理
+ */
+let SecurityController = class SecurityController extends midwayjs_base_1.BaseController {
+    /** 获取审计日志列表 */
+    async getAuditLogs(body = {}) {
+        try {
+            const logs = (0, agent_server_1.listAuditLogs)({
+                tool: body.tool,
+                level: body.level,
+                sessionId: body.sessionId,
+                limit: body.limit || 50,
+                since: body.since,
+                until: body.until,
+            });
+            return { ret: 0, data: logs, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 获取工具风险评估（调用 agent-server 的 assessToolRisk） */
+    async getToolRisks(body = {}) {
+        try {
+            if (body.toolName) {
+                // 评估单个工具
+                const assessment = (0, agent_server_1.assessToolRisk)(body.toolName);
+                return {
+                    ret: 0,
+                    data: {
+                        tool: body.toolName,
+                        riskLevel: assessment.level,
+                        description: assessment.reasons.join('; '),
+                        requiresApproval: assessment.requiresConfirmation,
+                    },
+                    msg: 'success',
+                };
+            }
+            // 无 toolName 时返回策略
+            const policy = (0, agent_server_1.getSecurityPolicy)();
+            return { ret: 0, data: policy, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 获取待审批列表 */
+    async getPendingApprovals() {
+        try {
+            const requests = (0, agent_server_1.listPermissionRequests)('pending');
+            return { ret: 0, data: requests, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 审批操作 */
+    async approveAction(id, body) {
+        try {
+            const result = (0, agent_server_1.decidePermissionRequest)(id, body.approved, body.reason);
+            if (!result) {
+                return { ret: 1, msg: 'Permission request not found' };
+            }
+            return {
+                ret: 0,
+                data: result,
+                msg: body.approved ? '已批准' : '已拒绝',
+            };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 更新工具安全策略 */
+    async updateToolPolicy(toolName, body) {
+        try {
+            // 更新安全策略
+            const updates = {};
+            if (body.riskLevel) {
+                updates.toolBaseRisk = updates.toolBaseRisk || {};
+                updates.toolBaseRisk[toolName] = body.riskLevel;
+            }
+            if (body.requiresApproval === false) {
+                updates.allowTools = updates.allowTools || [];
+                updates.allowTools.push(toolName);
+            }
+            else if (body.requiresApproval === true) {
+                updates.denyTools = updates.denyTools || [];
+                updates.denyTools.push(toolName);
+            }
+            const updated = (0, agent_server_1.updateSecurityPolicy)(updates);
+            return {
+                ret: 0,
+                data: { tool: toolName, ...body, policy: updated },
+                msg: '安全策略已更新',
+            };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+};
+exports.SecurityController = SecurityController;
+__decorate([
+    (0, core_1.Get)('/audit-logs'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], SecurityController.prototype, "getAuditLogs", null);
+__decorate([
+    (0, core_1.Get)('/tool-risks'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], SecurityController.prototype, "getToolRisks", null);
+__decorate([
+    (0, core_1.Get)('/approvals/pending'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], SecurityController.prototype, "getPendingApprovals", null);
+__decorate([
+    (0, core_1.Post)('/approvals/:id'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Param)('id')),
+    __param(1, (0, core_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [String, Object]),
+    __metadata("design:returntype", Promise)
+], SecurityController.prototype, "approveAction", null);
+__decorate([
+    (0, core_1.Put)('/tool-policy/:toolName'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Param)('toolName')),
+    __param(1, (0, core_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [String, Object]),
+    __metadata("design:returntype", Promise)
+], SecurityController.prototype, "updateToolPolicy", null);
+exports.SecurityController = SecurityController = __decorate([
+    (0, core_1.Provide)(),
+    (0, core_1.Controller)('/api/security')
+], SecurityController);

package/dist/controller/settings.controller.d.ts

@@ -0,0 +1,83 @@
+import { BaseController } from '@cicctencent/midwayjs-base';
+import { OpenAIService } from '../service/openai.service';
+import { AgentServerService } from '../service/agent-server.service';
+import { ApplicationService } from '../service/application.service';
+/**
+ * 全局设置 Controller
+ *
+ * 提供系统级配置管理，包括 AI 服务状态、模型配置概览、Agent 引擎状态等
+ * 使用 agent-server 的 SettingsProvider 进行设置持久化
+ */
+export declare class SettingsController extends BaseController {
+    openaiService: OpenAIService;
+    agentServerService: AgentServerService;
+    applicationService: ApplicationService;
+    openaiConfig: any;
+    /**
+     * 热更新 LLM 配置
+     * 在设置更新后调用，使新的模型配置立即生效
+     */
+    private refreshLLMConfig;
+    /** 获取全局设置概览（API Key 脱敏） */
+    getSettings(): Promise<{
+        ret: number;
+        data: any;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 获取 AI 服务状态 */
+    getAIStatus(): Promise<{
+        ret: number;
+        data: {
+            available: boolean;
+            model: any;
+            baseUrl: any;
+            fallbackModels: any;
+        };
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 获取 Agent 引擎状态 */
+    getAgentStatus(): Promise<{
+        ret: number;
+        data: {
+            available: boolean;
+            toolCount: number;
+            tools: {
+                name: any;
+                description: any;
+            }[];
+        };
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 更新全局设置（API Key 脱敏返回） */
+    updateSettings(body: any): Promise<{
+        ret: number;
+        data: any;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 重置全局设置为默认值 */
+    resetSettings(): Promise<{
+        ret: number;
+        data: any;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+}

package/dist/controller/settings.controller.js

@@ -0,0 +1,280 @@
+"use strict";
+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 __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+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;
+    };
+})();
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SettingsController = void 0;
+const core_1 = require("@midwayjs/core");
+const midwayjs_base_1 = require("@cicctencent/midwayjs-base");
+const openai_service_1 = require("../service/openai.service");
+const agent_server_service_1 = require("../service/agent-server.service");
+const application_service_1 = require("../service/application.service");
+const agent_server_1 = require("@cicctencent/agent-server");
+/** API Key 脱敏占位符 */
+const REDACTED_VALUE = '••••••••';
+/**
+ * 递归脱敏设置对象中的所有 apiKey 字段
+ * 将所有名为 apiKey 的字段值替换为 REDACTED_VALUE
+ */
+function redactSettings(settings) {
+    if (settings == null)
+        return settings;
+    if (typeof settings === 'string') {
+        return settings;
+    }
+    if (Array.isArray(settings)) {
+        return settings.map(item => redactSettings(item));
+    }
+    if (typeof settings === 'object') {
+        const result = {};
+        for (const key of Object.keys(settings)) {
+            if (key === 'apiKey' &&
+                typeof settings[key] === 'string' &&
+                settings[key].length > 0) {
+                result[key] = REDACTED_VALUE;
+            }
+            else {
+                result[key] = redactSettings(settings[key]);
+            }
+        }
+        return result;
+    }
+    return settings;
+}
+/**
+ * 全局设置 Controller
+ *
+ * 提供系统级配置管理，包括 AI 服务状态、模型配置概览、Agent 引擎状态等
+ * 使用 agent-server 的 SettingsProvider 进行设置持久化
+ */
+let SettingsController = class SettingsController extends midwayjs_base_1.BaseController {
+    /**
+     * 热更新 LLM 配置
+     * 在设置更新后调用，使新的模型配置立即生效
+     */
+    async refreshLLMConfig() {
+        try {
+            // 重新初始化 OpenAI 客户端以应用新的配置
+            if (this.openaiService.isAvailable()) {
+                const settings = (0, agent_server_1.getSettings)();
+                const defaultProvider = settings.modelProviders?.find(p => p.id === settings.defaultModelId);
+                if (defaultProvider?.apiKey) {
+                    // 重新初始化 OpenAI 客户端
+                    const { default: OpenAI } = await Promise.resolve().then(() => __importStar(require('openai')));
+                    this.openaiService.client = new OpenAI({
+                        apiKey: defaultProvider.apiKey,
+                        baseURL: defaultProvider.baseUrl || '',
+                    });
+                    console.log('[Settings] LLM 配置已热更新, model:', defaultProvider.model);
+                }
+            }
+        }
+        catch (error) {
+            console.warn('[Settings] LLM 配置热更新失败:', error.message);
+        }
+    }
+    /** 获取全局设置概览（API Key 脱敏） */
+    async getSettings() {
+        try {
+            const defaultApp = await this.applicationService.getDefault();
+            const settings = (0, agent_server_1.getSettings)();
+            return {
+                ret: 0,
+                data: redactSettings({
+                    ai: {
+                        available: this.openaiService.isAvailable(),
+                        model: this.openaiConfig?.model || '',
+                        baseUrl: this.openaiConfig?.baseUrl || '',
+                    },
+                    agent: {
+                        available: this.agentServerService.isAvailable(),
+                    },
+                    application: defaultApp
+                        ? {
+                            id: defaultApp.id,
+                            name: defaultApp.name,
+                            key: defaultApp.key,
+                        }
+                        : null,
+                    settings: settings,
+                }),
+                msg: 'success',
+            };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 获取 AI 服务状态 */
+    async getAIStatus() {
+        try {
+            return {
+                ret: 0,
+                data: {
+                    available: this.openaiService.isAvailable(),
+                    model: this.openaiConfig?.model || '',
+                    baseUrl: this.openaiConfig?.baseUrl || '',
+                    fallbackModels: this.openaiConfig?.fallbackModels || [],
+                },
+                msg: 'success',
+            };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 获取 Agent 引擎状态 */
+    async getAgentStatus() {
+        try {
+            const tools = this.agentServerService.getTools();
+            return {
+                ret: 0,
+                data: {
+                    available: this.agentServerService.isAvailable(),
+                    toolCount: tools.length,
+                    tools: tools.map(t => ({
+                        name: t.name,
+                        description: t.description,
+                    })),
+                },
+                msg: 'success',
+            };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 更新全局设置（API Key 脱敏返回） */
+    async updateSettings(body) {
+        try {
+            // 使用 agent-server 的 updateSettings 持久化
+            const updated = (0, agent_server_1.updateSettings)(body);
+            // 热更新 LLM 配置，使新的模型配置立即生效
+            await this.refreshLLMConfig();
+            return {
+                ret: 0,
+                data: redactSettings(updated),
+                msg: '设置已更新',
+            };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 重置全局设置为默认值 */
+    async resetSettings() {
+        try {
+            // 使用 agent-server 的 resetSettings 重置
+            const reset = (0, agent_server_1.resetSettings)();
+            // 热更新 LLM 配置
+            await this.refreshLLMConfig();
+            return {
+                ret: 0,
+                data: redactSettings(reset),
+                msg: '设置已重置为默认值',
+            };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+};
+exports.SettingsController = SettingsController;
+__decorate([
+    (0, core_1.Inject)(),
+    __metadata("design:type", openai_service_1.OpenAIService)
+], SettingsController.prototype, "openaiService", void 0);
+__decorate([
+    (0, core_1.Inject)(),
+    __metadata("design:type", agent_server_service_1.AgentServerService)
+], SettingsController.prototype, "agentServerService", void 0);
+__decorate([
+    (0, core_1.Inject)(),
+    __metadata("design:type", application_service_1.ApplicationService)
+], SettingsController.prototype, "applicationService", void 0);
+__decorate([
+    (0, core_1.Config)('openai'),
+    __metadata("design:type", Object)
+], SettingsController.prototype, "openaiConfig", void 0);
+__decorate([
+    (0, core_1.Get)('/'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], SettingsController.prototype, "getSettings", null);
+__decorate([
+    (0, core_1.Get)('/ai-status'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], SettingsController.prototype, "getAIStatus", null);
+__decorate([
+    (0, core_1.Get)('/agent-status'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], SettingsController.prototype, "getAgentStatus", null);
+__decorate([
+    (0, core_1.Put)('/'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], SettingsController.prototype, "updateSettings", null);
+__decorate([
+    (0, core_1.Post)('/reset'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], SettingsController.prototype, "resetSettings", null);
+exports.SettingsController = SettingsController = __decorate([
+    (0, core_1.Provide)(),
+    (0, core_1.Controller)('/api/settings')
+], SettingsController);

package/dist/core/ai-workstation.d.ts

@@ -0,0 +1,17 @@
+import type { AIFrameworkAdapterOptions, AIHttpHandler, AIJsonResponse, AIRequestContext, AISSEWriter } from './types';
+import { AIServiceContainer } from './service-container';
+export declare class AIWorkstationCore {
+    readonly container: AIServiceContainer;
+    constructor(options: AIFrameworkAdapterOptions);
+    get prefix(): string;
+    routes(): Record<string, AIHttpHandler>;
+    runAgentSync(ctx: AIRequestContext): Promise<AIJsonResponse>;
+    runAgentStream(ctx: AIRequestContext, writer: AISSEWriter): Promise<void>;
+    reconnectStream(ctx: AIRequestContext, writer: AISSEWriter): Promise<void>;
+    abortAgent(ctx: AIRequestContext): Promise<AIJsonResponse>;
+    submitTask(ctx: AIRequestContext): Promise<AIJsonResponse>;
+    health(_ctx: AIRequestContext): Promise<AIJsonResponse>;
+    private getRunInput;
+    private getPrompt;
+}
+export declare function createAIWorkstation(options: AIFrameworkAdapterOptions): AIWorkstationCore;

package/dist/core/ai-workstation.js

@@ -0,0 +1,129 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AIWorkstationCore = void 0;
+exports.createAIWorkstation = createAIWorkstation;
+const service_container_1 = require("./service-container");
+const sse_1 = require("./sse");
+class AIWorkstationCore {
+    constructor(options) {
+        this.container = (0, service_container_1.createAIServiceContainer)(options);
+    }
+    get prefix() {
+        return this.container.prefix;
+    }
+    routes() {
+        return {
+            'POST /agent/run/sync': ctx => this.runAgentSync(ctx),
+            'POST /agent/task': ctx => this.submitTask(ctx),
+            'GET /agent/health': ctx => this.health(ctx),
+        };
+    }
+    async runAgentSync(ctx) {
+        try {
+            const input = this.getRunInput(ctx);
+            const prompt = this.getPrompt(input);
+            const user = await this.container.resolveUser(ctx);
+            const applicationId = await this.container.resolveApplicationId(ctx, user);
+            const result = await this.container.services.agent.runAgent(prompt, {
+                userId: user.userId,
+                applicationId,
+                threadId: input.threadId,
+                profileId: input.profileId,
+                maxSteps: input.maxSteps,
+                timeout: input.timeout,
+                customContext: input.customContext,
+            });
+            return { ret: 0, data: result, msg: 'success' };
+        }
+        catch (error) {
+            await this.container.handleError(error, ctx);
+            return { ret: 1, msg: error.message || 'Agent run failed' };
+        }
+    }
+    async runAgentStream(ctx, writer) {
+        (0, sse_1.initSSEHeaders)(writer);
+        try {
+            const input = this.getRunInput(ctx);
+            const prompt = this.getPrompt(input);
+            const user = await this.container.resolveUser(ctx);
+            const applicationId = await this.container.resolveApplicationId(ctx, user);
+            const stream = this.container.services.agent.runAgentStream(prompt, {
+                userId: user.userId,
+                applicationId,
+                threadId: input.threadId,
+                profileId: input.profileId,
+                maxSteps: input.maxSteps,
+                timeout: input.timeout,
+                customContext: input.customContext,
+            });
+            await (0, sse_1.pipeSSEStream)(writer, stream);
+        }
+        catch (error) {
+            await this.container.handleError(error, ctx);
+            (0, sse_1.writeSSEEvent)(writer, {
+                type: 'error',
+                data: { message: error.message || 'Agent stream failed' },
+                timestamp: Date.now(),
+            });
+            if (!writer.writableEnded)
+                writer.end();
+        }
+    }
+    async reconnectStream(ctx, writer) {
+        const threadId = Number(ctx.params?.threadId || ctx.query?.threadId);
+        const reconnect = this.container.services.agent.reconnectStream;
+        if (!reconnect) {
+            (0, sse_1.writeSSEEvent)(writer, {
+                type: 'error',
+                data: { message: 'Reconnect stream is not supported by current agent service' },
+                timestamp: Date.now(),
+            });
+            writer.end();
+            return;
+        }
+        const user = await this.container.resolveUser(ctx);
+        await (0, sse_1.pipeSSEStream)(writer, reconnect.call(this.container.services.agent, threadId, user.userId));
+    }
+    async abortAgent(ctx) {
+        try {
+            const threadId = Number(ctx.params?.threadId || ctx.body?.threadId);
+            const user = await this.container.resolveUser(ctx);
+            const aborted = await this.container.services.agent.abortAgent?.(threadId, user.userId);
+            return { ret: 0, data: { aborted: !!aborted }, msg: aborted ? 'Agent 已中止' : '无法中止 Agent' };
+        }
+        catch (error) {
+            await this.container.handleError(error, ctx);
+            return { ret: 1, msg: error.message || 'Abort failed' };
+        }
+    }
+    async submitTask(ctx) {
+        return {
+            ret: 1,
+            msg: 'Task queue is framework-specific. Please wire AgentTaskQueueService in the adapter options.',
+        };
+    }
+    async health(_ctx) {
+        const agent = this.container.services.agent;
+        return {
+            ret: 0,
+            data: {
+                available: agent.isAvailable ? agent.isAvailable() : true,
+                tools: agent.getTools ? agent.getTools().map(tool => tool.definition?.function?.name || 'unknown') : [],
+            },
+            msg: 'success',
+        };
+    }
+    getRunInput(ctx) {
+        return (ctx.body || {});
+    }
+    getPrompt(input) {
+        const prompt = input.prompt || input.content;
+        if (!prompt)
+            throw new Error('prompt is required');
+        return prompt;
+    }
+}
+exports.AIWorkstationCore = AIWorkstationCore;
+function createAIWorkstation(options) {
+    return new AIWorkstationCore(options);
+}

package/dist/core/index.d.ts

@@ -0,0 +1,4 @@
+export * from './types';
+export * from './sse';
+export * from './service-container';
+export * from './ai-workstation';