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

Move public_utils to utils in tests

Browse Source 
Renamed conflicting utils.py as testing_utils.py

PiperOrigin-RevId: 761715808
This commit is contained in:
Selcuk Gun
2025-05-21 16:34:28 -07:00
committed by Copybara-Service
parent 09cb128cf9
commit 41b33d4a0a
22 changed files with 236 additions and 208 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -20,7 +20,7 @@ from google.adk.sessions import Session
from google.genai import types
import pytest
from ... import utils
from ... import testing_utils
@pytest.mark.asyncio
@@ -36,7 +36,9 @@ 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 = await testing_utils.create_invocation_context(
agent=agent
)
invocation_context.session = Session(
app_name="test_app",
user_id="test_user",
@@ -75,7 +77,9 @@ async def test_function_system_instruction():
name="agent",
instruction=build_function_instruction,
)
invocation_context = await utils.create_invocation_context(agent=agent)
invocation_context = await testing_utils.create_invocation_context(
agent=agent
)
invocation_context.session = Session(
app_name="test_app",
user_id="test_user",
@@ -117,7 +121,9 @@ async def test_async_function_system_instruction():
name="agent",
instruction=build_function_instruction,
)
invocation_context = await utils.create_invocation_context(agent=agent)
invocation_context = await testing_utils.create_invocation_context(
agent=agent
)
invocation_context.session = Session(
app_name="test_app",
user_id="test_user",
@@ -156,7 +162,9 @@ 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 = await testing_utils.create_invocation_context(
agent=sub_agent
)
invocation_context.session = Session(
app_name="test_app",
user_id="test_user",
@@ -198,7 +206,9 @@ 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 = await testing_utils.create_invocation_context(
agent=sub_agent
)
invocation_context.session = Session(
app_name="test_app",
user_id="test_user",
@@ -240,7 +250,9 @@ 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 = await testing_utils.create_invocation_context(
agent=sub_agent
)
invocation_context.session = Session(
app_name="test_app",
user_id="test_user",
@@ -272,7 +284,9 @@ 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 = await testing_utils.create_invocation_context(
agent=agent
)
invocation_context.session = Session(
app_name="test_app",
user_id="test_user",