Merge pull request lightspeed-core#1681 from tisnik/lcore-2111-type-hints-in-a2s-storage

tisnik · web-flow · commit 6af0a2a13356 · 2026-05-05T10:36:00.000+02:00
LCORE-2111: Type hints in a2a storage module
diff --git a/src/a2a_storage/postgres_context_store.py b/src/a2a_storage/postgres_context_store.py
@@ -83,7 +83,7 @@ async def get(self, context_id: str) -> Optional[str]:
                 a2a_context_table.c.context_id == context_id
             )
             result = await session.execute(stmt)
-            row = result.scalar_one_or_none()
+            row: Optional[str] = result.scalar_one_or_none()
 
             if row:
                 logger.debug("Context %s maps to conversation %s", context_id, row)
diff --git a/src/a2a_storage/sqlite_context_store.py b/src/a2a_storage/sqlite_context_store.py
@@ -82,7 +82,7 @@ async def get(self, context_id: str) -> Optional[str]:
                 a2a_context_table.c.context_id == context_id
             )
             result = await session.execute(stmt)
-            row = result.scalar_one_or_none()
+            row: Optional[str] = result.scalar_one_or_none()
 
             if row:
                 logger.debug("Context %s maps to conversation %s", context_id, row)

Original file line number	Diff line number	Diff line change
`@@ -83,7 +83,7 @@ async def get(self, context_id: str) -> Optional[str]:`
`83`	`83`	`a2a_context_table.c.context_id == context_id`
`84`	`84`	`)`
`85`	`85`	`result = await session.execute(stmt)`
`86`		`- row = result.scalar_one_or_none()`
	`86`	`+ row: Optional[str] = result.scalar_one_or_none()`
`87`	`87`
`88`	`88`	`if row:`
`89`	`89`	`logger.debug("Context %s maps to conversation %s", context_id, row)`
Original file line number	Diff line number	Diff line change
`@@ -82,7 +82,7 @@ async def get(self, context_id: str) -> Optional[str]:`
`82`	`82`	`a2a_context_table.c.context_id == context_id`
`83`	`83`	`)`
`84`	`84`	`result = await session.execute(stmt)`
`85`		`- row = result.scalar_one_or_none()`
	`85`	`+ row: Optional[str] = result.scalar_one_or_none()`
`86`	`86`
`87`	`87`	`if row:`
`88`	`88`	`logger.debug("Context %s maps to conversation %s", context_id, row)`