fix: make history param optional

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

Hide file tree

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

diff --git a/src/history/in-memory-custom-history.ts b/src/history/in-memory-custom-history.ts
@@ -5,15 +5,15 @@ import type { BaseMessage } from "@langchain/core/messages";
 export type CustomInMemoryChatMessageHistoryInput = {
   messages?: BaseMessage[];
   topLevelChatHistoryLength?: number;
-  modelNameWithProvider: string;
+  modelNameWithProvider?: string;
 };
 
 export class CustomInMemoryChatMessageHistory extends BaseListChatMessageHistory {
   lc_namespace = ["langchain", "stores", "message", "in_memory"];
 
   private messages: BaseMessage[] = [];
   private topLevelChatHistoryLength?: number;
-  private modelNameWithProvider: string;
+  private modelNameWithProvider?: string;
 
   constructor(fields: CustomInMemoryChatMessageHistoryInput) {
     const { modelNameWithProvider, messages, topLevelChatHistoryLength } = fields;

diff --git a/src/history/index.ts b/src/history/index.ts
@@ -5,17 +5,17 @@ import { InternalUpstashError } from "../error";
 
 type HistoryConfig = {
   redis?: Redis;
-  modelNameWithProvider: string;
+  modelNameWithProvider?: string;
 };
 type GetHistory = { sessionId: string; length?: number; sessionTTL?: number };
 
 export class History {
   private redis?: Redis;
-  private modelNameWithProvider: string;
+  private modelNameWithProvider?: string;
   private inMemoryChatHistory?: CustomInMemoryChatMessageHistory;
 
-  constructor(fields: HistoryConfig) {
-    const { modelNameWithProvider, redis } = fields;
+  constructor(fields?: HistoryConfig) {
+    const { modelNameWithProvider, redis } = fields ?? {};
 
     this.redis = redis;
     this.modelNameWithProvider = modelNameWithProvider;

diff --git a/src/history/redis-custom-history.ts b/src/history/redis-custom-history.ts
@@ -19,7 +19,7 @@ export type CustomUpstashRedisChatMessageHistoryInput = {
   config?: RedisConfigNodejs;
   client?: Redis;
   topLevelChatHistoryLength?: number;
-  modelNameWithProvider: string;
+  modelNameWithProvider?: string;
 };
 
 /**
@@ -39,7 +39,7 @@ export class CustomUpstashRedisChatMessageHistory extends BaseListChatMessageHis
   public client: Redis;
 
   private sessionId: string;
-  private modelNameWithProvider: string;
+  private modelNameWithProvider?: string;
 
   private sessionTTL?: number;
   private topLevelChatHistoryLength?: number;