Update fix.py

srdas · Oct 24, 2024 · 94ff6ae · 94ff6ae
1 parent 933a8d4
commit 94ff6ae
Showing 1 changed file with 0 additions and 13 deletions.
diff --git a/packages/jupyter-ai/jupyter_ai/chat_handlers/fix.py b/packages/jupyter-ai/jupyter_ai/chat_handlers/fix.py
@@ -81,19 +81,6 @@ def create_llm_chain(
         self.prompt_template = prompt_template
 
         runnable = prompt_template | llm  # type:ignore
-        if not llm.manages_history:
-            runnable = RunnableWithMessageHistory(
-                runnable=runnable,  #  type:ignore[arg-type]
-                get_session_history=self.get_llm_chat_memory,
-                input_messages_key="extra_instructions",
-                history_messages_key="history",
-                history_factory_config=[
-                    ConfigurableFieldSpec(
-                        id="last_human_msg",
-                        annotation=HumanChatMessage,
-                    ),
-                ],
-            )
         self.llm_chain = runnable
 
     async def process_message(self, message: HumanChatMessage):