fix(backend): generate topic when finish completion

AprilNEA · Dec 11, 2023 · 67376fb · 67376fb
1 parent c30081d
commit 67376fb
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/packages/backend/src/modules/chat/chat.controller.ts b/packages/backend/src/modules/chat/chat.controller.ts
@@ -164,7 +164,7 @@ export class ChatController {
       modelId: data.modelId,
       input: data.content,
       messages: chatSession.messages,
-      // key,
+      topic: chatSession.topic,
     });
   }
 }
diff --git a/packages/backend/src/modules/chat/chat.service.ts b/packages/backend/src/modules/chat/chat.service.ts
@@ -263,6 +263,7 @@ ${message}
     input,
     modelId,
     messages, // key,
+    topic,
   }: {
     userId: number;
     sessionId: string;
@@ -273,6 +274,7 @@ ${message}
     messages: ChatMessage[];
     /* Request API Key */
     // key: string;
+    topic?: string;
   }) {
     const { name: model } = await this.prisma.client.model.findUniqueOrThrow({
       where: { id: modelId },
@@ -338,6 +340,24 @@ ${message}
             }),
           ]);
           subscriber.complete();
+          /* 首次对话自动总结对话，
+           * First conversation automatically summarizes the conversation
+           */
+          if (!topic) {
+            await this.summarizeTopic(
+              [
+                ...histories,
+                { role: 'user', content: input },
+                {
+                  role: 'assistant',
+                  content: generated,
+                },
+              ]
+                .map((m) => `${m.role}: ${m.content}`)
+                .join('\n'),
+              sessionId,
+            );
+          }
         }
       })();
     });