Spaces:

bentobytes
/

LawGPT

Runtime error

bentobytes commited on Sep 26, 2023

Commit

3766553

1 Parent(s): 6b3f443

Update conversation.py

Files changed (1) hide show

conversation.py CHANGED Viewed

@@ -28,7 +28,9 @@ def create_conversation(query: str, chat_history: list, collection_name: str) ->
         memory = ConversationBufferMemory(
             memory_key='chat_history',
-            return_messages=False
         )
         cqa = ConversationalRetrievalChain.from_llm(
             llm=ChatOpenAI(temperature=0.0,

         memory = ConversationBufferMemory(
             memory_key='chat_history',
+            return_messages=False,
+            input_key='question',
+            output_key='answer',
         )
         cqa = ConversationalRetrievalChain.from_llm(
             llm=ChatOpenAI(temperature=0.0,