Add LLM task title generation and improve prompt timing

Author: charlesvien

apps/twig/src/main/services/agent/service.ts

@@ -28,7 +28,6 @@ import { logger } from "../../lib/logger.js";
 import { createTimingCollector } from "../../lib/timing.js";
 import { TypedEventEmitter } from "../../lib/typed-event-emitter.js";
 import type { FsService } from "../fs/service.js";
-import { getCurrentUserId, getPostHogClient } from "../posthog-analytics.js";
 import type { ProcessTrackingService } from "../process-tracking/service.js";
 import type { SleepService } from "../sleep/service.js";
 import {
@@ -468,32 +467,12 @@ export class AgentService extends TypedEventEmitter<AgentServiceEvents> {
     // Preview sessions don't persist logs — no real task exists
     const isPreview = taskId === "__preview__";
 
-    // OTEL log pipeline or legacy S3 writer if FF false
-    const useOtelPipeline = isPreview
-      ? false
-      : await tc.time("featureFlag", () =>
-          this.isFeatureFlagEnabled("twig-agent-logs-pipeline"),
-        );
-
-    log.info("Agent log transport", {
-      transport: isPreview ? "none" : useOtelPipeline ? "otel" : "s3",
-      taskId,
-      taskRunId,
-    });
-
     const agent = new Agent({
       posthog: {
         apiUrl: credentials.apiHost,
         getApiKey: () => this.getToken(credentials.apiKey),
         projectId: credentials.projectId,
       },
-      otelTransport: useOtelPipeline
-        ? {
-            host: credentials.apiHost,
-            apiKey: this.getToken(credentials.apiKey),
-            logsPath: "/i/v1/agent-logs",
-          }
-        : undefined,
       skipLogPersistence: isPreview,
       debug: !app.isPackaged,
       onLog: onAgentLog,
@@ -1011,20 +990,6 @@ For git operations while detached:
     return mockNodeDir;
   }
 
-  private async isFeatureFlagEnabled(flagKey: string): Promise<boolean> {
-    try {
-      const client = getPostHogClient();
-      const userId = getCurrentUserId();
-      if (!client || !userId) {
-        return false;
-      }
-      return (await client.isFeatureEnabled(flagKey, userId)) ?? false;
-    } catch (error) {
-      log.warn(`Error checking feature flag "${flagKey}":`, error);
-      return false;
-    }
-  }
-
   private cleanupMockNodeEnvironment(mockNodeDir: string): void {
     try {
       rmSync(mockNodeDir, { recursive: true, force: true });

apps/twig/src/renderer/features/sessions/service/service.ts

@@ -611,6 +611,11 @@ export class SessionService {
     }
 
     if (initialPrompt?.length) {
+      if (submittedAt) {
+        sessionStoreSetters.updateSession(taskRun.id, {
+          promptStartedAt: submittedAt,
+        });
+      }
       await tc.time("sendPrompt", () => this.sendPrompt(taskId, initialPrompt));
     }
 
@@ -816,9 +821,10 @@ export class SessionService {
     const msg = acpMsg.message;
 
     if (isJsonRpcRequest(msg) && msg.method === "session/prompt") {
+      // Preserve existing promptStartedAt if already set (e.g., submittedAt for initial prompts)
       sessionStoreSetters.updateSession(taskRunId, {
         isPromptPending: true,
-        promptStartedAt: acpMsg.ts,
+        promptStartedAt: session.promptStartedAt ?? acpMsg.ts,
       });
     }
 
@@ -1079,9 +1085,11 @@ export class SessionService {
     session: AgentSession,
     blocks: ContentBlock[],
   ): Promise<{ stopReason: string }> {
+    // Preserve existing promptStartedAt if set (e.g., from submittedAt for initial prompts)
+    const currentSession = sessionStoreSetters.getSessions()[session.taskRunId];
     sessionStoreSetters.updateSession(session.taskRunId, {
       isPromptPending: true,
-      promptStartedAt: Date.now(),
+      promptStartedAt: currentSession?.promptStartedAt ?? Date.now(),
     });
 
     try {

apps/twig/src/renderer/sagas/task/task-creation.ts

@@ -1,12 +1,15 @@
+import { useAuthStore } from "@features/auth/stores/authStore";
 import { buildPromptBlocks } from "@features/editor/utils/prompt-builder";
 import { getSessionService } from "@features/sessions/service/service";
 import { useWorkspaceStore } from "@features/workspace/stores/workspaceStore";
 import { Saga, type SagaLogger } from "@posthog/shared";
 import type { PostHogAPIClient } from "@renderer/api/posthogClient";
 import { logger } from "@renderer/lib/logger";
+import { queryClient } from "@renderer/lib/queryClient";
 import { useTaskDirectoryStore } from "@renderer/stores/taskDirectoryStore";
 import { trpcVanilla } from "@renderer/trpc";
 import { getTaskRepository } from "@renderer/utils/repository";
+import { getCloudUrlFromRegion } from "@shared/constants/oauth";
 import type {
   ExecutionMode,
   Task,
@@ -16,6 +19,81 @@ import type {
 
 const log = logger.scope("task-creation-saga");
 
+function truncateToTitle(content: string): string {
+  // Strip XML/HTML tags
+  const stripped = content.replace(/<[^>]*>/g, "").trim();
+  if (!stripped) return "Untitled";
+  if (stripped.length <= 80) return stripped;
+  // Truncate at word boundary
+  const truncated = stripped.slice(0, 80);
+  const lastSpace = truncated.lastIndexOf(" ");
+  return lastSpace > 20
+    ? `${truncated.slice(0, lastSpace)}...`
+    : `${truncated}...`;
+}
+
+const TITLE_SYSTEM_PROMPT = `You are a title generator. You output ONLY a task title. Nothing else.
+
+Convert the task description into a concise task title.
+- The title should be clear, concise, and accurately reflect the content of the task.
+- You should keep it short and simple, ideally no more than 6 words.
+- Avoid using jargon or overly technical terms unless absolutely necessary.
+- The title should be easy to understand for anyone reading it.
+- Use sentence case (capitalize only first word and proper nouns)
+-Remove: the, this, my, a, an
+- If possible, start with action verbs (Fix, Implement, Analyze, Debug, Update, Research, Review)
+- Keep exact: technical terms, numbers, filenames, HTTP codes, PR numbers
+- Never assume tech stack
+- Only output "Untitled" if the input is completely null/missing, not just unclear
+
+Examples:
+- "Fix the login bug in the authentication system" → Fix authentication login bug
+- "Schedule a meeting with stakeholders to discuss Q4 budget planning" → Schedule Q4 budget meeting
+- "Update user documentation for new API endpoints" → Update API documentation
+- "Research competitor pricing strategies for our product" → Research competitor pricing
+- "Review pull request #123" → Review pull request #123
+- "debug 500 errors in production" → Debug production 500 errors
+- "why is the payment flow failing" → Analyze payment flow failure
+- "So how about that weather huh" → "Weather chat"
+- "dsfkj sdkfj help me code" → "Coding help request"
+- "👋😊" → "Friendly greeting"
+- "aaaaaaaaaa" → "Repeated letters"
+- "   " → "Empty message"
+- "What's the best restaurant in NYC?" → "NYC restaurant recommendations"`;
+
+async function generateTaskTitle(
+  taskId: string,
+  description: string,
+  posthogClient: PostHogAPIClient,
+): Promise<void> {
+  try {
+    const authState = useAuthStore.getState();
+    const apiKey = authState.oauthAccessToken;
+    const cloudRegion = authState.cloudRegion;
+    if (!apiKey || !cloudRegion) return;
+
+    const apiHost = getCloudUrlFromRegion(cloudRegion);
+
+    const result = await trpcVanilla.llmGateway.prompt.mutate({
+      credentials: { apiKey, apiHost },
+      system: TITLE_SYSTEM_PROMPT,
+      messages: [{ role: "user", content: description }],
+    });
+
+    const title = result.content.trim();
+    if (!title) return;
+
+    await posthogClient.updateTask(taskId, { title });
+
+    // Update all cached task lists so the sidebar reflects the new title instantly
+    queryClient.setQueriesData<Task[]>({ queryKey: ["tasks", "list"] }, (old) =>
+      old?.map((task) => (task.id === taskId ? { ...task, title } : task)),
+    );
+  } catch (error) {
+    log.error("Failed to generate task title", { taskId, error });
+  }
+}
+
 // Adapt our logger to SagaLogger interface
 const sagaLogger: SagaLogger = {
   info: (message, data) => log.info(message, data),
@@ -78,6 +156,11 @@ export class TaskCreationSaga extends Saga<
         )
       : await this.createTask(input);
 
+    // Fire-and-forget: generate a proper LLM title for new tasks
+    if (!taskId) {
+      generateTaskTitle(task.id, input.content ?? "", this.deps.posthogClient);
+    }
+
     // Step 2: Resolve repoPath - input takes precedence, then stored mappings
     // Wait for workspace store to load first (it loads async on init)
     await this.readOnlyStep("wait_workspaces_loaded", () =>
@@ -282,6 +365,7 @@ export class TaskCreationSaga extends Saga<
       name: "task_creation",
       execute: async () => {
         const result = await this.deps.posthogClient.createTask({
+          title: truncateToTitle(input.content ?? ""),
           description: input.content ?? "",
           repository: repository ?? undefined,
           github_integration:

packages/agent/src/adapters/claude/claude-agent.ts

@@ -170,7 +170,7 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
 
     // Fire off MCP metadata fetch early — it populates a module-level cache
     // used later during permission checks, not needed by buildSessionOptions or query()
-    const mcpMetadataPromise = fetchMcpToolMetadata(mcpServers, this.logger);
+    fetchMcpToolMetadata(mcpServers, this.logger);
 
     const options = timeSync("buildSessionOptions", () =>
       buildSessionOptions({
@@ -222,17 +222,14 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
     // Slash commands: send update to client when ready (not blocking)
     getAvailableSlashCommands(q)
       .then((slashCommands) => {
-        this.sendAvailableCommandsUpdate(sessionId, slashCommands);
+        if (this.sessionId === sessionId) {
+          this.sendAvailableCommandsUpdate(sessionId, slashCommands);
+        }
       })
       .catch((err) => {
         this.logger.warn("Failed to fetch slash commands", { err });
       });
 
-    // MCP metadata: already running, just ensure errors are handled
-    mcpMetadataPromise.catch((err) => {
-      this.logger.warn("Failed to fetch MCP tool metadata", { err });
-    });
-
     session.modelId = modelOptions.currentModelId;
 
     // Fire-and-forget — trySetModel already swallows errors
@@ -272,7 +269,7 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
     const mcpServers = parseMcpServers(params);
 
     // Fire off MCP metadata fetch early — populates cache for permission checks
-    const mcpMetadataPromise = fetchMcpToolMetadata(mcpServers, this.logger);
+    fetchMcpToolMetadata(mcpServers, this.logger);
 
     const permissionMode: TwigExecutionMode =
       meta?.permissionMode &&
@@ -298,16 +295,14 @@ export class ClaudeAcpAgent extends BaseAcpAgent {
     // Defer slash commands and MCP metadata — not needed to return configOptions
     getAvailableSlashCommands(q)
       .then((slashCommands) => {
-        this.sendAvailableCommandsUpdate(sessionId, slashCommands);
+        if (this.sessionId === sessionId) {
+          this.sendAvailableCommandsUpdate(sessionId, slashCommands);
+        }
       })
       .catch((err) => {
         this.logger.warn("Failed to fetch slash commands on resume", { err });
       });
 
-    mcpMetadataPromise.catch((err) => {
-      this.logger.warn("Failed to fetch MCP tool metadata on resume", { err });
-    });
-
     return {
       configOptions: await this.buildConfigOptions(),
     };