Update agents.py

agents.py

@@ -10,6 +10,11 @@ from utils import function_to_schema
 
 MODEL = "gpt-4o-mini"
 
+_assistant = None
+
+def set_current_agent(agent):
+    _assistant = agent
+
 def create_triage_agent(client):
     return client.beta.assistants.create(
         name="Triage Agent",