Function create_agent_endpoints

autoagent/server.py:71–111

()

# 重写agent endpoints创建逻辑
@app.on_event("startup")
def create_agent_endpoints():
    for agent_name, agent_func in registry.agents.items():
        async def create_agent_endpoint(
            request: AgentRequest, 
            func=agent_func
        ) -> AgentResponse:
            try:
                # 创建agent实例
                agent = func(model=request.model)
                
                # 创建MetaChain实例
                mc = MetaChain()
                
                # 构建messages
                messages = [
                    {"role": "user", "content": request.query}
                ]
                
                # 运行agent
                response = mc.run(
                    agent=agent,
                    messages=messages,
                    context_storage=request.context_variables,
                    debug=True
                )
                
                return AgentResponse(
                    result=response.messages[-1]['content'],
                    messages=response.messages,
                    agent_name=agent.name
                )
                
            except Exception as e:
                raise HTTPException(
                    status_code=400,
                    detail=f"Agent execution failed: {str(e)}"
                )
        
        endpoint = create_agent_endpoint
        endpoint.__name__ = f"agent_{agent_name}"
        app.post(f"/agents/{agent_name}/run")(endpoint)

# 获取所有可用的agents信息
@app.get("/agents")