@agentproto/driver-agent-cli 2.1.0 → 2.2.0

package/dist/{chunk-QL3CS6E4.mjs → chunk-TFB4QYOT.mjs}

@@ -324,7 +324,7 @@ var printConfigSchema = z.object({
     flag: z.string(),
     kind: z.enum(["value", "boolean"])
   }).strict().optional(),
-  event_schema: z.enum(["claude-stream-json", "mastra-jsonl"]).optional()
+  event_schema: z.enum(["claude-stream-json", "mastra-jsonl", "antigravity-stream-json"]).optional()
 }).strict().optional();
 var agentCliFrontmatterSchema = z.object({
   name: z.string().min(1).max(80),
@@ -720,7 +720,11 @@ function createPrintSession(opts) {
           if (csid) capturedSessionId = csid;
           const sid = capturedSessionId || sessionId || "";
           const mapped = mapEvent(evt, sid, stderrLines, eventSchema, mastraState);
-          if (mapped) queue.push(mapped);
+          if (Array.isArray(mapped)) {
+            for (const m of mapped) queue.push(m);
+          } else if (mapped) {
+            queue.push(mapped);
+          }
         });
         rl.once("close", () => queue.end());
         for await (const mapped of queue) {
@@ -729,7 +733,7 @@ function createPrintSession(opts) {
         const exitCode = await waitForExit(child);
         if (capturedSessionId) sessionId = capturedSessionId;
         if (exitCode !== 0 && exitCode !== null) {
-          const binLabel = eventSchema === "mastra-jsonl" ? "mastracode" : "claude";
+          const binLabel = eventSchema === "mastra-jsonl" ? "mastracode" : eventSchema === "antigravity-stream-json" ? "agy" : "claude";
           const errEvt = {
             kind: "error",
             error: {
@@ -864,6 +868,16 @@ function captureSessionId(evt, schema) {
       if (evt.type === "result" && typeof evt.session_id === "string" && evt.session_id)
         return evt.session_id;
       return null;
+    case "antigravity-stream-json": {
+      if (evt.event === "init" && typeof evt.conversation_id === "string" && evt.conversation_id)
+        return evt.conversation_id;
+      if (evt.event === "result") {
+        const r = evt.result;
+        if (typeof r?.conversation_id === "string" && r.conversation_id)
+          return r.conversation_id;
+      }
+      return null;
+    }
     case "mastra-jsonl":
       if (evt.type === "result" && typeof evt.threadId === "string" && evt.threadId)
         return evt.threadId;
@@ -874,6 +888,8 @@ function mapEvent(evt, sessionId, stderrLines, schema, mastraState) {
   switch (schema) {
     case "claude-stream-json":
       return mapClaudeEvent(evt, sessionId, stderrLines);
+    case "antigravity-stream-json":
+      return mapAntigravityEvent(evt, sessionId, stderrLines);
     case "mastra-jsonl": {
       if (!mastraState) {
         return mapClaudeEvent(evt, sessionId, stderrLines);
@@ -928,6 +944,93 @@ function mapClaudeEvent(evt, sessionId, stderrLines) {
       return null;
   }
 }
+function mapAntigravityEvent(evt, sessionId, stderrLines) {
+  switch (evt.event) {
+    case "step_update": {
+      const su = evt.step_update;
+      if (!su) return null;
+      const stepType = su.step_type;
+      if (stepType === "agent_response" && typeof su.text_delta === "string" && su.text_delta) {
+        return { kind: "text-delta", sessionId, text: su.text_delta };
+      }
+      if (stepType === "tool" && su.state === "DONE") {
+        const ti = su.tool_info;
+        const toolName = typeof su.tool_name === "string" && su.tool_name ? su.tool_name : typeof ti?.name === "string" && ti.name ? ti.name : "?";
+        const toolCallId = typeof su.step_index === "number" ? `step-${su.step_index}` : "";
+        const err = ti?.error;
+        const call = {
+          kind: "tool-call",
+          sessionId,
+          toolCallId,
+          toolName,
+          arguments: ti?.parameters ?? {}
+        };
+        const result = {
+          kind: "tool-result",
+          sessionId,
+          toolCallId,
+          result: err ? typeof err.message === "string" ? err.message : err : ti?.output ?? null,
+          isError: !!err
+        };
+        return [call, result];
+      }
+      return null;
+    }
+    case "result": {
+      const r = evt.result;
+      if (!r) return null;
+      const status = typeof r.status === "string" ? r.status : "";
+      const out = [];
+      const usage = mapAntigravityUsage(r.usage, sessionId);
+      if (usage) out.push(usage);
+      if (status === "ERROR" || status === "INVALID") {
+        out.push({
+          kind: "error",
+          sessionId,
+          error: {
+            message: typeof r.error === "string" && r.error ? r.error : `agy run ended with status ${status || "ERROR"}`,
+            ...stderrLines.length ? { data: { stderr: stderrLines.join("\n") } } : {}
+          }
+        });
+      } else {
+        out.push({
+          kind: "turn-end",
+          sessionId,
+          reason: mapAntigravityStatus(status)
+        });
+      }
+      return out.length ? out : null;
+    }
+    // init and any unknown event carry no host-facing StreamEvent.
+    default:
+      return null;
+  }
+}
+function mapAntigravityStatus(raw) {
+  switch (raw) {
+    case "CANCELED":
+    case "INTERRUPTED":
+      return "cancelled";
+    // SUCCESS, WAITING, RUNNING (or anything else non-error): the turn ended.
+    default:
+      return "completed";
+  }
+}
+function mapAntigravityUsage(usage, sessionId) {
+  if (usage === null || typeof usage !== "object") return null;
+  const u = usage;
+  const tokensIn = typeof u.input_tokens === "number" ? u.input_tokens : void 0;
+  const tokensOut = typeof u.output_tokens === "number" ? u.output_tokens : void 0;
+  if (tokensIn === void 0 && tokensOut === void 0) return null;
+  return {
+    kind: "usage_update",
+    sessionId,
+    size: 0,
+    used: 0,
+    ...tokensIn !== void 0 ? { tokensIn } : {},
+    ...tokensOut !== void 0 ? { tokensOut } : {}
+  };
+}
 function createMastraMapperState() {
   return { lastTextLength: 0 };
 }
@@ -1558,15 +1661,22 @@ function createAgentCliRuntime(definition) {
         opts?.posture,
         config?.mode
       );
+      const isolateClaudeCodeConfig = definition.id === "claude-code";
       let claudeConfigDir;
-      if (permissionMode) {
+      if (isolateClaudeCodeConfig) {
         claudeConfigDir = mkdtempSync(
           join(tmpdir(), "agentproto-claude-config-")
         );
         writeFileSync(
-          join(claudeConfigDir, "settings.json"),
-          JSON.stringify({ permissions: { defaultMode: permissionMode } })
+          join(claudeConfigDir, ".claude.json"),
+          JSON.stringify({ mcpServers: {} })
         );
+        if (permissionMode) {
+          writeFileSync(
+            join(claudeConfigDir, "settings.json"),
+            JSON.stringify({ permissions: { defaultMode: permissionMode } })
+          );
+        }
         env.CLAUDE_CONFIG_DIR = claudeConfigDir;
       }
       if (definition.protocol === "print") {
@@ -1758,5 +1868,5 @@ function filterStringEnv(env) {
 }
 
 export { LEGACY_GATEWAY_MODE_IDS, LEGACY_POSTURE_MODE_IDS, RuntimeConfigError, agentCliFrontmatterSchema, applyModelCommand, autoAllowPermissionHandler, composeSpawn, createAcpProtocolArm, createAgentCliRuntime, createArmSessionControls, createMastraMapperState, createPrintSession, createProprietaryProtocolArm, defineAgentCli, inferLegacyModeKind, isLegacyExtractedModeId, mapMastraEvent, planModePermissionHandler, promptTurn, resolveContinuationStrategy, runtimeConfigSchema, toFileBasedMcpServers };
-//# sourceMappingURL=chunk-QL3CS6E4.mjs.map
-//# sourceMappingURL=chunk-QL3CS6E4.mjs.map
+//# sourceMappingURL=chunk-TFB4QYOT.mjs.map
+//# sourceMappingURL=chunk-TFB4QYOT.mjs.map