saving

ts/packages/cli/src/commands/agent.ts

@@ -0,0 +1,34 @@
+/**
+ * Agent CLI commands.
+ *
+ * Python reference: trustgraph-cli/trustgraph/cli/invoke_agent.py
+ */
+
+import type { Command } from "commander";
+import { createSocket, getOpts } from "./util.js";
+
+export function registerAgentCommands(program: Command): void {
+  program
+    .command("agent")
+    .description("Ask the TrustGraph agent a question")
+    .argument("<question>", "Question to ask")
+    .action(async (question: string, _opts, cmd) => {
+      const opts = getOpts(cmd);
+      const socket = await createSocket(opts);
+
+      try {
+        const resp = await socket.request("agent", { question }, {
+          flowId: opts.flow,
+          onChunk: (chunk) => {
+            const c = chunk as { answer?: string };
+            if (c.answer) process.stdout.write(c.answer);
+          },
+        });
+
+        const r = resp as { answer?: string };
+        if (r.answer) console.log(r.answer);
+      } finally {
+        await socket.close();
+      }
+    });
+}