Update main.py

main.py

@@ -7,7 +7,6 @@ from typing import List
 from pathlib import Path
 
 from langchain_huggingface import HuggingFaceEndpoint
-from langchain_community.chat_models.huggingface import ChatHuggingFace
 
 from langchain_openai import ChatOpenAI, OpenAI
 from langchain.schema.runnable.config import RunnableConfig
@@ -69,10 +68,9 @@ def create_agent(filename: str):
     os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
     
     repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
-    model_llm = HuggingFaceEndpoint(
+    llm = HuggingFaceEndpoint(
         repo_id=repo_id, task="text2text-generation", max_new_tokens=8000, temperature=0.01, streaming=True
     )
-    llm = ChatHuggingFace(llm=model_llm)
     # Read the CSV file into a Pandas DataFrame.
     if cl.user_session.get("createdb") == None:
         df = pd.read_csv(filename)