NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-12-18 19:32:21 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat! Update session service interface to be async.

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/unittests/flows/llm_flows/test_instructions.py
View File

@@ -36,7 +36,7 @@ async def test_build_system_instruction():
{{customer_int }, { non-identifier-float}}, \
{'key1': 'value1'} and {{'key2': 'value2'}}."""),
)
invocation_context = await utils.create_invocation_context(agent=agent)
invocation_context = 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 = await utils.create_invocation_context(agent=agent)
invocation_context = 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 = await utils.create_invocation_context(agent=agent)
invocation_context = 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 = await utils.create_invocation_context(agent=sub_agent)
invocation_context = 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 = await utils.create_invocation_context(agent=sub_agent)
invocation_context = 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 = await utils.create_invocation_context(agent=sub_agent)
invocation_context = 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 = await utils.create_invocation_context(agent=agent)
invocation_context = utils.create_invocation_context(agent=agent)
invocation_context.session = Session(
app_name="test_app",
user_id="test_user",