@automatalabs/pi-acp 0.1.0

package/dist/structured-output.js

@@ -0,0 +1,71 @@
+import { Type } from "typebox";
+import { adapterError } from "./errors.js";
+export const STRUCTURED_TOOL_NAME = "__acp_structured_output";
+export class StructuredOutputState {
+    tool;
+    schemaHolder = Type.Any();
+    captured;
+    baseActive = [];
+    constructor() {
+        const tool = {
+            name: STRUCTURED_TOOL_NAME,
+            label: "Structured output",
+            description: "Return the final response as structured data matching the requested schema.",
+            parameters: this.schemaHolder,
+            execute: async (_toolCallId, params) => {
+                this.captured = params;
+                return {
+                    content: [{ type: "text", text: "(structured output captured)" }],
+                    details: params,
+                    terminate: true,
+                };
+            },
+        };
+        this.tool = tool;
+    }
+    install(session) {
+        if (!session.getToolDefinition(STRUCTURED_TOOL_NAME)) {
+            throw adapterError("structured_tool_collision");
+        }
+        this.baseActive = session.getActiveToolNames().filter((name) => name !== STRUCTURED_TOOL_NAME);
+        session.setActiveToolsByName(this.baseActive);
+    }
+    arm(session, schema) {
+        if (typeof schema !== "object" || schema === null || Array.isArray(schema)) {
+            throw adapterError("invalid_output_schema");
+        }
+        this.captured = undefined;
+        const holder = this.schemaHolder;
+        for (const key of Object.keys(holder))
+            delete holder[key];
+        Object.assign(holder, schema);
+        try {
+            session.setActiveToolsByName([...this.baseActive, STRUCTURED_TOOL_NAME]);
+        }
+        catch (error) {
+            this.captured = undefined;
+            try {
+                session.setActiveToolsByName(this.baseActive);
+            }
+            catch (rollbackError) {
+                console.error("pi-acp structured-output arm rollback error:", rollbackError);
+            }
+            throw error;
+        }
+        return "Finish by calling __acp_structured_output with a value conforming to the requested output schema.";
+    }
+    disarm(session) {
+        session.setActiveToolsByName(this.baseActive);
+        this.captured = undefined;
+    }
+    takeJson() {
+        if (this.captured === undefined)
+            return undefined;
+        try {
+            return JSON.stringify(this.captured);
+        }
+        catch {
+            return undefined;
+        }
+    }
+}

package/dist/translate.d.ts

@@ -0,0 +1,20 @@
+import type { ContentBlock, SessionUpdate, ToolCallContent, ToolCallLocation, ToolKind } from "@agentclientprotocol/sdk";
+import type { AgentSessionEvent } from "@earendil-works/pi-coding-agent";
+interface PiResult {
+    content?: Array<{
+        type: "text";
+        text: string;
+    } | {
+        type: "image";
+        data: string;
+        mimeType: string;
+    }>;
+    details?: unknown;
+}
+export declare function mapKind(toolName: string): ToolKind;
+export declare function fileLocations(args: unknown): ToolCallLocation[] | undefined;
+export declare function contentItems(result: PiResult): ContentBlock[];
+export declare function toContent(result: PiResult): ToolCallContent[];
+export declare function translateEvent(event: AgentSessionEvent, failedResult?: PiResult): SessionUpdate[];
+export {};
+//# sourceMappingURL=translate.d.ts.map

package/dist/translate.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"translate.d.ts","sourceRoot":"","sources":["../src/translate.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,YAAY,EACZ,aAAa,EACb,eAAe,EACf,gBAAgB,EAChB,QAAQ,EACT,MAAM,0BAA0B,CAAC;AAClC,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAC;AAEzE,UAAU,QAAQ;IAChB,OAAO,CAAC,EAAE,KAAK,CACX;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,GAC9B;QAAE,IAAI,EAAE,OAAO,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CACpD,CAAC;IACF,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED,wBAAgB,OAAO,CAAC,QAAQ,EAAE,MAAM,GAAG,QAAQ,CAgBlD;AAED,wBAAgB,aAAa,CAAC,IAAI,EAAE,OAAO,GAAG,gBAAgB,EAAE,GAAG,SAAS,CAI3E;AAED,wBAAgB,YAAY,CAAC,MAAM,EAAE,QAAQ,GAAG,YAAY,EAAE,CAM7D;AAED,wBAAgB,SAAS,CAAC,MAAM,EAAE,QAAQ,GAAG,eAAe,EAAE,CAE7D;AA4BD,wBAAgB,cAAc,CAAC,KAAK,EAAE,iBAAiB,EAAE,YAAY,CAAC,EAAE,QAAQ,GAAG,aAAa,EAAE,CAsDjG"}

package/dist/translate.js

@@ -0,0 +1,110 @@
+export function mapKind(toolName) {
+    switch (toolName) {
+        case "read":
+        case "ls":
+            return "read";
+        case "edit":
+        case "write":
+            return "edit";
+        case "bash":
+            return "execute";
+        case "grep":
+        case "find":
+            return "search";
+        default:
+            return "other";
+    }
+}
+export function fileLocations(args) {
+    if (typeof args !== "object" || args === null)
+        return undefined;
+    const path = args.path;
+    return typeof path === "string" ? [{ path }] : undefined;
+}
+export function contentItems(result) {
+    return (result.content ?? []).map((item) => item.type === "text"
+        ? { type: "text", text: item.text }
+        : { type: "image", data: item.data, mimeType: item.mimeType });
+}
+export function toContent(result) {
+    return contentItems(result).map((content) => ({ type: "content", content }));
+}
+function translateAssistantEvent(event) {
+    switch (event.type) {
+        case "text_delta":
+            return [{ sessionUpdate: "agent_message_chunk", content: { type: "text", text: event.delta } }];
+        case "thinking_delta":
+            return [{ sessionUpdate: "agent_thought_chunk", content: { type: "text", text: event.delta } }];
+        case "start":
+        case "text_start":
+        case "text_end":
+        case "thinking_start":
+        case "thinking_end":
+        case "toolcall_start":
+        case "toolcall_delta":
+        case "toolcall_end":
+        case "done":
+        case "error":
+            return [];
+        default: {
+            const exhaustive = event;
+            return exhaustive;
+        }
+    }
+}
+export function translateEvent(event, failedResult) {
+    switch (event.type) {
+        case "message_update":
+            return translateAssistantEvent(event.assistantMessageEvent);
+        case "tool_execution_start":
+            return [{
+                    sessionUpdate: "tool_call",
+                    toolCallId: event.toolCallId,
+                    title: event.toolName,
+                    kind: mapKind(event.toolName),
+                    status: "pending",
+                    rawInput: event.args,
+                    locations: fileLocations(event.args),
+                    _meta: { toolName: event.toolName },
+                }];
+        case "tool_execution_update":
+            return [{
+                    sessionUpdate: "tool_call_update",
+                    toolCallId: event.toolCallId,
+                    status: "in_progress",
+                    content: toContent(event.partialResult),
+                }];
+        case "tool_execution_end": {
+            const result = failedResult ?? event.result;
+            const update = {
+                sessionUpdate: "tool_call_update",
+                toolCallId: event.toolCallId,
+                status: event.isError ? "failed" : "completed",
+                content: toContent(result),
+            };
+            if (result.details !== undefined)
+                update.rawOutput = result.details;
+            return [update];
+        }
+        case "agent_start":
+        case "agent_end":
+        case "turn_start":
+        case "turn_end":
+        case "message_start":
+        case "message_end":
+        case "agent_settled":
+        case "queue_update":
+        case "compaction_start":
+        case "compaction_end":
+        case "entry_appended":
+        case "session_info_changed":
+        case "thinking_level_changed":
+        case "auto_retry_start":
+        case "auto_retry_end":
+            return [];
+        default: {
+            const exhaustive = event;
+            return exhaustive;
+        }
+    }
+}

package/dist/usage.d.ts

@@ -0,0 +1,35 @@
+import type { PromptResponse, SessionUpdate, Usage } from "@agentclientprotocol/sdk";
+import type { AgentSession } from "@earendil-works/pi-coding-agent";
+interface PiUsage {
+    input: number;
+    output: number;
+    cacheRead: number;
+    cacheWrite: number;
+    totalTokens: number;
+    reasoning?: number;
+}
+export interface AssistantLike {
+    role: "assistant";
+    usage: PiUsage;
+    stopReason: string;
+    errorMessage?: string;
+    diagnostics?: Array<{
+        type: string;
+        timestamp: number;
+        error?: {
+            name?: string;
+            message: string;
+            stack?: string;
+            code?: string | number;
+        };
+        details?: unknown;
+    }>;
+}
+export declare function agentMessages(session: AgentSession): unknown[];
+export declare function assistantMessages(messages: readonly unknown[]): AssistantLike[];
+export declare function terminalAssistant(messages: readonly unknown[]): AssistantLike | undefined;
+export declare function promptUsage(messages: readonly unknown[]): Usage;
+export declare function usageUpdate(session: AgentSession): SessionUpdate;
+export declare function response(stopReason: PromptResponse["stopReason"], messages: readonly unknown[]): PromptResponse;
+export {};
+//# sourceMappingURL=usage.d.ts.map

package/dist/usage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage.d.ts","sourceRoot":"","sources":["../src/usage.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,aAAa,EAAE,KAAK,EAAE,MAAM,0BAA0B,CAAC;AACrF,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AAEpE,UAAU,OAAO;IACf,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,WAAW,CAAC;IAClB,KAAK,EAAE,OAAO,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE,KAAK,CAAC;QAClB,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,EAAE,MAAM,CAAC;QAClB,KAAK,CAAC,EAAE;YAAE,IAAI,CAAC,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,MAAM,CAAC;YAAC,KAAK,CAAC,EAAE,MAAM,CAAC;YAAC,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,CAAA;SAAE,CAAC;QACnF,OAAO,CAAC,EAAE,OAAO,CAAC;KACnB,CAAC,CAAC;CACJ;AAED,wBAAgB,aAAa,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,EAAE,CAE9D;AAED,wBAAgB,iBAAiB,CAAC,QAAQ,EAAE,SAAS,OAAO,EAAE,GAAG,aAAa,EAAE,CAK/E;AAED,wBAAgB,iBAAiB,CAAC,QAAQ,EAAE,SAAS,OAAO,EAAE,GAAG,aAAa,GAAG,SAAS,CAEzF;AAED,wBAAgB,WAAW,CAAC,QAAQ,EAAE,SAAS,OAAO,EAAE,GAAG,KAAK,CAe/D;AAED,wBAAgB,WAAW,CAAC,OAAO,EAAE,YAAY,GAAG,aAAa,CAQhE;AAED,wBAAgB,QAAQ,CAAC,UAAU,EAAE,cAAc,CAAC,YAAY,CAAC,EAAE,QAAQ,EAAE,SAAS,OAAO,EAAE,GAAG,cAAc,CAE/G"}

package/dist/usage.js

@@ -0,0 +1,36 @@
+export function agentMessages(session) {
+    return session.agent.state.messages;
+}
+export function assistantMessages(messages) {
+    return messages.filter((message) => typeof message === "object" && message !== null && message.role === "assistant");
+}
+export function terminalAssistant(messages) {
+    return assistantMessages(messages).at(-1);
+}
+export function promptUsage(messages) {
+    const assistants = assistantMessages(messages);
+    const sum = (key) => assistants.reduce((total, message) => total + (message.usage[key] ?? 0), 0);
+    const usage = {
+        inputTokens: sum("input"),
+        outputTokens: sum("output"),
+        cachedReadTokens: sum("cacheRead"),
+        cachedWriteTokens: sum("cacheWrite"),
+        totalTokens: sum("totalTokens"),
+    };
+    if (assistants.some((message) => message.usage.reasoning !== undefined)) {
+        usage.thoughtTokens = sum("reasoning");
+    }
+    return usage;
+}
+export function usageUpdate(session) {
+    const context = session.getContextUsage();
+    return {
+        sessionUpdate: "usage_update",
+        used: context?.tokens ?? 0,
+        size: context?.contextWindow ?? session.model?.contextWindow ?? 0,
+        cost: { amount: session.getSessionStats().cost, currency: "USD" },
+    };
+}
+export function response(stopReason, messages) {
+    return { stopReason, usage: promptUsage(messages) };
+}

package/package.json

@@ -0,0 +1,47 @@
+{
+  "name": "@automatalabs/pi-acp",
+  "version": "0.1.0",
+  "license": "Apache-2.0",
+  "engines": {
+    "node": ">=22.19.0"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/VikashLoomba/agentprism-workflows.git",
+    "directory": "packages/pi-acp"
+  },
+  "type": "module",
+  "bin": {
+    "pi-acp": "./dist/index.js"
+  },
+  "main": "./dist/lib.js",
+  "types": "./dist/lib.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/lib.d.ts",
+      "import": "./dist/lib.js",
+      "default": "./dist/lib.js"
+    }
+  },
+  "files": [
+    "dist"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "dependencies": {
+    "@agentclientprotocol/sdk": "1.2.1",
+    "@earendil-works/pi-coding-agent": "0.80.8",
+    "@modelcontextprotocol/sdk": "1.29.0",
+    "typebox": "1.3.2"
+  },
+  "devDependencies": {
+    "@earendil-works/pi-agent-core": "0.80.8",
+    "@earendil-works/pi-ai": "0.80.8"
+  },
+  "scripts": {
+    "build": "tsc -b",
+    "typecheck": "tsc --noEmit",
+    "test": "tsc -p tsconfig.type-tests.json && tsx --test \"test/**/*.test.ts\""
+  }
+}