c56bef871b
Sync docs with Docusaurus / sync (push) Waiting to run
Tests / Check if changed (push) Waiting to run
Tests / format (push) Blocked by required conditions
Tests / check-imports (push) Blocked by required conditions
Tests / Unit / macos-latest (push) Blocked by required conditions
Tests / Unit / ubuntu-latest (push) Blocked by required conditions
Tests / Unit / windows-latest (push) Blocked by required conditions
Tests / mypy (push) Blocked by required conditions
Tests / Integration / ubuntu-latest (push) Blocked by required conditions
Tests / Integration / macos-latest (push) Blocked by required conditions
Tests / Integration / windows-latest (push) Blocked by required conditions
Tests / notify-slack-on-failure (push) Blocked by required conditions
Tests / Mark tests as completed (push) Blocked by required conditions
Docker image release / Build base image (push) Waiting to run
CodeQL / Analyze (python) (push) Has been cancelled
Update Platform Components Table / update (push) Has been cancelled
476 lines
19 KiB
Markdown
476 lines
19 KiB
Markdown
---
|
|
title: "Agents"
|
|
id: experimental-agents-api
|
|
description: "Tool-using agents with provider-agnostic chat model support."
|
|
slug: "/experimental-agents-api"
|
|
---
|
|
|
|
<a id="haystack_experimental.components.agents.agent"></a>
|
|
|
|
## Module haystack\_experimental.components.agents.agent
|
|
|
|
<a id="haystack_experimental.components.agents.agent.Agent"></a>
|
|
|
|
### Agent
|
|
|
|
A Haystack component that implements a tool-using agent with provider-agnostic chat model support.
|
|
|
|
NOTE: This class extends Haystack's Agent component to add support for human-in-the-loop confirmation strategies.
|
|
|
|
The component processes messages and executes tools until an exit condition is met.
|
|
The exit condition can be triggered either by a direct text response or by invoking a specific designated tool.
|
|
Multiple exit conditions can be specified.
|
|
|
|
When you call an Agent without tools, it acts as a ChatGenerator, produces one response, then exits.
|
|
|
|
### Usage example
|
|
```python
|
|
from haystack.components.generators.chat import OpenAIChatGenerator
|
|
from haystack.dataclasses import ChatMessage
|
|
from haystack.tools.tool import Tool
|
|
|
|
from haystack_experimental.components.agents import Agent
|
|
from haystack_experimental.components.agents.human_in_the_loop import (
|
|
HumanInTheLoopStrategy,
|
|
AlwaysAskPolicy,
|
|
NeverAskPolicy,
|
|
SimpleConsoleUI,
|
|
)
|
|
|
|
calculator_tool = Tool(name="calculator", description="A tool for performing mathematical calculations.", ...)
|
|
search_tool = Tool(name="search", description="A tool for searching the web.", ...)
|
|
|
|
agent = Agent(
|
|
chat_generator=OpenAIChatGenerator(),
|
|
tools=[calculator_tool, search_tool],
|
|
confirmation_strategies={
|
|
calculator_tool.name: HumanInTheLoopStrategy(
|
|
confirmation_policy=NeverAskPolicy(), confirmation_ui=SimpleConsoleUI()
|
|
),
|
|
search_tool.name: HumanInTheLoopStrategy(
|
|
confirmation_policy=AlwaysAskPolicy(), confirmation_ui=SimpleConsoleUI()
|
|
),
|
|
},
|
|
)
|
|
|
|
# Run the agent
|
|
result = agent.run(
|
|
messages=[ChatMessage.from_user("Find information about Haystack")]
|
|
)
|
|
|
|
assert "messages" in result # Contains conversation history
|
|
```
|
|
|
|
<a id="haystack_experimental.components.agents.agent.Agent.__init__"></a>
|
|
|
|
#### Agent.\_\_init\_\_
|
|
|
|
```python
|
|
def __init__(*,
|
|
chat_generator: ChatGenerator,
|
|
tools: ToolsType | None = None,
|
|
system_prompt: str | None = None,
|
|
exit_conditions: list[str] | None = None,
|
|
state_schema: dict[str, Any] | None = None,
|
|
max_agent_steps: int = 100,
|
|
streaming_callback: StreamingCallbackT | None = None,
|
|
raise_on_tool_invocation_failure: bool = False,
|
|
confirmation_strategies: dict[str, ConfirmationStrategy]
|
|
| None = None,
|
|
tool_invoker_kwargs: dict[str, Any] | None = None,
|
|
chat_message_store: ChatMessageStore | None = None,
|
|
memory_store: MemoryStore | None = None) -> None
|
|
```
|
|
|
|
Initialize the agent component.
|
|
|
|
**Arguments**:
|
|
|
|
- `chat_generator`: An instance of the chat generator that your agent should use. It must support tools.
|
|
- `tools`: List of Tool objects or a Toolset that the agent can use.
|
|
- `system_prompt`: System prompt for the agent.
|
|
- `exit_conditions`: List of conditions that will cause the agent to return.
|
|
Can include "text" if the agent should return when it generates a message without tool calls,
|
|
or tool names that will cause the agent to return once the tool was executed. Defaults to ["text"].
|
|
- `state_schema`: The schema for the runtime state used by the tools.
|
|
- `max_agent_steps`: Maximum number of steps the agent will run before stopping. Defaults to 100.
|
|
If the agent exceeds this number of steps, it will stop and return the current state.
|
|
- `streaming_callback`: A callback that will be invoked when a response is streamed from the LLM.
|
|
The same callback can be configured to emit tool results when a tool is called.
|
|
- `raise_on_tool_invocation_failure`: Should the agent raise an exception when a tool invocation fails?
|
|
If set to False, the exception will be turned into a chat message and passed to the LLM.
|
|
- `tool_invoker_kwargs`: Additional keyword arguments to pass to the ToolInvoker.
|
|
- `chat_message_store`: The ChatMessageStore that the agent can use to store
|
|
and retrieve chat messages history.
|
|
- `memory_store`: The memory store that the agent can use to store and retrieve memories.
|
|
|
|
**Raises**:
|
|
|
|
- `TypeError`: If the chat_generator does not support tools parameter in its run method.
|
|
- `ValueError`: If the exit_conditions are not valid.
|
|
|
|
<a id="haystack_experimental.components.agents.agent.Agent.run"></a>
|
|
|
|
#### Agent.run
|
|
|
|
```python
|
|
def run(messages: list[ChatMessage],
|
|
streaming_callback: StreamingCallbackT | None = None,
|
|
*,
|
|
generation_kwargs: dict[str, Any] | None = None,
|
|
break_point: AgentBreakpoint | None = None,
|
|
snapshot: AgentSnapshot | None = None,
|
|
system_prompt: str | None = None,
|
|
tools: ToolsType | list[str] | None = None,
|
|
confirmation_strategy_context: dict[str, Any] | None = None,
|
|
chat_message_store_kwargs: dict[str, Any] | None = None,
|
|
memory_store_kwargs: dict[str, Any] | None = None,
|
|
**kwargs: Any) -> dict[str, Any]
|
|
```
|
|
|
|
Process messages and execute tools until an exit condition is met.
|
|
|
|
**Arguments**:
|
|
|
|
- `messages`: List of Haystack ChatMessage objects to process.
|
|
- `streaming_callback`: A callback that will be invoked when a response is streamed from the LLM.
|
|
The same callback can be configured to emit tool results when a tool is called.
|
|
- `generation_kwargs`: Additional keyword arguments for LLM. These parameters will
|
|
override the parameters passed during component initialization.
|
|
- `break_point`: An AgentBreakpoint, can be a Breakpoint for the "chat_generator" or a ToolBreakpoint
|
|
for "tool_invoker".
|
|
- `snapshot`: A dictionary containing a snapshot of a previously saved agent execution. The snapshot contains
|
|
the relevant information to restart the Agent execution from where it left off.
|
|
- `system_prompt`: System prompt for the agent. If provided, it overrides the default system prompt.
|
|
- `tools`: Optional list of Tool objects, a Toolset, or list of tool names to use for this run.
|
|
When passing tool names, tools are selected from the Agent's originally configured tools.
|
|
- `confirmation_strategy_context`: Optional dictionary for passing request-scoped resources
|
|
to confirmation strategies. Useful in web/server environments to provide per-request
|
|
objects (e.g., WebSocket connections, async queues, Redis pub/sub clients) that strategies
|
|
can use for non-blocking user interaction.
|
|
- `chat_message_store_kwargs`: Optional dictionary of keyword arguments to pass to the ChatMessageStore.
|
|
For example, it can include the `chat_history_id` and `last_k` parameters for retrieving chat history.
|
|
- `memory_store_kwargs`: Optional dictionary of keyword arguments to pass to the MemoryStore.
|
|
It can include:
|
|
- `user_id`: The user ID to search and add memories from.
|
|
- `run_id`: The run ID to search and add memories from.
|
|
- `agent_id`: The agent ID to search and add memories from.
|
|
- `search_criteria`: A dictionary of containing kwargs for the `search_memories` method.
|
|
This can include:
|
|
- `filters`: A dictionary of filters to search for memories.
|
|
- `query`: The query to search for memories.
|
|
Note: If you pass this, the user query passed to the agent will be
|
|
ignored for memory retrieval.
|
|
- `top_k`: The number of memories to return.
|
|
- `include_memory_metadata`: Whether to include the memory metadata in the ChatMessage.
|
|
- `kwargs`: Additional data to pass to the State schema used by the Agent.
|
|
The keys must match the schema defined in the Agent's `state_schema`.
|
|
|
|
**Raises**:
|
|
|
|
- `RuntimeError`: If the Agent component wasn't warmed up before calling `run()`.
|
|
- `BreakpointException`: If an agent breakpoint is triggered.
|
|
|
|
**Returns**:
|
|
|
|
A dictionary with the following keys:
|
|
- "messages": List of all messages exchanged during the agent's run.
|
|
- "last_message": The last message exchanged during the agent's run.
|
|
- Any additional keys defined in the `state_schema`.
|
|
|
|
<a id="haystack_experimental.components.agents.agent.Agent.run_async"></a>
|
|
|
|
#### Agent.run\_async
|
|
|
|
```python
|
|
async def run_async(messages: list[ChatMessage],
|
|
streaming_callback: StreamingCallbackT | None = None,
|
|
*,
|
|
generation_kwargs: dict[str, Any] | None = None,
|
|
break_point: AgentBreakpoint | None = None,
|
|
snapshot: AgentSnapshot | None = None,
|
|
system_prompt: str | None = None,
|
|
tools: ToolsType | list[str] | None = None,
|
|
confirmation_strategy_context: dict[str, Any]
|
|
| None = None,
|
|
chat_message_store_kwargs: dict[str, Any] | None = None,
|
|
memory_store_kwargs: dict[str, Any] | None = None,
|
|
**kwargs: Any) -> dict[str, Any]
|
|
```
|
|
|
|
Asynchronously process messages and execute tools until the exit condition is met.
|
|
|
|
This is the asynchronous version of the `run` method. It follows the same logic but uses
|
|
asynchronous operations where possible, such as calling the `run_async` method of the ChatGenerator
|
|
if available.
|
|
|
|
**Arguments**:
|
|
|
|
- `messages`: List of Haystack ChatMessage objects to process.
|
|
- `streaming_callback`: An asynchronous callback that will be invoked when a response is streamed from the
|
|
LLM. The same callback can be configured to emit tool results when a tool is called.
|
|
- `generation_kwargs`: Additional keyword arguments for LLM. These parameters will
|
|
override the parameters passed during component initialization.
|
|
- `break_point`: An AgentBreakpoint, can be a Breakpoint for the "chat_generator" or a ToolBreakpoint
|
|
for "tool_invoker".
|
|
- `snapshot`: A dictionary containing a snapshot of a previously saved agent execution. The snapshot contains
|
|
the relevant information to restart the Agent execution from where it left off.
|
|
- `system_prompt`: System prompt for the agent. If provided, it overrides the default system prompt.
|
|
- `tools`: Optional list of Tool objects, a Toolset, or list of tool names to use for this run.
|
|
- `confirmation_strategy_context`: Optional dictionary for passing request-scoped resources
|
|
to confirmation strategies. Useful in web/server environments to provide per-request
|
|
objects (e.g., WebSocket connections, async queues, Redis pub/sub clients) that strategies
|
|
can use for non-blocking user interaction.
|
|
- `chat_message_store_kwargs`: Optional dictionary of keyword arguments to pass to the ChatMessageStore.
|
|
For example, it can include the `chat_history_id` and `last_k` parameters for retrieving chat history.
|
|
- `kwargs`: Additional data to pass to the State schema used by the Agent.
|
|
- `memory_store_kwargs`: Optional dictionary of keyword arguments to pass to the MemoryStore.
|
|
It can include:
|
|
- `user_id`: The user ID to search and add memories from.
|
|
- `run_id`: The run ID to search and add memories from.
|
|
- `agent_id`: The agent ID to search and add memories from.
|
|
- `search_criteria`: A dictionary of containing kwargs for the `search_memories` method.
|
|
This can include:
|
|
- `filters`: A dictionary of filters to search for memories.
|
|
- `query`: The query to search for memories.
|
|
Note: If you pass this, the user query passed to the agent will be
|
|
ignored for memory retrieval.
|
|
- `top_k`: The number of memories to return.
|
|
- `include_memory_metadata`: Whether to include the memory metadata in the ChatMessage.
|
|
- `kwargs`: Additional data to pass to the State schema used by the Agent.
|
|
The keys must match the schema defined in the Agent's `state_schema`.
|
|
|
|
**Raises**:
|
|
|
|
- `RuntimeError`: If the Agent component wasn't warmed up before calling `run_async()`.
|
|
- `BreakpointException`: If an agent breakpoint is triggered.
|
|
|
|
**Returns**:
|
|
|
|
A dictionary with the following keys:
|
|
- "messages": List of all messages exchanged during the agent's run.
|
|
- "last_message": The last message exchanged during the agent's run.
|
|
- Any additional keys defined in the `state_schema`.
|
|
|
|
<a id="haystack_experimental.components.agents.agent.Agent.to_dict"></a>
|
|
|
|
#### Agent.to\_dict
|
|
|
|
```python
|
|
def to_dict() -> dict[str, Any]
|
|
```
|
|
|
|
Serialize the component to a dictionary.
|
|
|
|
**Returns**:
|
|
|
|
Dictionary with serialized data
|
|
|
|
<a id="haystack_experimental.components.agents.agent.Agent.from_dict"></a>
|
|
|
|
#### Agent.from\_dict
|
|
|
|
```python
|
|
@classmethod
|
|
def from_dict(cls, data: dict[str, Any]) -> "Agent"
|
|
```
|
|
|
|
Deserialize the agent from a dictionary.
|
|
|
|
**Arguments**:
|
|
|
|
- `data`: Dictionary to deserialize from
|
|
|
|
**Returns**:
|
|
|
|
Deserialized agent
|
|
|
|
<a id="haystack_experimental.components.agents.human_in_the_loop.breakpoint"></a>
|
|
|
|
## Module haystack\_experimental.components.agents.human\_in\_the\_loop.breakpoint
|
|
|
|
<a id="haystack_experimental.components.agents.human_in_the_loop.breakpoint.get_tool_calls_and_descriptions_from_snapshot"></a>
|
|
|
|
#### get\_tool\_calls\_and\_descriptions\_from\_snapshot
|
|
|
|
```python
|
|
def get_tool_calls_and_descriptions_from_snapshot(
|
|
agent_snapshot: AgentSnapshot,
|
|
breakpoint_tool_only: bool = True
|
|
) -> tuple[list[dict], dict[str, str]]
|
|
```
|
|
|
|
Extract tool calls and tool descriptions from an AgentSnapshot.
|
|
|
|
By default, only the tool call that caused the breakpoint is processed and its arguments are reconstructed.
|
|
This is useful for scenarios where you want to present the relevant tool call and its description
|
|
to a human for confirmation before execution.
|
|
|
|
**Arguments**:
|
|
|
|
- `agent_snapshot`: The AgentSnapshot from which to extract tool calls and descriptions.
|
|
- `breakpoint_tool_only`: If True, only the tool call that caused the breakpoint is returned. If False, all tool
|
|
calls are returned.
|
|
|
|
**Returns**:
|
|
|
|
A tuple containing a list of tool call dictionaries and a dictionary of tool descriptions
|
|
|
|
<a id="haystack_experimental.components.agents.human_in_the_loop.errors"></a>
|
|
|
|
## Module haystack\_experimental.components.agents.human\_in\_the\_loop.errors
|
|
|
|
<a id="haystack_experimental.components.agents.human_in_the_loop.errors.HITLBreakpointException"></a>
|
|
|
|
### HITLBreakpointException
|
|
|
|
Exception raised when a tool execution is paused by a ConfirmationStrategy (e.g. BreakpointConfirmationStrategy).
|
|
|
|
<a id="haystack_experimental.components.agents.human_in_the_loop.errors.HITLBreakpointException.__init__"></a>
|
|
|
|
#### HITLBreakpointException.\_\_init\_\_
|
|
|
|
```python
|
|
def __init__(message: str,
|
|
tool_name: str,
|
|
snapshot_file_path: str,
|
|
tool_call_id: str | None = None) -> None
|
|
```
|
|
|
|
Initialize the HITLBreakpointException.
|
|
|
|
**Arguments**:
|
|
|
|
- `message`: The exception message.
|
|
- `tool_name`: The name of the tool whose execution is paused.
|
|
- `snapshot_file_path`: The file path to the saved pipeline snapshot.
|
|
- `tool_call_id`: Optional unique identifier for the tool call. This can be used to track and correlate
|
|
the decision with a specific tool invocation.
|
|
|
|
<a id="haystack_experimental.components.agents.human_in_the_loop.strategies"></a>
|
|
|
|
## Module haystack\_experimental.components.agents.human\_in\_the\_loop.strategies
|
|
|
|
<a id="haystack_experimental.components.agents.human_in_the_loop.strategies.BreakpointConfirmationStrategy"></a>
|
|
|
|
### BreakpointConfirmationStrategy
|
|
|
|
Confirmation strategy that raises a tool breakpoint exception to pause execution and gather user feedback.
|
|
|
|
This strategy is designed for scenarios where immediate user interaction is not possible.
|
|
When a tool execution requires confirmation, it raises an `HITLBreakpointException`, which is caught by the Agent.
|
|
The Agent then serialize its current state, including the tool call details. This information can then be used to
|
|
notify a user to review and confirm the tool execution.
|
|
|
|
<a id="haystack_experimental.components.agents.human_in_the_loop.strategies.BreakpointConfirmationStrategy.__init__"></a>
|
|
|
|
#### BreakpointConfirmationStrategy.\_\_init\_\_
|
|
|
|
```python
|
|
def __init__(snapshot_file_path: str) -> None
|
|
```
|
|
|
|
Initialize the BreakpointConfirmationStrategy.
|
|
|
|
**Arguments**:
|
|
|
|
- `snapshot_file_path`: The path to the directory that the snapshot should be saved.
|
|
|
|
<a id="haystack_experimental.components.agents.human_in_the_loop.strategies.BreakpointConfirmationStrategy.run"></a>
|
|
|
|
#### BreakpointConfirmationStrategy.run
|
|
|
|
```python
|
|
def run(
|
|
*,
|
|
tool_name: str,
|
|
tool_description: str,
|
|
tool_params: dict[str, Any],
|
|
tool_call_id: str | None = None,
|
|
confirmation_strategy_context: dict[str, Any] | None = None
|
|
) -> ToolExecutionDecision
|
|
```
|
|
|
|
Run the breakpoint confirmation strategy for a given tool and its parameters.
|
|
|
|
**Arguments**:
|
|
|
|
- `tool_name`: The name of the tool to be executed.
|
|
- `tool_description`: The description of the tool.
|
|
- `tool_params`: The parameters to be passed to the tool.
|
|
- `tool_call_id`: Optional unique identifier for the tool call. This can be used to track and correlate the decision with a
|
|
specific tool invocation.
|
|
- `confirmation_strategy_context`: Optional dictionary for passing request-scoped resources. Not used by this strategy but included for
|
|
interface compatibility.
|
|
|
|
**Raises**:
|
|
|
|
- `HITLBreakpointException`: Always raises an `HITLBreakpointException` exception to signal that user confirmation is required.
|
|
|
|
**Returns**:
|
|
|
|
This method does not return; it always raises an exception.
|
|
|
|
<a id="haystack_experimental.components.agents.human_in_the_loop.strategies.BreakpointConfirmationStrategy.run_async"></a>
|
|
|
|
#### BreakpointConfirmationStrategy.run\_async
|
|
|
|
```python
|
|
async def run_async(
|
|
*,
|
|
tool_name: str,
|
|
tool_description: str,
|
|
tool_params: dict[str, Any],
|
|
tool_call_id: str | None = None,
|
|
confirmation_strategy_context: dict[str, Any] | None = None
|
|
) -> ToolExecutionDecision
|
|
```
|
|
|
|
Async version of run. Calls the sync run() method.
|
|
|
|
**Arguments**:
|
|
|
|
- `tool_name`: The name of the tool to be executed.
|
|
- `tool_description`: The description of the tool.
|
|
- `tool_params`: The parameters to be passed to the tool.
|
|
- `tool_call_id`: Optional unique identifier for the tool call.
|
|
- `confirmation_strategy_context`: Optional dictionary for passing request-scoped resources.
|
|
|
|
**Raises**:
|
|
|
|
- `HITLBreakpointException`: Always raises an `HITLBreakpointException` exception to signal that user confirmation is required.
|
|
|
|
**Returns**:
|
|
|
|
This method does not return; it always raises an exception.
|
|
|
|
<a id="haystack_experimental.components.agents.human_in_the_loop.strategies.BreakpointConfirmationStrategy.to_dict"></a>
|
|
|
|
#### BreakpointConfirmationStrategy.to\_dict
|
|
|
|
```python
|
|
def to_dict() -> dict[str, Any]
|
|
```
|
|
|
|
Serializes the BreakpointConfirmationStrategy to a dictionary.
|
|
|
|
<a id="haystack_experimental.components.agents.human_in_the_loop.strategies.BreakpointConfirmationStrategy.from_dict"></a>
|
|
|
|
#### BreakpointConfirmationStrategy.from\_dict
|
|
|
|
```python
|
|
@classmethod
|
|
def from_dict(cls, data: dict[str, Any]) -> "BreakpointConfirmationStrategy"
|
|
```
|
|
|
|
Deserializes the BreakpointConfirmationStrategy from a dictionary.
|
|
|
|
**Arguments**:
|
|
|
|
- `data`: Dictionary to deserialize from.
|
|
|
|
**Returns**:
|
|
|
|
Deserialized BreakpointConfirmationStrategy.
|
|
|