ajout system prompt

agent.py

@@ -80,10 +80,15 @@ class LangGraphAgent:
             MessagesPlaceholder(variable_name="agent_scratchpad"),
         ])
         
-        
         # Définir les nodes
         def assistant(state: MessagesState):
-            return {"messages": [self.llm.invoke(state["messages"])]}
+            # Utiliser le prompt pour formater les messages
+            messages = prompt.format_messages(
+                chat_history=state["messages"][:-1],
+                input=state["messages"][-1].content,
+                agent_scratchpad=[]
+            )
+            return {"messages": [self.llm.invoke(messages)]}
         
         # Créer le graphe
         builder = StateGraph(MessagesState)
@@ -132,4 +137,28 @@ class LangGraphAgent:
             
         except Exception as e:
             logger.error(f"Erreur lors du traitement de la question: {str(e)}")
-            return f"Désolé, une erreur s'est produite: {str(e)}" 
+            return f"Désolé, une erreur s'est produite: {str(e)}" 
+
+if __name__ == "__main__":
+    # Créer une instance de l'agent
+    agent = LangGraphAgent(verbose=True)
+    
+    # Question à tester
+    question = "How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia."
+    
+    # Obtenir la réponse
+    response = agent(question)
+    
+    # Afficher la réponse
+    print("\nQuestion:", question)
+    print("\nRéponse:", response) 
+
+    # Question à tester
+    question ="In the video https://www.youtube.com/watch?v=L1vXCYZAYYM, what is the highest number of bird species to be on camera simultaneously?"
+    
+    # Obtenir la réponse
+    response = agent(question)
+    
+    # Afficher la réponse
+    print("\nQuestion:", question)
+    print("\nRéponse:", response) 

app.py

@@ -9,8 +9,6 @@ from dotenv import load_dotenv
 from agent import LangGraphAgent
 import config
 
-# Charger les variables d'environnement
-load_dotenv()
 
 # --- Constants ---
 DEFAULT_API_URL = config.DEFAULT_API_URL