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

tests/unittests/flows/llm_flows/_test_examples.py

@@ -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 = utils.create_invocation_context(
+  invocation_context = await utils.create_invocation_context(
       agent=agent, user_content=""
   )
 
@@ -69,7 +69,7 @@ async def test_agent_examples():
       name="agent",
       examples=example_list,
   )
-  invocation_context = utils.create_invocation_context(
+  invocation_context = await 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 = utils.create_invocation_context(
+  invocation_context = await utils.create_invocation_context(
       agent=agent, user_content="test"
   )
 

tests/unittests/flows/llm_flows/test_async_tool_callbacks.py

@@ -81,7 +81,7 @@ async def invoke_tool_with_callbacks(
       before_tool_callback=before_cb,
       after_tool_callback=after_cb,
   )
-  invocation_context = utils.create_invocation_context(
+  invocation_context = await utils.create_invocation_context(
       agent=agent, user_content=""
   )
   # Build function call event

tests/unittests/flows/llm_flows/test_identity.py

@@ -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,

tests/unittests/flows/llm_flows/test_instructions.py

@@ -36,7 +36,7 @@ async def test_build_system_instruction():
 {{customer_int  }, {  non-identifier-float}}, \
 {'key1': 'value1'} and {{'key2': 'value2'}}."""),
   )
-  invocation_context = utils.create_invocation_context(agent=agent)
+  invocation_context = await utils.create_invocation_context(agent=agent)
   invocation_context.session = Session(
       app_name="test_app",
       user_id="test_user",
@@ -73,7 +73,7 @@ async def test_function_system_instruction():
       name="agent",
       instruction=build_function_instruction,
   )
-  invocation_context = utils.create_invocation_context(agent=agent)
+  invocation_context = await utils.create_invocation_context(agent=agent)
   invocation_context.session = Session(
       app_name="test_app",
       user_id="test_user",
@@ -111,7 +111,7 @@ async def test_async_function_system_instruction():
       name="agent",
       instruction=build_function_instruction,
   )
-  invocation_context = utils.create_invocation_context(agent=agent)
+  invocation_context = await utils.create_invocation_context(agent=agent)
   invocation_context.session = Session(
       app_name="test_app",
       user_id="test_user",
@@ -147,7 +147,7 @@ async def test_global_system_instruction():
       model="gemini-1.5-flash",
       config=types.GenerateContentConfig(system_instruction=""),
   )
-  invocation_context = utils.create_invocation_context(agent=sub_agent)
+  invocation_context = await utils.create_invocation_context(agent=sub_agent)
   invocation_context.session = Session(
       app_name="test_app",
       user_id="test_user",
@@ -189,7 +189,7 @@ async def test_function_global_system_instruction():
       model="gemini-1.5-flash",
       config=types.GenerateContentConfig(system_instruction=""),
   )
-  invocation_context = utils.create_invocation_context(agent=sub_agent)
+  invocation_context = await utils.create_invocation_context(agent=sub_agent)
   invocation_context.session = Session(
       app_name="test_app",
       user_id="test_user",
@@ -231,7 +231,7 @@ async def test_async_function_global_system_instruction():
       model="gemini-1.5-flash",
       config=types.GenerateContentConfig(system_instruction=""),
   )
-  invocation_context = utils.create_invocation_context(agent=sub_agent)
+  invocation_context = await utils.create_invocation_context(agent=sub_agent)
   invocation_context.session = Session(
       app_name="test_app",
       user_id="test_user",
@@ -263,7 +263,7 @@ async def test_build_system_instruction_with_namespace():
           """Use the echo_info tool to echo { customerId }, {app:key}, {user:key}, {a:key}."""
       ),
   )
-  invocation_context = utils.create_invocation_context(agent=agent)
+  invocation_context = await utils.create_invocation_context(agent=agent)
   invocation_context.session = Session(
       app_name="test_app",
       user_id="test_user",