tangger/matagent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

调整agent布局

Browse Source
This commit is contained in:
Yutang Li
2025-01-15 14:01:34 +08:00
parent 6fcac50416
commit 8725907ec3
7 changed files with 341 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
_backend/main.py
View File

@@ -27,8 +27,8 @@ model_client = OpenAIChatCompletionClient(
async def main(task: str = "") -> dict:
scientist_team = create_scientist_team()
engineer_team = create_engineer_team()
# await code_executor.start()
robot_platform = create_robot_team()
await code_executor.start()
robot_platform = create_robot_team(code_executor)
result = {}
user = UserProxyAgent("user", input_func=input)
@@ -45,8 +45,9 @@ async def main(task: str = "") -> dict:
1. User: A human agent to whom you transfer information whenever you need to confirm your execution steps to a human.
2. Engineer team: A team of professional engineers who are responsible for writing code, visualizing experimental schemes, converting experimental schemes to JSON, and more.
- The engineer team has the following members:
2.1 Structural engineer: Focus on converting natural language synthesis schemes to formatted synthesis schemes in JSON or XML format.
2.2 Data_Engineer: A professional data engineer will use Python to implement various machine learning algorithms to analyze and visualize data.
2.1 Structural engineer: A professional structural engineer who focus on converting natural language synthesis schemes to JSON or XML formated scheme, and then upload this JSON to S3 Storage.
2.2 Software_Engineer: A professional Python software engineer will use Python to implement tasks.
2.3 SandBox_Env: A computer terminal that performs no other action than running Python scripts (provided to it quoted in ```python code blocks), or sh shell scripts (provided to it quoted in ```sh code blocks).
3. Scientist team: A professional team of material scientists who are mainly responsible for consulting on material synthesis, structure, application and properties.
- The scientist team has the following members:
3.1 Synthesis Scientist: who is good at giving perfect and correct synthesis solutions.
@@ -110,6 +111,7 @@ async def main(task: str = "") -> dict:
selector_func=selector_func,
)
await Console(team.run_stream(task=task))
await code_executor.stop()
# async for message in team.run_stream(task=task):
# if isinstance(message, TextMessage):
# print(f"----------------{message.source}----------------\n {message.content}")