scriptoriumadmin/supermemory
1
0
Fork 0
mirror of https://github.com/supermemoryai/supermemory.git synced 2026-09-10 22:41:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

perf(middleware): await background task cancellation on timeout

Browse source
This commit is contained in:
Ishaan Gupta 2026-03-18 22:35:10 +05:30
parent c534008001
commit e97fdc338c
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
packages/openai-sdk-python/src/supermemory_openai/middleware.py
View file

@ -554,9 +554,12 @@ class SupermemoryOpenAIWrapper:
f"Background tasks did not complete within {timeout}s timeout"
)
# Cancel remaining tasks
for task in self._background_tasks:
if not task.done():
task.cancel()
tasks_to_cancel = [task for task in self._background_tasks if not task.done()]
for task in tasks_to_cancel:
task.cancel()
if tasks_to_cancel:
await asyncio.gather(*tasks_to_cancel, return_exceptions=True)
raise
def cancel_background_tasks(self) -> None: