This commit is contained in:
2025-02-07 10:21:16 +08:00
parent 2c9b1bba3a
commit 0ef44b002e
18 changed files with 29943 additions and 111 deletions

View File

@@ -7,7 +7,7 @@ from autogen_agentchat.teams import SelectorGroupChat, RoundRobinGroupChat
from autogen_agentchat.ui import Console
from autogen_agentchat.base import Handoff
from autogen_ext.models.openai import OpenAIChatCompletionClient
from constant import MODEL, OPENAI_API_KEY, OPENAI_BASE_URL, code_executor
from constant import MODEL, OPENAI_API_KEY, OPENAI_BASE_URL
from scientist_team import create_scientist_team
from engineer_team import create_engineer_team
from robot_platform import create_robot_team
@@ -121,7 +121,7 @@ async def main(task: str = "") -> dict:
selector_func=selector_func,
)
await Console(team.run_stream(task=task))
await code_executor.stop()
# await code_executor.stop()
# async for message in team.run_stream(task=task):
# if isinstance(message, TextMessage):
# print(f"----------------{message.source}----------------\n {message.content}")