codefuse-chatbot/coagent/connector/configs/agent_config.py

330 lines
10 KiB
Python

from enum import Enum
from .prompts import *
# from .prompts import (
# REACT_PROMPT_INPUT, CHECK_PROMPT_INPUT, EXECUTOR_PROMPT_INPUT, CONTEXT_PROMPT_INPUT, QUERY_CONTEXT_PROMPT_INPUT,PLAN_PROMPT_INPUT,
# RECOGNIZE_INTENTION_PROMPT,
# CHECKER_TEMPLATE_PROMPT,
# CONV_SUMMARY_PROMPT,
# QA_PROMPT, CODE_QA_PROMPT, QA_TEMPLATE_PROMPT,
# EXECUTOR_TEMPLATE_PROMPT,
# REFINE_TEMPLATE_PROMPT,
# SELECTOR_AGENT_TEMPLATE_PROMPT,
# PLANNER_TEMPLATE_PROMPT, GENERAL_PLANNER_PROMPT, DATA_PLANNER_PROMPT, TOOL_PLANNER_PROMPT,
# PRD_WRITER_METAGPT_PROMPT, DESIGN_WRITER_METAGPT_PROMPT, TASK_WRITER_METAGPT_PROMPT, CODE_WRITER_METAGPT_PROMPT,
# REACT_TEMPLATE_PROMPT,
# REACT_TOOL_PROMPT, REACT_CODE_PROMPT, REACT_TOOL_AND_CODE_PLANNER_PROMPT, REACT_TOOL_AND_CODE_PROMPT
# )
from .prompt_config import *
# BASE_PROMPT_CONFIGS, EXECUTOR_PROMPT_CONFIGS, SELECTOR_PROMPT_CONFIGS, BASE_NOTOOLPROMPT_CONFIGS
class AgentType:
REACT = "ReactAgent"
EXECUTOR = "ExecutorAgent"
ONE_STEP = "BaseAgent"
DEFAULT = "BaseAgent"
SELECTOR = "SelectorAgent"
AGETN_CONFIGS = {
"baseGroup": {
"role": {
"role_prompt": SELECTOR_AGENT_TEMPLATE_PROMPT,
"role_type": "assistant",
"role_name": "baseGroup",
"role_desc": "",
"agent_type": "SelectorAgent"
},
"prompt_config": SELECTOR_PROMPT_CONFIGS,
"group_agents": ["tool_react", "code_react"],
"chat_turn": 1,
},
"checker": {
"role": {
"role_prompt": CHECKER_TEMPLATE_PROMPT,
"role_type": "assistant",
"role_name": "checker",
"role_desc": "",
"agent_type": "BaseAgent"
},
"prompt_config": BASE_PROMPT_CONFIGS,
"chat_turn": 1,
},
"conv_summary": {
"role": {
"role_prompt": CONV_SUMMARY_PROMPT,
"role_type": "assistant",
"role_name": "conv_summary",
"role_desc": "",
"agent_type": "BaseAgent"
},
"prompt_config": BASE_PROMPT_CONFIGS,
"chat_turn": 1,
},
"general_planner": {
"role": {
"role_prompt": PLANNER_TEMPLATE_PROMPT,
"role_type": "assistant",
"role_name": "general_planner",
"role_desc": "",
"agent_type": "BaseAgent"
},
"prompt_config": BASE_PROMPT_CONFIGS,
"chat_turn": 1,
},
"executor": {
"role": {
"role_prompt": EXECUTOR_TEMPLATE_PROMPT,
"role_type": "assistant",
"role_name": "executor",
"role_desc": "",
"agent_type": "ExecutorAgent",
},
"prompt_config": EXECUTOR_PROMPT_CONFIGS,
"stop": "\n**Observation:**",
"chat_turn": 1,
},
"base_refiner": {
"role": {
"role_prompt": REFINE_TEMPLATE_PROMPT,
"role_type": "assistant",
"role_name": "base_refiner",
"role_desc": "",
"agent_type": "BaseAgent"
},
"prompt_config": BASE_PROMPT_CONFIGS,
"chat_turn": 1,
},
"planner": {
"role": {
"role_prompt": DATA_PLANNER_PROMPT,
"role_type": "assistant",
"role_name": "planner",
"role_desc": "",
"agent_type": "BaseAgent"
},
"prompt_config": BASE_PROMPT_CONFIGS,
"chat_turn": 1,
},
"intention_recognizer": {
"role": {
"role_prompt": RECOGNIZE_INTENTION_PROMPT,
"role_type": "assistant",
"role_name": "intention_recognizer",
"role_desc": "",
"agent_type": "BaseAgent"
},
"prompt_config": BASE_PROMPT_CONFIGS,
"chat_turn": 1,
},
"tool_planner": {
"role": {
"role_prompt": TOOL_PLANNER_PROMPT,
"role_type": "assistant",
"role_name": "tool_planner",
"role_desc": "",
"agent_type": "BaseAgent"
},
"prompt_config": BASE_PROMPT_CONFIGS,
"chat_turn": 1,
},
"tool_and_code_react": {
"role": {
"role_prompt": REACT_TOOL_AND_CODE_PROMPT,
"role_type": "assistant",
"role_name": "tool_and_code_react",
"role_desc": "",
"agent_type": "ReactAgent",
},
"prompt_config": BASE_PROMPT_CONFIGS,
"stop": "\n**Observation:**",
"chat_turn": 7,
},
"tool_and_code_planner": {
"role": {
"role_prompt": REACT_TOOL_AND_CODE_PLANNER_PROMPT,
"role_type": "assistant",
"role_name": "tool_and_code_planner",
"role_desc": "",
"agent_type": "BaseAgent"
},
"prompt_config": BASE_PROMPT_CONFIGS,
"chat_turn": 1,
},
"tool_react": {
"role": {
"role_prompt": REACT_TOOL_PROMPT,
"role_type": "assistant",
"role_name": "tool_react",
"role_desc": "",
"agent_type": "ReactAgent"
},
"prompt_config": BASE_PROMPT_CONFIGS,
"chat_turn": 5,
"stop": "\n**Observation:**"
},
"code_react": {
"role": {
"role_prompt": REACT_CODE_PROMPT,
"role_type": "assistant",
"role_name": "code_react",
"role_desc": "",
"agent_type": "ReactAgent"
},
"prompt_config": BASE_NOTOOLPROMPT_CONFIGS,
"chat_turn": 5,
"stop": "\n**Observation:**"
},
"qaer": {
"role": {
"role_prompt": QA_TEMPLATE_PROMPT,
"role_type": "assistant",
"role_name": "qaer",
"role_desc": "",
"agent_type": "BaseAgent"
},
"prompt_config": BASE_PROMPT_CONFIGS,
"chat_turn": 1,
},
"code_qaer": {
"role": {
"role_prompt": CODE_QA_PROMPT,
"role_type": "assistant",
"role_name": "code_qaer",
"role_desc": "",
"agent_type": "BaseAgent"
},
"prompt_config": BASE_PROMPT_CONFIGS,
"chat_turn": 1,
},
"searcher": {
"role": {
"role_prompt": QA_TEMPLATE_PROMPT,
"role_type": "assistant",
"role_name": "searcher",
"role_desc": "",
"agent_type": "BaseAgent"
},
"prompt_config": BASE_PROMPT_CONFIGS,
"chat_turn": 1,
},
"metaGPT_PRD": {
"role": {
"role_prompt": PRD_WRITER_METAGPT_PROMPT,
"role_type": "assistant",
"role_name": "metaGPT_PRD",
"role_desc": "",
"agent_type": "BaseAgent"
},
"prompt_config": BASE_PROMPT_CONFIGS,
"chat_turn": 1,
"focus_agents": [],
"focus_message_keys": [],
},
"metaGPT_DESIGN": {
"role": {
"role_prompt": DESIGN_WRITER_METAGPT_PROMPT,
"role_type": "assistant",
"role_name": "metaGPT_DESIGN",
"role_desc": "",
"agent_type": "BaseAgent"
},
"prompt_config": BASE_PROMPT_CONFIGS,
"chat_turn": 1,
"focus_agents": ["metaGPT_PRD"],
"focus_message_keys": [],
},
"metaGPT_TASK": {
"role": {
"role_prompt": TASK_WRITER_METAGPT_PROMPT,
"role_type": "assistant",
"role_name": "metaGPT_TASK",
"role_desc": "",
"agent_type": "BaseAgent"
},
"prompt_config": BASE_PROMPT_CONFIGS,
"chat_turn": 1,
"focus_agents": ["metaGPT_DESIGN"],
"focus_message_keys": [],
},
"metaGPT_CODER": {
"role": {
"role_prompt": CODE_WRITER_METAGPT_PROMPT,
"role_type": "assistant",
"role_name": "metaGPT_CODER",
"role_desc": "",
"agent_type": "ExecutorAgent"
},
"prompt_config": EXECUTOR_PROMPT_CONFIGS,
"chat_turn": 1,
"focus_agents": ["metaGPT_DESIGN", "metaGPT_TASK"],
"focus_message_keys": [],
},
"class2Docer": {
"role": {
"role_prompt": Class2Doc_PROMPT,
"role_type": "assistant",
"role_name": "class2Docer",
"role_desc": "",
"agent_type": "CodeGenDocer"
},
"prompt_config": CODE2DOC_PROMPT_CONFIGS,
"prompt_manager_type": "Code2DocPM",
"chat_turn": 1,
"focus_agents": [],
"focus_message_keys": [],
},
"func2Docer": {
"role": {
"role_prompt": Func2Doc_PROMPT,
"role_type": "assistant",
"role_name": "func2Docer",
"role_desc": "",
"agent_type": "CodeGenDocer"
},
"prompt_config": CODE2DOC_PROMPT_CONFIGS,
"prompt_manager_type": "Code2DocPM",
"chat_turn": 1,
"focus_agents": [],
"focus_message_keys": [],
},
"code2DocsGrouper": {
"role": {
"role_prompt": Code2DocGroup_PROMPT,
"role_type": "assistant",
"role_name": "code2DocsGrouper",
"role_desc": "",
"agent_type": "SelectorAgent"
},
"prompt_config": CODE2DOC_GROUP_PROMPT_CONFIGS,
"group_agents": ["class2Docer", "func2Docer"],
"chat_turn": 1,
},
"Code2TestJudger": {
"role": {
"role_prompt": judgeCode2Tests_PROMPT,
"role_type": "assistant",
"role_name": "Code2TestJudger",
"role_desc": "",
"agent_type": "CodeRetrieval"
},
"prompt_config": CODE2TESTS_PROMPT_CONFIGS,
"prompt_manager_type": "CodeRetrievalPM",
"chat_turn": 1,
},
"code2Tests": {
"role": {
"role_prompt": code2Tests_PROMPT,
"role_type": "assistant",
"role_name": "code2Tests",
"role_desc": "",
"agent_type": "CodeRetrieval"
},
"prompt_config": CODE2TESTS_PROMPT_CONFIGS,
"prompt_manager_type": "CodeRetrievalPM",
"chat_turn": 1,
},
}