diff --git a/veadk/memory/short_term_memory.py b/veadk/memory/short_term_memory.py
index a161376a..db964cdf 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,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.
+                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.
@@ -175,7 +180,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(