From bb67df8f427d10fa13cb56d3b519f90cb861c543 Mon Sep 17 00:00:00 2001
From: martin legrand <martin.legrand@epitech.eu>
Date: Sun, 6 Apr 2025 19:46:52 +0200
Subject: [PATCH] feat : signal handler now agent specific instead of global

---
 main.py                         | 6 ------
 server/app.py                   | 1 +
 sources/agents/browser_agent.py | 2 +-
 sources/agents/code_agent.py    | 2 +-
 sources/router.py               | 7 ++++++-
 5 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/main.py b/main.py
index 587b184..f03416b 100755
--- a/main.py
+++ b/main.py
@@ -1,7 +1,6 @@
 #!/usr/bin python3
 
 import sys
-import signal
 import argparse
 import configparser
 
@@ -17,12 +16,7 @@ warnings.filterwarnings("ignore")
 config = configparser.ConfigParser()
 config.read('config.ini')
 
-def handleInterrupt(signum, frame):
-    sys.exit(0)
-
 def main():
-    signal.signal(signal.SIGINT, handler=handleInterrupt)
-
     pretty_print("Initializing...", color="status")
     provider = Provider(provider_name=config["MAIN"]["provider_name"],
                         model=config["MAIN"]["provider_model"],
diff --git a/server/app.py b/server/app.py
index 2245c66..85e0f04 100644
--- a/server/app.py
+++ b/server/app.py
@@ -41,6 +41,7 @@ def setup():
 def get_updated_sentence():
     if not generator:
         return jsonify({"error": "Generator not initialized"}), 405
+    print(generator.get_status())
     return generator.get_status()
 
 if __name__ == '__main__':
diff --git a/sources/agents/browser_agent.py b/sources/agents/browser_agent.py
index 355c439..c616f49 100644
--- a/sources/agents/browser_agent.py
+++ b/sources/agents/browser_agent.py
@@ -296,7 +296,7 @@ class BrowserAgent(Agent):
         prompt = self.make_newsearch_prompt(user_prompt, search_result)
         unvisited = [None]
         while not complete:
-            answer, reasoning = self.llm_decide(prompt, show_reasoning = True)
+            answer, reasoning = self.llm_decide(prompt, show_reasoning = False)
 
             extracted_form = self.extract_form(answer)
             if len(extracted_form) > 0:
diff --git a/sources/agents/code_agent.py b/sources/agents/code_agent.py
index 25faff0..396e21b 100644
--- a/sources/agents/code_agent.py
+++ b/sources/agents/code_agent.py
@@ -51,7 +51,7 @@ class CoderAgent(Agent):
             self.wait_message(speech_module)
             answer, reasoning = self.llm_request()
             if clarify_trigger in answer:
-                return answer.replace(clarify_trigger, ""), reasoning
+                return answer, reasoning
             if not "```" in answer:
                 self.last_answer = answer
                 break
diff --git a/sources/router.py b/sources/router.py
index 6c98132..b152413 100644
--- a/sources/router.py
+++ b/sources/router.py
@@ -162,6 +162,7 @@ class AgentRouter:
             ("Search my drive for a file called vacation_photos_2023.jpg.", "files"),
             ("Help me organize my desktop files into folders by type.", "files"),
             ("What’s your favorite movie and why?", "talk"),
+            ("what directory are you in ?", "files"),
             ("Search my drive for a file named budget_2024.xlsx", "files"),
             ("Write a Python function to sort a list of dictionaries by key", "code"),
             ("Find the latest updates on quantum computing on the web", "web"),
@@ -330,7 +331,11 @@ class AgentRouter:
         Returns:
         str: The estimated complexity
         """
-        predictions = self.complexity_classifier.predict(text)
+        try:
+            predictions = self.complexity_classifier.predict(text)
+        except Exception as e:
+            pretty_print(f"Error in estimate_complexity: {str(e)}", color="failure")
+            return "LOW"
         predictions = sorted(predictions, key=lambda x: x[1], reverse=True)
         if len(predictions) == 0:
             return "LOW"