Feat : text to speech & flight api

sources/tools/BashInterpreter.py

@@ -26,6 +26,8 @@ class BashInterpreter(Tools):
     
         concat_output = ""
         for command in commands:
+            if "python3" in command:
+                continue # because stubborn AI always want to run python3 with bash when it write code
             try:
                 process = subprocess.Popen(
                     command,

sources/tools/flightSearch.py

@@ -0,0 +1,83 @@
+import os
+import requests
+import dotenv
+
+dotenv.load_dotenv()
+
+if __name__ == "__main__":
+    from tools import Tools
+else:
+    from sources.tools.tools import Tools
+
+class FlightSearch(Tools):
+    def __init__(self, api_key: str = None):
+        """
+        A tool to search for flight information using a flight number via AviationStack API.
+        """
+        super().__init__()
+        self.tag = "flight_search"
+        self.api_key = api_key or os.getenv("AVIATIONSTACK_API_KEY")
+
+    def execute(self, blocks: str, safety: bool = True) -> str:
+        if self.api_key is None:
+            return "Error: No AviationStack API key provided."
+        
+        for block in blocks:
+            flight_number = block.strip()
+            if not flight_number:
+                return "Error: No flight number provided."
+
+            try:
+                url = "http://api.aviationstack.com/v1/flights"
+                params = {
+                    "access_key": self.api_key,
+                    "flight_iata": flight_number,
+                    "limit": 1
+                }
+                response = requests.get(url, params=params)
+                response.raise_for_status()
+
+                data = response.json()
+                if "data" in data and len(data["data"]) > 0:
+                    flight = data["data"][0]
+                    # Extract key flight information
+                    flight_status = flight.get("flight_status", "Unknown")
+                    departure = flight.get("departure", {})
+                    arrival = flight.get("arrival", {})
+                    airline = flight.get("airline", {}).get("name", "Unknown")
+                    
+                    departure_airport = departure.get("airport", "Unknown")
+                    departure_time = departure.get("scheduled", "Unknown")
+                    arrival_airport = arrival.get("airport", "Unknown")
+                    arrival_time = arrival.get("scheduled", "Unknown")
+
+                    return (
+                        f"Flight: {flight_number}\n"
+                        f"Airline: {airline}\n"
+                        f"Status: {flight_status}\n"
+                        f"Departure: {departure_airport} at {departure_time}\n"
+                        f"Arrival: {arrival_airport} at {arrival_time}"
+                    )
+                else:
+                    return f"No flight information found for {flight_number}"
+            except requests.RequestException as e:
+                return f"Error during flight search: {str(e)}"
+            except Exception as e:
+                return f"Unexpected error: {str(e)}"
+        return "No flight search performed"
+
+    def execution_failure_check(self, output: str) -> bool:
+        return output.startswith("Error") or "No flight information found" in output
+
+    def interpreter_feedback(self, output: str) -> str:
+        if self.execution_failure_check(output):
+            return f"Flight search failed: {output}"
+        return f"Flight information:\n{output}"
+
+
+if __name__ == "__main__":
+    flight_tool = FlightSearch()
+    flight_number = "AA123"
+    result = flight_tool.execute([flight_number], safety=True)
+    feedback = flight_tool.interpreter_feedback(result)
+    print(feedback)

sources/tools/webSearch.py

@@ -1,6 +1,9 @@
 
 import os
 import requests
+import dotenv
+
+dotenv.load_dotenv()
 
 if __name__ == "__main__":
     from tools import Tools
@@ -15,10 +18,10 @@ class webSearch(Tools):
         super().__init__()
         self.tag = "web_search"
         self.api_key = api_key or os.getenv("SERPAPI_KEY")  # Requires a SerpApi key
-        if not self.api_key:
-            raise ValueError("SerpApi key is required for webSearch tool. Set SERPAPI_KEY environment variable or pass it to the constructor.")
 
     def execute(self, blocks: str, safety: bool = True) -> str:
+        if self.api_key is None:
+            return "Error: No SerpApi key provided."
         for block in blocks:
             query = block.strip()
             if not query:
@@ -60,7 +63,7 @@ class webSearch(Tools):
 
 
 if __name__ == "__main__":
-    search_tool = webSearch(api_key="c4da252b63b0fc3cbf2c7dd98b931ae632aecf3feacbbfe099e17872eb192c44")
+    search_tool = webSearch(api_key=os.getenv("SERPAPI_KEY"))
     query = "when did covid start"
     result = search_tool.execute(query, safety=True)
     feedback = search_tool.interpreter_feedback(result)