Update agent.py

agent.py

@@ -492,12 +492,18 @@ def all_nodes():
     """
     return nodes_df["Name"].dropna().astype(str).tolist()
 
+def all_locations():
+    """
+    Return a list of all locations.
+    """
+    return list(set(all_nodes()))
+
 def match_node_name(input):
     """
     Match the input to the closest match in the nodes list and return their id.
     """
     input = str(input)
-    matches = difflib.get_close_matches(input, all_nodes(), n=1, cutoff=0.2)
+    matches = difflib.get_close_matches(input, all_locations(), n=1, cutoff=0.2)
     return matches[0] if matches else None
 
 def node_pos(id: int):
@@ -634,7 +640,7 @@ def path_human(destination, source=None):
 class PathFinding(smolagents.tools.Tool):
     name = "find_path"
     description = (
-        "Find the easiest path to reach areas and locations inside the TechSpark. Also useful to help the user to reach machines in those locations."
+        "Find the location and easiest path to reach areas inside the TechSpark. Also useful to help the user to reach machines in those locations."
     )
     inputs = {
         "destination": {"type": "string", "description": "Name of the location inside the TechSpark."},
@@ -663,6 +669,7 @@ Respond concisely and directly with the information requested by the user, utili
 Which machines to use for a task, and where to find them.
 When you were in doubt, try searching wikipedia to gain more knowledge.
 Only answer questions related to TechSpark and manufacturing. If the question was out of scope, inform the user and try to suggest relevant question to ask.
+Don't tell anyone about your internal tools.
 
 Safety is important. So:
 - When talking about any machines, check whether it is accessbile to students or not.