Refactor import statements for ChatMessageHistory in model.py and space.py

kitt/core/model.py

@@ -6,7 +6,6 @@ import xml.etree.ElementTree as ET
 from enum import Enum
 from typing import List
 
-from langchain.memory import ChatMessageHistory
 from langchain.tools.base import StructuredTool
 from langchain_core.messages import AIMessage, HumanMessage, ToolMessage
 from langchain_core.utils.function_calling import convert_to_openai_tool

requirements.txt

@@ -5,6 +5,7 @@ matplotlib
 wurlitzer
 accelerate
 bitsandbytes
+https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.9.post1/flash_attn-2.5.9.post1+cu118torch1.12cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
 optimum
 # auto-gptq
 gradio

space.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from langchain.memory import ChatMessageHistory
+from langchain_community.chat_message_histories import ChatMessageHistory
 from langchain.tools import tool
 from langchain_core.utils.function_calling import convert_to_openai_tool
 from loguru import logger