update

Author: howieleung

sdk/ai/azure-ai-agents/azure/ai/agents/aio/operations/_patch.py

@@ -1934,6 +1934,8 @@ async def create_and_poll(
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
+        curr_time = time.monotonic()
+
         if body is not _Unset:
             if isinstance(body, dict):
                 vector_store_file_batch = await super().create(
@@ -1961,6 +1963,9 @@ async def create_and_poll(
             )
 
         while vector_store_file_batch.status == "in_progress":
+            if timeout is not None and (time.monotonic() - curr_time - sleep_interval) >= timeout:
+                raise TimeoutError("Timeout reached. Stopping polling.")
+
             await asyncio.sleep(sleep_interval)
             vector_store_file_batch = await super().get(
                 vector_store_id=vector_store_id, batch_id=vector_store_file_batch.id