@ottocode/server 0.1.192 → 0.1.193

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@ottocode/server",
-	"version": "0.1.192",
+	"version": "0.1.193",
 	"description": "HTTP API server for ottocode",
 	"type": "module",
 	"main": "./src/index.ts",
@@ -29,8 +29,8 @@
 		"typecheck": "tsc --noEmit"
 	},
 	"dependencies": {
-		"@ottocode/sdk": "0.1.192",
-		"@ottocode/database": "0.1.192",
+		"@ottocode/sdk": "0.1.193",
+		"@ottocode/database": "0.1.193",
 		"drizzle-orm": "^0.44.5",
 		"hono": "^4.9.9",
 		"zod": "^4.1.8"

package/src/routes/auth.ts

@@ -161,6 +161,27 @@ export function registerAuthRoutes(app: Hono) {
 		}
 	});
 
+	app.get('/v1/auth/setu/export', async (c) => {
+		try {
+			const projectRoot = process.cwd();
+			const wallet = await getSetuWallet(projectRoot);
+
+			if (!wallet) {
+				return c.json({ error: 'Setu wallet not configured' }, 404);
+			}
+
+			return c.json({
+				success: true,
+				publicKey: wallet.publicKey,
+				privateKey: wallet.privateKey,
+			});
+		} catch (error) {
+			logger.error('Failed to export Setu wallet', error);
+			const errorResponse = serializeError(error);
+			return c.json(errorResponse, errorResponse.error.status || 500);
+		}
+	});
+
 	app.post('/v1/auth/:provider', async (c) => {
 		try {
 			const provider = c.req.param('provider') as ProviderId;

package/src/runtime/agent/runner.ts

@@ -1,11 +1,12 @@
 import { hasToolCall, streamText } from 'ai';
-import { messageParts } from '@ottocode/database/schema';
+import { messages, messageParts } from '@ottocode/database/schema';
 import { eq } from 'drizzle-orm';
 import { publish, subscribe } from '../../events/bus.ts';
 import { debugLog, time } from '../debug/index.ts';
 import { toErrorPayload } from '../errors/handling.ts';
 import {
 	type RunOpts,
+	enqueueAssistantRun,
 	setRunning,
 	dequeueJob,
 	cleanupSession,
@@ -293,13 +294,71 @@ async function runAssistant(opts: RunOpts) {
 		await cleanupEmptyTextParts(opts, db);
 		firstToolTimer.end({ seen: firstToolSeen() });
 
+		let streamFinishReason: string | undefined;
+		try {
+			streamFinishReason = await result.finishReason;
+		} catch {
+			streamFinishReason = undefined;
+		}
+
 		debugLog(
-			`[RUNNER] Stream finished. finishSeen=${_finishObserved}, firstToolSeen=${fs}`,
+			`[RUNNER] Stream finished. finishSeen=${_finishObserved}, firstToolSeen=${fs}, finishReason=${streamFinishReason}`,
 		);
 
-		if (!_finishObserved && fs) {
+		const wasTruncated = streamFinishReason === 'length';
+
+		const shouldContinue = !_finishObserved && (wasTruncated || fs);
+
+		if (shouldContinue) {
+			debugLog(
+				`[RUNNER] WARNING: Stream ended without finish. finishReason=${streamFinishReason}, firstToolSeen=${fs}. Auto-continuing.`,
+			);
+
+			const MAX_CONTINUATIONS = 10;
+			const count = opts.continuationCount ?? 0;
+			if (count < MAX_CONTINUATIONS) {
+				debugLog(
+					`[RUNNER] Auto-continuing (${count + 1}/${MAX_CONTINUATIONS})...`,
+				);
+
+				try {
+					await completeAssistantMessage({}, opts, db);
+				} catch (err) {
+					debugLog(
+						`[RUNNER] completeAssistantMessage failed before continuation: ${err instanceof Error ? err.message : String(err)}`,
+					);
+				}
+
+				const continuationMessageId = crypto.randomUUID();
+				await db.insert(messages).values({
+					id: continuationMessageId,
+					sessionId: opts.sessionId,
+					role: 'assistant',
+					status: 'pending',
+					agent: opts.agent,
+					provider: opts.provider,
+					model: opts.model,
+					createdAt: Date.now(),
+				});
+
+				publish({
+					type: 'message.created',
+					sessionId: opts.sessionId,
+					payload: { id: continuationMessageId, role: 'assistant' },
+				});
+
+				enqueueAssistantRun(
+					{
+						...opts,
+						assistantMessageId: continuationMessageId,
+						continuationCount: count + 1,
+					},
+					runSessionLoop,
+				);
+				return;
+			}
 			debugLog(
-				`[RUNNER] WARNING: Stream ended without finish tool being called. Model was mid-execution (tools were used). This is likely an unclean stream termination from the provider.`,
+				`[RUNNER] Max continuations (${MAX_CONTINUATIONS}) reached, stopping.`,
 			);
 		}
 	} catch (err) {

package/src/runtime/session/queue.ts

@@ -17,6 +17,7 @@ export type RunOpts = {
 	compactionContext?: string;
 	toolApprovalMode?: ToolApprovalMode;
 	compactionRetries?: number;
+	continuationCount?: number;
 };
 
 export type QueuedMessage = {