stagent 0.3.5 → 0.3.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "stagent",
-  "version": "0.3.5",
+  "version": "0.3.6",
   "description": "Governed AI agent workspace for supervised local execution, workflows, documents, and provider runtimes.",
   "keywords": [
     "ai",

package/src/app/api/settings/runtime/route.ts

@@ -0,0 +1,25 @@
+import { NextRequest, NextResponse } from "next/server";
+import { getSetting, setSetting } from "@/lib/settings/helpers";
+import { SETTINGS_KEYS } from "@/lib/constants/settings";
+
+export async function GET() {
+  const sdkTimeoutSeconds = await getSetting(SETTINGS_KEYS.SDK_TIMEOUT_SECONDS);
+  return NextResponse.json({
+    sdkTimeoutSeconds: sdkTimeoutSeconds ?? "60",
+  });
+}
+
+export async function POST(req: NextRequest) {
+  const body = await req.json();
+  const seconds = parseInt(body.sdkTimeoutSeconds, 10);
+
+  if (isNaN(seconds) || seconds < 10 || seconds > 300) {
+    return NextResponse.json(
+      { error: "sdkTimeoutSeconds must be between 10 and 300" },
+      { status: 400 }
+    );
+  }
+
+  await setSetting(SETTINGS_KEYS.SDK_TIMEOUT_SECONDS, String(seconds));
+  return NextResponse.json({ sdkTimeoutSeconds: String(seconds) });
+}

package/src/app/settings/page.tsx

@@ -4,6 +4,7 @@ import { PermissionsSections } from "@/components/settings/permissions-sections"
 import { DataManagementSection } from "@/components/settings/data-management-section";
 import { BudgetGuardrailsSection } from "@/components/settings/budget-guardrails-section";
 import { ChatSettingsSection } from "@/components/settings/chat-settings-section";
+import { RuntimeTimeoutSection } from "@/components/settings/runtime-timeout-section";
 import { PageShell } from "@/components/shared/page-shell";
 
 export const dynamic = "force-dynamic";
@@ -15,6 +16,7 @@ export default function SettingsPage() {
         <AuthConfigSection />
         <OpenAIRuntimeSection />
         <ChatSettingsSection />
+        <RuntimeTimeoutSection />
         <BudgetGuardrailsSection />
         <PermissionsSections />
         <DataManagementSection />

package/src/components/settings/runtime-timeout-section.tsx

@@ -0,0 +1,90 @@
+"use client";
+
+import { useEffect, useState, useCallback } from "react";
+import { Timer } from "lucide-react";
+import { toast } from "sonner";
+import {
+  Card,
+  CardContent,
+  CardDescription,
+  CardHeader,
+  CardTitle,
+} from "@/components/ui/card";
+import { Input } from "@/components/ui/input";
+import { FormSectionCard } from "@/components/shared/form-section-card";
+
+const SETTING_KEY = "runtime.sdkTimeoutSeconds";
+const DEFAULT_TIMEOUT = 60;
+
+export function RuntimeTimeoutSection() {
+  const [timeout, setTimeout_] = useState(String(DEFAULT_TIMEOUT));
+  const [saving, setSaving] = useState(false);
+
+  const fetchTimeout = useCallback(async () => {
+    try {
+      const res = await fetch("/api/settings/runtime");
+      if (res.ok) {
+        const data = await res.json();
+        if (data.sdkTimeoutSeconds) setTimeout_(data.sdkTimeoutSeconds);
+      }
+    } catch {
+      // Use default
+    }
+  }, []);
+
+  useEffect(() => {
+    fetchTimeout();
+  }, [fetchTimeout]);
+
+  const handleSave = async (value: string) => {
+    const num = parseInt(value, 10);
+    if (isNaN(num) || num < 10 || num > 300) {
+      toast.error("Timeout must be between 10 and 300 seconds");
+      return;
+    }
+
+    setSaving(true);
+    try {
+      await fetch("/api/settings/runtime", {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({ sdkTimeoutSeconds: String(num) }),
+      });
+      setTimeout_(String(num));
+      toast.success("Timeout updated");
+    } catch {
+      toast.error("Failed to save timeout");
+    } finally {
+      setSaving(false);
+    }
+  };
+
+  return (
+    <Card>
+      <CardHeader>
+        <CardTitle>Runtime</CardTitle>
+        <CardDescription>
+          Configure runtime behavior for AI operations.
+        </CardDescription>
+      </CardHeader>
+      <CardContent>
+        <FormSectionCard
+          icon={Timer}
+          title="SDK Timeout"
+          hint="Maximum seconds to wait for AI responses (task assist, completions). Min 10, max 300."
+        >
+          <Input
+            type="number"
+            min={10}
+            max={300}
+            value={timeout}
+            onChange={(e) => setTimeout_(e.target.value)}
+            onBlur={(e) => handleSave(e.target.value)}
+            disabled={saving}
+            className="w-32"
+          />
+        </FormSectionCard>
+      </CardContent>
+    </Card>
+  );
+}

package/src/lib/agents/runtime/claude.ts

@@ -10,6 +10,8 @@ import { executeClaudeTask, resumeClaudeTask } from "@/lib/agents/claude-agent";
 import { getRuntimeCapabilities, getRuntimeCatalogEntry } from "./catalog";
 import { buildClaudeSdkEnv } from "./claude-sdk";
 import { getLaunchCwd } from "@/lib/environment/workspace-context";
+import { getSetting } from "@/lib/settings/helpers";
+import { SETTINGS_KEYS } from "@/lib/constants/settings";
 import type {
   AgentRuntimeAdapter,
   RuntimeConnectionResult,
@@ -89,6 +91,21 @@ async function collectResultText(
   return { resultText, usage };
 }
 
+/** Read the user-configurable SDK timeout (in ms). Falls back to 60s. */
+async function getSdkTimeout(): Promise<number> {
+  const raw = await getSetting(SETTINGS_KEYS.SDK_TIMEOUT_SECONDS);
+  const seconds = raw ? parseInt(raw, 10) : 60;
+  return (isNaN(seconds) || seconds < 10 ? 60 : seconds) * 1000;
+}
+
+/** Check if an error is an abort/timeout error from the SDK. */
+function isAbortError(error: unknown): boolean {
+  return (
+    error instanceof Error &&
+    (error.name === "AbortError" || error.message.includes("aborted"))
+  );
+}
+
 async function runSingleProfileTest(
   profileId: string,
   test: { task: string; expectedKeywords: string[] }
@@ -106,7 +123,8 @@ async function runSingleProfileTest(
   const prompt = `${payload.instructions}\n\n---\n\nTask: ${test.task}\n\nProvide a brief analysis (2-3 paragraphs max). Include specific terminology relevant to your domain.`;
   const authEnv = await getAuthEnv();
   const abortController = new AbortController();
-  const timeout = setTimeout(() => abortController.abort(), 30_000);
+  const sdkTimeoutMs = await getSdkTimeout();
+  const timeout = setTimeout(() => abortController.abort(), sdkTimeoutMs);
   const startedAt = new Date();
   let usage: UsageSnapshot = {};
   let ledgerRecorded = false;
@@ -265,7 +283,8 @@ export async function runMetaCompletion(input: {
   const startedAt = new Date();
   let usage: UsageSnapshot = {};
   const abortController = new AbortController();
-  const timeout = setTimeout(() => abortController.abort(), 60_000);
+  const sdkTimeoutMs = await getSdkTimeout();
+  const timeout = setTimeout(() => abortController.abort(), sdkTimeoutMs);
 
   try {
     const response = query({
@@ -312,6 +331,9 @@ export async function runMetaCompletion(input: {
       startedAt,
       finishedAt: new Date(),
     });
+    if (isAbortError(error)) {
+      throw new Error("Request timed out. You can increase the timeout in Settings → Runtime.");
+    }
     throw error;
   } finally {
     clearTimeout(timeout);
@@ -336,7 +358,8 @@ async function runClaudeTaskAssist(
   let usage: UsageSnapshot = {};
 
   const abortController = new AbortController();
-  const timeout = setTimeout(() => abortController.abort(), 30_000);
+  const sdkTimeoutMs = await getSdkTimeout();
+  const timeout = setTimeout(() => abortController.abort(), sdkTimeoutMs);
 
   try {
     const response = query({
@@ -394,6 +417,9 @@ async function runClaudeTaskAssist(
       startedAt,
       finishedAt: new Date(),
     });
+    if (isAbortError(error)) {
+      throw new Error("Request timed out. You can increase the timeout in Settings → Runtime.");
+    }
     throw error;
   } finally {
     clearTimeout(timeout);

package/src/lib/chat/tools/settings-tools.ts

@@ -28,6 +28,7 @@ export function settingsTools(_ctx: ToolContext) {
           const keys = [
             "auth_method",
             "default_runtime",
+            "runtime.sdkTimeoutSeconds",
             "budget_max_tokens_per_task",
             "budget_max_cost_per_task",
             "budget_max_daily_cost",

package/src/lib/constants/settings.ts

@@ -8,6 +8,7 @@ export const SETTINGS_KEYS = {
   BUDGET_POLICY: "usage.budgetPolicy",
   BUDGET_WARNING_STATE: "usage.budgetWarningState",
   PRICING_REGISTRY: "usage.pricingRegistry",
+  SDK_TIMEOUT_SECONDS: "runtime.sdkTimeoutSeconds",
 } as const;
 
 export type AuthMethod = "api_key" | "oauth";