mirror of
https://github.com/EvolutionAPI/adk-python.git
synced 2025-07-14 01:41:25 -06:00
5b3204c356
Also keep the sync version in the InMemorySessionService as create_session_sync() as a temporary migration option. PiperOrigin-RevId: 759224250
114 lines
3.1 KiB
Python
114 lines
3.1 KiB
Python
# Copyright 2025 Google LLC
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import abc
|
|
from typing import Any
|
|
from typing import Optional
|
|
|
|
from pydantic import BaseModel
|
|
from pydantic import Field
|
|
|
|
from ..events.event import Event
|
|
from .session import Session
|
|
from .state import State
|
|
|
|
|
|
class GetSessionConfig(BaseModel):
|
|
"""The configuration of getting a session."""
|
|
|
|
num_recent_events: Optional[int] = None
|
|
after_timestamp: Optional[float] = None
|
|
|
|
|
|
class ListSessionsResponse(BaseModel):
|
|
"""The response of listing sessions.
|
|
|
|
The events and states are not set within each Session object.
|
|
"""
|
|
|
|
sessions: list[Session] = Field(default_factory=list)
|
|
|
|
|
|
class BaseSessionService(abc.ABC):
|
|
"""Base class for session services.
|
|
|
|
The service provides a set of methods for managing sessions and events.
|
|
"""
|
|
|
|
@abc.abstractmethod
|
|
async def create_session(
|
|
self,
|
|
*,
|
|
app_name: str,
|
|
user_id: str,
|
|
state: Optional[dict[str, Any]] = None,
|
|
session_id: Optional[str] = None,
|
|
) -> Session:
|
|
"""Creates a new session.
|
|
|
|
Args:
|
|
app_name: the name of the app.
|
|
user_id: the id of the user.
|
|
state: the initial state of the session.
|
|
session_id: the client-provided id of the session. If not provided, a
|
|
generated ID will be used.
|
|
|
|
Returns:
|
|
session: The newly created session instance.
|
|
"""
|
|
|
|
@abc.abstractmethod
|
|
async def get_session(
|
|
self,
|
|
*,
|
|
app_name: str,
|
|
user_id: str,
|
|
session_id: str,
|
|
config: Optional[GetSessionConfig] = None,
|
|
) -> Optional[Session]:
|
|
"""Gets a session."""
|
|
|
|
@abc.abstractmethod
|
|
async def list_sessions(
|
|
self, *, app_name: str, user_id: str
|
|
) -> ListSessionsResponse:
|
|
"""Lists all the sessions."""
|
|
|
|
@abc.abstractmethod
|
|
async def delete_session(
|
|
self, *, app_name: str, user_id: str, session_id: str
|
|
) -> None:
|
|
"""Deletes a session."""
|
|
|
|
async def close_session(self, *, session: Session):
|
|
"""Closes a session."""
|
|
# TODO: determine whether we want to finalize the session here.
|
|
|
|
async def append_event(self, session: Session, event: Event) -> Event:
|
|
"""Appends an event to a session object."""
|
|
if event.partial:
|
|
return event
|
|
self.__update_session_state(session, event)
|
|
session.events.append(event)
|
|
return event
|
|
|
|
def __update_session_state(self, session: Session, event: Event):
|
|
"""Updates the session state based on the event."""
|
|
if not event.actions or not event.actions.state_delta:
|
|
return
|
|
for key, value in event.actions.state_delta.items():
|
|
if key.startswith(State.TEMP_PREFIX):
|
|
continue
|
|
session.state.update({key: value})
|