Jason Zhang
0c79d1a8da
Rename `contributing/samples/plugin/` to `contributing/samples/plugins/` so the samples directory matches the `plugins/` naming already used everywhere else in the repository for this feature area: the `google.adk.plugins` source package (`src/google/adk/plugins/`), the guide at `docs/guides/plugins/`, and the unit tests at `tests/unittests/plugins/`. It also matches the plural naming used by every sibling sample group (`tools/`, `models/`, `workflows/`, `patterns/`, and others). The moved files themselves are unchanged. Also update the paths that pointed into the renamed directory: - Repoint the two "Related samples" links in the reflect-and-retry tool plugin guide, which the move would otherwise break. - Fix the run commands in the two sample READMEs and in the debug logging agent docstring. These already omitted the grouping directory and so did not work before the rename either; they now refer to paths that exist. Co-authored-by: Jason Zhang <jasoncz@google.com> PiperOrigin-RevId: 960009471
65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
# Copyright 2026 Google LLC
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import asyncio
|
|
|
|
from google.adk import Agent
|
|
from google.adk.runners import InMemoryRunner
|
|
from google.adk.tools.tool_context import ToolContext
|
|
from google.genai import types
|
|
|
|
# [Step 2] Import the plugin.
|
|
from .count_plugin import CountInvocationPlugin
|
|
|
|
|
|
async def hello_world(tool_context: ToolContext, query: str):
|
|
print(f'Hello world: query is [{query}]')
|
|
|
|
|
|
root_agent = Agent(
|
|
name='hello_world',
|
|
description='Prints hello world with user query.',
|
|
instruction="""Use hello_world tool to print hello world and user query.
|
|
""",
|
|
tools=[hello_world],
|
|
)
|
|
|
|
|
|
async def main():
|
|
"""Main entry point for the agent."""
|
|
prompt = 'hello world'
|
|
runner = InMemoryRunner(
|
|
agent=root_agent,
|
|
app_name='test_app_with_plugin',
|
|
# [Step 2] Add your plugin here. You can add multiple plugins.
|
|
plugins=[CountInvocationPlugin()],
|
|
)
|
|
session = await runner.session_service.create_session(
|
|
user_id='user',
|
|
app_name='test_app_with_plugin',
|
|
)
|
|
|
|
async for event in runner.run_async(
|
|
user_id='user',
|
|
session_id=session.id,
|
|
new_message=types.Content(
|
|
role='user', parts=[types.Part.from_text(text=prompt)]
|
|
),
|
|
):
|
|
print(f'** Got event from {event.author}')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
asyncio.run(main())
|