From 5ab548a25b3abcaeaa0f3e09b0471bcec79f281a Mon Sep 17 00:00:00 2001 From: kedar49 Date: Thu, 14 May 2026 01:28:26 +0530 Subject: [PATCH] fix(samples): use configurable model name for triaging agents to avoid 404 --- contributing/samples/adk_answering_agent/agent.py | 3 ++- contributing/samples/adk_answering_agent/settings.py | 1 + contributing/samples/adk_issue_formatting_agent/agent.py | 3 ++- contributing/samples/adk_issue_formatting_agent/settings.py | 1 + contributing/samples/adk_pr_triaging_agent/agent.py | 3 ++- contributing/samples/adk_pr_triaging_agent/settings.py | 1 + contributing/samples/adk_triaging_agent/agent.py | 3 ++- contributing/samples/adk_triaging_agent/settings.py | 1 + 8 files changed, 12 insertions(+), 4 deletions(-) diff --git a/contributing/samples/adk_answering_agent/agent.py b/contributing/samples/adk_answering_agent/agent.py index ff82e5029a..05a7dc4539 100644 --- a/contributing/samples/adk_answering_agent/agent.py +++ b/contributing/samples/adk_answering_agent/agent.py @@ -15,6 +15,7 @@ from adk_answering_agent.gemini_assistant.agent import root_agent as gemini_assistant_agent from adk_answering_agent.settings import BOT_RESPONSE_LABEL from adk_answering_agent.settings import IS_INTERACTIVE +from adk_answering_agent.settings import LLM_MODEL_NAME from adk_answering_agent.settings import OWNER from adk_answering_agent.settings import REPO from adk_answering_agent.settings import VERTEXAI_DATASTORE_ID @@ -38,7 +39,7 @@ root_agent = Agent( - model="gemini-2.5-pro", + model=LLM_MODEL_NAME, name="adk_answering_agent", description="Answer questions about ADK repo.", instruction=f""" diff --git a/contributing/samples/adk_answering_agent/settings.py b/contributing/samples/adk_answering_agent/settings.py index e7b1f8275e..af4cfe37bf 100644 --- a/contributing/samples/adk_answering_agent/settings.py +++ b/contributing/samples/adk_answering_agent/settings.py @@ -39,6 +39,7 @@ OWNER = os.getenv("OWNER", "google") REPO = os.getenv("REPO", "adk-python") +LLM_MODEL_NAME = os.getenv("LLM_MODEL_NAME", "gemini-2.5-flash") BOT_RESPONSE_LABEL = os.getenv("BOT_RESPONSE_LABEL", "bot responded") DISCUSSION_NUMBER = os.getenv("DISCUSSION_NUMBER") diff --git a/contributing/samples/adk_issue_formatting_agent/agent.py b/contributing/samples/adk_issue_formatting_agent/agent.py index ae906f3993..52f88b24b2 100644 --- a/contributing/samples/adk_issue_formatting_agent/agent.py +++ b/contributing/samples/adk_issue_formatting_agent/agent.py @@ -17,6 +17,7 @@ from adk_issue_formatting_agent.settings import GITHUB_BASE_URL from adk_issue_formatting_agent.settings import IS_INTERACTIVE +from adk_issue_formatting_agent.settings import LLM_MODEL_NAME from adk_issue_formatting_agent.settings import OWNER from adk_issue_formatting_agent.settings import REPO from adk_issue_formatting_agent.utils import error_response @@ -132,7 +133,7 @@ def list_comments_on_issue(issue_number: int) -> dict[str, any]: root_agent = Agent( - model="gemini-2.5-pro", + model=LLM_MODEL_NAME, name="adk_issue_formatting_assistant", description="Check ADK issue format and content.", instruction=f""" diff --git a/contributing/samples/adk_issue_formatting_agent/settings.py b/contributing/samples/adk_issue_formatting_agent/settings.py index ed5b1c49b2..3812b2623c 100644 --- a/contributing/samples/adk_issue_formatting_agent/settings.py +++ b/contributing/samples/adk_issue_formatting_agent/settings.py @@ -26,6 +26,7 @@ OWNER = os.getenv("OWNER", "google") REPO = os.getenv("REPO", "adk-python") +LLM_MODEL_NAME = os.getenv("LLM_MODEL_NAME", "gemini-2.5-flash") EVENT_NAME = os.getenv("EVENT_NAME") ISSUE_NUMBER = os.getenv("ISSUE_NUMBER") ISSUE_COUNT_TO_PROCESS = os.getenv("ISSUE_COUNT_TO_PROCESS") diff --git a/contributing/samples/adk_pr_triaging_agent/agent.py b/contributing/samples/adk_pr_triaging_agent/agent.py index 7f849fe358..4c7ec8ad00 100644 --- a/contributing/samples/adk_pr_triaging_agent/agent.py +++ b/contributing/samples/adk_pr_triaging_agent/agent.py @@ -17,6 +17,7 @@ from adk_pr_triaging_agent.settings import GITHUB_BASE_URL from adk_pr_triaging_agent.settings import IS_INTERACTIVE +from adk_pr_triaging_agent.settings import LLM_MODEL_NAME from adk_pr_triaging_agent.settings import OWNER from adk_pr_triaging_agent.settings import REPO from adk_pr_triaging_agent.utils import error_response @@ -220,7 +221,7 @@ def add_comment_to_pr(pr_number: int, comment: str) -> dict[str, Any]: root_agent = Agent( - model="gemini-2.5-pro", + model=LLM_MODEL_NAME, name="adk_pr_triaging_assistant", description="Triage ADK pull requests.", instruction=f""" diff --git a/contributing/samples/adk_pr_triaging_agent/settings.py b/contributing/samples/adk_pr_triaging_agent/settings.py index 844dbdc67b..a1be9137e9 100644 --- a/contributing/samples/adk_pr_triaging_agent/settings.py +++ b/contributing/samples/adk_pr_triaging_agent/settings.py @@ -27,6 +27,7 @@ OWNER = os.getenv("OWNER", "google") REPO = os.getenv("REPO", "adk-python") +LLM_MODEL_NAME = os.getenv("LLM_MODEL_NAME", "gemini-2.5-flash") PULL_REQUEST_NUMBER = os.getenv("PULL_REQUEST_NUMBER") IS_INTERACTIVE = os.environ.get("INTERACTIVE", "1").lower() in ["true", "1"] diff --git a/contributing/samples/adk_triaging_agent/agent.py b/contributing/samples/adk_triaging_agent/agent.py index 5797361af8..d554864af6 100644 --- a/contributing/samples/adk_triaging_agent/agent.py +++ b/contributing/samples/adk_triaging_agent/agent.py @@ -16,6 +16,7 @@ from adk_triaging_agent.settings import GITHUB_BASE_URL from adk_triaging_agent.settings import IS_INTERACTIVE +from adk_triaging_agent.settings import LLM_MODEL_NAME from adk_triaging_agent.settings import OWNER from adk_triaging_agent.settings import REPO from adk_triaging_agent.utils import error_response @@ -244,7 +245,7 @@ def change_issue_type(issue_number: int, issue_type: str) -> dict[str, Any]: root_agent = Agent( - model="gemini-2.5-pro", + model=LLM_MODEL_NAME, name="adk_triaging_assistant", description="Triage ADK issues.", instruction=f""" diff --git a/contributing/samples/adk_triaging_agent/settings.py b/contributing/samples/adk_triaging_agent/settings.py index fdc8b4e033..326b20d7d3 100644 --- a/contributing/samples/adk_triaging_agent/settings.py +++ b/contributing/samples/adk_triaging_agent/settings.py @@ -26,6 +26,7 @@ OWNER = os.getenv("OWNER", "google") REPO = os.getenv("REPO", "adk-python") +LLM_MODEL_NAME = os.getenv("LLM_MODEL_NAME", "gemini-2.5-flash") EVENT_NAME = os.getenv("EVENT_NAME") ISSUE_NUMBER = os.getenv("ISSUE_NUMBER") ISSUE_TITLE = os.getenv("ISSUE_TITLE")