Skip to content

Conversation

@jayy-77
Copy link

@jayy-77 jayy-77 commented Feb 6, 2026

Please ensure you have read the contribution guide before creating a pull request.

Link to Issue or Description of Change

fixes #3778

1. Link to an existing issue (if applicable):

  • Closes: #issue_number
  • Related: #issue_number

2. Or, if no issue exists, describe the change:

If applicable, please follow the issue templates to provide as much detail as
possible.

Problem:
A clear and concise description of what the problem is.

Solution:
A clear and concise description of what you want to happen and why you choose
this solution.

Testing Plan

Please describe the tests that you ran to verify your changes. This is required
for all PRs that are not small documentation or typo fixes.

Unit Tests:

  • I have added or updated unit tests for my change.
  • All unit tests pass locally.

Please include a summary of passed pytest results.

Manual End-to-End (E2E) Tests:

Please provide instructions on how to manually test your changes, including any
necessary setup or configuration. Please provide logs or screenshots to help
reviewers better understand the fix.

Checklist

  • I have read the CONTRIBUTING.md document.
  • I have performed a self-review of my own code.
  • I have commented my code, particularly in hard-to-understand areas.
  • I have added tests that prove my fix is effective or that my feature works.
  • New and existing unit tests pass locally with my changes.
  • I have manually tested my changes end-to-end.
  • Any dependent changes have been merged and published in downstream modules.

Additional context

Add any other context or screenshots about the feature request here.

@gemini-code-assist
Copy link
Contributor

Summary of Changes

Hello @jayy-77, I'm Gemini Code Assist1! I'm currently reviewing this pull request and will post my feedback shortly. In the meantime, here's a summary to help you and other reviewers quickly get up to speed!

This pull request introduces the IfAgent (Conditional Agent) into the ADK framework, significantly enhancing the ability to create dynamic and adaptive agent workflows. By allowing the execution path to be determined at runtime based on a user-defined condition, this feature enables more sophisticated decision-making and flexible orchestration of agents. The IfAgent seamlessly integrates with the existing configuration system, supporting both asynchronous and live execution, and provides robust state management for resumable operations.

Highlights

  • Introduction of IfAgent: A new agent type, IfAgent, has been added to enable conditional branching in agent workflows.
  • Dynamic Workflow Control: This agent allows for runtime evaluation of a user-defined condition, directing execution to one of two sub-agents (if_true or if_false) based on the boolean result.
  • YAML Configuration Support: IfAgent can be configured through YAML, where the condition is specified as a code reference (module and function), and the two sub-agents are defined.
  • Resumable Workflow Integration: The IfAgent includes state management capabilities, allowing it to save and load its execution state for resumable workflows.

🧠 New Feature in Public Preview: You can now enable Memory to help Gemini Code Assist learn from your team's feedback. This makes future code reviews more consistent and personalized to your project's style. Click here to enable Memory in your admin console.

Changelog
  • src/google/adk/agents/init.py
    • Imported the new IfAgent class.
    • Exported IfAgent by adding it to the all list.
  • src/google/adk/agents/agent_config.py
    • Imported the new IfAgentConfig.
    • Registered "IfAgent" in the _ADK_AGENT_CLASSES set.
    • Integrated IfAgentConfig into the ConfigsUnion for agent configuration discrimination.
  • src/google/adk/agents/if_agent.py
    • New file: Implements the IfAgent class, which evaluates a condition and delegates to one of two sub-agents.
    • Defines IfAgentState for managing the agent's state during execution.
    • Includes methods for parsing configuration, evaluating conditions, and running in both asynchronous (_run_async_impl) and live (_run_live_impl) modes.
  • src/google/adk/agents/if_agent_config.py
    • New file: Defines the IfAgentConfig Pydantic model for YAML serialization.
    • Specifies the condition field, allowing it to reference a callable function for dynamic evaluation.
Activity
Using Gemini Code Assist

The full guide for Gemini Code Assist can be found on our documentation page, here are some quick tips.

Invoking Gemini

You can request assistance from Gemini at any point by creating a comment using either /gemini <command> or @gemini-code-assist <command>. Below is a summary of the supported commands on the current page.

Feature Command Description
Code Review /gemini review Performs a code review for the current pull request in its current state.
Pull Request Summary /gemini summary Provides a summary of the current pull request in its current state.
Comment @gemini-code-assist Responds in comments when explicitly tagged, both in pull request comments and review comments.
Help /gemini help Displays a list of available commands.

Customization

To customize Gemini Code Assist for GitHub experience, repository maintainers can create a configuration file and/or provide a custom code review style guide (such as PEP-8 for Python) by creating and adding files to a .gemini/ folder in the base of the repository. Detailed instructions can be found here.

Limitations & Feedback

Gemini Code Assist may make mistakes. Please leave feedback on any instances where its feedback is incorrect or counter productive. You can react with 👍 and 👎 on @gemini-code-assist comments. If you're interested in giving your feedback about your experience with Gemini Code Assist for Github and other Google products, sign up here.

You can also get AI-powered code generation, chat, as well as code reviews directly in the IDE at no cost with the Gemini Code Assist IDE Extension.

Footnotes

  1. Review the Privacy Notices, Generative AI Prohibited Use Policy, Terms of Service, and learn how to configure Gemini Code Assist in GitHub here. Gemini can make mistakes, so double check it and use code with caution.

@adk-bot adk-bot added the core [Component] This issue is related to the core interface and implementation label Feb 6, 2026
@adk-bot
Copy link
Collaborator

adk-bot commented Feb 6, 2026

Response from ADK Triaging Agent

Hello @jayy-77, thank you for your contribution!

To help us review your PR, could you please fill out the PR description template? In particular, the "Testing Plan" section is required for new features. Providing details on how you've tested this change will help the review process go more smoothly.

You can find more information in our contribution guidelines.

Thanks!

Copy link
Contributor

@gemini-code-assist gemini-code-assist bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request introduces the IfAgent, a conditional branching agent that allows for runtime workflow decisions based on a condition function. It also includes the necessary configuration classes and integrations into the existing agent framework. The implementation is generally well-structured and documented. I've identified a couple of areas for improvement related to exception handling and minor code redundancy.

'IfAgent %s: condition evaluated to %s', self.name, result
)
return bool(result)
except Exception as e: # pylint: disable=broad-except
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

Catching a broad Exception can mask specific issues and make debugging more difficult. Consider catching more specific exceptions that you expect to handle, or re-raising the exception after logging if it's an unexpected error. While defaulting to False is a fallback, understanding the root cause of failures is important.

Comment on lines +221 to +223
chosen_agent = (
self.if_true_agent if condition_result else self.if_false_agent
)
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

The chosen_agent is already determined and assigned on lines 206-208. This re-assignment is redundant and can be removed for cleaner code.

    # Execute the chosen sub-agent
    # chosen_agent is already determined above

Comment on lines +248 to +250
chosen_agent = (
self.if_true_agent if condition_result else self.if_false_agent
)
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

Similar to _run_async_impl, the chosen_agent is already determined on lines 248-250. This re-assignment is redundant and can be removed.

    # Execute the chosen sub-agent in live mode
    # chosen_agent is already determined above

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

core [Component] This issue is related to the core interface and implementation

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Add ConditionalAgent / BranchAgent for Runtime Workflow Branching

2 participants