Improved error handling for existing threads

VRSEN · Aug 13, 2024 · c24be1c · c24be1c
1 parent a32329f
commit c24be1c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/agency_swarm/threads/thread.py b/agency_swarm/threads/thread.py
@@ -404,6 +404,11 @@ def create_message(self, message: str, role: str = "user", attachments: List[dic
                     thread_id=f"thread_{thread_id}",
                     run_id=f"run_{run_id}"
                 )
+                self.run = self.client.beta.threads.runs.poll(
+                    thread_id=self.id,
+                    run_id=self.run.id,
+                    poll_interval_ms=500,
+                )
                 return self.client.beta.threads.messages.create(
                     thread_id=self.id,
                     role=role,