zhitalk 0.0.4 → 0.0.6

package/dist/agent/commands.js

@@ -1,51 +1,45 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.commands = exports.threadId = void 0;
-const crypto_1 = require("crypto");
-const cli_table3_1 = __importDefault(require("cli-table3"));
-const colors_1 = require("./colors");
-const db_1 = require("./db");
-const agent_1 = require("./agent");
-exports.threadId = (0, crypto_1.randomUUID)();
-exports.commands = new Map();
-exports.commands.set('new', {
+import { randomUUID } from 'crypto';
+import Table from 'cli-table3';
+import { color } from './colors.js';
+import { listRecentSessions, threadIdExists } from './db.js';
+import { compressContext } from './agent.js';
+export let threadId = randomUUID();
+export const commands = new Map();
+commands.set('new', {
     name: 'new',
     description: 'Start a new chat session',
     execute() {
-        exports.threadId = (0, crypto_1.randomUUID)();
-        console.log(colors_1.color.goodbye(`\nNew session started ${exports.threadId}.\n`));
+        threadId = randomUUID();
+        console.log(color.goodbye(`\nNew session started ${threadId}.\n`));
     },
 });
-exports.commands.set('rewind', {
+commands.set('rewind', {
     name: 'rewind',
     description: 'Restore a chat session by thread_id',
     execute(args) {
         const id = args[0];
         if (!id) {
-            console.log(colors_1.color.error('\nError: please provide a thread_id. Usage: /rewind <thread_id>\n'));
+            console.log(color.error('\nError: please provide a thread_id. Usage: /rewind <thread_id>\n'));
             return;
         }
-        if (!(0, db_1.threadIdExists)(id)) {
-            console.log(colors_1.color.error(`\nError: thread_id "${id}" not found.\n`));
+        if (!threadIdExists(id)) {
+            console.log(color.error(`\nError: thread_id "${id}" not found.\n`));
             return;
         }
-        exports.threadId = id;
-        console.log(colors_1.color.goodbye(`\nRestored session ${exports.threadId}.\n`));
+        threadId = id;
+        console.log(color.goodbye(`\nRestored session ${threadId}.\n`));
     },
 });
-exports.commands.set('sessions', {
+commands.set('sessions', {
     name: 'sessions',
     description: 'List recent chat sessions',
     execute() {
-        const sessions = (0, db_1.listRecentSessions)();
+        const sessions = listRecentSessions();
         if (sessions.length === 0) {
             console.log('\n暂无聊天记录\n');
             return;
         }
-        const table = new cli_table3_1.default({
+        const table = new Table({
             head: ['thread_id', '最后用户输入的问题', '时间'],
         });
         for (const s of sessions) {
@@ -54,13 +48,13 @@ exports.commands.set('sessions', {
         console.log('\n' + table.toString() + '\n');
     },
 });
-exports.commands.set('compact', {
+commands.set('compact', {
     name: 'compact',
     description: 'Compress context for the current session',
     async execute() {
-        const result = await (0, agent_1.compressContext)(exports.threadId);
+        const result = await compressContext(threadId);
         if (result.didCompress) {
-            console.log(colors_1.color.goodbye(`\nContext compressed (count: ${result.count}).\n`));
+            console.log(color.goodbye(`\nContext compressed (count: ${result.count}).\n`));
         }
         else {
             console.log('\nNo context to compress.\n');

package/dist/agent/config.js

@@ -1,90 +1,48 @@
-"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 __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 });
-exports.CONFIG_PATH = exports.WORKSPACE_DIR = void 0;
-exports.loadConfig = loadConfig;
-exports.getModelConfig = getModelConfig;
-exports.getEnv = getEnv;
-exports.getHooksConfig = getHooksConfig;
-exports.getMCPServerConfig = getMCPServerConfig;
-exports.clearConfigCache = clearConfigCache;
-const fs = __importStar(require("fs"));
-const path = __importStar(require("path"));
-const os = __importStar(require("os"));
-exports.WORKSPACE_DIR = path.join(os.homedir(), '.zhitalk');
-exports.CONFIG_PATH = path.join(exports.WORKSPACE_DIR, 'zhitalk.json');
+import * as fs from 'fs';
+import * as path from 'path';
+import * as os from 'os';
+export const WORKSPACE_DIR = path.join(os.homedir(), '.zhitalk');
+export const CONFIG_PATH = path.join(WORKSPACE_DIR, 'zhitalk.json');
 let cachedConfig = null;
-function loadConfig() {
+export function loadConfig() {
     if (cachedConfig) {
         return cachedConfig;
     }
-    if (!fs.existsSync(exports.CONFIG_PATH)) {
-        throw new Error(`找不到配置文件 ${exports.CONFIG_PATH}\n\n请创建该文件并配置相关信息，例如：\n{\n  "model": {\n    "model": "kimi-k2.6",\n    "apiKey": "your-api-key",\n    "baseURL": "https://api.moonshot.cn/v1"\n  },\n  "env": {\n    "TAVILY_API_KEY": "your-tavily-api-key"\n  }\n}`);
+    if (!fs.existsSync(CONFIG_PATH)) {
+        throw new Error(`找不到配置文件 ${CONFIG_PATH}\n\n请创建该文件并配置相关信息，例如：\n{\n  "model": {\n    "model": "kimi-k2.6",\n    "apiKey": "your-api-key",\n    "baseURL": "https://api.moonshot.cn/v1"\n  },\n  "env": {\n    "TAVILY_API_KEY": "your-tavily-api-key"\n  }\n}`);
     }
     let config;
     try {
-        const content = fs.readFileSync(exports.CONFIG_PATH, 'utf-8');
+        const content = fs.readFileSync(CONFIG_PATH, 'utf-8');
         config = JSON.parse(content);
     }
     catch (e) {
-        throw new Error(`配置文件 ${exports.CONFIG_PATH} 解析失败：${e instanceof Error ? e.message : String(e)}`);
+        throw new Error(`配置文件 ${CONFIG_PATH} 解析失败：${e instanceof Error ? e.message : String(e)}`);
     }
     cachedConfig = config;
     return config;
 }
-function getModelConfig() {
+export function getModelConfig() {
     const config = loadConfig();
     if (!config.model || typeof config.model !== 'object') {
-        throw new Error(`配置文件 ${exports.CONFIG_PATH} 中缺少 model 对象，请参考 https://zhitalk.chat/#config 修改配置`);
+        throw new Error(`配置文件 ${CONFIG_PATH} 中缺少 model 对象，请参考 https://zhitalk.chat/#config 修改配置`);
     }
     if (!config.model.model) {
-        throw new Error(`配置文件 ${exports.CONFIG_PATH} 中缺少 model.model 字段，请参考 https://zhitalk.chat/#config 修改配置`);
+        throw new Error(`配置文件 ${CONFIG_PATH} 中缺少 model.model 字段，请参考 https://zhitalk.chat/#config 修改配置`);
     }
     if (!config.model.apiKey) {
-        throw new Error(`配置文件 ${exports.CONFIG_PATH} 中缺少 model.apiKey 字段，请参考 https://zhitalk.chat/#config 修改配置`);
+        throw new Error(`配置文件 ${CONFIG_PATH} 中缺少 model.apiKey 字段，请参考 https://zhitalk.chat/#config 修改配置`);
     }
     if (!config.model.baseURL) {
-        throw new Error(`配置文件 ${exports.CONFIG_PATH} 中缺少 model.baseURL 字段，请参考 https://zhitalk.chat/#config 修改配置`);
+        throw new Error(`配置文件 ${CONFIG_PATH} 中缺少 model.baseURL 字段，请参考 https://zhitalk.chat/#config 修改配置`);
     }
     return config.model;
 }
-function getEnv(key) {
+export function getEnv(key) {
     const config = loadConfig();
     return config.env?.[key];
 }
-function getHooksConfig() {
+export function getHooksConfig() {
     try {
         const config = loadConfig();
         return { hooks: config.hooks || {} };
@@ -93,9 +51,9 @@ function getHooksConfig() {
         return { hooks: {} };
     }
 }
-function getMCPServerConfig() {
+export function getMCPServerConfig() {
     try {
-        const content = fs.readFileSync(exports.CONFIG_PATH, 'utf-8');
+        const content = fs.readFileSync(CONFIG_PATH, 'utf-8');
         const parsed = JSON.parse(content);
         if (!parsed.mcpServers || typeof parsed.mcpServers !== 'object') {
             // console.warn('[MCP] zhitalk.json missing "mcpServers" field, skipping')
@@ -111,6 +69,6 @@ function getMCPServerConfig() {
         return {};
     }
 }
-function clearConfigCache() {
+export function clearConfigCache() {
     cachedConfig = null;
 }

package/dist/agent/context.js

@@ -1,10 +1,6 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.getModelContextLimit = getModelContextLimit;
-exports.compressMessages = compressMessages;
-const messages_1 = require("@langchain/core/messages");
-const model_1 = require("./model");
-const config_1 = require("./config");
+import { HumanMessage, } from '@langchain/core/messages';
+import { createModel } from './model.js';
+import { getModelConfig } from './config.js';
 const MODEL_CONTEXT_LIMITS = {
     // Moonshot / Kimi
     'moonshot-v1-8k': 8192,
@@ -38,8 +34,8 @@ const MODEL_CONTEXT_LIMITS = {
     'mimo-7b': 32768,
     'mimo-v2.5': 1048576,
 };
-function getModelContextLimit() {
-    const modelName = (0, config_1.getModelConfig)().model || '';
+export function getModelContextLimit() {
+    const modelName = getModelConfig().model || '';
     return MODEL_CONTEXT_LIMITS[modelName.toLowerCase()] || 128000;
 }
 function formatMessagesForCompression(messages) {
@@ -56,7 +52,7 @@ function formatMessagesForCompression(messages) {
     })
         .join('\n\n');
 }
-async function compressMessages(messages, existingSummary) {
+export async function compressMessages(messages, existingSummary) {
     if (messages.length === 0) {
         return existingSummary || '';
     }
@@ -95,8 +91,8 @@ async function compressMessages(messages, existingSummary) {
 对话内容如下：
 
 ${text}`;
-    const model = (0, model_1.createModel)({ streaming: false });
-    const response = await model.invoke([new messages_1.HumanMessage(prompt)]);
+    const model = createModel({ streaming: false });
+    const response = await model.invoke([new HumanMessage(prompt)]);
     const newSummary = typeof response.content === 'string'
         ? response.content
         : JSON.stringify(response.content);

package/dist/agent/db.js

@@ -1,27 +1,17 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.DB_PATH = void 0;
-exports.searchMemories = searchMemories;
-exports.threadIdExists = threadIdExists;
-exports.initDb = initDb;
-exports.listRecentSessions = listRecentSessions;
-const better_sqlite3_1 = __importDefault(require("better-sqlite3"));
-const fs_1 = require("fs");
-const path_1 = require("path");
-const config_1 = require("./config");
-const utils_1 = require("./utils");
-exports.DB_PATH = (0, path_1.join)(config_1.WORKSPACE_DIR, '.data', 'checkpointer.db');
-(0, fs_1.mkdirSync)((0, path_1.dirname)(exports.DB_PATH), { recursive: true });
-function searchMemories(query, limit = 10) {
+import Database from 'better-sqlite3';
+import { mkdirSync } from 'fs';
+import { dirname, join } from 'path';
+import { WORKSPACE_DIR } from './config.js';
+import { formatRelativeTime, truncate } from './utils.js';
+export const DB_PATH = join(WORKSPACE_DIR, '.data', 'checkpointer.db');
+mkdirSync(dirname(DB_PATH), { recursive: true });
+export function searchMemories(query, limit = 10) {
     const trimmedQueries = query.map((q) => q.trim()).filter((q) => q.length > 0);
     if (trimmedQueries.length === 0) {
         return [];
     }
     const queryStr = trimmedQueries.join(' OR ').replace(/-/g, ' ');
-    const db = new better_sqlite3_1.default(exports.DB_PATH);
+    const db = new Database(DB_PATH);
     try {
         const rows = db.prepare(`
       WITH ranked AS (
@@ -65,8 +55,8 @@ function searchMemories(query, limit = 10) {
         db.close();
     }
 }
-function threadIdExists(threadId) {
-    const db = new better_sqlite3_1.default(exports.DB_PATH);
+export function threadIdExists(threadId) {
+    const db = new Database(DB_PATH);
     try {
         const row = db.prepare(`
       SELECT 1 FROM checkpoints WHERE thread_id = ? LIMIT 1
@@ -77,8 +67,8 @@ function threadIdExists(threadId) {
         db.close();
     }
 }
-function initDb() {
-    const db = new better_sqlite3_1.default(exports.DB_PATH);
+export function initDb() {
+    const db = new Database(DB_PATH);
     try {
         db.exec(`
       CREATE TABLE IF NOT EXISTS memory (
@@ -122,8 +112,8 @@ function initDb() {
         db.close();
     }
 }
-function listRecentSessions() {
-    const db = new better_sqlite3_1.default(exports.DB_PATH);
+export function listRecentSessions() {
+    const db = new Database(DB_PATH);
     try {
         const rows = db.prepare(`
       WITH thread_last_ts AS (
@@ -149,8 +139,8 @@ function listRecentSessions() {
     `).all();
         return rows.map((r) => ({
             thread_id: r.thread_id,
-            last_question: (0, utils_1.truncate)(r.last_question, 50),
-            last_ts: (0, utils_1.formatRelativeTime)(r.last_ts),
+            last_question: truncate(r.last_question, 50),
+            last_ts: formatRelativeTime(r.last_ts),
         }));
     }
     finally {

package/dist/agent/hooks.d.ts

@@ -1,4 +1,4 @@
-import { HookConfig, HooksConfig } from './config';
+import { HookConfig, HooksConfig } from './config.js';
 export type { HookConfig, HooksConfig };
 export type HookResult = {
     action: 'continue';

package/dist/agent/hooks.js

@@ -1,28 +1,19 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.loadHooksConfig = loadHooksConfig;
-exports.clearHooksCache = clearHooksCache;
-exports.matchHooks = matchHooks;
-exports.runHook = runHook;
-exports.runPreToolUseHooks = runPreToolUseHooks;
-exports.runPostToolUseHooks = runPostToolUseHooks;
-exports.runSessionStartHooks = runSessionStartHooks;
-const child_process_1 = require("child_process");
-const util_1 = require("util");
-const config_1 = require("./config");
-const execAsync = (0, util_1.promisify)(child_process_1.exec);
-function loadHooksConfig() {
-    return (0, config_1.getHooksConfig)();
+import { exec } from 'child_process';
+import { promisify } from 'util';
+import { getHooksConfig, clearConfigCache, } from './config.js';
+const execAsync = promisify(exec);
+export function loadHooksConfig() {
+    return getHooksConfig();
 }
-function clearHooksCache() {
-    (0, config_1.clearConfigCache)();
+export function clearHooksCache() {
+    clearConfigCache();
 }
-function matchHooks(hooks, toolName) {
+export function matchHooks(hooks, toolName) {
     if (!hooks)
         return [];
     return hooks.filter((h) => h.matcher === '*' || toolName.includes(h.matcher));
 }
-async function runHook(hook, env, timeout = 30000) {
+export async function runHook(hook, env, timeout = 30000) {
     const hookType = env.ZHITALK_HOOK_TYPE || 'hook';
     console.log(`[Hook ${hookType}] ${hook.command}`);
     try {
@@ -48,7 +39,7 @@ async function runHook(hook, env, timeout = 30000) {
         };
     }
 }
-async function runPreToolUseHooks(context) {
+export async function runPreToolUseHooks(context) {
     const config = loadHooksConfig();
     const hooks = matchHooks(config.hooks.PreToolUse, context.toolName);
     const injectMessages = [];
@@ -72,7 +63,7 @@ async function runPreToolUseHooks(context) {
     }
     return { action: 'continue' };
 }
-async function runPostToolUseHooks(context) {
+export async function runPostToolUseHooks(context) {
     const config = loadHooksConfig();
     const hooks = matchHooks(config.hooks.PostToolUse, context.toolName);
     const injectMessages = [];
@@ -97,7 +88,7 @@ async function runPostToolUseHooks(context) {
     }
     return { action: 'continue' };
 }
-async function runSessionStartHooks(threadId) {
+export async function runSessionStartHooks(threadId) {
     const config = loadHooksConfig();
     const hooks = config.hooks.SessionStart || [];
     for (const hook of hooks) {

package/dist/agent/mcp/client.js

@@ -1,24 +1,20 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.loadMcpConfig = loadMcpConfig;
-exports.connectMcpServer = connectMcpServer;
-exports.initializeMcpClients = initializeMcpClients;
-exports.callMcpTool = callMcpTool;
-exports.disconnectAllMcpClients = disconnectAllMcpClients;
-const index_js_1 = require("@modelcontextprotocol/sdk/client/index.js");
-const stdio_js_1 = require("@modelcontextprotocol/sdk/client/stdio.js");
-const streamableHttp_js_1 = require("@modelcontextprotocol/sdk/client/streamableHttp.js");
-const colors_1 = require("../colors");
-const config_1 = require("../config");
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js';
+import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
+import { color } from '../colors.js';
+import { getMCPServerConfig } from '../config.js';
 const CONNECT_TIMEOUT_MS = 60000;
-function loadMcpConfig() {
-    return (0, config_1.getMCPServerConfig)();
+export function loadMcpConfig() {
+    return getMCPServerConfig();
 }
 async function withTimeout(promise, ms, label) {
-    return Promise.race([
-        promise,
-        new Promise((_, reject) => setTimeout(() => reject(new Error(`${label} timed out after ${ms}ms`)), ms)),
-    ]);
+    return new Promise((resolve, reject) => {
+        const timer = setTimeout(() => reject(new Error(`${label} timed out after ${ms}ms`)), ms);
+        promise
+            .then(resolve)
+            .catch(reject)
+            .finally(() => clearTimeout(timer));
+    });
 }
 function createMcpTransport(config) {
     if (config.url) {
@@ -26,10 +22,10 @@ function createMcpTransport(config) {
         const requestInit = config.headers
             ? { headers: config.headers }
             : undefined;
-        return new streamableHttp_js_1.StreamableHTTPClientTransport(url, { requestInit });
+        return new StreamableHTTPClientTransport(url, { requestInit });
     }
     if (config.command) {
-        return new stdio_js_1.StdioClientTransport({
+        return new StdioClientTransport({
             command: config.command,
             args: config.args,
             env: config.env,
@@ -37,9 +33,9 @@ function createMcpTransport(config) {
     }
     throw new Error('Invalid MCP server config: must have either "command" or "url"');
 }
-async function connectMcpServer(name, config) {
+export async function connectMcpServer(name, config) {
     const transport = createMcpTransport(config);
-    const client = new index_js_1.Client({ name: 'zhitalk', version: '0.0.1' }, { capabilities: {} });
+    const client = new Client({ name: 'zhitalk', version: '0.0.1' }, { capabilities: {} });
     await withTimeout(client.connect(transport), CONNECT_TIMEOUT_MS, `MCP server "${name}" connect`);
     const toolsResult = await withTimeout(client.listTools(), CONNECT_TIMEOUT_MS, `MCP server "${name}" listTools`);
     const tools = (toolsResult.tools || []).map((t) => ({
@@ -49,19 +45,19 @@ async function connectMcpServer(name, config) {
     }));
     return { name, client, transport, tools };
 }
-async function initializeMcpClients() {
+export async function initializeMcpClients() {
     const config = loadMcpConfig();
     const entries = Object.entries(config);
     if (entries.length === 0) {
         return [];
     }
-    console.log(colors_1.color.info(`[MCP] Connecting to ${entries.length} server(s)...`));
+    console.log(color.info(`[MCP] Connecting to ${entries.length} server(s)...`));
     const results = await Promise.allSettled(entries.map(([name, cfg]) => connectMcpServer(name, cfg).catch((err) => {
         const msg = err.message || '';
         const hint = msg.includes('timed out')
             ? ' (Hint: if using npx, the first run may need to download the package. Consider pre-installing with `npm install -g <package>` or increasing timeout.)'
             : '';
-        console.error(colors_1.color.error(`[MCP] Failed to connect "${name}": ${msg}${hint}`));
+        console.error(color.error(`[MCP] Failed to connect "${name}": ${msg}${hint}`));
         throw err;
     })));
     const connections = [];
@@ -69,16 +65,16 @@ async function initializeMcpClients() {
         const result = results[i];
         if (result.status === 'fulfilled') {
             const conn = result.value;
-            console.log(colors_1.color.good(`[MCP] Connected "${conn.name}" with ${conn.tools.length} tool(s)`));
+            console.log(color.good(`[MCP] Connected "${conn.name}" with ${conn.tools.length} tool(s)`));
             connections.push(conn);
         }
         else {
-            console.error(colors_1.color.error(`[MCP] Skipped "${entries[i][0]}" due to connection error`));
+            console.error(color.error(`[MCP] Skipped "${entries[i][0]}" due to connection error`));
         }
     }
     return connections;
 }
-async function callMcpTool(client, name, args) {
+export async function callMcpTool(client, name, args) {
     const result = await client.callTool({ name, arguments: args });
     if (result.isError) {
         const text = extractTextFromResult(result);
@@ -101,7 +97,7 @@ function extractTextFromResult(result) {
     }
     return texts.join('\n');
 }
-async function disconnectAllMcpClients(connections) {
+export async function disconnectAllMcpClients(connections) {
     await Promise.allSettled(connections.map(async (conn) => {
         try {
             await conn.transport.close();

package/dist/agent/mcp/index.d.ts

@@ -1,4 +1,4 @@
-import type { ZhitalkTool } from '../tools';
+import type { ZhitalkTool } from '../tools.js';
 export declare function initMcpTools(): Promise<ZhitalkTool[]>;
 export declare function getMcpTools(): ZhitalkTool[];
 export declare function shutdownMcp(): Promise<void>;

package/dist/agent/mcp/index.js

@@ -1,27 +1,22 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.initMcpTools = initMcpTools;
-exports.getMcpTools = getMcpTools;
-exports.shutdownMcp = shutdownMcp;
-const client_1 = require("./client");
-const wrapper_1 = require("./wrapper");
+import { initializeMcpClients, disconnectAllMcpClients } from './client.js';
+import { wrapMcpTool } from './wrapper.js';
 let mcpConnections = [];
 let mcpTools = [];
-async function initMcpTools() {
-    mcpConnections = await (0, client_1.initializeMcpClients)();
+export async function initMcpTools() {
+    mcpConnections = await initializeMcpClients();
     mcpTools = [];
     for (const conn of mcpConnections) {
         for (const mcpTool of conn.tools) {
-            mcpTools.push((0, wrapper_1.wrapMcpTool)(conn, mcpTool));
+            mcpTools.push(wrapMcpTool(conn, mcpTool));
         }
     }
     return mcpTools;
 }
-function getMcpTools() {
+export function getMcpTools() {
     return mcpTools;
 }
-async function shutdownMcp() {
-    await (0, client_1.disconnectAllMcpClients)(mcpConnections);
+export async function shutdownMcp() {
+    await disconnectAllMcpClients(mcpConnections);
     mcpConnections = [];
     mcpTools = [];
 }

package/dist/agent/mcp/wrapper.d.ts

@@ -1,5 +1,5 @@
-import { type McpServerConnection } from './client';
-import type { ZhitalkTool } from '../tools';
+import { type McpServerConnection } from './client.js';
+import type { ZhitalkTool } from '../tools.js';
 export declare function wrapMcpTool(connection: McpServerConnection, mcpTool: {
     name: string;
     description?: string;

package/dist/agent/mcp/wrapper.js

@@ -1,10 +1,7 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.wrapMcpTool = wrapMcpTool;
-const tools_1 = require("@langchain/core/tools");
-const client_1 = require("./client");
+import { tool } from '@langchain/core/tools';
+import { callMcpTool } from './client.js';
 const MCP_TOOL_PERMISSION_LEVEL = 'mcp';
-function wrapMcpTool(connection, mcpTool) {
+export function wrapMcpTool(connection, mcpTool) {
     const prefixedName = `${connection.name}_${mcpTool.name}`;
     const schemaJson = JSON.stringify(mcpTool.inputSchema, null, 2);
     const description = `[MCP:${connection.name}] ${mcpTool.description || 'No description'}\n\nParameters schema:\n${schemaJson}`;
@@ -12,10 +9,10 @@ function wrapMcpTool(connection, mcpTool) {
     // sees the real parameter requirements instead of a shapeless passthrough.
     const schema = mcpTool.inputSchema;
     const impl = async (args) => {
-        const result = await (0, client_1.callMcpTool)(connection.client, mcpTool.name, args);
+        const result = await callMcpTool(connection.client, mcpTool.name, args);
         return result;
     };
-    const t = (0, tools_1.tool)(impl, {
+    const t = tool(impl, {
         name: prefixedName,
         description,
         schema,

package/dist/agent/model.js

@@ -1,12 +1,8 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.createModel = createModel;
-exports.checkModel = checkModel;
-const openai_1 = require("@langchain/openai");
-const config_1 = require("./config");
-const modelConfig = (0, config_1.getModelConfig)();
-function createModel(options) {
-    return new openai_1.ChatOpenAI({
+import { ChatOpenAI } from '@langchain/openai';
+import { getModelConfig } from './config.js';
+const modelConfig = getModelConfig();
+export function createModel(options) {
+    return new ChatOpenAI({
         model: modelConfig.model,
         apiKey: modelConfig.apiKey,
         configuration: {
@@ -18,7 +14,7 @@ function createModel(options) {
         },
     });
 }
-async function checkModel() {
+export async function checkModel() {
     if (modelConfig.apiKey.length < 20) {
         try {
             const model = createModel({ streaming: false });