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: 759252188
This commit is contained in:
Google Team Member
2025-05-15 12:23:33 -07:00
committed by Copybara-Service
parent 5b3204c356
commit 1804ca39a6
23 changed files with 268 additions and 264 deletions

View File

@@ -31,7 +31,7 @@ async def test_no_examples():
config=types.GenerateContentConfig(system_instruction=""),
)
agent = Agent(model="gemini-1.5-flash", name="agent", examples=[])
invocation_context = await utils.create_invocation_context(
invocation_context = utils.create_invocation_context(
agent=agent, user_content=""
)
@@ -69,7 +69,7 @@ async def test_agent_examples():
name="agent",
examples=example_list,
)
invocation_context = await utils.create_invocation_context(
invocation_context = utils.create_invocation_context(
agent=agent, user_content="test"
)
@@ -122,7 +122,7 @@ async def test_agent_base_example_provider():
name="agent",
examples=provider,
)
invocation_context = await utils.create_invocation_context(
invocation_context = utils.create_invocation_context(
agent=agent, user_content="test"
)