chore: Allows BaseAgent in the constructor of InMemoryRunner.

PiperOrigin-RevId: 759818175

Author: Jacksunwei

contributing/samples/hello_world/main.py

@@ -17,11 +17,9 @@
 
 import agent
 from dotenv import load_dotenv
-from google.adk import Runner
 from google.adk.agents.run_config import RunConfig
-from google.adk.artifacts import InMemoryArtifactService
 from google.adk.cli.utils import logs
-from google.adk.sessions import InMemorySessionService
+from google.adk.runners import InMemoryRunner
 from google.adk.sessions import Session
 from google.genai import types
 
@@ -32,15 +30,11 @@
 async def main():
   app_name = 'my_app'
   user_id_1 = 'user1'
-  session_service = InMemorySessionService()
-  artifact_service = InMemoryArtifactService()
-  runner = Runner(
-      app_name=app_name,
+  runner = InMemoryRunner(
       agent=agent.root_agent,
-      artifact_service=artifact_service,
-      session_service=session_service,
+      app_name=app_name,
   )
-  session_11 = await session_service.create_session(
+  session_11 = await runner.session_service.create_session(
       app_name=app_name, user_id=user_id_1
   )
 
@@ -85,7 +79,7 @@ async def run_prompt_bytes(session: Session, new_message: str):
   await run_prompt(session_11, 'What numbers did I got?')
   await run_prompt_bytes(session_11, 'Hi bytes')
   print(
-      await artifact_service.list_artifact_keys(
+      await runner.artifact_service.list_artifact_keys(
           app_name=app_name, user_id=user_id_1, session_id=session_11.id
       )
   )

src/google/adk/runners.py

@@ -23,7 +23,6 @@
 from typing import Optional
 import warnings
 
-from deprecated import deprecated
 from google.genai import types
 
 from .agents.active_streaming_tool import ActiveStreamingTool
@@ -33,7 +32,6 @@
 from .agents.live_request_queue import LiveRequestQueue
 from .agents.llm_agent import LlmAgent
 from .agents.run_config import RunConfig
-from .agents.run_config import StreamingMode
 from .artifacts.base_artifact_service import BaseArtifactService
 from .artifacts.in_memory_artifact_service import InMemoryArtifactService
 from .events.event import Event
@@ -475,7 +473,7 @@ class InMemoryRunner(Runner):
         session service for the runner.
   """
 
-  def __init__(self, agent: LlmAgent, *, app_name: str = 'InMemoryRunner'):
+  def __init__(self, agent: BaseAgent, *, app_name: str = 'InMemoryRunner'):
     """Initializes the InMemoryRunner.
 
     Args: