atom-lite 0.4.1 → 0.6.2

package/dist/agent/turn-recorder.js

@@ -0,0 +1,74 @@
+export class TurnRecorder {
+    session;
+    onEvent;
+    extensionDispatch;
+    constructor(options) {
+        this.session = options.session;
+        this.onEvent = options.onEvent;
+        this.extensionDispatch = options.extensionDispatch;
+    }
+    async recordAssistant(response, streamedText) {
+        let message = response.message;
+        void this.extensionDispatch?.("message_start", { type: "message_start", message });
+        const results = await this.extensionDispatch?.("message_end", {
+            type: "message_end",
+            message,
+        });
+        const replacement = Array.isArray(results)
+            ? results.find((x) => x && x.message)?.message
+            : results?.message;
+        if (replacement &&
+            typeof replacement === "object" &&
+            replacement.role === message.role) {
+            message = replacement;
+        }
+        this.session.appendMessage(message);
+        const text = streamedText ||
+            (Array.isArray(message.content)
+                ? message.content
+                    .map((p) => (p?.type === "text" ? p.text : ""))
+                    .join("")
+                : typeof message.content === "string"
+                    ? message.content
+                    : "");
+        if (response.usage) {
+            this.onEvent?.({
+                type: "usage",
+                input: response.usage.input,
+                output: response.usage.output,
+            });
+        }
+        const toolCalls = response.toolCalls ?? [];
+        if (toolCalls.length === 0) {
+            this.onEvent?.({ type: "final", text });
+        }
+        else {
+            for (const call of toolCalls) {
+                this.onEvent?.({
+                    type: "tool_call",
+                    id: call.id,
+                    name: call.name,
+                    args: call.arguments,
+                });
+            }
+        }
+        return text;
+    }
+    recordToolResult(result) {
+        this.session.appendMessage({
+            role: "tool",
+            toolCallId: result.toolCallId,
+            name: result.name,
+            content: result.content,
+        });
+        this.onEvent?.({
+            type: "tool_result",
+            toolCallId: result.toolCallId,
+            name: result.name,
+            content: result.content,
+            isError: result.isError,
+            diff: result.diff,
+        });
+    }
+}
+//# sourceMappingURL=turn-recorder.js.map

package/dist/agent/turn-recorder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"turn-recorder.js","sourceRoot":"","sources":["../../src/agent/turn-recorder.ts"],"names":[],"mappings":"AAQA,MAAM,OAAO,YAAY;IACf,OAAO,CAAiC;IACxC,OAAO,CAAiC;IACxC,iBAAiB,CAA2C;IAEpE,YAAY,OAA4B;QACtC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAC/B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAC/B,IAAI,CAAC,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,QAIrB,EAAE,YAAoB;QACrB,IAAI,OAAO,GAAY,QAAQ,CAAC,OAAO,CAAC;QACxC,KAAK,IAAI,CAAC,iBAAiB,EAAE,CAAC,eAAe,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,CAAC,CAAC;QACnF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC,aAAa,EAAE;YAC5D,IAAI,EAAE,aAAa;YACnB,OAAO;SACR,CAAC,CAAC;QACH,MAAM,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;YACxC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO;YACnD,CAAC,CAAE,OAAe,EAAE,OAAO,CAAC;QAC9B,IACE,WAAW;YACX,OAAO,WAAW,KAAK,QAAQ;YAC9B,WAAmB,CAAC,IAAI,KAAK,OAAO,CAAC,IAAI,EAC1C,CAAC;YACD,OAAO,GAAG,WAAsB,CAAC;QACnC,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QACpC,MAAM,IAAI,GACR,YAAY;YACZ,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC;gBAC7B,CAAC,CAAC,OAAO,CAAC,OAAO;qBACZ,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;qBACnD,IAAI,CAAC,EAAE,CAAC;gBACb,CAAC,CAAC,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ;oBACnC,CAAC,CAAC,OAAO,CAAC,OAAO;oBACjB,CAAC,CAAC,EAAE,CAAC,CAAC;QACZ,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;YACnB,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,KAAK;gBAC3B,MAAM,EAAE,QAAQ,CAAC,KAAK,CAAC,MAAM;aAC9B,CAAC,CAAC;QACL,CAAC;QACD,MAAM,SAAS,GAAG,QAAQ,CAAC,SAAS,IAAI,EAAE,CAAC;QAC3C,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3B,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1C,CAAC;aAAM,CAAC;YACN,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;gBAC7B,IAAI,CAAC,OAAO,EAAE,CAAC;oBACb,IAAI,EAAE,WAAW;oBACjB,EAAE,EAAE,IAAI,CAAC,EAAE;oBACX,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,IAAI,EAAE,IAAI,CAAC,SAAS;iBACrB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,gBAAgB,CAAC,MAMhB;QACC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC;YACzB,IAAI,EAAE,MAAM;YACZ,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,OAAO,EAAE,MAAM,CAAC,OAAO;SACxB,CAAC,CAAC;QACH,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,IAAI,EAAE,aAAa;YACnB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,IAAI,EAAE,MAAM,CAAC,IAAI;SAClB,CAAC,CAAC;IACL,CAAC;CACF"}

package/dist/agent/types.d.ts

@@ -50,6 +50,11 @@ export type AgentEvent = {
 } | {
     type: "skill_loaded";
     name: string;
+} | {
+    type: "budget_warning";
+    scope: "turn" | "session";
+    used: number;
+    limit: number;
 } | {
     type: "error";
     message: string;
@@ -62,12 +67,22 @@ export interface AgentRuntimeOptions {
     limits?: Partial<AgentLimits>;
     systemPrompt?: string;
     approvals?: ApprovalContext;
+    /** Initial agent mode. Defaults to yolo for programmatic use. */
+    mode?: "plan" | "yolo";
+    /** Optional session token budget (hard stop when exceeded). */
+    maxTokensPerSession?: number | null;
+    /** Optional per-turn token soft warning threshold. */
+    maxTokensPerTurn?: number | null;
     contextManager?: ContextManager;
     onEvent?: (event: AgentEvent) => void;
     effort?: string;
     bashConcurrency?: "serial" | "parallel";
     /** Selected model's context window, when known. */
     contextWindow?: number;
+    /** Optional steering queue drained into the turn between iterations. */
+    steeringQueue?: string[];
+    /** Optional dispatcher for Pi extension events. */
+    extensionDispatch?: (event: string, payload: any) => Promise<any[]> | any[];
 }
 export interface ContextUsage {
     /** Tokens occupied after the latest assistant response, when known. */