chatlab-cli 0.24.1 → 0.25.1

package/dist/chat-command-5XMRLXZM.mjs

@@ -0,0 +1,166 @@
+import { createRequire as __createRequire } from 'module';
+import { dirname as __pathDirname } from 'path';
+import { fileURLToPath as __fileURLToPath } from 'url';
+const require = __createRequire(import.meta.url);
+const __filename = __fileURLToPath(import.meta.url);
+const __dirname = __pathDirname(__filename);
+import {
+  createCliRunAgentStream
+} from "./chunk-WYHGWTA4.mjs";
+import "./chunk-JE3FOFF7.mjs";
+import "./chunk-7VG6QGQ2.mjs";
+import "./chunk-VSOWNKPA.mjs";
+
+// src/ai/chat-command.ts
+import { createInterface } from "readline/promises";
+function write(stream, text) {
+  stream.write(text);
+}
+function buildTitle(question) {
+  const trimmed = question?.trim();
+  if (!trimmed) return "CLI Chat";
+  return trimmed.length > 50 ? `${trimmed.slice(0, 50)}...` : trimmed;
+}
+function createAgentStreamError(error) {
+  if (error instanceof Error) return error;
+  if (typeof error === "string" && error) return new Error(error);
+  if (typeof error === "object" && error !== null) {
+    const record = error;
+    const streamError = new Error(
+      typeof record.message === "string" && record.message ? record.message : "Agent stream failed"
+    );
+    if (typeof record.name === "string" && record.name) streamError.name = record.name;
+    if (typeof record.stack === "string" && record.stack) streamError.stack = record.stack;
+    return streamError;
+  }
+  return new Error("Agent stream failed");
+}
+function resolveSingleSessionId(dbManager) {
+  const sessionIds = dbManager.listSessionIds();
+  if (sessionIds.length === 1) return sessionIds[0];
+  if (sessionIds.length === 0) {
+    throw new Error("No chat sessions found. Import data first or run `chatlab sessions` to verify available sessions.");
+  }
+  throw new Error("Missing --session-id. Run `chatlab sessions` to choose a session ID.");
+}
+function assertSessionExists(dbManager, sessionId) {
+  const db = dbManager.open(sessionId);
+  if (!db) {
+    throw new Error(`Session ${sessionId} not found`);
+  }
+}
+function resolveAIChatTarget(options, deps) {
+  if (options.aiChatId) {
+    const aiChat2 = deps.aiChatManager.getAIChat(options.aiChatId);
+    if (!aiChat2) {
+      throw new Error(`AI chat ${options.aiChatId} not found`);
+    }
+    if (options.sessionId && options.sessionId !== aiChat2.sessionId) {
+      throw new Error(`AI chat ${options.aiChatId} belongs to session ${aiChat2.sessionId}, not ${options.sessionId}`);
+    }
+    assertSessionExists(deps.dbManager, aiChat2.sessionId);
+    return { sessionId: aiChat2.sessionId, aiChatId: aiChat2.id, assistantId: aiChat2.assistantId, created: false };
+  }
+  const sessionId = options.sessionId ?? resolveSingleSessionId(deps.dbManager);
+  assertSessionExists(deps.dbManager, sessionId);
+  const aiChat = deps.aiChatManager.createAIChat(sessionId, buildTitle(options.question), "general_cn");
+  return { sessionId, aiChatId: aiChat.id, assistantId: aiChat.assistantId, created: true };
+}
+async function runChatTurn(options, deps) {
+  const stdout = deps.stdout ?? process.stdout;
+  const target = resolveAIChatTarget(options, deps);
+  const startedAt = Date.now();
+  let answer = "";
+  let tokenUsage = null;
+  let streamError = null;
+  const runAgentStream = (deps.createRunAgentStream ?? createCliRunAgentStream)(deps.dbManager, deps.aiChatManager);
+  await runAgentStream(
+    {
+      userMessage: options.question,
+      sessionId: target.sessionId,
+      aiChatId: target.aiChatId,
+      assistantId: target.assistantId,
+      chatType: "group",
+      locale: options.locale ?? "zh-CN"
+    },
+    (chunk) => {
+      if (chunk.type === "error") {
+        streamError = createAgentStreamError(chunk.error);
+        return;
+      }
+      if (chunk.type === "content" && chunk.content) {
+        answer += chunk.content;
+        if (!options.json && options.stream !== false) write(stdout, chunk.content);
+      }
+      if (chunk.type === "done" && chunk.usage) {
+        tokenUsage = chunk.usage;
+      }
+    },
+    new AbortController().signal
+  );
+  if (streamError) throw streamError;
+  if (!options.json && options.stream !== false && answer && !answer.endsWith("\n")) {
+    write(stdout, "\n");
+  }
+  deps.aiChatManager.addMessage(target.aiChatId, "user", options.question);
+  deps.aiChatManager.addMessage(
+    target.aiChatId,
+    "assistant",
+    answer,
+    void 0,
+    void 0,
+    void 0,
+    tokenUsage ?? void 0
+  );
+  return {
+    sessionId: target.sessionId,
+    aiChatId: target.aiChatId,
+    question: options.question,
+    answer,
+    usage: {
+      durationMs: Date.now() - startedAt,
+      tokenUsage
+    }
+  };
+}
+async function runChatCommand(options, deps) {
+  const stdout = deps.stdout ?? process.stdout;
+  const stderr = deps.stderr ?? process.stderr;
+  if (options.question?.trim()) {
+    const result = await runChatTurn({ ...options, question: options.question.trim() }, deps);
+    if (options.json) {
+      write(stdout, `${JSON.stringify(result, null, 2)}
+`);
+    } else if (options.stream === false) {
+      write(stdout, `${result.answer}
+`);
+    }
+    return;
+  }
+  const rl = createInterface({
+    input: deps.stdin ?? process.stdin,
+    output: stdout
+  });
+  try {
+    let aiChatId = options.aiChatId;
+    while (true) {
+      const question = (await rl.question("chatlab> ")).trim();
+      if (!question || question === "exit" || question === "quit") break;
+      try {
+        const result = await runChatTurn({ ...options, aiChatId, question, json: false, stream: true }, deps);
+        aiChatId = result.aiChatId;
+      } catch (error) {
+        write(stderr, `${error instanceof Error ? error.message : String(error)}
+`);
+      }
+    }
+  } finally {
+    rl.close();
+  }
+}
+export {
+  resolveAIChatTarget,
+  runChatCommand,
+  runChatTurn
+};
+//# sourceMappingURL=chat-command-5XMRLXZM.mjs.map

package/dist/chat-command-5XMRLXZM.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/ai/chat-command.ts"],"sourcesContent":["import { createInterface } from 'node:readline/promises'\nimport type { Readable, Writable } from 'node:stream'\nimport type { DatabaseManager, AIChatManager, AgentStreamChunk, TokenUsageData } from '@openchatlab/node-runtime'\nimport type { PathProvider } from '@openchatlab/core'\nimport { createCliRunAgentStream } from './agent-stream-runner'\n\nexport interface ChatCommandOptions {\n  sessionId?: string\n  aiChatId?: string\n  question?: string\n  json?: boolean\n  stream?: boolean\n  locale?: string\n}\n\nexport interface ChatCommandDeps {\n  dbManager: DatabaseManager\n  pathProvider: PathProvider\n  aiChatManager: AIChatManager\n  stdout?: Pick<Writable, 'write'>\n  stderr?: Pick<Writable, 'write'>\n  stdin?: Readable\n  createRunAgentStream?: typeof createCliRunAgentStream\n}\n\nexport interface ResolvedAIChatTarget {\n  sessionId: string\n  aiChatId: string\n  assistantId: string\n  created: boolean\n}\n\nexport interface ChatTurnResult {\n  sessionId: string\n  aiChatId: string\n  question: string\n  answer: string\n  usage: {\n    durationMs: number\n    tokenUsage: TokenUsageData | null\n  }\n}\n\nfunction write(stream: Pick<Writable, 'write'>, text: string): void {\n  stream.write(text)\n}\n\nfunction buildTitle(question?: string): string {\n  const trimmed = question?.trim()\n  if (!trimmed) return 'CLI Chat'\n  return trimmed.length > 50 ? `${trimmed.slice(0, 50)}...` : trimmed\n}\n\nfunction createAgentStreamError(error: unknown): Error {\n  if (error instanceof Error) return error\n  if (typeof error === 'string' && error) return new Error(error)\n  if (typeof error === 'object' && error !== null) {\n    const record = error as { name?: unknown; message?: unknown; stack?: unknown }\n    const streamError = new Error(\n      typeof record.message === 'string' && record.message ? record.message : 'Agent stream failed'\n    )\n    if (typeof record.name === 'string' && record.name) streamError.name = record.name\n    if (typeof record.stack === 'string' && record.stack) streamError.stack = record.stack\n    return streamError\n  }\n  return new Error('Agent stream failed')\n}\n\nfunction resolveSingleSessionId(dbManager: DatabaseManager): string {\n  const sessionIds = dbManager.listSessionIds()\n  if (sessionIds.length === 1) return sessionIds[0]!\n  if (sessionIds.length === 0) {\n    throw new Error('No chat sessions found. Import data first or run `chatlab sessions` to verify available sessions.')\n  }\n  throw new Error('Missing --session-id. Run `chatlab sessions` to choose a session ID.')\n}\n\nfunction assertSessionExists(dbManager: DatabaseManager, sessionId: string): void {\n  const db = dbManager.open(sessionId)\n  if (!db) {\n    throw new Error(`Session ${sessionId} not found`)\n  }\n}\n\nexport function resolveAIChatTarget(\n  options: Pick<ChatCommandOptions, 'sessionId' | 'aiChatId' | 'question'>,\n  deps: Pick<ChatCommandDeps, 'dbManager' | 'aiChatManager'>\n): ResolvedAIChatTarget {\n  if (options.aiChatId) {\n    const aiChat = deps.aiChatManager.getAIChat(options.aiChatId)\n    if (!aiChat) {\n      throw new Error(`AI chat ${options.aiChatId} not found`)\n    }\n    if (options.sessionId && options.sessionId !== aiChat.sessionId) {\n      throw new Error(`AI chat ${options.aiChatId} belongs to session ${aiChat.sessionId}, not ${options.sessionId}`)\n    }\n    assertSessionExists(deps.dbManager, aiChat.sessionId)\n    return { sessionId: aiChat.sessionId, aiChatId: aiChat.id, assistantId: aiChat.assistantId, created: false }\n  }\n\n  const sessionId = options.sessionId ?? resolveSingleSessionId(deps.dbManager)\n  assertSessionExists(deps.dbManager, sessionId)\n  const aiChat = deps.aiChatManager.createAIChat(sessionId, buildTitle(options.question), 'general_cn')\n  return { sessionId, aiChatId: aiChat.id, assistantId: aiChat.assistantId, created: true }\n}\n\nexport async function runChatTurn(\n  options: Required<Pick<ChatCommandOptions, 'question'>> &\n    Pick<ChatCommandOptions, 'sessionId' | 'aiChatId' | 'json' | 'stream' | 'locale'>,\n  deps: ChatCommandDeps\n): Promise<ChatTurnResult> {\n  const stdout = deps.stdout ?? process.stdout\n  const target = resolveAIChatTarget(options, deps)\n  const startedAt = Date.now()\n  let answer = ''\n  let tokenUsage: TokenUsageData | null = null\n  let streamError: Error | null = null\n\n  const runAgentStream = (deps.createRunAgentStream ?? createCliRunAgentStream)(deps.dbManager, deps.aiChatManager)\n\n  await runAgentStream(\n    {\n      userMessage: options.question,\n      sessionId: target.sessionId,\n      aiChatId: target.aiChatId,\n      assistantId: target.assistantId,\n      chatType: 'group',\n      locale: options.locale ?? 'zh-CN',\n    },\n    (chunk: AgentStreamChunk) => {\n      if (chunk.type === 'error') {\n        streamError = createAgentStreamError(chunk.error)\n        return\n      }\n      if (chunk.type === 'content' && chunk.content) {\n        answer += chunk.content\n        if (!options.json && options.stream !== false) write(stdout, chunk.content)\n      }\n      if (chunk.type === 'done' && chunk.usage) {\n        tokenUsage = chunk.usage\n      }\n    },\n    new AbortController().signal\n  )\n\n  if (streamError) throw streamError\n\n  if (!options.json && options.stream !== false && answer && !answer.endsWith('\\n')) {\n    write(stdout, '\\n')\n  }\n\n  deps.aiChatManager.addMessage(target.aiChatId, 'user', options.question)\n  deps.aiChatManager.addMessage(\n    target.aiChatId,\n    'assistant',\n    answer,\n    undefined,\n    undefined,\n    undefined,\n    tokenUsage ?? undefined\n  )\n\n  return {\n    sessionId: target.sessionId,\n    aiChatId: target.aiChatId,\n    question: options.question,\n    answer,\n    usage: {\n      durationMs: Date.now() - startedAt,\n      tokenUsage,\n    },\n  }\n}\n\nexport async function runChatCommand(options: ChatCommandOptions, deps: ChatCommandDeps): Promise<void> {\n  const stdout = deps.stdout ?? process.stdout\n  const stderr = deps.stderr ?? process.stderr\n\n  if (options.question?.trim()) {\n    const result = await runChatTurn({ ...options, question: options.question.trim() }, deps)\n    if (options.json) {\n      write(stdout, `${JSON.stringify(result, null, 2)}\\n`)\n    } else if (options.stream === false) {\n      write(stdout, `${result.answer}\\n`)\n    }\n    return\n  }\n\n  const rl = createInterface({\n    input: deps.stdin ?? process.stdin,\n    output: stdout as Writable,\n  })\n\n  try {\n    let aiChatId = options.aiChatId\n    while (true) {\n      const question = (await rl.question('chatlab> ')).trim()\n      if (!question || question === 'exit' || question === 'quit') break\n      try {\n        const result = await runChatTurn({ ...options, aiChatId, question, json: false, stream: true }, deps)\n        aiChatId = result.aiChatId\n      } catch (error) {\n        write(stderr, `${error instanceof Error ? error.message : String(error)}\\n`)\n      }\n    }\n  } finally {\n    rl.close()\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;AAAA,SAAS,uBAAuB;AA2ChC,SAAS,MAAM,QAAiC,MAAoB;AAClE,SAAO,MAAM,IAAI;AACnB;AAEA,SAAS,WAAW,UAA2B;AAC7C,QAAM,UAAU,UAAU,KAAK;AAC/B,MAAI,CAAC,QAAS,QAAO;AACrB,SAAO,QAAQ,SAAS,KAAK,GAAG,QAAQ,MAAM,GAAG,EAAE,CAAC,QAAQ;AAC9D;AAEA,SAAS,uBAAuB,OAAuB;AACrD,MAAI,iBAAiB,MAAO,QAAO;AACnC,MAAI,OAAO,UAAU,YAAY,MAAO,QAAO,IAAI,MAAM,KAAK;AAC9D,MAAI,OAAO,UAAU,YAAY,UAAU,MAAM;AAC/C,UAAM,SAAS;AACf,UAAM,cAAc,IAAI;AAAA,MACtB,OAAO,OAAO,YAAY,YAAY,OAAO,UAAU,OAAO,UAAU;AAAA,IAC1E;AACA,QAAI,OAAO,OAAO,SAAS,YAAY,OAAO,KAAM,aAAY,OAAO,OAAO;AAC9E,QAAI,OAAO,OAAO,UAAU,YAAY,OAAO,MAAO,aAAY,QAAQ,OAAO;AACjF,WAAO;AAAA,EACT;AACA,SAAO,IAAI,MAAM,qBAAqB;AACxC;AAEA,SAAS,uBAAuB,WAAoC;AAClE,QAAM,aAAa,UAAU,eAAe;AAC5C,MAAI,WAAW,WAAW,EAAG,QAAO,WAAW,CAAC;AAChD,MAAI,WAAW,WAAW,GAAG;AAC3B,UAAM,IAAI,MAAM,mGAAmG;AAAA,EACrH;AACA,QAAM,IAAI,MAAM,sEAAsE;AACxF;AAEA,SAAS,oBAAoB,WAA4B,WAAyB;AAChF,QAAM,KAAK,UAAU,KAAK,SAAS;AACnC,MAAI,CAAC,IAAI;AACP,UAAM,IAAI,MAAM,WAAW,SAAS,YAAY;AAAA,EAClD;AACF;AAEO,SAAS,oBACd,SACA,MACsB;AACtB,MAAI,QAAQ,UAAU;AACpB,UAAMA,UAAS,KAAK,cAAc,UAAU,QAAQ,QAAQ;AAC5D,QAAI,CAACA,SAAQ;AACX,YAAM,IAAI,MAAM,WAAW,QAAQ,QAAQ,YAAY;AAAA,IACzD;AACA,QAAI,QAAQ,aAAa,QAAQ,cAAcA,QAAO,WAAW;AAC/D,YAAM,IAAI,MAAM,WAAW,QAAQ,QAAQ,uBAAuBA,QAAO,SAAS,SAAS,QAAQ,SAAS,EAAE;AAAA,IAChH;AACA,wBAAoB,KAAK,WAAWA,QAAO,SAAS;AACpD,WAAO,EAAE,WAAWA,QAAO,WAAW,UAAUA,QAAO,IAAI,aAAaA,QAAO,aAAa,SAAS,MAAM;AAAA,EAC7G;AAEA,QAAM,YAAY,QAAQ,aAAa,uBAAuB,KAAK,SAAS;AAC5E,sBAAoB,KAAK,WAAW,SAAS;AAC7C,QAAM,SAAS,KAAK,cAAc,aAAa,WAAW,WAAW,QAAQ,QAAQ,GAAG,YAAY;AACpG,SAAO,EAAE,WAAW,UAAU,OAAO,IAAI,aAAa,OAAO,aAAa,SAAS,KAAK;AAC1F;AAEA,eAAsB,YACpB,SAEA,MACyB;AACzB,QAAM,SAAS,KAAK,UAAU,QAAQ;AACtC,QAAM,SAAS,oBAAoB,SAAS,IAAI;AAChD,QAAM,YAAY,KAAK,IAAI;AAC3B,MAAI,SAAS;AACb,MAAI,aAAoC;AACxC,MAAI,cAA4B;AAEhC,QAAM,kBAAkB,KAAK,wBAAwB,yBAAyB,KAAK,WAAW,KAAK,aAAa;AAEhH,QAAM;AAAA,IACJ;AAAA,MACE,aAAa,QAAQ;AAAA,MACrB,WAAW,OAAO;AAAA,MAClB,UAAU,OAAO;AAAA,MACjB,aAAa,OAAO;AAAA,MACpB,UAAU;AAAA,MACV,QAAQ,QAAQ,UAAU;AAAA,IAC5B;AAAA,IACA,CAAC,UAA4B;AAC3B,UAAI,MAAM,SAAS,SAAS;AAC1B,sBAAc,uBAAuB,MAAM,KAAK;AAChD;AAAA,MACF;AACA,UAAI,MAAM,SAAS,aAAa,MAAM,SAAS;AAC7C,kBAAU,MAAM;AAChB,YAAI,CAAC,QAAQ,QAAQ,QAAQ,WAAW,MAAO,OAAM,QAAQ,MAAM,OAAO;AAAA,MAC5E;AACA,UAAI,MAAM,SAAS,UAAU,MAAM,OAAO;AACxC,qBAAa,MAAM;AAAA,MACrB;AAAA,IACF;AAAA,IACA,IAAI,gBAAgB,EAAE;AAAA,EACxB;AAEA,MAAI,YAAa,OAAM;AAEvB,MAAI,CAAC,QAAQ,QAAQ,QAAQ,WAAW,SAAS,UAAU,CAAC,OAAO,SAAS,IAAI,GAAG;AACjF,UAAM,QAAQ,IAAI;AAAA,EACpB;AAEA,OAAK,cAAc,WAAW,OAAO,UAAU,QAAQ,QAAQ,QAAQ;AACvE,OAAK,cAAc;AAAA,IACjB,OAAO;AAAA,IACP;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA,cAAc;AAAA,EAChB;AAEA,SAAO;AAAA,IACL,WAAW,OAAO;AAAA,IAClB,UAAU,OAAO;AAAA,IACjB,UAAU,QAAQ;AAAA,IAClB;AAAA,IACA,OAAO;AAAA,MACL,YAAY,KAAK,IAAI,IAAI;AAAA,MACzB;AAAA,IACF;AAAA,EACF;AACF;AAEA,eAAsB,eAAe,SAA6B,MAAsC;AACtG,QAAM,SAAS,KAAK,UAAU,QAAQ;AACtC,QAAM,SAAS,KAAK,UAAU,QAAQ;AAEtC,MAAI,QAAQ,UAAU,KAAK,GAAG;AAC5B,UAAM,SAAS,MAAM,YAAY,EAAE,GAAG,SAAS,UAAU,QAAQ,SAAS,KAAK,EAAE,GAAG,IAAI;AACxF,QAAI,QAAQ,MAAM;AAChB,YAAM,QAAQ,GAAG,KAAK,UAAU,QAAQ,MAAM,CAAC,CAAC;AAAA,CAAI;AAAA,IACtD,WAAW,QAAQ,WAAW,OAAO;AACnC,YAAM,QAAQ,GAAG,OAAO,MAAM;AAAA,CAAI;AAAA,IACpC;AACA;AAAA,EACF;AAEA,QAAM,KAAK,gBAAgB;AAAA,IACzB,OAAO,KAAK,SAAS,QAAQ;AAAA,IAC7B,QAAQ;AAAA,EACV,CAAC;AAED,MAAI;AACF,QAAI,WAAW,QAAQ;AACvB,WAAO,MAAM;AACX,YAAM,YAAY,MAAM,GAAG,SAAS,WAAW,GAAG,KAAK;AACvD,UAAI,CAAC,YAAY,aAAa,UAAU,aAAa,OAAQ;AAC7D,UAAI;AACF,cAAM,SAAS,MAAM,YAAY,EAAE,GAAG,SAAS,UAAU,UAAU,MAAM,OAAO,QAAQ,KAAK,GAAG,IAAI;AACpG,mBAAW,OAAO;AAAA,MACpB,SAAS,OAAO;AACd,cAAM,QAAQ,GAAG,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK,CAAC;AAAA,CAAI;AAAA,MAC7E;AAAA,IACF;AAAA,EACF,UAAE;AACA,OAAG,MAAM;AAAA,EACX;AACF;","names":["aiChat"]}