@townco/agent 0.1.19 → 0.1.21

package/dist/acp-server/adapter.js

@@ -1,77 +1,81 @@
 import * as acp from "@agentclientprotocol/sdk";
 /** Adapts an Agent to speak the ACP protocol */
 export class AgentAcpAdapter {
-	connection;
-	sessions;
-	agent;
-	constructor(agent, connection) {
-		this.connection = connection;
-		this.sessions = new Map();
-		this.agent = agent;
-	}
-	async initialize(_params) {
-		return {
-			protocolVersion: acp.PROTOCOL_VERSION,
-			agentCapabilities: {
-				loadSession: false,
-			},
-		};
-	}
-	async newSession(_params) {
-		const sessionId = Math.random().toString(36).substring(2);
-		this.sessions.set(sessionId, {
-			pendingPrompt: null,
-			messages: [],
-		});
-		return {
-			sessionId,
-		};
-	}
-	async authenticate(_params) {
-		// No auth needed - return empty response
-		return {};
-	}
-	async setSessionMode(_params) {
-		// Session mode changes are no-op for us (not related to coding)
-		return {};
-	}
-	async prompt(params) {
-		let session = this.sessions.get(params.sessionId);
-		// If session not found (e.g., after server restart), create a new one
-		if (!session) {
-			console.log(
-				`Session ${params.sessionId} not found, creating new session`,
-			);
-			session = {
-				pendingPrompt: null,
-				messages: [],
-			};
-			this.sessions.set(params.sessionId, session);
-		}
-		session.pendingPrompt?.abort();
-		session.pendingPrompt = new AbortController();
-		try {
-			for await (const msg of this.agent.invoke({
-				prompt: params.prompt,
-				sessionId: params.sessionId,
-			})) {
-				this.connection.sessionUpdate({
-					sessionId: params.sessionId,
-					update: msg,
-				});
-			}
-		} catch (err) {
-			if (session.pendingPrompt.signal.aborted) {
-				return { stopReason: "cancelled" };
-			}
-			throw err;
-		}
-		session.pendingPrompt = null;
-		return {
-			stopReason: "end_turn",
-		};
-	}
-	async cancel(params) {
-		this.sessions.get(params.sessionId)?.pendingPrompt?.abort();
-	}
+    connection;
+    sessions;
+    agent;
+    constructor(agent, connection) {
+        this.connection = connection;
+        this.sessions = new Map();
+        this.agent = agent;
+    }
+    async initialize(_params) {
+        return {
+            protocolVersion: acp.PROTOCOL_VERSION,
+            agentCapabilities: {
+                loadSession: false,
+            },
+        };
+    }
+    async newSession(_params) {
+        const sessionId = Math.random().toString(36).substring(2);
+        this.sessions.set(sessionId, {
+            pendingPrompt: null,
+            messages: [],
+        });
+        return {
+            sessionId,
+        };
+    }
+    async authenticate(_params) {
+        // No auth needed - return empty response
+        return {};
+    }
+    async setSessionMode(_params) {
+        // Session mode changes are no-op for us (not related to coding)
+        return {};
+    }
+    async prompt(params) {
+        let session = this.sessions.get(params.sessionId);
+        // If session not found (e.g., after server restart), create a new one
+        if (!session) {
+            console.log(`Session ${params.sessionId} not found, creating new session`);
+            session = {
+                pendingPrompt: null,
+                messages: [],
+            };
+            this.sessions.set(params.sessionId, session);
+        }
+        session.pendingPrompt?.abort();
+        session.pendingPrompt = new AbortController();
+        // Generate a unique messageId for this assistant response
+        const messageId = Math.random().toString(36).substring(2);
+        try {
+            for await (const msg of this.agent.invoke({
+                prompt: params.prompt,
+                sessionId: params.sessionId,
+                messageId,
+            })) {
+                // The agent may emit extended types (like tool_output) that aren't in ACP SDK yet
+                // The http transport will handle routing these appropriately
+                this.connection.sessionUpdate({
+                    sessionId: params.sessionId,
+                    update: msg,
+                });
+            }
+        }
+        catch (err) {
+            if (session.pendingPrompt.signal.aborted) {
+                return { stopReason: "cancelled" };
+            }
+            throw err;
+        }
+        session.pendingPrompt = null;
+        return {
+            stopReason: "end_turn",
+        };
+    }
+    async cancel(params) {
+        this.sessions.get(params.sessionId)?.pendingPrompt?.abort();
+    }
 }