@xalia/agent 0.6.9 → 0.6.11

package/src/tool/chatMain.ts

@@ -1,594 +0,0 @@
-/* eslint-disable @typescript-eslint/only-throw-error */
-
-import chalk from "chalk";
-import { command, optional, option, string, number, subcommands } from "cmd-ts";
-import { stdout } from "process";
-import * as fs from "fs";
-import { strict as assert } from "assert";
-
-import { configuration } from "@xalia/xmcp/tool";
-import { getLogger } from "@xalia/xmcp/sdk";
-
-import {
-  ChatClient,
-  SessionClient,
-  IChatClientEventHandler,
-} from "../chat/client";
-import {
-  runServer,
-  resolveSessionIdFromIdentifier,
-} from "../chat/server/server";
-import {
-  ServerToClient,
-  decodeAssistantMessageParam,
-} from "../chat/protocol/messages";
-import { Database, createSessionParticipantMap } from "../chat/data/database";
-
-import { IPrompt, Prompt, ScriptPrompt } from "./prompt";
-import * as options from "./options";
-import { CommandPrompt } from "./commandPrompt";
-import { NODE_PLATFORM } from "./nodePlatform";
-import { loadImageAsDataUrl, loadSessionFileAsDataUrl } from "./files";
-import { isFileMetaData } from "../chat/data/dbSessionFileModels";
-
-const logger = getLogger();
-
-const listSessions = command({
-  name: "delete-session",
-  args: {
-    supabaseUrl: options.supabaseUrl,
-    supabaseKey: options.supabaseKey,
-  },
-  handler: async ({ supabaseUrl, supabaseKey }): Promise<void> => {
-    const db = new Database(supabaseUrl, supabaseKey);
-    console.log(JSON.stringify(await db.sessionsGet()));
-  },
-});
-
-const clearSessions = command({
-  name: "clear-sessions",
-  args: {
-    supabaseUrl: options.supabaseUrl,
-    supabaseKey: options.supabaseKey,
-  },
-  handler: async ({ supabaseUrl, supabaseKey }): Promise<void> => {
-    const db = new Database(supabaseUrl, supabaseKey);
-    await db.clearSessions();
-  },
-});
-
-const deleteSession = command({
-  name: "delete-session",
-  args: {
-    supabaseUrl: options.supabaseUrl,
-    supabaseKey: options.supabaseKey,
-    session: option({
-      type: string,
-      long: "session",
-      description:
-        "Session identifier (id, name, user/name or name of new session)",
-      env: "SESSION",
-    }),
-  },
-  handler: async ({ supabaseUrl, supabaseKey, session }): Promise<void> => {
-    const db = new Database(supabaseUrl, supabaseKey);
-    const sessionId = await resolveSessionIdFromIdentifier(db, session);
-    if (sessionId) {
-      await db.sessionDeleteById(sessionId);
-    } else {
-      throw `No such session ${session}`;
-    }
-  },
-});
-
-const listParticipants = command({
-  name: "list-participants",
-  args: {
-    supabaseUrl: options.supabaseUrl,
-    supabaseKey: options.supabaseKey,
-    session: option({
-      type: string,
-      long: "session",
-      description:
-        "Session identifier (id, name, user/name or name of new session)",
-      env: "SESSION",
-    }),
-  },
-  handler: async ({ supabaseUrl, supabaseKey, session }): Promise<void> => {
-    const db = new Database(supabaseUrl, supabaseKey);
-    const sessionId = await resolveSessionIdFromIdentifier(db, session);
-    if (sessionId) {
-      const participants = await db.sessionGetParticipants(sessionId);
-
-      console.log(`participants: ${JSON.stringify(participants)}`);
-      const users = createSessionParticipantMap(participants);
-      console.log(JSON.stringify(users));
-    } else {
-      throw `No such session ${session}`;
-    }
-  },
-});
-
-const server = command({
-  name: "server",
-  args: {
-    port: option({
-      type: number,
-      long: "port",
-      short: "p",
-      env: "CHAT_SERVER_PORT",
-      defaultValue: () => 5003,
-    }),
-    supabaseUrl: options.supabaseUrl,
-    supabaseKey: options.supabaseKey,
-    xmcpUrl: options.xmcpUrl,
-    pidFile: options.pidFile,
-  },
-  // eslint-disable-next-line @typescript-eslint/require-await
-  handler: async ({
-    port,
-    supabaseUrl,
-    supabaseKey,
-    xmcpUrl,
-    pidFile,
-  }): Promise<void> => {
-    if (pidFile) {
-      fs.writeFileSync(pidFile, process.pid.toString());
-    }
-
-    void runServer(port, supabaseUrl, supabaseKey, xmcpUrl);
-  },
-});
-
-const client = command({
-  name: "client",
-  args: {
-    url: option({
-      type: string,
-      long: "url",
-      short: "h",
-      env: "CHAT_URL",
-      description: "Chat server url",
-      defaultValue: () => "ws://localhost:5003",
-    }),
-    apiKey: options.apiKey,
-    sessionId: option({
-      type: optional(string),
-      long: "session-id",
-      description: "Join existing session with the given UUID",
-      env: "SESSION_UUID",
-    }),
-    teamId: option({
-      type: optional(string),
-      long: "team-id",
-      description:
-        "Team uuid, if not provided, a user session will be created.",
-      env: "TEAM_ID",
-    }),
-    agentProfileId: option({
-      type: optional(string),
-      long: "agent-profile-id",
-      description: "Create new session with agent profile (uuid)",
-      env: "AGENT_PROFILE_ID",
-    }),
-    sessionTitle: option({
-      type: optional(string),
-      long: "session-title",
-      description: "Title for new session",
-      env: "SESSION_TITLE",
-    }),
-    script: option({
-      type: optional(string),
-      long: "script",
-      description: "Script (file) to execute and then exit.",
-    }),
-  },
-  handler: async ({
-    url,
-    apiKey,
-    sessionId,
-    agentProfileId,
-    sessionTitle,
-    teamId,
-    script,
-  }): Promise<void> => {
-    if (!apiKey) {
-      // TODO: configFIle param list in ../main.ts
-      const sudomcpConfig = configuration.loadEnsureConfig();
-      apiKey = sudomcpConfig.api_key;
-    }
-
-    const repl: IPrompt = (() => {
-      if (script) {
-        const scriptLines = fs.readFileSync(script, "utf8").split("\n");
-        return new ScriptPrompt(scriptLines);
-      }
-      return new Prompt();
-    })();
-
-    const helpText = `
-  ${chalk.yellow("/aa <server> <tool> [1]")}   Auto approve
-  ${chalk.yellow("/error <msg>")}        Force server error
-  ${chalk.yellow("/ap <team> <user>")}   Add-Participant
-  ${chalk.yellow("/rp <team> <user>")}   Remove-Participant
-  ${chalk.yellow("/lp")}                 List-Participants
-  ${chalk.yellow("/si [<file>]")}        Write Session id to file
-  ${chalk.yellow("/ct <name>")}          Create Team
-  ${chalk.yellow("/ci [<file>]")}        Write team id to file
-  ${chalk.yellow("/sw [file:a.png|data:....]")}  Set workspace (no args = clear)
-  ${chalk.yellow("/list-files")}         List file-manager files
-  ${chalk.yellow("/put-file <name> <file:...|data:...>")}  Add file
-  ${chalk.yellow("/delete-file <name>")} Delete file
-  ${chalk.yellow("/get-file <name>")}    Get file contents
-  ${chalk.yellow("/share-session <file>")} Share and write access token
-  ${chalk.yellow("/pause-agent <0|1>")}  Pause (or unpause) the agent
-  ${chalk.yellow("/add-custom-mcp-server <server-name> <url>")} Add MCP server
-  ${chalk.yellow("/remove-custom-mcp-server <server-name>")} Remove MCP server
-  ${chalk.yellow("/list-custom-mcp-servers")} list custom mcp servers
-  ${chalk.yellow("/get-resource <server-name> <uri>")} get MCP server resource
-`;
-
-    const cmdPrompt = new CommandPrompt(repl, helpText);
-    const eventHandler = getCLIEventHandler(cmdPrompt);
-
-    const [client, sessionClient] = await (async () => {
-      const newClient = await ChatClient.init(url, apiKey, eventHandler);
-      if (sessionId) {
-        if (agentProfileId || sessionTitle) {
-          throw "t agent-profile-id/session-title set joining existing session";
-        }
-
-        console.log(`joining existing session: ${sessionId} ...`);
-        const newSessionClient = await newClient.connectToSession(sessionId);
-        return [newClient, newSessionClient];
-      }
-
-      const newSessionClient = await newClient.createNewSession(
-        sessionTitle || "New Chat",
-        agentProfileId,
-        teamId
-      );
-      return [newClient, newSessionClient];
-    })();
-    logger.debug("client created");
-
-    const onUnknownCommand = async (cmds: string[]): Promise<void> => {
-      switch (cmds[0]) {
-        case "aa":
-          sessionClient.setAutoApproval(cmds[1], cmds[2], !!cmds[3]);
-          break;
-        case "error":
-          sessionClient.sendMessage({ type: "fatal_error", message: cmds[1] });
-          break;
-        case "ap": // Add Participant
-        case "au":
-          client.addTeamMember(cmds[1], cmds[2]);
-          break;
-        case "rp": // Remove Participant
-        case "ru":
-          client.removeTeamMember(cmds[1], cmds[2]);
-          break;
-        case "lp": // List Participants
-        case "lu":
-          console.log(JSON.stringify(sessionClient.getParticipants()));
-          break;
-        case "si": // Write Session id to file
-          fs.writeFileSync(cmds[1], sessionClient.getSessionUUID());
-          break;
-        case "ct": // Create Team
-          client.createNewTeam(cmds[1], []);
-          break;
-        case "ci": // Write team id to file
-          fs.writeFileSync(cmds[1], client.getCurrentTeamId() ?? "");
-          break;
-        case "sw": // Set workspace
-          setWorkspaceFromPrompt(sessionClient, cmds);
-          break;
-        case "list-files": // List files
-          listFiles(sessionClient, cmds[1]);
-          break;
-        case "put-file": // Add file
-          putFile(sessionClient, cmds.slice(1));
-          break;
-        case "delete-file": // Delete file
-          deleteFile(sessionClient, cmds[1]);
-          break;
-        case "get-file": // Get file contents
-          await getFileContents(sessionClient, cmds[1]);
-          break;
-        case "share-session":
-          await shareSession(sessionClient, cmds[1]);
-          break;
-        case "pause-agent":
-          pauseAgent(sessionClient, cmds[1]);
-          break;
-        case "add-custom-mcp-server":
-          addCustomMcpServer(client, cmds[1], cmds[2]);
-          break;
-        case "remove-custom-mcp-server":
-          removeCustomMcpServer(client, cmds[1]);
-          break;
-        case "list-custom-mcp-servers":
-          listCustomMcpServers(client);
-          break;
-        case "get-resource":
-          await getResource(sessionClient, cmds[1], cmds[2]);
-          break;
-        default:
-          console.log(`error: Unknown command ${cmds[0]}`);
-      }
-    };
-
-    // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition
-    while (true) {
-      const [msgText, imageFile] = await cmdPrompt.getNextPrompt(
-        sessionClient,
-        sessionClient.getSudoMcpServerManager(),
-        onUnknownCommand
-      );
-      assert(imageFile === undefined);
-
-      if (msgText === undefined) {
-        logger.debug("exiting...");
-        logger.info("exiting...");
-        client.shutdown();
-        return;
-      }
-
-      // TODO: other prompts
-
-      if (msgText.length > 0) {
-        logger.debug(`sending message: ${msgText}`);
-        sessionClient.userMessage(msgText);
-      } else {
-        logger.debug("(ignoring empty message)");
-      }
-    }
-  },
-});
-
-function listFiles(
-  sessionClient: SessionClient,
-  filename: string | undefined
-): void {
-  const sessionFiles = sessionClient.getSessionFiles();
-  const fileList = sessionFiles.listFiles();
-  const fileListStr = JSON.stringify(fileList, undefined, 2);
-  console.log(`Session files:\n${fileListStr}`);
-  if (filename) {
-    fs.writeFileSync(filename, fileListStr);
-  }
-}
-
-function putFile(sessionClient: SessionClient, tail: string[]): void {
-  if (tail.length < 2) {
-    throw new Error(`usage: put-file <name> <content ....>`);
-  }
-  const name = tail[0];
-  let content = tail.slice(1).join(" ");
-  if (content.startsWith("file:")) {
-    content = loadSessionFileAsDataUrl(content.slice(5));
-  }
-
-  const sessionFiles = sessionClient.getSessionFiles();
-  sessionFiles.putFileContent(name, undefined, content);
-}
-
-function deleteFile(sessionClient: SessionClient, name: string) {
-  const sessionFiles = sessionClient.getSessionFiles();
-  sessionFiles.deleteFile(name);
-}
-
-async function getFileContents(sessionClient: SessionClient, name: string) {
-  const sessionFiles = sessionClient.getSessionFiles();
-  const dataUrl = await sessionFiles.getFileContent(name);
-  try {
-    void NODE_PLATFORM.renderHTML(`<img src="${dataUrl}" />`);
-  } catch {
-    console.log(`(render failed) file ${name}: ${dataUrl}`);
-  }
-}
-
-async function shareSession(sessionClient: SessionClient, filename: string) {
-  const accessToken = await sessionClient.shareSession();
-  fs.writeFileSync(filename, accessToken);
-}
-
-function pauseAgent(sessionClient: SessionClient, pause: string) {
-  sessionClient.setAgentPaused(!!parseInt(pause));
-}
-
-function addCustomMcpServer(chatClient: ChatClient, name: string, url: string) {
-  chatClient.addCustomMcpServer(name, "Temporary description", url);
-}
-
-function removeCustomMcpServer(chatClient: ChatClient, name: string) {
-  chatClient.removeCustomMcpServer(name);
-}
-
-function listCustomMcpServers(chatClient: ChatClient) {
-  const customMcpServers: Record<string, string> =
-    chatClient.getCustomMcpServers();
-  console.log(
-    `Custom mcp serevrs:\n${JSON.stringify(customMcpServers, undefined, 2)}`
-  );
-}
-
-async function getResource(
-  sessionClient: SessionClient,
-  serverName: string,
-  resourceUrl: string
-) {
-  const contents = await sessionClient.getMcpResource(serverName, resourceUrl);
-  console.log(
-    `resource: (${serverName}) ${resourceUrl}: ${JSON.stringify(contents)}`
-  );
-}
-
-function setWorkspaceFromPrompt(
-  sessionClient: SessionClient,
-  cmds: string[]
-): void {
-  let msg = "";
-  let image = undefined;
-  if (!cmds[1]) {
-    sessionClient.setWorkspace(undefined, undefined);
-    return;
-  }
-  if (cmds[1].startsWith("file:")) {
-    image = loadImageAsDataUrl(cmds[1].slice(5));
-    msg = cmds.slice(2).join(" ");
-    console.log(`setting workspace: ${msg} (image: ${image})`);
-  } else {
-    msg = cmds.slice(1).join(" ");
-    console.log(`setting workspace: ${msg}`);
-  }
-
-  sessionClient.setWorkspace(msg, image);
-}
-
-function getCLIEventHandler(
-  cmdPrompt?: CommandPrompt
-): IChatClientEventHandler {
-  let numErrors = 0;
-
-  const onClosed = async () => {
-    logger.debug("[onClosed]");
-    if (cmdPrompt) {
-      // TODO: check if this fully shuts down
-      await cmdPrompt.shutdown();
-    }
-
-    // If there have been errors, exit with an error code.
-    process.exit(numErrors);
-  };
-
-  const onError = (message: string) => {
-    numErrors++;
-    logger.debug("[onError]");
-    console.error(message);
-  };
-
-  let startMsg = true;
-  let startReasoning = true;
-  const onMessage = async (msg: ServerToClient, client: ChatClient) => {
-    let sessionClient: SessionClient | undefined = undefined;
-    try {
-      sessionClient = client.getCurrentSession("[chatMain.onMessage]");
-    } catch (error) {
-      logger.error(String(error));
-      return;
-    }
-
-    switch (msg.type) {
-      case "user_msg":
-        stdout.write(`[${msg.user_nickname}]: ${msg.message || ""}\n`);
-        if (msg.imageB64) {
-          stdout.write(`[${msg.user_nickname}]: IMAGE: ${msg.imageB64}\n`);
-        }
-        break;
-      case "agent_msg":
-        {
-          const text = decodeAssistantMessageParam(msg.message);
-          if (text) {
-            stdout.write(`[AGENT]: ${text}\n`);
-          }
-        }
-        break;
-      case "agent_msg_chunk":
-        if (startMsg) {
-          if (!startReasoning) {
-            stdout.write(chalk.grey("]\n"));
-          }
-          stdout.write("[AGENT]: ");
-          startMsg = false;
-        }
-        stdout.write(msg.message);
-        if (msg.end) {
-          stdout.write("\n");
-          startMsg = true;
-          startReasoning = true;
-        }
-        break;
-      case "agent_reasoning_chunk":
-        if (startMsg) {
-          stdout.write(chalk.grey(`[${msg.reasoning}`));
-          startReasoning = false;
-        } else {
-          stdout.write(chalk.grey(msg.reasoning));
-        }
-        break;
-      case "approve_tool_call":
-        {
-          const result = cmdPrompt
-            ? await cmdPrompt.promptToolCall(msg.tool_call)
-            : true;
-          logger.debug(`cmdPrompt.promptToolCall returned: ${String(result)}`);
-          sessionClient.toolCallApprovalResult(
-            msg.id,
-            result,
-            false /*  auto-approve */
-          );
-        }
-        break;
-      case "tool_call_result":
-        {
-          console.log(`tool call result: ${JSON.stringify(msg.result)}`);
-
-          const metadata = msg.result._meta;
-          if (isFileMetaData(metadata)) {
-            console.log(`filemanager file: ${JSON.stringify(metadata)}`);
-
-            // SessionFiles should already hold the file information.
-
-            const sessionFiles = sessionClient.getSessionFiles();
-            const { name } = sessionFiles.decodeSessionFileUrl(
-              msg.result.content as string
-            );
-            if (!sessionFiles.descriptors.has(name)) {
-              throw new Error(`no ${name} in SessionFiles`);
-            }
-          }
-        }
-        break;
-      case "render_html":
-        await NODE_PLATFORM.renderHTML(msg.html);
-        break;
-      case "authenticate":
-        NODE_PLATFORM.openUrl(
-          msg.url,
-          new Promise<boolean>((r) => {
-            r(true);
-          }),
-          msg.display_name
-        );
-        break;
-      case "session_error":
-        stdout.write(
-          chalk.red(`session_error(${msg.session_id}): ${msg.message}`)
-        );
-        break;
-      default:
-        stdout.write(`(unrecognised) ${JSON.stringify(msg.type)}\n`);
-        break;
-    }
-  };
-
-  return {
-    onMessage,
-    onError,
-    onClosed: onClosed,
-  };
-}
-
-export const chatMain = subcommands({
-  name: "chat",
-  cmds: {
-    server,
-    client,
-    "clear-sessions": clearSessions,
-    "delete-session": deleteSession,
-    "list-sessions": listSessions,
-    "list-participants": listParticipants,
-  },
-});