diff --git a/src/mindpilot/app/agent/agent_api.py b/src/mindpilot/app/agent/agent_api.py index 4b06812..1a2dffd 100644 --- a/src/mindpilot/app/agent/agent_api.py +++ b/src/mindpilot/app/agent/agent_api.py @@ -55,21 +55,21 @@ def create_agent( FROM agents WHERE id = ? ''', (agent_id,)) - agent_data = cursor.fetchone() + agent = cursor.fetchone() conn.close() - if agent_data: + if agent: agent_dict = { - "id": agent_data[0], - "agent_name": agent_data[1], - "agent_abstract": agent_data[2], - "agent_info": agent_data[3], - "temperature": agent_data[4], - "max_tokens": agent_data[5], - "tool_config": agent_data[6].split(',') if agent_data[6] else [], - "kb_name": agent_data[7].split(',') if agent_data[7] else [], - "avatar": agent_data[8] + "id": agent[0], + "agent_name": agent[1], + "agent_abstract": agent[2], + "agent_info": agent[3], + "temperature": agent[4], + "max_tokens": agent[5], + "tool_config": agent[6].split(',') if agent[6] else [], + "kb_name": agent[7].split(',') if agent[7] else [], + "avatar": agent[8] } return BaseResponse(code=200, msg=f"已新增Agent {agent_name}", data=agent_dict) else: