fix: fix agent short_term_memory parameter passthrough issue

Author: sjy3

.gitignore

@@ -199,4 +199,6 @@ cython_debug/
 **./output
 
 *.mp3
-*.pcm
+*.pcm
+
+.trae

veadk/runner.py

@@ -122,20 +122,20 @@ def intercept_new_message(process_func):
     def decorator(func):
         @functools.wraps(func)
         async def wrapper(
-            self,
-            *,
-            user_id: str,
-            session_id: str,
-            new_message: types.Content,
-            **kwargs,
+                self,
+                *,
+                user_id: str,
+                session_id: str,
+                new_message: types.Content,
+                **kwargs,
         ):
             await pre_run_process(self, process_func, new_message, user_id, session_id)
 
             async for event in func(
-                user_id=user_id,
-                session_id=session_id,
-                new_message=new_message,
-                **kwargs,
+                    user_id=user_id,
+                    session_id=session_id,
+                    new_message=new_message,
+                    **kwargs,
             ):
                 if event is None:
                     logger.error(f"Event is None with new_message: {new_message}")
@@ -172,10 +172,10 @@ async def wrapper(
 
 
 def _convert_messages(
-    messages: RunnerMessage,
-    app_name: str,
-    user_id: str,
-    session_id: str,
+        messages: RunnerMessage,
+        app_name: str,
+        user_id: str,
+        session_id: str,
 ) -> list:
     """Convert a VeADK ``RunnerMessage`` into a list of Google ADK messages.
 
@@ -251,7 +251,7 @@ def _convert_messages(
 
 
 async def _upload_image_to_tos(
-    part: genai.types.Part, app_name: str, user_id: str, session_id: str
+        part: genai.types.Part, app_name: str, user_id: str, session_id: str
 ) -> None:
     """Upload inline media data in a message part to TOS and rewrite its URL.
 
@@ -316,15 +316,15 @@ class Runner(ADKRunner):
     """
 
     def __init__(
-        self,
-        agent: BaseAgent | Agent | None = None,
-        short_term_memory: ShortTermMemory | None = None,
-        app_name: str | None = None,
-        user_id: str = "veadk_default_user",
-        upload_inline_data_to_tos: bool = False,
-        run_processor: "BaseRunProcessor | None" = None,
-        *args,
-        **kwargs,
+            self,
+            agent: BaseAgent | Agent | None = None,
+            short_term_memory: ShortTermMemory | None = None,
+            app_name: str | None = None,
+            user_id: str = "veadk_default_user",
+            upload_inline_data_to_tos: bool = False,
+            run_processor: "BaseRunProcessor | None" = None,
+            *args,
+            **kwargs,
     ) -> None:
         """Initialize a Runner instance.
 
@@ -355,13 +355,16 @@ def __init__(
         Raises:
             None
         """
+
         self.user_id = user_id
         self.long_term_memory = None
-        self.short_term_memory = short_term_memory
         self.upload_inline_data_to_tos = upload_inline_data_to_tos
         credential_service = kwargs.pop("credential_service", None)
         session_service = kwargs.pop("session_service", None)
         memory_service = kwargs.pop("memory_service", None)
+        if not short_term_memory:
+            short_term_memory = agent.short_term_memory
+        self.short_term_memory = short_term_memory
 
         # Handle run_processor: priority is runner arg > agent.run_processor > NoOpRunProcessor
         if run_processor is not None:
@@ -426,14 +429,14 @@ def __init__(
         )
 
     async def run(
-        self,
-        messages: RunnerMessage,
-        user_id: str = "",
-        session_id: str = f"tmp-session-{formatted_timestamp()}",
-        run_config: RunConfig | None = None,
-        save_tracing_data: bool = False,
-        upload_inline_data_to_tos: bool = False,
-        run_processor: "BaseRunProcessor | None" = None,
+            self,
+            messages: RunnerMessage,
+            user_id: str = "",
+            session_id: str = f"tmp-session-{formatted_timestamp()}",
+            run_config: RunConfig | None = None,
+            save_tracing_data: bool = False,
+            upload_inline_data_to_tos: bool = False,
+            run_processor: "BaseRunProcessor | None" = None,
     ):
         """Run a conversation with multi-turn text and multimodal inputs.
 
@@ -503,20 +506,20 @@ async def run(
                 )
                 async def event_generator():
                     async for event in self.run_async(
-                        user_id=user_id,
-                        session_id=session_id,
-                        new_message=converted_message,
-                        run_config=run_config,
+                            user_id=user_id,
+                            session_id=session_id,
+                            new_message=converted_message,
+                            run_config=run_config,
                     ):
                         yield event
 
                 async for event in event_generator():
                     if event.content is not None and event.content.parts:
                         for part in event.content.parts:
                             if (
-                                not part.thought
-                                and part.text
-                                and len(part.text.strip()) > 0
+                                    not part.thought
+                                    and part.text
+                                    and len(part.text.strip()) > 0
                             ):
                                 final_output = part.text
                                 break
@@ -630,7 +633,7 @@ def save_tracing_file(self, session_id: str) -> str:
             ```
         """
         if not isinstance(
-            self.agent, (Agent, SequentialAgent, ParallelAgent, LoopAgent)
+                self.agent, (Agent, SequentialAgent, ParallelAgent, LoopAgent)
         ):
             logger.warning(
                 (
@@ -689,7 +692,7 @@ async def save_eval_set(self, session_id: str, eval_set_id: str = "default") ->
         return eval_set_path
 
     async def save_session_to_long_term_memory(
-        self, session_id: str, user_id: str = "", app_name: str = "", **kwargs
+            self, session_id: str, user_id: str = "", app_name: str = "", **kwargs
     ) -> None:
         """Save the specified session to long-term memory.