@averyyy/pi-coding-agent 0.80.3-piclient.6 → 0.80.3-piclient.8

package/dist/core/agent-session.js

@@ -67,6 +67,9 @@ const INTRA_TURN_COMPACTION_MARKER_TEXT = "Continue from the compaction summary
 const VALIDATION_HINT_MESSAGE_TYPE = "pi:validation-hint";
 const VALIDATION_HINT_FILE_LIMIT = 12;
 const VALIDATION_HINT_FAILURE_CHAR_LIMIT = 1200;
+const AUTO_SESSION_NAME_INPUT_CHAR_LIMIT = 4000;
+const AUTO_SESSION_NAME_MAX_CHARS = 80;
+const AUTO_SESSION_NAME_PROMPT = "Name this coding session from the transcript below. Return only a concise title, in the user's language when clear. No quotes, no markdown, no explanation. Do not call tools.";
 function isPiServerMode() {
     return process.env.PI_SERVER_MODE === "true";
 }
@@ -79,6 +82,16 @@ function getStringField(input, key) {
     const value = input[key];
     return typeof value === "string" && value.trim().length > 0 ? value : undefined;
 }
+function getPiServerFailurePhase(message) {
+    for (const diagnostic of message.diagnostics ?? []) {
+        if (diagnostic.type !== "pi_server_failure")
+            continue;
+        const phase = diagnostic.details?.phase;
+        if (typeof phase === "string")
+            return phase;
+    }
+    return undefined;
+}
 function toolTextContent(content) {
     return content
         .filter((part) => part.type === "text")
@@ -92,6 +105,41 @@ function compactValidationSummary(text) {
         return trimmed;
     return `...${trimmed.slice(trimmed.length - VALIDATION_HINT_FAILURE_CHAR_LIMIT + 3)}`;
 }
+function messageTextContent(message) {
+    if (!("content" in message))
+        return "";
+    const content = message.content;
+    if (typeof content === "string")
+        return content;
+    if (!Array.isArray(content))
+        return "";
+    return content
+        .filter((part) => part.type === "text")
+        .map((part) => part.text)
+        .join("\n")
+        .trim();
+}
+function truncateForSessionNameInput(text) {
+    if (text.length <= AUTO_SESSION_NAME_INPUT_CHAR_LIMIT)
+        return text;
+    return text.slice(0, AUTO_SESSION_NAME_INPUT_CHAR_LIMIT).trim();
+}
+function sanitizeGeneratedSessionName(text) {
+    const firstLine = text
+        .split(/\r?\n/u)
+        .map((line) => line.trim())
+        .find((line) => line.length > 0);
+    if (!firstLine)
+        return undefined;
+    const title = firstLine
+        .replace(/^title:\s*/iu, "")
+        .replace(/^["'`]+|["'`]+$/g, "")
+        .replace(/\s+/g, " ")
+        .trim();
+    if (!title)
+        return undefined;
+    return title.length > AUTO_SESSION_NAME_MAX_CHARS ? title.slice(0, AUTO_SESSION_NAME_MAX_CHARS).trim() : title;
+}
 // ============================================================================
 // AgentSession Class
 // ============================================================================
@@ -144,6 +192,7 @@ export class AgentSession {
     _extensionShutdownHandler;
     _extensionErrorListener;
     _extensionErrorUnsubscriber;
+    _autoSessionNameAttempted = false;
     // Model registry for API key resolution
     _modelRegistry;
     // Tool registry for extension getTools/setTools
@@ -883,12 +932,79 @@ export class AgentSession {
                     throw error;
                 }
             }
+            await this._maybeGenerateSessionName();
         }
         finally {
             this._systemPromptOverride = undefined;
             this._flushPendingBashMessages();
         }
     }
+    _sessionHasInfoEntry() {
+        return this.sessionManager.getEntries().some((entry) => entry.type === "session_info");
+    }
+    _buildSessionNameTranscript() {
+        const lines = [];
+        for (const message of this.agent.state.messages) {
+            if (message.role !== "user" && message.role !== "assistant")
+                continue;
+            const text = messageTextContent(message);
+            if (!text)
+                continue;
+            lines.push(`${message.role === "user" ? "User" : "Assistant"}: ${text}`);
+        }
+        const transcript = lines.join("\n\n").trim();
+        return transcript ? truncateForSessionNameInput(transcript) : undefined;
+    }
+    _shouldGenerateSessionName() {
+        if (this._autoSessionNameAttempted || this._sessionHasInfoEntry())
+            return false;
+        const userMessages = this.agent.state.messages.filter((message) => message.role === "user");
+        if (userMessages.length !== 1)
+            return false;
+        const lastAssistant = this._findLastAssistantMessage();
+        return lastAssistant !== undefined && !this._isTerminalAssistantFailure(lastAssistant);
+    }
+    async _maybeGenerateSessionName() {
+        if (!this._shouldGenerateSessionName())
+            return;
+        this._autoSessionNameAttempted = true;
+        const model = this.model;
+        const transcript = this._buildSessionNameTranscript();
+        if (!model || !transcript)
+            return;
+        try {
+            const stream = await this.agent.streamFn(model, {
+                systemPrompt: this.systemPrompt,
+                messages: [
+                    {
+                        role: "user",
+                        content: [{ type: "text", text: `${AUTO_SESSION_NAME_PROMPT}\n\n${transcript}` }],
+                        timestamp: Date.now(),
+                    },
+                ],
+                tools: this.agent.state.tools,
+            }, {
+                maxTokens: 32,
+                reasoning: undefined,
+                sessionId: this.sessionId,
+                transport: this.agent.transport,
+                thinkingBudgets: this.agent.thinkingBudgets,
+                maxRetryDelayMs: this.agent.maxRetryDelayMs,
+            });
+            for await (const _event of stream) {
+                // Drain the title stream without surfacing it as a chat message.
+            }
+            const response = await stream.result();
+            if (response.stopReason === "error" || response.stopReason === "aborted")
+                return;
+            const title = sanitizeGeneratedSessionName(messageTextContent(response));
+            if (title)
+                this.setSessionName(title);
+        }
+        catch {
+            // Session naming is a post-turn convenience; the completed assistant turn stays authoritative.
+        }
+    }
     async _handlePostAgentRun() {
         const msg = this._lastAssistantMessage;
         this._lastAssistantMessage = undefined;
@@ -2345,6 +2461,9 @@ export class AgentSession {
             return false;
         if (this._isPostAssistantPiServerSyncFailure(message))
             return false;
+        const piServerPhase = getPiServerFailurePhase(message);
+        if (piServerPhase && piServerPhase !== "provider_stream")
+            return false;
         if (message.errorMessage && this._isNonRetryableProviderLimitError(message.errorMessage))
             return false;
         return isRetryableAssistantError(message);