@musnows/scriverse 1.1.1 → 1.1.2

package/dist/ai.js

@@ -2438,6 +2438,7 @@ export class AiManager {
     retryPolicy;
     retrySleep;
     liteLlmPriceCache;
+    conversationTitleGenerations = new Map();
     taskControllers = new Map();
     autoRunStarting = new Map();
     autoRunTimers = new Map();
@@ -5141,13 +5142,13 @@ export class AiManager {
         const titleSettings = this.store.getWorkAiSettings(input.workId);
         const titleModelId = typeof titleSettings.titleGenerationModelId === "string" ? titleSettings.titleGenerationModelId : "";
         const defaultTitle = firstUserContent ? defaultAiConversationTitle(firstUserContent) : "";
-        const isCompletingSecondAssistantTurn = conversationBefore?.messages.at(-1)?.role === "user"
-            && userMessages.length === 2
-            && assistantMessages.length === 1;
+        const isCompletingFirstAssistantTurn = conversationBefore?.messages.at(-1)?.role === "user"
+            && userMessages.length === 1
+            && assistantMessages.length === 0;
         const shouldGenerateTitle = Boolean(input.conversationId
             && firstUserContent
             && titleModelId
-            && isCompletingSecondAssistantTurn
+            && isCompletingFirstAssistantTurn
             && (conversationBefore?.title === "新对话" || conversationBefore?.title === defaultTitle));
         const processStartedAt = process.hrtime.bigint();
         let persistedConversationMessage = null;
@@ -5258,14 +5259,26 @@ export class AiManager {
                 });
             }
         }
+        let conversationTitleGenerationStarted = false;
         if (shouldGenerateTitle && conversationMessage && input.conversationId) {
-            void this.generateConversationTitle(input.workId, input.conversationId, titleModelId, [
+            conversationTitleGenerationStarted = true;
+            const generation = this.generateConversationTitle(input.workId, input.conversationId, titleModelId, [
                 ...(conversationBefore?.messages ?? []),
                 { role: "assistant", content: generated.content }
             ], defaultTitle).catch((error) => {
                 logger.warn("ai.conversation_title.failed", { workId: input.workId, conversationId: input.conversationId, error: aiErrorForLog(error) });
+                return null;
+            });
+            this.conversationTitleGenerations.set(input.conversationId, generation);
+            void generation.then(() => {
+                if (this.conversationTitleGenerations.get(input.conversationId) === generation) {
+                    this.conversationTitleGenerations.delete(input.conversationId);
+                }
             });
         }
+        const updatedConversation = input.conversationId
+            ? this.store.getAiConversationSummary(input.conversationId)
+            : null;
         return {
             callId: generated.callId,
             content: generated.content,
@@ -5277,10 +5290,15 @@ export class AiManager {
             toolCalls: generated.toolCalls,
             processSteps: generated.processSteps,
             contextUsage: generated.contextUsage,
+            conversationTitle: updatedConversation?.title ?? "新对话",
+            conversationTitleGenerationStarted,
             roleplayMemoriesCommitted: committedRoleplayMemories,
             ...(conversationMessage ? { conversationMessage } : {})
         };
     }
+    async waitForConversationTitle(conversationId) {
+        await this.conversationTitleGenerations.get(conversationId);
+    }
     async resumeUserQuestion(input, stream = {}) {
         const answeredItems = (input.answers ?? []).map((answer) => ({
             question: String(answer.question ?? ""),