debugging

gaia_graph.py

@@ -9,10 +9,10 @@ from dotenv import load_dotenv
 from langchain.tools import Tool
 from langchain.agents import initialize_agent, AgentType
 from langchain_openai import ChatOpenAI
-from langgraph.graph import StateGraph
+from langgraph.graph import StateGraph, END
 
+import openai  # Import openai to catch RateLimitError
 from tenacity import retry, wait_exponential, stop_after_attempt, retry_if_exception_type
-import openai.error
 
 # ─── Load Environment Variables ──────────────────────────────────────────────
 load_dotenv()
@@ -75,10 +75,10 @@ search_tool = Tool(
     description="Useful for answering factual questions using a search engine.",
 )
 
-# ─── Create LLM Agent with retry ──────────────────────────────────────────────
+# ─── Create LLM Agent ────────────────────────────────────────────────────────
 llm = ChatOpenAI(
     temperature=0.0,
-    model="gpt-4o-mini",  # switched to gpt-4o-mini for higher token limits
+    model="gpt-4o",
     openai_api_key=OPENAI_API_KEY
 )
 
@@ -90,14 +90,6 @@ agent_executor = initialize_agent(
     handle_parsing_errors=True,
 )
 
-@retry(
-    wait=wait_exponential(multiplier=1, min=1, max=10),
-    stop=stop_after_attempt(5),
-    retry=retry_if_exception_type(openai.error.RateLimitError)
-)
-def run_agent_with_retry(prompt: str) -> str:
-    return agent_executor.run(prompt)
-
 # ─── Clean Output ────────────────────────────────────────────────────────────
 def clean_answer(ans: str) -> str:
     if "```" in ans:
@@ -113,12 +105,18 @@ class GaiaState(TypedDict):
     question: str
     answer: str
 
-# ─── Define Node Function with retry ──────────────────────────────────────────
+# ─── Retry on Rate Limit Error ───────────────────────────────────────────────
+@retry(
+    wait=wait_exponential(multiplier=1, min=1, max=10),
+    stop=stop_after_attempt(5),
+    retry=retry_if_exception_type(openai.error.RateLimitError)
+)
+def run_agent_with_retry(question: str) -> str:
+    return agent_executor.run(question)
+
+# ─── Define Node Function ────────────────────────────────────────────────────
 def agent_node(state: GaiaState) -> GaiaState:
-    try:
-        raw = run_agent_with_retry(state["question"])
-    except openai.error.RateLimitError:
-        raw = "Error: rate limit exceeded after retries."
+    raw = run_agent_with_retry(state["question"])
     return {"question": state["question"], "answer": clean_answer(raw)}
 
 # ─── Build LangGraph ─────────────────────────────────────────────────────────

requirements.txt

@@ -37,3 +37,4 @@ PyYAML
 hf-xet~=1.1.1
 langchain-openai
 tenacity
+openai