fix: add model name with provider to history

upstash · May 31, 2024 · cf8bf82 · cf8bf82
1 parent ca2144a
commit cf8bf82
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 6 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@upstash/rag-chat",
-  "version": "0.0.27-alpha",
+  "version": "0.0.29-alpha",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",
   "types": "./dist/index.d.ts",

diff --git a/src/history/redis-custom-history.ts b/src/history/redis-custom-history.ts
@@ -96,11 +96,13 @@ export class CustomUpstashRedisChatMessageHistory extends BaseListChatMessageHis
    * @returns Promise resolving to void.
    */
   async addMessage(message: BaseMessage): Promise<void> {
+    message.response_metadata = {
+      ...message.response_metadata,
+      modelNameWithProvider: this.modelNameWithProvider,
+    };
     const messageToAdd = mapChatMessagesToStoredMessages([message]);
-    await this.client.lpush(
-      this.sessionId,
-      JSON.stringify({ ...messageToAdd[0], modelNameWithProvider: this.modelNameWithProvider })
-    );
+
+    await this.client.lpush(this.sessionId, JSON.stringify(messageToAdd[0]));
     if (this.sessionTTL) {
       await this.client.expire(this.sessionId, this.sessionTTL);
     }

diff --git a/src/rag-chat-base.ts b/src/rag-chat-base.ts
@@ -49,7 +49,7 @@ export class RAGChatBase {
   }
 
   /** This method first gets required params, then returns another function depending on streaming param input */
-  chainCall(chatOptions: ChatOptions, question: string, facts: string) {
+  protected chainCall(chatOptions: ChatOptions, question: string, facts: string) {
     const formattedHistoryChain = RunnableSequence.from<CustomInputValues>([
       {
         chat_history: (input) => formatChatHistory(input.chat_history ?? []),