asdf

graph.py

@@ -13,6 +13,8 @@ from langgraph.graph import StateGraph, END, add_messages
 from langchain_community.tools import TavilySearchResults
 from pydantic import BaseModel, Field
 from trafilatura import extract
+from langchain_anthropic import ChatAnthropic
+
 
 logger = logging.getLogger(__name__)
 ASSISTANT_SYSTEM_PROMPT_BASE = """"""
@@ -45,7 +47,14 @@ if search_enabled:
 else:
     print("TAVILY_API_KEY environment variable not found. Websearch disabled")
 
-weak_model = ChatOpenAI(model="gpt-4o-mini", tags=["assistant"])
+weak_model = ChatAnthropic(
+            model="claude-3-5-sonnet-20240620",
+            temperature=0,
+            max_tokens=1024,
+            timeout=None,
+            max_retries=2,
+            # other params...
+        )
 model = weak_model
 assistant_model = weak_model