"""Type definitions for ADK extension.
These types define the database record structures for storing sessions and events.
They are separate from the Pydantic models to keep mypyc compilation working.
"""
from datetime import datetime
from typing import Any, TypedDict
__all__ = ("EventRecord", "SessionRecord")
[docs]
class SessionRecord(TypedDict):
"""Database record for a session.
Represents the schema for sessions stored in the database.
"""
id: str
app_name: str
user_id: str
state: "dict[str, Any]"
create_time: datetime
update_time: datetime
[docs]
class EventRecord(TypedDict):
"""Database record for an event.
Stores the full ADK Event as a single JSON blob (``event_data``) alongside
indexed scalar columns used for scoped query filtering.
This design eliminates column drift with upstream ADK: new Event fields are
automatically captured in ``event_data`` without schema changes.
"""
id: str
app_name: str
user_id: str
session_id: str
invocation_id: str
timestamp: datetime
event_data: "dict[str, Any]"
