NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-12-25 14:37:45 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

add toolset base class and allow llm agent to accept toolset as tools

Browse Source 
PiperOrigin-RevId: 756605470
This commit is contained in:
Xiang (Sean) Zhou
2025-05-08 22:27:13 -07:00
committed by Copybara-Service
parent 8963300518
commit 4d7298e4f2
10 changed files with 129 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/google/adk/flows/llm_flows/base_llm_flow.py
View File

@@ -29,6 +29,7 @@ from ...agents.base_agent import BaseAgent
from ...agents.callback_context import CallbackContext
from ...agents.invocation_context import InvocationContext
from ...agents.live_request_queue import LiveRequestQueue
from ...agents.readonly_context import ReadonlyContext
from ...agents.run_config import StreamingMode
from ...agents.transcription_entry import TranscriptionEntry
from ...events.event import Event
@@ -296,7 +297,9 @@ class BaseLlmFlow(ABC):
yield event
# Run processors for tools.
for tool in agent.canonical_tools:
for tool in await agent.canonical_tools(
ReadonlyContext(invocation_context)
):
tool_context = ToolContext(invocation_context)
await tool.process_llm_request(
tool_context=tool_context, llm_request=llm_request