From df7ef67ba43bc63c29abb02ef9e38569204a5fe6 Mon Sep 17 00:00:00 2001 From: caiyuwang Date: Tue, 17 Mar 2026 18:09:55 +0800 Subject: [PATCH 1/2] feat: support initial state when creating session in short term memory --- veadk/memory/short_term_memory.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/veadk/memory/short_term_memory.py b/veadk/memory/short_term_memory.py index a161376a..92756806 100644 --- a/veadk/memory/short_term_memory.py +++ b/veadk/memory/short_term_memory.py @@ -138,6 +138,7 @@ async def create_session( app_name: str, user_id: str, session_id: str, + state: dict | None = None, ) -> Session | None: """Create or retrieve a user session. @@ -151,6 +152,7 @@ async def create_session( app_name (str): The name of the application associated with the session. user_id (str): The unique identifier of the user. session_id (str): The unique identifier of the session to be created or retrieved. + state (dict | None): The initial state of the session. Returns: Session | None: The retrieved or newly created `Session` object, or `None` if the session creation failed. @@ -175,7 +177,7 @@ async def create_session( return session else: return await self._session_service.create_session( - app_name=app_name, user_id=user_id, session_id=session_id + app_name=app_name, user_id=user_id, session_id=session_id, state=state ) async def generate_profile( From 52051d14b4ba55c841ae2911bdd9c5cc4fc0b7b1 Mon Sep 17 00:00:00 2001 From: caiyuwang Date: Tue, 17 Mar 2026 19:11:10 +0800 Subject: [PATCH 2/2] docs: clarify state parameter if session exists --- veadk/memory/short_term_memory.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/veadk/memory/short_term_memory.py b/veadk/memory/short_term_memory.py index 92756806..db964cdf 100644 --- a/veadk/memory/short_term_memory.py +++ b/veadk/memory/short_term_memory.py @@ -152,7 +152,10 @@ async def create_session( app_name (str): The name of the application associated with the session. user_id (str): The unique identifier of the user. session_id (str): The unique identifier of the session to be created or retrieved. - state (dict | None): The initial state of the session. + state (dict | None): + The initial state of the session. + If a session with the given `session_id` already exists, + this argument is ignored and the existing session state is preserved. Returns: Session | None: The retrieved or newly created `Session` object, or `None` if the session creation failed.