Update agents.py

agents.py

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