Update gaia_agent.py

gaia_agent.py

@@ -1,4 +1,4 @@
-from langchain_ollama import ChatOllama
+# from langchain_ollama import ChatOllama
 from langchain_core.messages import SystemMessage, HumanMessage
 from langgraph.graph import START, StateGraph, MessagesState
 from langgraph.prebuilt import ToolNode, tools_condition
@@ -154,12 +154,12 @@ sys_msg = SystemMessage(content=GAIA_SYSTEM_PROMPT)
 
 
 # Build graph function
-def build_graph(provider: str = "ollama"):
+def build_graph(provider: str = "huggingface"):
     """Build the graph"""
     # Load environment variables from .env file
-    if provider == "ollama":
-        chat = ChatOllama(model="llama3.1")
-    elif provider == "huggingface":
+    # if provider == "ollama":
+    #     chat = ChatOllama(model="llama3.1")
+    if provider == "huggingface":
         llm = HuggingFaceEndpoint(
             repo_id="Qwen/Qwen2.5-Coder-32B-Instruct"
         )