Skip to content

Commit 73ef275

Browse files
committed
fix: improve type hinting for memory extension modules
1 parent 73e7843 commit 73ef275

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

src/agents/extensions/memory/__init__.py

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,18 @@
88

99
from __future__ import annotations
1010

11-
from typing import Any
11+
from typing import TYPE_CHECKING, Any
12+
13+
if TYPE_CHECKING:
14+
from .advanced_sqlite_session import AdvancedSQLiteSession
15+
from .dapr_session import (
16+
DAPR_CONSISTENCY_EVENTUAL,
17+
DAPR_CONSISTENCY_STRONG,
18+
DaprSession,
19+
)
20+
from .encrypt_session import EncryptedSession
21+
from .redis_session import RedisSession
22+
from .sqlalchemy_session import SQLAlchemySession
1223

1324
__all__: list[str] = [
1425
"AdvancedSQLiteSession",

0 commit comments

Comments
 (0)