taolib.testing.multi_agent.agents.factory

智能体工厂。

用于创建各种类型的智能体。

Attributes

_global_factory

Classes

AgentFactory

智能体工厂。

Functions

get_agent_factory(→ AgentFactory)	获取全局智能体工厂。
set_agent_factory(→ None)	设置全局智能体工厂。

Module Contents

class taolib.testing.multi_agent.agents.factory.AgentFactory(llm_manager: taolib.testing.multi_agent.llm.LLMManager | None = None)

智能体工厂。

_llm_manager = None

_templates: Dict[str, taolib.testing.multi_agent.agents.templates.AgentTemplate]

_agent_classes: Dict[taolib.testing.multi_agent.models.AgentType, Type[taolib.testing.multi_agent.agents.base.BaseAgent]]

register_template(template: taolib.testing.multi_agent.agents.templates.AgentTemplate) → None

注册智能体模板。

参数:

template -- 智能体模板

get_template(template_id: str) → taolib.testing.multi_agent.agents.templates.AgentTemplate | None

获取智能体模板。

参数:

template_id -- 模板ID

返回:

模板,如果不存在则返回None

返回类型:

Optional[AgentTemplate]

get_all_templates() → Dict[str, taolib.testing.multi_agent.agents.templates.AgentTemplate]

获取所有模板。

返回:

所有模板的字典

返回类型:

Dict[str, AgentTemplate]

async create_agent(agent_create: taolib.testing.multi_agent.models.AgentCreate, agent_id: str | None = None) → taolib.testing.multi_agent.agents.base.BaseAgent

创建智能体。

参数:

• agent_create -- 智能体创建配置

• agent_id -- 智能体ID,如果为None则自动生成

返回:

创建的智能体

返回类型:

BaseAgent

抛出:

AgentError -- 无法创建智能体

async create_agent_from_template(template_id: str, agent_name: str | None = None, agent_id: str | None = None) → taolib.testing.multi_agent.agents.base.BaseAgent

从模板创建智能体。

参数:

• template_id -- 模板ID

• agent_name -- 智能体名称,如果为None则使用模板名称

• agent_id -- 智能体ID,如果为None则自动生成

返回:

创建的智能体

返回类型:

BaseAgent

抛出:

AgentError -- 模板不存在

async create_main_agent(name: str = '主智能体', agent_id: str | None = None) → taolib.testing.multi_agent.agents.main_agent.MainAgent

创建主智能体。

参数:

• name -- 智能体名称

• agent_id -- 智能体ID,如果为None则自动生成

返回:

创建的主智能体

返回类型:

MainAgent

taolib.testing.multi_agent.agents.factory._global_factory: AgentFactory | None = None

taolib.testing.multi_agent.agents.factory.get_agent_factory() → AgentFactory

获取全局智能体工厂。

返回:

智能体工厂

返回类型:

AgentFactory

taolib.testing.multi_agent.agents.factory.set_agent_factory(factory: AgentFactory) → None

设置全局智能体工厂。

参数:

factory -- 智能体工厂