Import ContextRepo defined in faststream_lifespan (#49)

* Import ContextRepo defined in faststream_lifespan * Import ContextRepo defined in faststream_lifespan
airtai · Aug 21, 2024 · 42be2e2 · 42be2e2
1 parent 37d3738
commit 42be2e2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fastagency/db/prisma.py b/fastagency/db/prisma.py
@@ -4,13 +4,13 @@
 from typing import TYPE_CHECKING, Any, AsyncGenerator, Dict, List, Optional, Union
 from uuid import UUID
 
+from faststream import ContextRepo
 from prisma import Prisma  # type: ignore[attr-defined]
 
 from .base import BackendDBProtocol, DefaultDB, FrontendDBProtocol, KeyNotFoundError
 
 if TYPE_CHECKING:
     from fastapi import FastAPI
-    from faststream import ContextRepo
 
 
 class PrismaBaseDB:
@@ -212,6 +212,6 @@ async def fastapi_lifespan(app: "FastAPI") -> AsyncGenerator[None, None]:
 
 
 @asynccontextmanager
-async def faststream_lifespan(context: "ContextRepo") -> AsyncGenerator[None, None]:
+async def faststream_lifespan(context: ContextRepo) -> AsyncGenerator[None, None]:
     async with _lifespan():
         yield