Update main.py

main.py

@@ -7,7 +7,7 @@ from typing import List
 from pathlib import Path
 from langchain_huggingface import HuggingFaceEndpoint
 from langchain.schema import StrOutputParser
-from langchain_anthropic import AnthropicLLM
+from langchain_anthropic import AnthropicLLM, ChatAnthropic
 
 from langchain.agents import AgentExecutor
 from langchain.agents.agent_types import AgentType
@@ -30,7 +30,7 @@ async def LLMistral():
 
 async def LLM():
     os.environ['ANTHROPIC_API_KEY'] = os.environ['ANTHROPIC_API_KEY']
-    llm = AnthropicLLM(model='claude-2.1',temperature=0, streaming=True, verbose=True)
+    llm = ChatAnthropic(model='claude-3-5-sonnet-20240620',temperature=0)
     return llm
     
 @cl.set_chat_profiles
@@ -60,9 +60,8 @@ async def on_message(message: cl.Message):
     agent = create_csv_agent(
         model,
         "./public/ExpeCFA_LP_CAA.csv",
-        verbose=False,
+        verbose=True,
         allow_dangerous_code=True,
-        handle_parsing_errors=True,
         agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION
     )