Update app.py

app.py

@@ -9,15 +9,38 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def get_error_tickets(arg1:str, arg2:int)-> str: #it's import to specify the return type
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that makes searches the databank reagarding existing problems
+def search_error_tickets(machine_name: str, error_code: int) -> str:
+    """
+    Simulates a search for error tickets in a system for a given machine and error code.
+    
     Args:
-        arg1: a machine name like MachineXY
-        arg2: a error code of the machine
+        machine_name: The name or identifier of the machine (e.g., "MachineXY").
+        error_code: The error code to search for.
+        
+    Returns:
+        A string summarizing the simulated search results.
     """
+    # Simulated database of error tickets
+    simulated_tickets = [
+        {"ticket_id": 2032, "machine": "MachineXY", "error_code": 404, "description": "Not Found"},
+        {"ticket_id": 2033, "machine": "MachineXY", "error_code": 500, "description": "Internal Server Error"},
+        {"ticket_id": 2034, "machine": "MachineAB", "error_code": 404, "description": "Not Found"},
+        {"ticket_id": 2035, "machine": "MachineXY", "error_code": 404, "description": "Resource missing"},
+    ]
     
-
+    # Filter the simulated tickets based on the provided machine name and error code.
+    matching_tickets = [
+        ticket for ticket in simulated_tickets
+        if ticket["machine"].lower() == machine_name.lower() and ticket["error_code"] == error_code
+    ]
+    
+    if matching_tickets:
+        results = "\n".join(
+            [f"Ticket {ticket['ticket_id']}: {ticket['description']}" for ticket in matching_tickets]
+        )
+        return f"Found the following tickets for machine '{machine_name}' with error code {error_code}:\n{results}"
+    else:
+        return f"No tickets found for machine '{machine_name}' with error code {error_code}."
     
     return f"if have found existing tickets for {arg1} with error code {arg2}, please check the ticketsystem"
 
@@ -58,7 +81,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_error_tickets], ## add your tools here (don't remove final answer)
+    tools=[final_answer, search_error_tickets], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,