@pasko70/pibo 1.8.0 → 1.8.2

package/dist/data/telemetry.js

@@ -8,6 +8,20 @@ export class TelemetryStore {
     constructor(db) {
         this.db = db;
     }
+    transaction(action) {
+        if (this.db.isTransaction)
+            return action();
+        this.db.exec("BEGIN IMMEDIATE");
+        try {
+            const result = action();
+            this.db.exec("COMMIT");
+            return result;
+        }
+        catch (error) {
+            this.db.exec("ROLLBACK");
+            throw error;
+        }
+    }
     listSessions(input = {}) {
         return listTelemetrySessions(this.db, input);
     }

package/dist/gateway/request.js

@@ -75,6 +75,7 @@ export async function sendGatewayMessageAndWaitForReply(event, options = {}) {
         let settled = false;
         let response;
         let reply;
+        let messageFinished = false;
         const timeout = setTimeout(() => {
             finish(new Error(`Timed out waiting for assistant reply from session "${event.piboSessionId}"`));
         }, timeoutMs);
@@ -91,14 +92,21 @@ export async function sendGatewayMessageAndWaitForReply(event, options = {}) {
                 resolve(result);
             }
         };
+        const finishCompletedReply = () => {
+            if (!response?.ok || !messageFinished)
+                return;
+            finish(reply
+                ? { response, reply }
+                : new Error(`Session "${eventWithId.piboSessionId}" finished without an assistant reply`));
+        };
         const handleFrame = (frame) => {
             if (frame.type === "res" && frame.id === id) {
                 response = frame;
                 if (!frame.ok) {
                     finish(new Error(frame.error?.message ?? "Gateway rejected the message"));
                 }
-                else if (reply) {
-                    finish({ response, reply });
+                else {
+                    finishCompletedReply();
                 }
                 return;
             }
@@ -111,13 +119,14 @@ export async function sendGatewayMessageAndWaitForReply(event, options = {}) {
                 finish(new Error(output.error));
                 return;
             }
-            if (output.type === "assistant_message" &&
-                output.piboSessionId === eventWithId.piboSessionId &&
-                output.eventId === eventWithId.id) {
+            if (output.piboSessionId !== eventWithId.piboSessionId || !("eventId" in output) || output.eventId !== eventWithId.id)
+                return;
+            if (output.type === "assistant_message") {
                 reply = output;
-                if (response?.ok) {
-                    finish({ response, reply });
-                }
+            }
+            else if (output.type === "message_finished") {
+                messageFinished = true;
+                finishCompletedReply();
             }
         };
         socket.once("connect", () => {

package/dist/ralph/service.js

@@ -252,7 +252,7 @@ export class PiboRalphService {
             },
         });
         this.store.attachRunSession(job.id, run.id, session.id);
-        const finalAnswer = await this.emitMessageAndWait(session.id, buildRalphPrompt(job), { isCancelled: () => this.cancelledRuns.has(run.id) });
+        const finalAnswer = await this.emitMessageAndWait(session.id, buildRalphPrompt(job));
         return { piboSessionId: session.id, finalAnswer };
     }
     resolveTarget(job) { if (job.target.kind === 'room') {
@@ -263,7 +263,7 @@ export class PiboRalphService {
             throw new Error('Target room is archived');
         return { roomId: room.id, workspace: room.workspace ?? getDefaultPiboWorkspace() };
     } const room = this.roomService.ensureDefaultRoom({ name: 'Shared Chat' }); return { roomId: room.id, workspace: room.workspace ?? getDefaultPiboWorkspace() }; }
-    async emitMessageAndWait(piboSessionId, text, options = {}) {
+    async emitMessageAndWait(piboSessionId, text) {
         const eventId = `ralph_msg_${randomUUID()}`;
         return await new Promise((resolve, reject) => {
             let settled = false;
@@ -308,9 +308,7 @@ export class PiboRalphService {
                     finish(lastSessionError ? new Error(lastSessionError) : undefined);
                 if (event.type === 'session_error') {
                     lastSessionError = event.error;
-                    const providerAttempt = event.errorDetails?.origin === 'provider' && Boolean(event.errorDetails.api || event.errorDetails.provider || event.errorDetails.model);
-                    if (!providerAttempt || options.isCancelled?.())
-                        finish(new Error(event.error));
+                    finish(new Error(event.error));
                 }
             });
             this.options.context.emit({ type: 'message', piboSessionId, id: eventId, source: 'service', text }).catch((error) => finish(error instanceof Error ? error : new Error(String(error))));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pasko70/pibo",
-  "version": "1.8.0",
+  "version": "1.8.2",
   "type": "module",
   "imports": {
     "vscode": "./src/apps/chat-vscode/extension/src/vscode-shim.js"