ajout de conditions

agent.py

@@ -154,7 +154,7 @@ def create_agent_and_answer(dict_data, question) -> str:
 
 
 research_agent = create_react_agent(
-    model=llm_4o,
+    model=llm_4_1,
     tools=[wiki_search, arvix_search],
     prompt=(
         "You are a research agent.\n\n"
@@ -168,7 +168,7 @@ research_agent = create_react_agent(
 )
 
 web_search_openai_agent = create_react_agent(
-    model=llm_4o,
+    model=llm_4_1,
     tools=[web_search_openai_tool],
     prompt=(
         "You are a websearch agent.\n\n"
@@ -236,7 +236,7 @@ supervisor = create_supervisor(
     model=init_chat_model("openai:gpt-4.1", api_key = api_open_ai_agent_key),
     agents=[research_agent, web_search_openai_agent, agent_excel_new, reflexion_agent],
     prompt=(
-        "You are a supervisor managing three agents:\n"
+        "You are a supervisor managing four agents:\n"
         "- research_agent: Specialised in ArXiv and Wikipedia. Assign research-related tasks to this agent.\n"
         "- reflexion: Called when the supervisor need a reflexion, not general knowledge. Can handle  math-related tasks such as solving equations, performing calculations or working on abstract maths subject such as matrix or demonstrating subjects.\n"
         "- web_search_openai_agent: Can browse the web to find up-to-date and relevant information. Assign web-related tasks to this agent.\n"

fonctions.py

@@ -4,10 +4,51 @@ import re
 import pandas as pd
 from openai import OpenAI
 import os
+from langchain_core.messages import convert_to_messages
 
 
 api_open_ai_agent_key=os.environ["OPENAI_API_KEY"]
 client = OpenAI(api_key=api_open_ai_agent_key)
+
+
+def pretty_print_message(message, indent=False):
+    pretty_message = message.pretty_repr(html=True)
+    if not indent:
+        print(pretty_message)
+        return
+
+    indented = "\n".join("\t" + c for c in pretty_message.split("\n"))
+    print(indented)
+
+
+def pretty_print_messages(update, last_message=False):
+    is_subgraph = False
+    if isinstance(update, tuple):
+        ns, update = update
+        # skip parent graph updates in the printouts
+        if len(ns) == 0:
+            return
+
+        graph_id = ns[-1].split(":")[0]
+        print(f"Update from subgraph {graph_id}:")
+        print("\n")
+        is_subgraph = True
+
+    for node_name, node_update in update.items():
+        update_label = f"Update from node {node_name}:"
+        if is_subgraph:
+            update_label = "\t" + update_label
+
+        print(update_label)
+        print("\n")
+
+        messages = convert_to_messages(node_update["messages"])
+        if last_message:
+            messages = messages[-1:]
+
+        for m in messages:
+            pretty_print_message(m, indent=is_subgraph)
+        print("\n")
 def clean_response(response):
     match = re.search(r'FINAL ANSWER:\s*(.+)', response['supervisor']['messages'][-1].content)
     answer = match.group(1).strip() if match else None
@@ -21,7 +62,7 @@ def response_from_agent(supervisor, question):
     ):
 
         response = chunk
-
+        pretty_print_messages(chunk)
 
     response = clean_response(response)
     return response
@@ -72,4 +113,5 @@ def load_data(question):
         return transcription.text
 
     else :
-        return 'there is no attached file'
+        return 'there is no attached file'
+

requirements.txt

@@ -13,4 +13,5 @@ tabulate
 pandas
 pydub
 torch
-numpy
+numpy
+openpyxl