62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
from langgraph.graph import END, START, StateGraph
|
||
from langgraph.prebuilt import tools_condition
|
||
|
||
from src.agents.common.toolagent import ToolAgent
|
||
|
||
from .context import Context
|
||
from .state import State
|
||
from .tools import get_tools
|
||
|
||
|
||
class SampleMultiAgent(ToolAgent):
|
||
name = "MultiAgent智能体"
|
||
description = "Supervisor智能体，具有调用其他子智能体的能力(在工具中添加)"
|
||
|
||
# TODO[已完成]: 通过将其他agent封装为工具的方式添加了多智能体调度
|
||
'''
|
||
你是一个多智能体核心，通过多智能体调用的方式帮助用户完成一系列任务：
|
||
|
||
1.当你需要知识库问答功能时，请调用对话聊天智能体实现
|
||
2.当你需要加密计算的时候，请调用加密计算智能体实现
|
||
'''
|
||
|
||
def __init__(self, **kwargs):
|
||
super().__init__(**kwargs)
|
||
self.graph = None
|
||
self.checkpointer = None
|
||
self.context_schema = Context
|
||
self.agent_tools = None
|
||
|
||
def get_tools(self):
|
||
return get_tools()
|
||
|
||
async def get_graph(self, **kwargs):
|
||
"""构建图"""
|
||
if self.graph:
|
||
return self.graph
|
||
|
||
builder = StateGraph(State, context_schema=self.context_schema)
|
||
builder.add_node("chatbot", self.llm_call)
|
||
builder.add_node("tools", self.dynamic_tools_node)
|
||
builder.add_edge(START, "chatbot")
|
||
builder.add_conditional_edges(
|
||
"chatbot",
|
||
tools_condition,
|
||
)
|
||
builder.add_edge("tools", "chatbot")
|
||
builder.add_edge("chatbot", END)
|
||
|
||
self.checkpointer = await self._get_checkpointer()
|
||
graph = builder.compile(checkpointer=self.checkpointer, name=self.name)
|
||
self.graph = graph
|
||
return graph
|
||
|
||
|
||
def main():
|
||
pass
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|
||
# asyncio.run(main())
|