fix _run_pipeline

deeppavlov · Dec 5, 2023 · 5c22cc7 · 5c22cc7
1 parent de97a12
commit 5c22cc7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/dff/pipeline/pipeline/pipeline.py b/dff/pipeline/pipeline/pipeline.py
@@ -322,7 +322,9 @@ async def _run_pipeline(self, request: Message, ctx_id: Optional[Hashable] = Non
         :param ctx_id: Current dialog id; if `None`, new dialog will be created.
         :return: Dialog `Context`.
         """
-        if isinstance(self.context_storage, DBContextStorage):
+        if ctx_id is None:
+            ctx = Context()
+        elif isinstance(self.context_storage, DBContextStorage):
             ctx = await self.context_storage.get_async(ctx_id, Context(id=ctx_id))
         else:
             ctx = self.context_storage.get(ctx_id, Context(id=ctx_id))