|
|
"""Load agent.""" |
|
|
|
|
|
from typing import Any, Optional, Sequence |
|
|
|
|
|
from langchain_core._api import deprecated |
|
|
from langchain_core.callbacks import BaseCallbackManager |
|
|
from langchain_core.language_models import BaseLanguageModel |
|
|
from langchain_core.tools import BaseTool |
|
|
|
|
|
from langchain._api.deprecation import AGENT_DEPRECATION_WARNING |
|
|
from langchain.agents.agent import AgentExecutor |
|
|
from langchain.agents.agent_types import AgentType |
|
|
from langchain.agents.loading import AGENT_TO_CLASS, load_agent |
|
|
|
|
|
|
|
|
@deprecated( |
|
|
"0.1.0", |
|
|
message=AGENT_DEPRECATION_WARNING, |
|
|
removal="1.0", |
|
|
) |
|
|
def initialize_agent( |
|
|
tools: Sequence[BaseTool], |
|
|
llm: BaseLanguageModel, |
|
|
agent: Optional[AgentType] = None, |
|
|
callback_manager: Optional[BaseCallbackManager] = None, |
|
|
agent_path: Optional[str] = None, |
|
|
agent_kwargs: Optional[dict] = None, |
|
|
*, |
|
|
tags: Optional[Sequence[str]] = None, |
|
|
**kwargs: Any, |
|
|
) -> AgentExecutor: |
|
|
"""Load an agent executor given tools and LLM. |
|
|
|
|
|
Args: |
|
|
tools: List of tools this agent has access to. |
|
|
llm: Language model to use as the agent. |
|
|
agent: Agent type to use. If None and agent_path is also None, will default |
|
|
to AgentType.ZERO_SHOT_REACT_DESCRIPTION. Defaults to None. |
|
|
callback_manager: CallbackManager to use. Global callback manager is used if |
|
|
not provided. Defaults to None. |
|
|
agent_path: Path to serialized agent to use. If None and agent is also None, |
|
|
will default to AgentType.ZERO_SHOT_REACT_DESCRIPTION. Defaults to None. |
|
|
agent_kwargs: Additional keyword arguments to pass to the underlying agent. |
|
|
Defaults to None. |
|
|
tags: Tags to apply to the traced runs. Defaults to None. |
|
|
kwargs: Additional keyword arguments passed to the agent executor. |
|
|
|
|
|
Returns: |
|
|
An agent executor. |
|
|
|
|
|
Raises: |
|
|
ValueError: If both `agent` and `agent_path` are specified. |
|
|
ValueError: If `agent` is not a valid agent type. |
|
|
ValueError: If both `agent` and `agent_path` are None. |
|
|
""" |
|
|
tags_ = list(tags) if tags else [] |
|
|
if agent is None and agent_path is None: |
|
|
agent = AgentType.ZERO_SHOT_REACT_DESCRIPTION |
|
|
if agent is not None and agent_path is not None: |
|
|
raise ValueError( |
|
|
"Both `agent` and `agent_path` are specified, " |
|
|
"but at most only one should be." |
|
|
) |
|
|
if agent is not None: |
|
|
if agent not in AGENT_TO_CLASS: |
|
|
raise ValueError( |
|
|
f"Got unknown agent type: {agent}. " |
|
|
f"Valid types are: {AGENT_TO_CLASS.keys()}." |
|
|
) |
|
|
tags_.append(agent.value if isinstance(agent, AgentType) else agent) |
|
|
agent_cls = AGENT_TO_CLASS[agent] |
|
|
agent_kwargs = agent_kwargs or {} |
|
|
agent_obj = agent_cls.from_llm_and_tools( |
|
|
llm, tools, callback_manager=callback_manager, **agent_kwargs |
|
|
) |
|
|
elif agent_path is not None: |
|
|
agent_obj = load_agent( |
|
|
agent_path, llm=llm, tools=tools, callback_manager=callback_manager |
|
|
) |
|
|
try: |
|
|
|
|
|
tags_.append(agent_obj._agent_type) |
|
|
except NotImplementedError: |
|
|
pass |
|
|
else: |
|
|
raise ValueError( |
|
|
"Somehow both `agent` and `agent_path` are None, " |
|
|
"this should never happen." |
|
|
) |
|
|
return AgentExecutor.from_agent_and_tools( |
|
|
agent=agent_obj, |
|
|
tools=tools, |
|
|
callback_manager=callback_manager, |
|
|
tags=tags_, |
|
|
**kwargs, |
|
|
) |
|
|
|
