aiq.agent.dual_node

Attributes

log

Classes

DualNodeAgent

Helper class that provides a standard way to create an ABC using

Module Contents

log

class DualNodeAgent(

llm: langchain_core.language_models.BaseChatModel,

tools: list[langchain_core.tools.BaseTool],

callbacks: list[langchain_core.callbacks.AsyncCallbackHandler] = None,

detailed_logs: bool = False,

)

Bases: aiq.agent.base.BaseAgent

Helper class that provides a standard way to create an ABC using inheritance.

abstractmethod agent_node(state: pydantic.BaseModel) → pydantic.BaseModel

Async:

abstractmethod tool_node(state: pydantic.BaseModel) → pydantic.BaseModel

Async:

abstractmethod conditional_edge(state: pydantic.BaseModel) → str

Async:

async _build_graph(state_schema) → langgraph.graph.graph.CompiledGraph