Skip to content

Commit ff39b2a

Browse files
yeesiancopybara-github
authored andcommitted
chore: Add support for ADK Agents as a supported type for agent engine
PiperOrigin-RevId: 785565944
1 parent a656288 commit ff39b2a

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

vertexai/agent_engines/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
# We just want to re-export certain classes
2525
# pylint: disable=g-multiple-import,g-importing-member
2626
from vertexai.agent_engines._agent_engines import (
27+
_AgentEngineInterface,
2728
AgentEngine,
2829
Cloneable,
2930
ModuleAgent,
@@ -60,7 +61,7 @@ def get(resource_name: str) -> AgentEngine:
6061

6162

6263
def create(
63-
agent_engine: Optional[Union[Queryable, OperationRegistrable]] = None,
64+
agent_engine: Optional[_AgentEngineInterface] = None,
6465
*,
6566
requirements: Optional[Union[str, Sequence[str]]] = None,
6667
display_name: Optional[str] = None,

0 commit comments

Comments
 (0)