Skip to content

Commit a46d67d

Browse files
committed
feat: add few more logging statements
1 parent 4de22e9 commit a46d67d

2 files changed

Lines changed: 6 additions & 0 deletions

File tree

adk-chatkit/src/adk_chatkit/_store.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -212,6 +212,7 @@ async def add_thread_item(self, thread_id: str, item: ThreadItem, context: ADKCo
212212
actions=actions_with_update,
213213
timestamp=datetime.now().timestamp(),
214214
)
215+
_LOGGER.info(f"Appending system event to session {session.id} for adding thread item {item.id}")
215216
await self._session_service.append_event(session, system_event)
216217

217218
async def save_attachment(self, attachment: Attachment, context: ADKContext) -> None:

examples/backend/src/backend/_refreshed_session_service.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,22 +1,27 @@
1+
import logging
12
from typing import Any
23

34
from google.adk.events import Event
45
from google.adk.sessions import Session
56
from google.adk.sessions.database_session_service import DatabaseSessionService
67

8+
_LOGGER = logging.getLogger(__name__)
9+
710

811
class RefreshedSessionService(DatabaseSessionService):
912
def __init__(self, db_url: str, **kwargs: Any) -> None:
1013
super().__init__(db_url, **kwargs)
1114

1215
async def append_event(self, session: Session, event: Event) -> Event:
1316
"""Append event with session refresh to prevent stale session errors."""
17+
_LOGGER.info("Appending event to session %s", session.id)
1418
# Refresh the session before appending to get the latest state
1519
refreshed_session = await self.get_session(
1620
app_name=session.app_name, user_id=session.user_id, session_id=session.id
1721
)
1822

1923
if refreshed_session:
24+
_LOGGER.info("Session %s refreshed before appending event ", session.id)
2025
session.last_update_time = refreshed_session.last_update_time
2126
session.events = refreshed_session.events
2227
session.state = refreshed_session.state

0 commit comments

Comments
 (0)