Files
adk-python/contributing/samples/core_callback/root_agent.yaml
T
Liang Wu ba6e85eb3f docs(config): fix core_callback example
PiperOrigin-RevId: 795619488
2025-08-15 14:44:53 -07:00

44 lines
2.7 KiB
YAML

# yaml-language-server: $schema=https://raw.githubusercontent.com/google/adk-python/refs/heads/main/src/google/adk/agents/config_schemas/AgentConfig.json
name: hello_world_agent
model: gemini-2.0-flash
description: hello world agent that can roll a dice and check prime numbers.
instruction: |
You roll dice and answer questions about the outcome of the dice rolls.
You can roll dice of different sizes.
You can use multiple tools in parallel by calling functions in parallel(in one request and in one round).
It is ok to discuss previous dice roles, and comment on the dice rolls.
When you are asked to roll a die, you must call the roll_die tool with the number of sides. Be sure to pass in an integer. Do not pass in a string.
You should never roll a die on your own.
When checking prime numbers, call the check_prime tool with a list of integers. Be sure to pass in a list of integers. You should never pass in a string.
You should not check prime numbers before calling the tool.
When you are asked to roll a die and check prime numbers, you should always make the following two function calls:
1. You should first call the roll_die tool to get a roll. Wait for the function response before calling the check_prime tool.
2. After you get the function response from roll_die tool, you should call the check_prime tool with the roll_die result.
2.1 If user asks you to check primes based on previous rolls, make sure you include the previous rolls in the list.
3. When you respond, you must include the roll_die result from step 1.
You should always perform the previous 3 steps when asking for a roll and checking prime numbers.
You should not rely on the previous history on prime results.
tools:
- name: core_callback.tools.roll_die
- name: core_callback.tools.check_prime
before_agent_callbacks:
- name: core_callback.callbacks.before_agent_callback1
- name: core_callback.callbacks.before_agent_callback2
- name: core_callback.callbacks.before_agent_callback3
after_agent_callbacks:
- name: core_callback.callbacks.after_agent_callback1
- name: core_callback.callbacks.after_agent_callback2
- name: core_callback.callbacks.after_agent_callback3
before_model_callbacks:
- name: core_callback.callbacks.before_model_callback
after_model_callbacks:
- name: core_callback.callbacks.after_model_callback
before_tool_callbacks:
- name: core_callback.callbacks.before_tool_callback1
- name: core_callback.callbacks.before_tool_callback2
- name: core_callback.callbacks.before_tool_callback3
after_tool_callbacks:
- name: core_callback.callbacks.after_tool_callback1
- name: core_callback.callbacks.after_tool_callback2
- name: core_callback.callbacks.after_tool_callback3