Merge pull request #300 from TBXark/dev

fix: cohere history bug
TBXark · Oct 31, 2024 · a9b84c2 · a9b84c2
2 parents 49d9cb1 + eefdd92
commit a9b84c2
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 6 deletions.
diff --git a/dist/buildinfo.json b/dist/buildinfo.json
diff --git a/dist/index.js b/dist/index.js
diff --git a/dist/timestamp b/dist/timestamp
diff --git a/src/agent/cohere.ts b/src/agent/cohere.ts
@@ -62,12 +62,14 @@ export class Cohere implements ChatAgent {
             'Accept': onStream !== null ? 'text/event-stream' : 'application/json',
         };
 
+        const messages = [...history || [], { role: 'user', content: message }];
+
         const body = {
             message,
             model: context.COHERE_CHAT_MODEL,
             stream: onStream != null,
             preamble: prompt,
-            chat_history: history?.map(this.render),
+            chat_history: messages.map(this.render),
         };
         if (!body.preamble) {
             delete body.preamble;