fix: NextChatAgent prompt拼接逻辑

TBXark · Nov 14, 2024 · f373dde · f373dde
1 parent 226c08d
commit f373dde
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/packages/lib/next/src/index.ts b/packages/lib/next/src/index.ts
@@ -108,9 +108,14 @@ export class NextChatAgent implements ChatAgent {
         if (!model) {
             throw new Error('Model not found');
         }
+        if (params.prompt) {
+            params.messages.unshift({
+                role: 'assistant',
+                content: params.prompt,
+            });
+        }
         return requestChatCompletionsV2({
             model: this.providerCreator(context).languageModel(model),
-            prompt: params.prompt,
             messages: params.messages,
         }, onStream);
     };