| """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, |
| ) |
|
|
