Update tools.py

tools.py

@@ -1,16 +1,36 @@
+current_agent, current_thread = None, None
+
+###
+
+def set_current_agent(agent):
+    current_agent = agent
+
+def set_current_thread(thread):
+    current_thread = thread
+
+def get_current_agent():
+    return current_agent
+
+def get_current_thread():
+    return current_thread
+
+###
+
 def transfer_to_sales_agent():
     """Use for anything sales or buying related."""
-    _assistant = sales_agent
+    current_agent = sales_agent
 
 def transfer_to_issues_repairs_agent():
     """Use for issues, repairs, or refunds."""
-    _assistant = issues_repairs_agent
+    current_agent = issues_repairs_agent
 
 def transfer_to_triage_agent():
     """Call this if the user brings up a topic outside of your purview,
     including escalating to human."""
-    _assistant = triage_agent
-    
+    current_agent = triage_agent
+
+###
+
 def escalate_to_human(summary):
     """Only call this if explicitly asked to."""
     print("Escalating to human agent...")
@@ -19,6 +39,8 @@ def escalate_to_human(summary):
     print("=========================\n")
     exit()
 
+###
+
 def execute_order(product, price: int):
     """Price should be in USD."""
     print("\n\n=== Order Summary ===")