Update gaia_agent.py

gaia_agent.py

@@ -8,6 +8,7 @@ from langchain_community.document_loaders import WikipediaLoader
 from langchain_community.document_loaders import ArxivLoader
 from langchain_core.tools import tool
 from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
+from langchain_xai import ChatXAI
 from math import sqrt
 
 ### =============== MATHEMATICAL TOOLS =============== ###
@@ -165,6 +166,11 @@ def build_graph(provider: str = "huggingface"):
         )
 
         chat = ChatHuggingFace(llm=llm, verbose=True)
+    elif provider == "xai":
+        chat = ChatXAI(
+            model="grok-3-latest",
+            temperature=0,
+        )
     else:
         raise ValueError("Invalid provider. Choose 'ollama' or 'huggingface'.")
     # Bind tools to LLM
@@ -193,7 +199,7 @@ def build_graph(provider: str = "huggingface"):
 if __name__ == "__main__":
     question = "Examine the video at https://www.youtube.com/watch?v=1htKBjuUWec.\n\nWhat does Teal'c say in response to the question \"Isn't that hot?\""
     # fixed_answer = "extremely"
-    graph = build_graph(provider="huggingface")
+    graph = build_graph(provider="xai")
     messages = [HumanMessage(content=question)]
     messages = graph.invoke({"messages": messages})
     for m in messages["messages"]: