reflexion pour maths

agent.py

@@ -15,7 +15,46 @@ llm_4o = ChatOpenAI(
     model_name="gpt-4o",
     openai_api_key=api_open_ai_agent_key,  # ou variable d’environnement
 )
+llm_reasoning = ChatOpenAI(
+    model_name = "o3-2025-04-16",
+    openai_api_key=api_open_ai_agent_key,  # ou variable d’environnement
+)
+
+
+llm_reasoning_small = ChatOpenAI(
+    model_name = "o4-mini-2025-04-16",
+    openai_api_key=api_open_ai_agent_key,  # ou variable d’environnement
+)
+
+text_exemple = """
+Set: {x, y, z}
+
+Table:
+   * | x | y | z
+   --------------
+   x | x | y | z
+   y | y | x | x
+   z | z | x | y
+
+→ z * y = x, but y * z = x → equal  
+→ y * z = x, but z * y = x → equal  
+→ y * y = x, but y * y = x → equal  
+→ All pairs commute → No counter-example
+
+Answer: The operation is commutative.
+"""
+exemple_2 = """
+Table:
+   * | a | b
+   ------------
+   a | a | b
+   b | a | a
+
+→ a * b = b, but b * a = a ≠ b  
+→ Counter-example: a, b  
+Answer: a, b
 
+"""
 
 @tool
 def wiki_search(query: str) -> str:
@@ -112,22 +151,27 @@ web_search_openai_agent = create_react_agent(
     name="web_search_openai_agent",
 )
 
-
-
 math_agent = create_react_agent(
-    model=llm_4o,
+    model=llm_reasoning_small,
 
     tools=[add, multiply, divide],
     prompt=(
-        "You are a math agent.\n\n"
+        "You are a mathematical reasoning agent.\n\n"
         "INSTRUCTIONS:\n"
-        "- Assist ONLY with math-related tasks\n"
-        "- After you're done with your tasks, respond to the supervisor directly\n"
-        "- Respond ONLY with the results of your work, do NOT include ANY other text."
-    ),
-    name="math_agent",
-)
+        "- You are given mathematical structures such as sets, operations, and tables.\n"
+        "- You must analyze them for properties like commutativity, associativity, identity, etc.\n"
+        "- When given a Cayley table, check if x * y == y * x for all pairs to test commutativity.\n"
+        "- Return a precise answer, including any counter-example pairs if they exist.\n"
+        "- If a counter-example exists, return the set of involved elements in alphabetical order as a comma-separated list.\n"
+        "- Do not rely on numeric calculation only — work symbolically when needed.\n"
+        f"- Here are two examples : Example 1:{text_exemple}.\n"
+        f"Example 2:{exemple_2}."),
+
+
+
+
 
+    name="math_agent")
 
 supervisor = create_supervisor(
     model=init_chat_model("openai:gpt-4o", api_key = api_open_ai_agent_key),

requirements.txt

@@ -4,4 +4,7 @@ langchain
 langgraph
 langchain-community
 langgraph-supervisor
-langchain-openai
+langchain-openai
+wikipedia
+arxiv
+pymupdf