diff --git a/vertexai/agent_engines/templates/langchain.py b/vertexai/agent_engines/templates/langchain.py index 49325b7480..31dabcbdd3 100644 --- a/vertexai/agent_engines/templates/langchain.py +++ b/vertexai/agent_engines/templates/langchain.py @@ -21,7 +21,6 @@ Iterable, Mapping, Optional, - Sequence, Union, ) @@ -46,9 +45,9 @@ try: from langchain_google_vertexai.functions_utils import _ToolsType - _ToolLike = _ToolsType + _ToolsType = _ToolsType except ImportError: - _ToolLike = Any + _ToolsType = Any try: from opentelemetry.sdk import trace @@ -116,7 +115,7 @@ def _default_runnable_builder( model: "BaseLanguageModel", *, system_instruction: Optional[str] = None, - tools: Optional[Sequence["_ToolLike"]] = None, + tools: Optional["_ToolsType"] = None, prompt: Optional["RunnableSerializable"] = None, output_parser: Optional["RunnableSerializable"] = None, chat_history: Optional["GetSessionHistoryCallable"] = None, @@ -320,7 +319,7 @@ def _validate_callable_parameters_are_annotated(callable: Callable): ) -def _validate_tools(tools: Sequence["_ToolLike"]): +def _validate_tools(tools: "_ToolsType"): """Validates that the tools are usable for tool calling.""" for tool in tools: if isinstance(tool, Callable): @@ -370,7 +369,7 @@ def __init__( *, system_instruction: Optional[str] = None, prompt: Optional["RunnableSerializable"] = None, - tools: Optional[Sequence["_ToolLike"]] = None, + tools: Optional["_ToolsType"] = None, output_parser: Optional["RunnableSerializable"] = None, chat_history: Optional["GetSessionHistoryCallable"] = None, model_kwargs: Optional[Mapping[str, Any]] = None, diff --git a/vertexai/preview/reasoning_engines/templates/langchain.py b/vertexai/preview/reasoning_engines/templates/langchain.py index a8bb11305e..eddb105d5b 100644 --- a/vertexai/preview/reasoning_engines/templates/langchain.py +++ b/vertexai/preview/reasoning_engines/templates/langchain.py @@ -21,7 +21,6 @@ Iterable, Mapping, Optional, - Sequence, Union, ) @@ -46,9 +45,9 @@ try: from langchain_google_vertexai.functions_utils import _ToolsType - _ToolLike = _ToolsType + _ToolsType = _ToolsType except ImportError: - _ToolLike = Any + _ToolsType = Any try: from opentelemetry.sdk import trace @@ -116,7 +115,7 @@ def _default_runnable_builder( model: "BaseLanguageModel", *, system_instruction: Optional[str] = None, - tools: Optional[Sequence["_ToolLike"]] = None, + tools: Optional["_ToolsType"] = None, prompt: Optional["RunnableSerializable"] = None, output_parser: Optional["RunnableSerializable"] = None, chat_history: Optional["GetSessionHistoryCallable"] = None, @@ -234,7 +233,7 @@ def _validate_callable_parameters_are_annotated(callable: Callable): ) -def _validate_tools(tools: Sequence["_ToolLike"]): +def _validate_tools(tools: "_ToolsType"): """Validates that the tools are usable for tool calling.""" for tool in tools: if isinstance(tool, Callable): @@ -284,7 +283,7 @@ def __init__( *, system_instruction: Optional[str] = None, prompt: Optional["RunnableSerializable"] = None, - tools: Optional[Sequence["_ToolLike"]] = None, + tools: Optional["_ToolsType"] = None, output_parser: Optional["RunnableSerializable"] = None, chat_history: Optional["GetSessionHistoryCallable"] = None, model_kwargs: Optional[Mapping[str, Any]] = None,