feat! Update session service interface to be async.

Also keep the sync version in the InMemorySessionService as create_session_sync() as a temporary migration option.

PiperOrigin-RevId: 759224250
This commit is contained in:
Shangjie Chen
2025-05-15 11:16:43 -07:00
committed by Copybara-Service
parent d161a2c3f7
commit 5b3204c356
23 changed files with 264 additions and 268 deletions

View File

@@ -28,7 +28,7 @@ async def test_no_description():
config=types.GenerateContentConfig(system_instruction=""),
)
agent = Agent(model="gemini-1.5-flash", name="agent")
invocation_context = utils.create_invocation_context(agent=agent)
invocation_context = await utils.create_invocation_context(agent=agent)
async for _ in identity.request_processor.run_async(
invocation_context,
@@ -52,7 +52,7 @@ async def test_with_description():
name="agent",
description="test description",
)
invocation_context = utils.create_invocation_context(agent=agent)
invocation_context = await utils.create_invocation_context(agent=agent)
async for _ in identity.request_processor.run_async(
invocation_context,