chore: import upstream snapshot with attribution
This commit is contained in:
wehub-resource-sync
@@ -0,0 +1,57 @@
|
||||
import tkinter as tk
|
||||
from tkinter import filedialog
|
||||
import shutil
|
||||
import os
|
||||
from rich.console import Console
|
||||
|
||||
def select_and_copy_files(dest_dir, console: Console, docker_files_dir: str):
|
||||
# 创建 tkinter 根窗口但隐藏它
|
||||
root = tk.Tk()
|
||||
root.withdraw()
|
||||
|
||||
# 打开文件选择对话框
|
||||
files = filedialog.askopenfilenames(
|
||||
title='Select files to copy',
|
||||
filetypes=[
|
||||
# ('Text files', '*.txt'),
|
||||
('All files', '*.*'),
|
||||
('PDF files', '*.pdf'),
|
||||
('Docx files', '*.docx'),
|
||||
('Txt files', '*.txt'),
|
||||
('Zip files', '*.zip'),
|
||||
('Text files', '*.txt'),
|
||||
]
|
||||
)
|
||||
|
||||
if not files:
|
||||
print("No files selected")
|
||||
return
|
||||
|
||||
# 选择目标文件夹
|
||||
# dest_dir = filedialog.askdirectory(
|
||||
# title='Select destination folder'
|
||||
# )
|
||||
|
||||
if not dest_dir:
|
||||
print("No destination folder selected")
|
||||
return
|
||||
|
||||
# 复制文件
|
||||
upload_infos = []
|
||||
for file_path in files:
|
||||
file_name = os.path.basename(file_path)
|
||||
dest_path = os.path.join(dest_dir, file_name)
|
||||
docker_dest_path = os.path.join(docker_files_dir, file_name)
|
||||
try:
|
||||
shutil.copy2(file_path, dest_path)
|
||||
msg = f"Uploaded: {file_name} to {docker_dest_path}"
|
||||
upload_infos.append(msg)
|
||||
console.print(f"[bold green]{msg}[/bold green]")
|
||||
except Exception as e:
|
||||
console.print(f"[bold red]Error uploading {file_name}: {e}[/bold red]")
|
||||
|
||||
console.print(f"[bold green]Successfully uploaded {len(files)} files[/bold green]")
|
||||
return upload_infos
|
||||
if __name__ == "__main__":
|
||||
dest_dir = "/Users/tangjiabin/Documents/reasoning/metachain/workspace_meta_showcase/showcase_nl2agent_showcase/workplace"
|
||||
select_and_copy_files(dest_dir)
|
||||
@@ -0,0 +1,256 @@
|
||||
from autoagent import MetaChain
|
||||
from autoagent.util import UserCompleter
|
||||
from prompt_toolkit import PromptSession
|
||||
from prompt_toolkit.formatted_text import HTML
|
||||
from prompt_toolkit.styles import Style
|
||||
from autoagent.logger import LoggerManager, MetaChainLogger
|
||||
from rich.console import Console
|
||||
from rich.panel import Panel
|
||||
from autoagent.agents.meta_agent.agent_former import get_agent_former_agent
|
||||
from autoagent.agents.meta_agent.tool_editor import get_tool_editor_agent
|
||||
from autoagent.agents.meta_agent.agent_creator import get_agent_creator_agent
|
||||
import re
|
||||
from autoagent.agents.meta_agent.form_complie import parse_agent_form
|
||||
|
||||
|
||||
def extract_agents_content(text):
|
||||
pattern = r'(<agents>.*?</agents>)'
|
||||
# re.DOTALL 让 . 也能匹配换行符
|
||||
match = re.search(pattern, text, re.DOTALL)
|
||||
if match:
|
||||
return match.group(1)
|
||||
return None
|
||||
|
||||
def agent_profiling(agent_former, client, messages, context_variables, requirements, debug):
|
||||
messages.append({"role": "user", "content": requirements+ """
|
||||
Directly output the form in the XML format without ANY other text.
|
||||
"""})
|
||||
|
||||
response = client.run(agent_former, messages, context_variables, debug=debug)
|
||||
output_xml_form = response.messages[-1]["content"]
|
||||
messages.extend(response.messages)
|
||||
agent_form = None
|
||||
|
||||
MAX_RETRY = 3
|
||||
for i in range(MAX_RETRY):
|
||||
try:
|
||||
output_xml_form = extract_agents_content(output_xml_form)
|
||||
assert output_xml_form is not None, "No the XML form should be found in the output with the tag <agents>...</agents>."
|
||||
agent_form = parse_agent_form(output_xml_form)
|
||||
break
|
||||
except Exception as e:
|
||||
print(f"Error parsing XML to agent form: {e}. Retry {i+1}/{MAX_RETRY}")
|
||||
messages.append({"role": "user", "content": f"Error parsing XML to agent form: {e}\nNote that there are some special restrictions for creating agent form, please try again."})
|
||||
response = client.run(agent_former, messages, context_variables, debug=debug)
|
||||
output_xml_form = response.messages[-1]["content"]
|
||||
messages.extend(response.messages)
|
||||
return agent_form, output_xml_form, messages
|
||||
|
||||
def tool_editing(tool_editor_agent, client, messages, context_variables, agent_form, output_xml_form, debug, suggestions = ""):
|
||||
def case_resolved(task_response: str, context_variables: dict):
|
||||
"""
|
||||
Use this tools when ALL desired tools are created and tested successfully. You can NOT use this tool if tools are not created or tested successfully by running the tools.
|
||||
|
||||
Args:
|
||||
task_response: the response of creating the tool which contains the completion status of the tool.
|
||||
"""
|
||||
return f"Case resolved. ALL desired tools are created and tested successfully. Details: {task_response}"
|
||||
def case_not_resolved(task_response: str, context_variables: dict):
|
||||
"""
|
||||
Use this tools when you encounter irresistible errors after trying your best with multiple attempts for creating the desired tool. You can NOT use this tool before you have tried your best.
|
||||
|
||||
Args:
|
||||
task_response: the reason why the tool is not created or tested successfully.
|
||||
"""
|
||||
return f"Case not resolved. Some desired tools are not created or tested successfully. Details: {task_response}"
|
||||
tool_editor_agent.functions.extend([case_resolved, case_not_resolved])
|
||||
MAX_RETRY = 3
|
||||
|
||||
if suggestions != "":
|
||||
suggestions = "[IMPORTANT] Here are some suggestions for creating the tools: " + suggestions
|
||||
|
||||
agents = agent_form.agents
|
||||
new_tools = []
|
||||
for agent in agents:
|
||||
if len(agent.tools.new) > 0:
|
||||
|
||||
for idx, tool in enumerate(agent.tools.new):
|
||||
new_tools.append(f"{idx+1}. Tool name: {tool.name}, Tool description: {tool.description}")
|
||||
if len(new_tools) == 0:
|
||||
return "Case resolved. ALL desired tools are created and tested successfully.", messages
|
||||
new_tools_str = "\n".join(new_tools)
|
||||
messages.append({"role": "user", "content": f"""\
|
||||
Your task is to create a list of new tools for me, the tools are:
|
||||
{new_tools_str}
|
||||
{suggestions}
|
||||
|
||||
Please create these new tools for me, note that you can NOT stop util you have created all the tools and tested them using `run_tool` successfully.
|
||||
|
||||
If ALL tools are created and tested successfully, you can stop and use `case_resolved` tool. Otherwise, you should continue to create the tools. After you have tried your best, you can use `case_not_resolved` tool to give the reason why the tool is not created or tested successfully.
|
||||
|
||||
[IMPORTANT] ALL tools MUST be tested successfully by running the tools using `run_tool` before you stop.
|
||||
"""})
|
||||
response = client.run(tool_editor_agent, messages, context_variables, debug=debug)
|
||||
content = response.messages[-1]["content"]
|
||||
for i in range(MAX_RETRY):
|
||||
if content.startswith("Case resolved"):
|
||||
return content, messages
|
||||
messages.append({"role": "user", "content": f"""\
|
||||
Your task is to create a list of new tools for me, the tools are:
|
||||
{new_tools_str}
|
||||
|
||||
Please create these new tools for me, note that you can NOT stop util you have created all the tools and tested them using `run_tool` successfully.
|
||||
The last attempt failed with the following error: {content}, please try again to create the tools.
|
||||
"""})
|
||||
response = client.run(tool_editor_agent, messages, context_variables, debug=debug)
|
||||
content = response.messages[-1]["content"]
|
||||
if i == MAX_RETRY:
|
||||
return f"{content}\nSome desired tools are not created or tested successfully with {MAX_RETRY} attempts.", messages
|
||||
|
||||
def agent_editing(agent_creator_agent, client, messages, context_variables, agent_form, output_xml_form, requirements, task, debug, suggestions = ""):
|
||||
MAX_RETRY = 3
|
||||
if suggestions != "":
|
||||
suggestions = "[IMPORTANT] Here are some suggestions for creating the agent(s): " + suggestions
|
||||
def case_resolved(task_response: str, context_variables: dict):
|
||||
"""
|
||||
Use this tools when the desired agent(s) is created and tested successfully. You can NOT use this tool if the agent(s) is not created or tested successfully by running the agent(s).
|
||||
"""
|
||||
return f"Case resolved. The desired agent(s) is created and tested successfully. : {task_response}"
|
||||
def case_not_resolved(task_response: str, context_variables: dict):
|
||||
"""
|
||||
Use this tools when you encounter irresistible errors after trying your best with multiple attempts for creating the desired agent(s). You can NOT use this tool before you have tried your best.
|
||||
"""
|
||||
return f"Case not resolved. The desired agent(s) is not created or tested successfully. Details: {task_response}"
|
||||
agent_creator_agent.functions.extend([case_resolved, case_not_resolved])
|
||||
messages.append({"role": "user", "content": f"""\
|
||||
The user's request to create agent(s) is: {requirements}
|
||||
Given the completed agent form with XML format: {output_xml_form}
|
||||
After previous attempts, you have created new tools that required by the desired agent(s).
|
||||
|
||||
Your task is to create the desired agent(s) for me, note that you may create ONE single agent or multiple agents connected by orchestrator agent.
|
||||
|
||||
After you have created the agent(s), you should test the agent(s) by running the agent(s) using `run_agent` tool to complete the user's task:
|
||||
{task}
|
||||
|
||||
Note that you can NOT stop util you have created the agent(s) and tested it successfully.
|
||||
{suggestions}
|
||||
"""})
|
||||
response = client.run(agent_creator_agent, messages, context_variables, debug=debug)
|
||||
content = response.messages[-1]["content"]
|
||||
for i in range(MAX_RETRY):
|
||||
if content.startswith("Case resolved"):
|
||||
return content, messages
|
||||
messages.append({"role": "user", "content": f"""\
|
||||
The user's request to create agent(s) is: {requirements}
|
||||
Given the completed agent form with XML format: {output_xml_form}
|
||||
After previous attempts, you have created new tools that required by the desired agent(s).
|
||||
|
||||
Your task is to create the desired agent(s) for me, note that you may create ONE single agent or multiple agents connected by orchestrator agent.
|
||||
|
||||
After you have created the agent(s), you should test the agent(s) by running the agent(s) using `run_agent` tool to complete the user's task:
|
||||
{task}
|
||||
|
||||
Note that you can NOT stop util you have created the agent(s) and tested it successfully.
|
||||
The last attempt failed with the following error: {content}, please try again to create the desired agent(s).
|
||||
{suggestions}
|
||||
"""})
|
||||
response = client.run(agent_creator_agent, messages, context_variables, debug=debug)
|
||||
content = response.messages[-1]["content"]
|
||||
if i == MAX_RETRY:
|
||||
return f"{content}\nThe desired agent(s) is not created or tested successfully with {MAX_RETRY} attempts.", messages
|
||||
|
||||
|
||||
def meta_agent(model: str, context_variables: dict, debug: bool = True):
|
||||
logger = LoggerManager.get_logger()
|
||||
# generate agent form
|
||||
agent_former = get_agent_former_agent(model)
|
||||
tool_editor_agent = get_tool_editor_agent(model)
|
||||
agent_creator_agent = get_agent_creator_agent(model)
|
||||
# enter agent
|
||||
agent = agent_former
|
||||
agents = {agent_former.name.replace(' ', '_'): agent_former, tool_editor_agent.name.replace(' ', '_'): tool_editor_agent, agent_creator_agent.name.replace(' ', '_'): agent_creator_agent}
|
||||
style = Style.from_dict({
|
||||
'bottom-toolbar': 'bg:#333333 #ffffff',
|
||||
})
|
||||
# 创建会话
|
||||
session = PromptSession(
|
||||
completer=UserCompleter(agents.keys()),
|
||||
complete_while_typing=True,
|
||||
style=style
|
||||
)
|
||||
client = MetaChain(log_path=logger)
|
||||
console = Console()
|
||||
messages = []
|
||||
|
||||
last_message = "Tell me what do you want to create with `Agent Chain`?"
|
||||
|
||||
while True:
|
||||
query = session.prompt(
|
||||
f'{last_message} (type "exit" to quit, press "Enter" to continue): ',
|
||||
bottom_toolbar=HTML('<b>Prompt:</b> Enter <b>@</b> to mention Agents'),
|
||||
)
|
||||
if query.strip().lower() == 'exit':
|
||||
|
||||
logo_text = "Agent Chain completed. See you next time! :waving_hand:"
|
||||
console.print(Panel(logo_text, style="bold salmon1", expand=True))
|
||||
break
|
||||
words = query.split()
|
||||
console.print(f"[bold green]Your request: {query}[/bold green]", end=" ")
|
||||
for word in words:
|
||||
if word.startswith('@') and word[1:] in agents.keys():
|
||||
# print(f"[bold magenta]{word}[bold magenta]", end=' ')
|
||||
agent = agents[word.replace('@', '')]
|
||||
else:
|
||||
# print(word, end=' ')
|
||||
pass
|
||||
print()
|
||||
agent_name = agent.name
|
||||
console.print(f"[bold green][bold magenta]@{agent_name}[/bold magenta] will help you, be patient...[/bold green]")
|
||||
|
||||
match agent_name:
|
||||
case 'Agent Former Agent':
|
||||
if query == "":
|
||||
console.print(f"[bold red]There MUST be a request to create the agent form.[/bold red]")
|
||||
continue
|
||||
requirements = query
|
||||
agent_form, output_xml_form, messages = agent_profiling(agent_former, client, messages, context_variables, requirements, debug)
|
||||
if agent_form is None:
|
||||
console.print(f"[bold red][bold magenta]@{agent_name}[/bold magenta] has not created agent form successfully, please modify your requirements again.[/bold red]")
|
||||
last_message = "Tell me what do you want to create with `Agent Chain`?"
|
||||
continue
|
||||
|
||||
agent = tool_editor_agent
|
||||
console.print(f"[bold green][bold magenta]@{agent_name}[/bold magenta] has created agent form successfully with the following details:\n[/bold green][bold blue]{output_xml_form}[/bold blue]")
|
||||
last_message = "It is time to create the desired tools, do you have any suggestions for creating the tools?"
|
||||
case 'Tool Editor Agent':
|
||||
suggestions = query
|
||||
tool_response, messages = tool_editing(tool_editor_agent, client, messages, context_variables, agent_form, output_xml_form, debug, suggestions)
|
||||
if tool_response.startswith("Case not resolved"):
|
||||
console.print(f"[bold red][bold magenta]@{agent_name}[/bold magenta] has not created tools successfully with the following error: {tool_response}[/bold red]")
|
||||
agent = tool_editor_agent
|
||||
last_message = "The tools are not created successfully, do you have any suggestions for creating the tools?"
|
||||
continue
|
||||
elif tool_response.startswith("Case resolved"):
|
||||
agent = agent_creator_agent
|
||||
console.print(f"[bold green][bold magenta]@{agent_name}[/bold magenta] has created tools successfully with the following details:\n[/bold green][bold blue]{tool_response}[/bold blue]")
|
||||
last_message = "It is time to create the desired agent(s), do you have any suggestions for creating the agent(s)?"
|
||||
else:
|
||||
raise ValueError(f"Unknown tool response: {tool_response}")
|
||||
|
||||
case 'Agent Creator Agent':
|
||||
suggestions = query
|
||||
default_value='Come up with a task for the agent(s) to test your created agent(s), and use `run_agent` tool to test your created agent(s).' # 这里设置你想要的默认值
|
||||
task = session.prompt(
|
||||
'It is time to create the desired agent(s), what task do you want to complete with the agent(s)? (Press Enter if none): ',
|
||||
|
||||
)
|
||||
task = default_value if not task.strip() else task
|
||||
agent_response, messages = agent_editing(agent_creator_agent, client, messages, context_variables, agent_form, output_xml_form, requirements, task, debug, suggestions)
|
||||
if agent_response.startswith("Case not resolved"):
|
||||
console.print(f"[bold red][bold magenta]@{agent_name}[/bold magenta] has not created agent(s) successfully with the following error: {agent_response}[/bold red]")
|
||||
agent = agent_creator_agent
|
||||
last_message = "The agent(s) are not created successfully, do you have any suggestions for creating the agent(s)?"
|
||||
continue
|
||||
else:
|
||||
console.print(f"[bold green][bold magenta]@{agent_name}[/bold magenta] has created agent(s) successfully with the following details:\n[/bold green][bold blue]{agent_response}[/bold blue]")
|
||||
last_message = "Tell me what do you want to create with `Agent Chain`?"
|
||||
@@ -0,0 +1,194 @@
|
||||
from autoagent import MetaChain
|
||||
from autoagent.util import ask_text, single_select_menu, print_markdown, debug_print, UserCompleter
|
||||
from prompt_toolkit import PromptSession
|
||||
from prompt_toolkit.formatted_text import HTML
|
||||
from prompt_toolkit.styles import Style
|
||||
from autoagent.logger import LoggerManager, MetaChainLogger
|
||||
from rich.console import Console
|
||||
from rich.panel import Panel
|
||||
from autoagent.agents.meta_agent.workflow_former import get_workflow_former_agent
|
||||
from autoagent.agents.meta_agent.workflow_creator import get_workflow_creator_agent
|
||||
import re
|
||||
from autoagent.agents.meta_agent.worklow_form_complie import parse_workflow_form, WorkflowForm
|
||||
|
||||
def workflow_profiling(workflow_former, client, messages, context_variables, requirements, debug):
|
||||
messages.append({"role": "user", "content": requirements + """
|
||||
Directly output the form in the XML format without ANY other text.
|
||||
"""})
|
||||
response = client.run(workflow_former, messages, context_variables, debug=debug)
|
||||
output_xml_form = response.messages[-1]["content"]
|
||||
messages.extend(response.messages)
|
||||
|
||||
MAX_RETRY = 3
|
||||
for i in range(MAX_RETRY):
|
||||
workflow_form = parse_workflow_form(output_xml_form)
|
||||
if isinstance(workflow_form, WorkflowForm):
|
||||
break
|
||||
elif isinstance(workflow_form, str):
|
||||
print(f"Error parsing XML to workflow form: {workflow_form}. Retry {i+1}/{MAX_RETRY}")
|
||||
messages.append({"role": "user", "content": f"Error parsing XML to workflow form, the error message is: {workflow_form}\nNote that there are some special restrictions for creating workflow form, please try again."})
|
||||
response = client.run(workflow_former, messages, context_variables, debug=debug)
|
||||
output_xml_form = response.messages[-1]["content"]
|
||||
messages.extend(response.messages)
|
||||
else:
|
||||
raise ValueError(f"Unexpected error: {workflow_form}")
|
||||
return workflow_form, output_xml_form, messages
|
||||
|
||||
def workflow_editing(workflow_creator_agent, client, messages, context_variables, workflow_form, output_xml_form, requirements, task, debug, suggestions = ""):
|
||||
MAX_RETRY = 3
|
||||
if suggestions != "":
|
||||
suggestions = "[IMPORTANT] Here are some suggestions for creating the workflow: " + suggestions
|
||||
agents = workflow_form.agents
|
||||
new_agents = []
|
||||
for agent in agents:
|
||||
if agent.category == "new":
|
||||
new_agents.append(agent)
|
||||
|
||||
if len(new_agents) != 0:
|
||||
new_agent_str = "AGENT CREATION INSTRUCTIONS:\nBefore you create the workflow, you need to create the following new agents in the workflow:\n"
|
||||
for agent in new_agents:
|
||||
new_agent_str += f"Agent name: {agent.name}\nAgent description: {agent.description}\n"
|
||||
new_agent_str += f"Agent tools: {agent.tools}\n" if agent.tools else "Agent tools: []\n"
|
||||
else:
|
||||
new_agent_str = ""
|
||||
|
||||
def case_resolved(task_response: str, context_variables: dict):
|
||||
"""
|
||||
Use this tools when the desired workflow is created and tested successfully. You can NOT use this tool if the workflow is not created or tested successfully by running the workflow.
|
||||
"""
|
||||
return f"Case resolved. The desired workflow is created and tested successfully. : {task_response}"
|
||||
def case_not_resolved(task_response: str, context_variables: dict):
|
||||
"""
|
||||
Use this tools when you encounter irresistible errors after trying your best with multiple attempts for creating the desired workflow. You can NOT use this tool before you have tried your best.
|
||||
"""
|
||||
return f"Case not resolved. The desired workflow is not created or tested successfully. Details: {task_response}"
|
||||
workflow_creator_agent.functions.extend([case_resolved, case_not_resolved])
|
||||
messages.append({"role": "user", "content": f"""\
|
||||
WORKFLOW CREATION INSTRUCTIONS:
|
||||
The user's request to create workflow is: {requirements}
|
||||
Given the completed workflow form with XML format: {output_xml_form}
|
||||
|
||||
TASK:
|
||||
Your task is to create the workflow for me, and then test the workflow by running the workflow using `run_workflow` tool to complete the user's task:
|
||||
{task}
|
||||
|
||||
{new_agent_str}
|
||||
|
||||
TERMINATION INSTRUCTIONS:
|
||||
After you have created the workflow and tested it successfully, you can use the `case_resolved` tool to indicate the case is resolved, otherwise you should try your best to create the workflow. And ONLY after you have tried multiple times, you can use the `case_not_resolved` tool to indicate the case is not resolved and give the reason.
|
||||
|
||||
Remember: you can NOT stop util you have created the workflow and tested it successfully.
|
||||
{suggestions}
|
||||
"""})
|
||||
response = client.run(workflow_creator_agent, messages, context_variables, debug=debug)
|
||||
content = response.messages[-1]["content"]
|
||||
for i in range(MAX_RETRY):
|
||||
if content.startswith("Case resolved"):
|
||||
return content, messages
|
||||
messages.append({"role": "user", "content": f"""\
|
||||
WORKFLOW CREATION INSTRUCTIONS:
|
||||
The user's request to create workflow is: {requirements}
|
||||
Given the completed workflow form with XML format: {output_xml_form}
|
||||
|
||||
TASK:
|
||||
Your task is to create the workflow for me, and then test the workflow by running the workflow using `run_workflow` tool to complete the user's task:
|
||||
{task}
|
||||
|
||||
{new_agent_str}
|
||||
|
||||
TERMINATION INSTRUCTIONS:
|
||||
After you have created the workflow and tested it successfully, you can use the `case_resolved` tool to indicate the case is resolved, otherwise you should try your best to create the workflow. And ONLY after you have tried multiple times, you can use the `case_not_resolved` tool to indicate the case is not resolved and give the reason.
|
||||
|
||||
Remember: you can NOT stop util you have created the workflow and tested it successfully.
|
||||
|
||||
FEEDBACK:
|
||||
The last attempt failed with the following error: {content}, please try again to create the desired workflow.
|
||||
{suggestions}
|
||||
"""})
|
||||
response = client.run(workflow_creator_agent, messages, context_variables, debug=debug)
|
||||
content = response.messages[-1]["content"]
|
||||
if i == MAX_RETRY:
|
||||
return f"The desired workflow is not created or tested successfully with {MAX_RETRY} attempts.", messages
|
||||
|
||||
|
||||
|
||||
|
||||
def meta_workflow(model: str, context_variables: dict, debug: bool = True):
|
||||
print('\033[s\033[?25l', end='') # Save cursor position and hide cursor
|
||||
logger = LoggerManager.get_logger()
|
||||
workflow_former = get_workflow_former_agent(model)
|
||||
workflow_creator_agent = get_workflow_creator_agent(model)
|
||||
|
||||
agent = workflow_former
|
||||
agents = {workflow_former.name.replace(' ', '_'): workflow_former, workflow_creator_agent.name.replace(' ', '_'): workflow_creator_agent}
|
||||
style = Style.from_dict({
|
||||
'bottom-toolbar': 'bg:#333333 #ffffff',
|
||||
})
|
||||
# 创建会话
|
||||
session = PromptSession(
|
||||
completer=UserCompleter(agents.keys()),
|
||||
complete_while_typing=True,
|
||||
style=style
|
||||
)
|
||||
|
||||
client = MetaChain(log_path=logger)
|
||||
console = Console()
|
||||
messages = []
|
||||
|
||||
last_message = "Tell me what do you want to create with `Workflow Chain`?"
|
||||
|
||||
while True:
|
||||
query = session.prompt(
|
||||
f'{last_message} (type "exit" to quit, press "Enter" to continue): ',
|
||||
bottom_toolbar=HTML('<b>Prompt:</b> Enter <b>@</b> to mention Agents'),
|
||||
)
|
||||
if query.strip().lower() == 'exit':
|
||||
|
||||
logo_text = "Workflow Chain completed. See you next time! :waving_hand:"
|
||||
console.print(Panel(logo_text, style="bold salmon1", expand=True))
|
||||
break
|
||||
words = query.split()
|
||||
console.print(f"[bold green]Your request: {query}[/bold green]", end=" ")
|
||||
for word in words:
|
||||
if word.startswith('@') and word[1:] in agents.keys():
|
||||
# print(f"[bold magenta]{word}[bold magenta]", end=' ')
|
||||
agent = agents[word.replace('@', '')]
|
||||
else:
|
||||
# print(word, end=' ')
|
||||
pass
|
||||
print()
|
||||
agent_name = agent.name
|
||||
console.print(f"[bold green][bold magenta]@{agent_name}[/bold magenta] will help you, be patient...[/bold green]")
|
||||
match agent_name:
|
||||
case "Workflow Former Agent":
|
||||
if query == "":
|
||||
console.print(f"[bold red]There MUST be a request to create the agent form.[/bold red]")
|
||||
continue
|
||||
requirements = query
|
||||
workflow_form, output_xml_form, messages = workflow_profiling(workflow_former, client, messages, context_variables, requirements, debug)
|
||||
if workflow_form is None:
|
||||
console.print(f"[bold red][bold magenta]@{agent_name}[/bold magenta] has not created workflow form successfully, please modify your requirements again.[/bold red]")
|
||||
last_message = "Tell me what do you want to create with `Workflow Chain`?"
|
||||
continue
|
||||
agent = workflow_creator_agent
|
||||
context_variables["workflow_form"] = workflow_form
|
||||
console.print(f"[bold green][bold magenta]@{agent_name}[/bold magenta] has created workflow form successfully with the following details:\n[/bold green][bold blue]{output_xml_form}[/bold blue]")
|
||||
last_message = "It is time to create the desired workflow, do you have any suggestions for creating the workflow?"
|
||||
case "Workflow Creator Agent":
|
||||
suggestions = query
|
||||
default_value='Come up with a task for the workflow to test your created workflow, and use `run_workflow` tool to test your created workflow.' # 这里设置你想要的默认值
|
||||
task = session.prompt(
|
||||
'It is time to create the desired workflow, what task do you want to complete with the workflow? (Press Enter if none): ',
|
||||
|
||||
)
|
||||
task = default_value if not task.strip() else task
|
||||
agent_response, messages = workflow_editing(workflow_creator_agent, client, messages, context_variables, workflow_form, output_xml_form, requirements, task, debug, suggestions)
|
||||
if agent_response.startswith("Case not resolved"):
|
||||
console.print(f"[bold red][bold magenta]@{agent_name}[/bold magenta] has not created workflow successfully with the following error: {agent_response}[/bold red]")
|
||||
agent = workflow_creator_agent
|
||||
else:
|
||||
console.print(f"[bold green][bold magenta]@{agent_name}[/bold magenta] has created workflow successfully with the following details:\n[/bold green][bold blue]{agent_response}[/bold blue]")
|
||||
last_message = "Tell me what do you want to create with `Workflow Chain` next?"
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user