langgraph-api 0.0.41__py3-none-any.whl → 0.0.43__py3-none-any.whl

langgraph_api/api/assistants.py

@@ -124,6 +124,7 @@ async def create_assistant(request: ApiRequest) -> ApiResponse:
             metadata=payload.get("metadata") or {},
             if_exists=payload.get("if_exists") or "raise",
             name=payload.get("name") or "Untitled",
+            description=payload.get("description"),
         )
     return ApiResponse(await fetchone(assistant, not_found_code=409))
 
@@ -312,6 +313,7 @@ async def patch_assistant(
             graph_id=payload.get("graph_id"),
             metadata=payload.get("metadata"),
             name=payload.get("name"),
+            description=payload.get("description"),
         )
     return ApiResponse(await fetchone(assistant))
 

langgraph_api/api/runs.py

@@ -273,7 +273,7 @@ async def wait_run_stateless(request: ApiRequest):
             except TimeoutError:
                 yield b"\n"
             except asyncio.CancelledError:
-                stream.cancel()
+                stream.cancel("Run stream cancelled")
                 await stream
                 raise
 

langgraph_api/asyncio.py

@@ -63,7 +63,9 @@ async def wait_if_not_done(coro: Coroutine[Any, Any, T], done: ValueEvent) -> T:
         async with asyncio.TaskGroup() as tg:
             coro_task = tg.create_task(coro)
             done_task = tg.create_task(done.wait())
-            coro_task.add_done_callback(lambda _: done_task.cancel())
+            coro_task.add_done_callback(
+                lambda _: done_task.cancel("Coro task completed")
+            )
             done_task.add_done_callback(lambda _: coro_task.cancel(done._value))
             try:
                 return await coro_task
@@ -131,6 +133,7 @@ class SimpleTaskGroup(AbstractAsyncContextManager["SimpleTaskGroup"]):
         cancel: bool = False,
         wait: bool = True,
         taskset: set[asyncio.Task] | None = None,
+        taskgroup_name: str | None = None,
     ) -> None:
         self.tasks = taskset if taskset is not None else set()
         self.cancel = cancel
@@ -140,6 +143,7 @@ class SimpleTaskGroup(AbstractAsyncContextManager["SimpleTaskGroup"]):
                 task.add_done_callback(partial(self._create_task_done_callback, ()))
         for coro in coros:
             self.create_task(coro)
+        self.taskgroup_name = f" {taskgroup_name} " if taskgroup_name else ""
 
     def _create_task_done_callback(
         self, ignore_exceptions: tuple[Exception, ...], task: asyncio.Task
@@ -175,7 +179,7 @@ class SimpleTaskGroup(AbstractAsyncContextManager["SimpleTaskGroup"]):
         # cancel all tasks
         if self.cancel:
             for task in tasks:
-                task.cancel()
+                task.cancel(f"Task group{self.taskgroup_name}cancelled.")
         # wait for all tasks
         if self.wait:
             await asyncio.gather(*tasks, return_exceptions=True)

langgraph_api/config.py

@@ -105,6 +105,20 @@ class StoreConfig(TypedDict, total=False):
     ttl: TTLConfig
 
 
+class CheckpointerConfig(TypedDict, total=False):
+    """Configuration for the built-in checkpointer, which handles checkpointing of state.
+
+    If omitted, no checkpointer is set up (the object store will still be present, however).
+    """
+
+    ttl: ThreadTTLConfig | None
+    """Optional. Defines the TTL (time-to-live) behavior configuration.
+    
+    If provided, the checkpointer will apply TTL settings according to the configuration.
+    If omitted, no TTL behavior is configured.
+    """
+
+
 # env
 
 env = Config()
@@ -223,9 +237,14 @@ def _parse_thread_ttl(value: str | None) -> ThreadTTLConfig | None:
     }
 
 
+CHECKPOINTER_CONFIG: CheckpointerConfig | None = env(
+    "LANGGRAPH_CHECKPOINTER", cast=_parse_json, default=None
+)
 THREAD_TTL: ThreadTTLConfig | None = env(
     "LANGGRAPH_THREAD_TTL", cast=_parse_thread_ttl, default=None
 )
+if THREAD_TTL is None and CHECKPOINTER_CONFIG is not None:
+    THREAD_TTL = CHECKPOINTER_CONFIG.get("ttl")
 
 N_JOBS_PER_WORKER = env("N_JOBS_PER_WORKER", cast=int, default=10)
 BG_JOB_TIMEOUT_SECS = env("BG_JOB_TIMEOUT_SECS", cast=float, default=3600)

langgraph_api/graph.py

@@ -291,7 +291,7 @@ def _handle_exception(task: asyncio.Task) -> None:
 async def stop_remote_graphs() -> None:
     logger.info("Cancelling remote graphs")
     for task in js_bg_tasks:
-        task.cancel()
+        task.cancel("Stopping remote graphs.")
 
 
 def verify_graphs() -> None:

langgraph_api/lifespan.py

@@ -47,7 +47,9 @@ async def lifespan(
     await start_pool()
     await collect_graphs_from_env(True)
     try:
-        async with SimpleTaskGroup(cancel=True, taskset=taskset) as tg:
+        async with SimpleTaskGroup(
+            cancel=True, taskset=taskset, taskgroup_name="Lifespan"
+        ) as tg:
             tg.create_task(metadata_loop())
             if config.N_JOBS_PER_WORKER > 0:
                 tg.create_task(queue())

langgraph_api/schema.py

@@ -73,6 +73,8 @@ class Assistant(TypedDict):
     """The ID of the graph."""
     name: str
     """The name of the assistant."""
+    description: str | None
+    """The description of the assistant."""
     config: Config
     """The assistant config."""
     created_at: datetime

langgraph_api/thread_ttl.py

@@ -25,7 +25,8 @@ async def thread_ttl_sweep_loop():
         )
     sweep_interval_minutes = thread_ttl_config.get("sweep_interval_minutes", 5)
     await logger.ainfo(
-        "Starting thread TTL sweeper",
+        f"Starting thread TTL sweeper with interval {sweep_interval_minutes} minutes",
+        strategy=strategy,
         interval_minutes=sweep_interval_minutes,
     )
 
@@ -38,7 +39,7 @@ async def thread_ttl_sweep_loop():
                 threads_processed, threads_deleted = await Threads.sweep_ttl(conn)
                 if threads_processed > 0:
                     await logger.ainfo(
-                        "Thread TTL sweep completed",
+                        f"Thread TTL sweep completed. Processed {threads_processed}",
                         threads_processed=threads_processed,
                         threads_deleted=threads_deleted,
                     )

{langgraph_api-0.0.41.dist-info → langgraph_api-0.0.43.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langgraph-api
-Version: 0.0.41
+Version: 0.0.43
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos

{langgraph_api-0.0.41.dist-info → langgraph_api-0.0.43.dist-info}/RECORD

@@ -1,15 +1,15 @@
 LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
 langgraph_api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_api/api/__init__.py,sha256=qNcg8QJydef0gM-vYJlxITMRZw-9r1vw8zqm2raqqYE,5493
-langgraph_api/api/assistants.py,sha256=0s6Bihek8g7hPkldFoikIe1zfbRd2SBzrBLfY4smL0o,14249
+langgraph_api/api/assistants.py,sha256=WwaBtx1MpGn9gdJ8P9fkorJHMVrJKHt1nvtw0OCZcdw,14353
 langgraph_api/api/mcp.py,sha256=KbR19dtFCpJEiKYj3IfepAuJij8YZVELuVp7JY_yu_o,13754
 langgraph_api/api/meta.py,sha256=ifJ_Ki0Qf2DYbmY6OKlqKhLGxbt55gm0lEqH1A0cJbw,2790
 langgraph_api/api/openapi.py,sha256=f9gfmWN2AMKNUpLCpSgZuw_aeOF9jCXPdOtFT5PaTWM,10960
-langgraph_api/api/runs.py,sha256=uijgtrw_FSf1lZHsx8pM-CIj_ur2O88Y7ys-CJZ4SNQ,17988
+langgraph_api/api/runs.py,sha256=qAXfgZjjaMBfwPnlnAogvemtKZZeshNSIMQqcW20tGs,18010
 langgraph_api/api/store.py,sha256=XNNZFRlvgReidq9u7mg-i7pjwz21BdP9Qw3Jr5Ra9Fk,5447
 langgraph_api/api/threads.py,sha256=QbAy7MRupWKDUhmC6_LKU2ExJbLJ6Z-CJG2gSpcMXtc,9163
 langgraph_api/api/ui.py,sha256=LiOZVewKOPbKEykCm30hCEaOA7vuS_Ti5hB32EEy4vw,2082
-langgraph_api/asyncio.py,sha256=hVuAxWTHoUyNqTzcIEKTkAvh7HFrdGK6WIDowDxORAE,8397
+langgraph_api/asyncio.py,sha256=h0eZ7aoDGnJpoxnHLZABVlj1jQ78UxjgiHntTmAEWek,8613
 langgraph_api/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_api/auth/custom.py,sha256=sPYkF-xTdGIn_WMSE2tfR_azg_azzXPRWwUfCye6GQ8,21401
 langgraph_api/auth/langsmith/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -20,10 +20,10 @@ langgraph_api/auth/noop.py,sha256=Bk6Nf3p8D_iMVy_OyfPlyiJp_aEwzL-sHrbxoXpCbac,58
 langgraph_api/auth/studio_user.py,sha256=FzFQRROKDlA9JjtBuwyZvk6Mbwno5M9RVYjDO6FU3F8,186
 langgraph_api/cli.py,sha256=X76TYnErQPjRkZhK-n6gthCUmtItkwqVrRPvwZswF5Y,12335
 langgraph_api/command.py,sha256=3O9v3i0OPa96ARyJ_oJbLXkfO8rPgDhLCswgO9koTFA,768
-langgraph_api/config.py,sha256=y-dmkmd929WDosIO0nUmcQicZN2XozJrVHzeBGtLhUg,9418
+langgraph_api/config.py,sha256=N6P_6jtr6rNGC7T5lSWoCbaTQhXbWi-F4YmH0NyPbAI,10141
 langgraph_api/cron_scheduler.py,sha256=9yzbbGxzNgJdIg4ZT7yu2oTwT_wRuPxD1c2sbbd52xs,2630
 langgraph_api/errors.py,sha256=Bu_i5drgNTyJcLiyrwVE_6-XrSU50BHf9TDpttki9wQ,1690
-langgraph_api/graph.py,sha256=jpMMMgn8uHqDc3gPu02tF7B8af0TV153t2F_kodcBvI,18799
+langgraph_api/graph.py,sha256=7NKAjtslNZq65J1tYhvJfpcP_P8JVTRra7vOfGuUWvc,18824
 langgraph_api/http.py,sha256=gYbxxjY8aLnsXeJymcJ7G7Nj_yToOGpPYQqmZ1_ggfA,5240
 langgraph_api/js/.gitignore,sha256=l5yI6G_V6F1600I1IjiUKn87f4uYIrBAYU1MOyBBhg4,59
 langgraph_api/js/base.py,sha256=xkBp5bwRrbpMFaAMViEU-qIlnsJuu3X_G8sa1pqNZK0,227
@@ -61,7 +61,7 @@ langgraph_api/js/tests/graphs/yarn.lock,sha256=i2AAIgXA3XBLM8-oU45wgUefCSG-Tne4g
 langgraph_api/js/tests/parser.test.mts,sha256=dEC8KTqKygeb1u39ZvpPqCT4HtfPD947nLmITt2buxA,27883
 langgraph_api/js/tests/utils.mts,sha256=q1V9gvT63v95onlfK9W4iv3n9ZJO3h-0RD9TdDYuRyY,439
 langgraph_api/js/yarn.lock,sha256=JtMDtRVX9kr9lK0rbWRQsGHYbkQSmzNa4NwTunBv58U,82773
-langgraph_api/lifespan.py,sha256=0F4Xn6IC4wUfvBWbm6KCJ-jiF0S7iWDOHJYJZ-A3o3s,2739
+langgraph_api/lifespan.py,sha256=rrmxgNrAxWa_D1H8AZTBKXoQoyiLeDv_C5z-byuxTWM,2788
 langgraph_api/logging.py,sha256=JJIzbNIgLCN6ClQ3tA-Mm5ffuBGvpRDSZsEvnIlsuu4,3693
 langgraph_api/metadata.py,sha256=5Mu3MUtUc-iIocU3X2SZDoGIqnUmTdT3517MhP94npI,3495
 langgraph_api/middleware/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -72,13 +72,13 @@ langgraph_api/models/run.py,sha256=85-pyyvosFAot8WrWl2QjCje-2c4lhattYvoAEMqztA,1
 langgraph_api/patch.py,sha256=82xjuFqY7tgrUm-k1XWHI6k8S6QovSD0zhe--8_xW4o,1296
 langgraph_api/queue_entrypoint.py,sha256=4xICUxXarNV8DhnaqAMhVi3xCmyVKCL3J5NzHxPA9Xc,1835
 langgraph_api/route.py,sha256=fM4qYCGbmH0a3_cV8uKocb1sLklehxO6HhdRXqLK6OM,4421
-langgraph_api/schema.py,sha256=hNbg6ep2wiGBBtBJVNBgMYA8uC33AfaqhRXXVUY_At8,5361
+langgraph_api/schema.py,sha256=Frh_YOC3S1cDAMPUVanNi78ooSXK2WFpu9YkIVz5h14,5433
 langgraph_api/serde.py,sha256=D5t_HeABMYKRAsoXWaWG0IsdaYM8yOXIez2wJUTIgT8,3963
 langgraph_api/server.py,sha256=bnXOOYztQmqR-QVpEFoRWB5Fzd33PuEIlwBK2R7W8NE,4849
 langgraph_api/sse.py,sha256=2wNodCOP2eg7a9mpSu0S3FQ0CHk2BBV_vv0UtIgJIcc,4034
 langgraph_api/state.py,sha256=8jx4IoTCOjTJuwzuXJKKFwo1VseHjNnw_CCq4x1SW14,2284
 langgraph_api/stream.py,sha256=lhjnom-T8GbquUZry-KSkajnqYjElaIERhPiXPtpw1E,11354
-langgraph_api/thread_ttl.py,sha256=ubkWMymTR7p9nGWd60-WKEOQ20ZgIkWB6WGstQUmRS4,1658
+langgraph_api/thread_ttl.py,sha256=QS9C7cx63KBh82_I6DqOb42-5mLTqTvneL8FJeX3aqU,1765
 langgraph_api/utils.py,sha256=92mSti9GfGdMRRWyESKQW5yV-75Z9icGHnIrBYvdypU,3619
 langgraph_api/validation.py,sha256=Qo3EkSRXzIRe3GRuqRWbElTcUXRMXMyA1w0VbMvdwME,4934
 langgraph_api/webhook.py,sha256=1ncwO0rIZcj-Df9sxSnFEzd1gP1bfS4okeZQS8NSRoE,1382
@@ -88,17 +88,16 @@ langgraph_license/middleware.py,sha256=_ODIYzQkymr6W9_Fp9wtf1kAQspnpsmr53xuzyF2G
 langgraph_license/validation.py,sha256=Uu_G8UGO_WTlLsBEY0gTVWjRR4czYGfw5YAD3HLZoj0,203
 langgraph_storage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_storage/checkpoint.py,sha256=Qq0y6vdh27qdF3h5nOLT5CcX9Rj2bcFqkVOMeCaGoK4,4036
-langgraph_storage/database.py,sha256=ys2glLtQaspfIY-_EgqEOT3b7a5WqBwuE_9I5VAh6FM,5821
+langgraph_storage/database.py,sha256=sZjZvMcvbr_6dX0d1YrYEccVuQozIfkiWt8bdlXGVYU,5849
 langgraph_storage/inmem_stream.py,sha256=LjJSAxsh_E0ywqEMzdWJk8Hy_Jn9oQByzycss-fANng,3264
-langgraph_storage/ops.py,sha256=6NxeirfuDAKghHc328zejnbTsijg6U80Rlt5rXTC2dc,75392
-langgraph_storage/queue.py,sha256=wyBcY1V_8zah0Z2e5hpovs_dsIPPPfUJQCly2rxFWvo,7039
+langgraph_storage/ops.py,sha256=S2qXgfyuCr5qkKLqIV-kZNRK924bBSLZ2qdLtExfi-M,75939
+langgraph_storage/queue.py,sha256=d2hykynk3rqrixEdrD_zl0Zyo63wJs3S4SJOzFeF-SA,7153
 langgraph_storage/retry.py,sha256=XmldOP4e_H5s264CagJRVnQMDFcEJR_dldVR1Hm5XvM,763
 langgraph_storage/store.py,sha256=JB9jZ87GE19MVN9wgl3-esgR2eIkeipws9q6qsPWkgc,3399
-langgraph_storage/ttl_dict.py,sha256=FlpEY8EANeXWKo_G5nmIotPquABZGyIJyk6HD9u6vqY,1533
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
 openapi.json,sha256=-25y3NRQ88e_944UXo76Goa34HJhC7pj6I9tjYUwvuE,131492
-langgraph_api-0.0.41.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.0.41.dist-info/METADATA,sha256=0yewVWyEzw_HfLoOEBCQS-OymzT8q0JUg_6ux-SQE8Y,4167
-langgraph_api-0.0.41.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-langgraph_api-0.0.41.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
-langgraph_api-0.0.41.dist-info/RECORD,,
+langgraph_api-0.0.43.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.0.43.dist-info/METADATA,sha256=GVLyrH4YCy2zRADllPQpBtuD8BHQwVaBitZPZlefgeQ,4167
+langgraph_api-0.0.43.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+langgraph_api-0.0.43.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
+langgraph_api-0.0.43.dist-info/RECORD,,

langgraph_storage/database.py

@@ -23,6 +23,7 @@ class Assistant(TypedDict):
     assistant_id: UUID
     graph_id: str
     name: str
+    description: str | None
     created_at: NotRequired[datetime]
     updated_at: NotRequired[datetime]
     config: dict[str, Any]

langgraph_storage/ops.py

@@ -177,6 +177,7 @@ class Assistants(Authenticated):
         if_exists: OnConflictBehavior,
         name: str,
         ctx: Auth.types.BaseAuthContext | None = None,
+        description: str | None = None,
     ) -> AsyncIterator[Assistant]:
         """Insert an assistant."""
         assistant_id = _ensure_uuid(assistant_id)
@@ -224,6 +225,7 @@ class Assistants(Authenticated):
             "created_at": now,
             "updated_at": now,
             "version": 1,
+            "description": description,
         }
         new_version = {
             "assistant_id": assistant_id,
@@ -251,16 +253,20 @@ class Assistants(Authenticated):
         graph_id: str | None = None,
         metadata: MetadataInput | None = None,
         name: str | None = None,
+        description: str | None = None,
         ctx: Auth.types.BaseAuthContext | None = None,
     ) -> AsyncIterator[Assistant]:
         """Update an assistant.
 
         Args:
+            conn: The connection to the in-memory store.
             assistant_id: The assistant ID.
             graph_id: The graph ID.
             config: The assistant config.
             metadata: The assistant metadata.
             name: The assistant name.
+            description: The assistant description.
+            ctx: The auth context.
 
         Returns:
             return the updated assistant model.
@@ -317,6 +323,9 @@ class Assistants(Authenticated):
             "metadata": metadata if metadata is not None else assistant["metadata"],
             "created_at": now,
             "name": name if name is not None else assistant["name"],
+            "description": description
+            if description is not None
+            else assistant["description"],
         }
         conn.store["assistant_versions"].append(new_version_entry)
 
@@ -327,6 +336,9 @@ class Assistants(Authenticated):
                 "config": new_version_entry["config"],
                 "metadata": new_version_entry["metadata"],
                 "name": name if name is not None else assistant["name"],
+                "description": description
+                if description is not None
+                else assistant["description"],
                 "updated_at": now,
                 "version": new_version,
             }
@@ -1350,7 +1362,7 @@ class Runs(Authenticated):
         # Get queue for this run
         queue = await Runs.Stream.subscribe(run_id)
 
-        async with SimpleTaskGroup(cancel=True) as tg:
+        async with SimpleTaskGroup(cancel=True, taskgroup_name="Runs.enter") as tg:
             done = ValueEvent()
             tg.create_task(listen_for_cancellation(queue, run_id, done))
 

langgraph_storage/queue.py

@@ -141,9 +141,9 @@ async def queue():
             bb.deactivate()
         logger.info("Shutting down background workers")
         for task in WORKERS:
-            task.cancel()
+            task.cancel("Shutting down background workers.")
         for task in WEBHOOKS:
-            task.cancel()
+            task.cancel("Shutting down webhooks for background workers.")
         await asyncio.wait_for(
             asyncio.gather(*WORKERS, *WEBHOOKS, return_exceptions=True),
             SHUTDOWN_GRACE_PERIOD_SECS,
@@ -156,20 +156,21 @@ def _enable_blockbuster() -> BlockBuster:
     bb.activate()
     # Note, we've cached this call in langsmith==0.3.21 so it shouldn't raise anyway
     # but we don't want to raise teh minbound just for that.
-    for func in ["os.stat", "io.TextIOWrapper.read"]:
+    bb.functions["os.stat"].deactivate()
+    for func in ("io.TextIOWrapper.read",):
         bb.functions[func].can_block_in("langsmith/client.py", "_default_retry_config")
         # Only triggers in python 3.11 for getting subgraphs
         # Will be unnecessary once we cache the assistant schemas
         bb.functions[func].can_block_in(
             "langgraph/pregel/utils.py", "get_function_nonlocals"
         )
-    for module, func in [
+    for module, func in (
         ("uvicorn/lifespan/on.py", "startup"),
         ("uvicorn/lifespan/on.py", "shutdown"),
         ("ansitowin32.py", "write_plain_text"),
         ("logging/__init__.py", "flush"),
         ("logging/__init__.py", "emit"),
-    ]:
+    ):
         bb.functions["io.TextIOWrapper.write"].can_block_in(module, func)
         bb.functions["io.BufferedWriter.write"].can_block_in(module, func)
     return bb

langgraph_storage/ttl_dict.py

@@ -1,54 +0,0 @@
-import asyncio
-
-
-# TODO: Make this class thread-safe
-class TTLDict:
-    def __init__(self, timeout=10):
-        self._store = {}
-        self._timeouts = {}
-        self._tasks = {}  # Track tasks for key expiry
-        self._timeout = timeout
-
-    async def _remove_key_after_timeout(self, key):
-        await asyncio.sleep(self._timeouts[key])
-        if key in self._store:
-            del self._store[key]
-            del self._timeouts[key]
-
-    def set(self, key, value, timeout=None):
-        self._store[key] = value
-        key_timeout = timeout if timeout is not None else self._timeout
-        self._timeouts[key] = key_timeout
-
-        # Cancel the previous task if it exists
-        if key in self._tasks:
-            self._tasks[key].cancel()
-
-        # Schedule the removal of the key
-        self._tasks[key] = asyncio.create_task(self._remove_key_after_timeout(key))
-
-    def get(self, key, default=None):
-        return self._store.get(key, default)
-
-    def __contains__(self, key):
-        return key in self._store
-
-    def remove(self, key):
-        if key in self._store:
-            del self._store[key]
-            del self._timeouts[key]
-            if key in self._tasks:
-                self._tasks[key].cancel()
-                del self._tasks[key]
-
-    def keys(self):
-        return self._store.keys()
-
-    def values(self):
-        return self._store.values()
-
-    def items(self):
-        return self._store.items()
-
-    def __repr__(self):
-        return f"{self.__class__.__name__}({self._store})"