blun-king-cli 9.1.472 → 9.1.473

package/bin/agent-api-http-adapter.mjs

@@ -271,6 +271,12 @@ function createAgentApiHttpAdapter(options = {}) {
     if (scopeOwners?.size === 0) owners.delete(key);
   }
 
+  function trackRunningScope(key, principalId, internalSessionId) {
+    const scopeOwners = owners.get(key) ?? new Map();
+    scopeOwners.set(principalId, internalSessionId);
+    owners.set(key, scopeOwners);
+  }
+
   async function handleStart(input, principalId) {
     const body = requireBody(input.body);
     requireKing(body.model);
@@ -295,20 +301,24 @@ function createAgentApiHttpAdapter(options = {}) {
       );
     }
     const internalSessionId = session.id.trim();
-    const response = publicCompletion(await controller.startTurn({
-      session,
-      publicSessionId: sessionId,
-      requestId,
-      prompt: normalizePrompt(body.input),
-      tools: normalizeTools(body.tools),
-    }));
-    trackScope(
-      scopeKey(sessionId, requestId),
-      principalId,
-      internalSessionId,
-      response,
-    );
-    return jsonResponse(200, response);
+    const prompt = normalizePrompt(body.input);
+    const tools = normalizeTools(body.tools);
+    const key = scopeKey(sessionId, requestId);
+    trackRunningScope(key, principalId, internalSessionId);
+    try {
+      const response = publicCompletion(await controller.startTurn({
+        session,
+        publicSessionId: sessionId,
+        requestId,
+        prompt,
+        tools,
+      }));
+      trackScope(key, principalId, internalSessionId, response);
+      return jsonResponse(200, response);
+    } catch (error) {
+      trackScope(key, principalId, internalSessionId, null);
+      throw error;
+    }
   }
 
   async function handleContinue(input, principalId, requestId) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "blun-king-cli",
-  "version": "9.1.472",
+  "version": "9.1.473",
   "description": "BLUN CLI - your own AI agent with a Telegram channel. Get it done. With BLUN.",
   "license": "MIT",
   "bin": {