Migrate team one to new subscriptions (#560)

Co-authored-by: afourney <adam.fourney@gmail.com>
This commit is contained in:
Jack Gerrits
2024-09-19 14:10:41 -04:00
committed by GitHub
parent 8018677234
commit 19f83debb1
15 changed files with 61 additions and 56 deletions

View File

@@ -10,7 +10,6 @@ import logging
from autogen_core.application import SingleThreadedAgentRuntime
from autogen_core.application.logging import EVENT_LOGGER_NAME
from autogen_core.base import AgentId, AgentProxy
from autogen_core.components import DefaultSubscription
from autogen_core.components.code_executor._impl.docker_command_line_code_executor import DockerCommandLineCodeExecutor
from team_one.agents.coder import Coder, Executor
from team_one.agents.orchestrator import RoundRobinOrchestrator
@@ -25,29 +24,27 @@ async def main() -> None:
async with DockerCommandLineCodeExecutor() as code_executor:
# Register agents.
await runtime.register(
"Coder", lambda: Coder(model_client=create_completion_client_from_env()), lambda: [DefaultSubscription()]
)
await Coder.register(runtime, "Coder", lambda: Coder(model_client=create_completion_client_from_env()))
coder = AgentProxy(AgentId("Coder", "default"), runtime)
await runtime.register(
await Executor.register(
runtime,
"Executor",
lambda: Executor("A agent for executing code", executor=code_executor),
lambda: [DefaultSubscription()],
)
executor = AgentProxy(AgentId("Executor", "default"), runtime)
await runtime.register(
await UserProxy.register(
runtime,
"UserProxy",
lambda: UserProxy(description="The current user interacting with you."),
lambda: [DefaultSubscription()],
)
user_proxy = AgentProxy(AgentId("UserProxy", "default"), runtime)
await runtime.register(
await RoundRobinOrchestrator.register(
runtime,
"orchestrator",
lambda: RoundRobinOrchestrator([coder, executor, user_proxy]),
lambda: [DefaultSubscription()],
)
runtime.start()