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

6
tests/unittests/flows/llm_flows/test_async_tool_callbacks.py
View File

@@ -29,7 +29,7 @@ from google.adk.tools.tool_context import ToolContext
from google.genai import types
import pytest
from ... import utils
from ... import testing_utils
class CallbackType(Enum):
@@ -73,7 +73,7 @@ async def invoke_tool_with_callbacks(
return {"initial": "response"}
tool = FunctionTool(simple_fn)
model = utils.MockModel.create(responses=[])
model = testing_utils.MockModel.create(responses=[])
agent = Agent(
name="agent",
model=model,
@@ -81,7 +81,7 @@ async def invoke_tool_with_callbacks(
before_tool_callback=before_cb,
after_tool_callback=after_cb,
)
invocation_context = await utils.create_invocation_context(
invocation_context = await testing_utils.create_invocation_context(
agent=agent, user_content=""
)
# Build function call event