Larry Yin commited on
Commit
2dee946
1 Parent(s): d40491e

Converts memory_token_limit to int

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ llm = ChatOpenAI(model_name=llm_name, temperature=0)
44
  retriever=vectordb.as_retriever()
45
  memory = ConversationTokenBufferMemory(
46
  llm = llm,
47
- max_token_limit=memory_token_limit,
48
  memory_key="chat_history",
49
  return_messages=True
50
  )
 
44
  retriever=vectordb.as_retriever()
45
  memory = ConversationTokenBufferMemory(
46
  llm = llm,
47
+ max_token_limit=int(memory_token_limit),
48
  memory_key="chat_history",
49
  return_messages=True
50
  )