fix: Fix tool-bridge slot leak when session.abort() throws inside runAgentTurn (#73)

[KochC]

Summary

Wrapped the entire post-bridge-acquisition body of runAgentTurn() in an outer try/finally that calls releaseToolBridge(bridge), so the pool slot is always returned even if session.create(), event.subscribe(), or session.promptAsync() throw before the event-loop is entered.

Closes #73

Diff

diff --git a/index.js b/index.js
index d55980c..65d511c 100644
--- a/index.js
+++ b/index.js
@@ -752,59 +752,62 @@ async function runAgentTurn(client, model, messages, system, callerTools, onChun
     toolsMap = buildToolsMap(baseTools, bridge)
   }
 
-  const session = await client.session.create({ body: { title: `Proxy: ${model.id}` } })
-  const sessionID = session.data.id
-  const prompt = buildPrompt(messages)
-  const toolIDSet = bridge ? new Set(bridge.toolIDs) : null
+  // Wrap all work that happens after bridge acquisition in a try/finally so the
+  // slot is always returned to the pool even if session.create, event.subscribe,
+  // or session.promptAsync throw before the event-loop runs.
+  try {
+    const session = await client.session.create({ body: { title: `Proxy: ${model.id}` } })
+    const sessionID = session.data.id
+    const prompt = buildPrompt(messages)
+    const toolIDSet = bridge ? new Set(bridge.toolIDs) : null
 
-  // Subscribe to the event stream before sending the prompt so we don't miss events.
-  const { stream } = await client.event.subscribe()
+    // Subscribe to the event stream before sending the prompt so we don't miss events.
+    const { stream } = await client.event.subscribe()
 
-  await client.session.promptAsync({
-    path: { id: sessionID },
-    body: {
-      model: { providerID: model.providerID, modelID: model.modelID },
-      system,
-      tools: toolsMap,
-      parts: [{ type: "text", text: prompt }],
-    },
-  })
+    await client.session.promptAsync({
+      path: { id: sessionID },
+      body: {
+        model: { providerID: model.providerID, modelID: model.modelID },
+        system,
+        tools: toolsMap,
+        parts: [{ type: "text", text: prompt }],
+      },
+    })
 
-  let errorMessage = null
-  let content = ""
-  const toolCallsByID = new Map()
-  let toolMessageID = null
+    let errorMessage = null
+    let content = ""
+    const toolCallsByID = new Map()
+    let toolMessageID = null
 
-  const recordToolPart = (part) => { ... }
+    const recordToolPart = (part) => { ... }
 
-  try {
     for await (const event of stream) { ... }
-  } finally {
-    releaseToolBridge(bridge)
-  }
 
-  // ... build result and return
+    // ... build result and return
+    return { sessionID, content, toolCalls, tokens, finish }
+  } finally {
+    releaseToolBridge(bridge)
   }
 }

---

Automated fix by OpenCode Agent