Initial impl of new register and subscriptions (#542)

* Initial impl of new register and subscriptions

* progress

* test fixes, main issue was unbound self in routed agent
This commit is contained in:
Jack Gerrits
2024-09-18 14:41:04 -04:00
committed by GitHub
parent a479a5e692
commit 093e261158
18 changed files with 423 additions and 156 deletions

View File

@@ -121,16 +121,16 @@
"import asyncio\n",
"\n",
"from autogen_core.application import WorkerAgentRuntime\n",
"from autogen_core.base import MESSAGE_TYPE_REGISTRY, AgentId, try_get_known_serializers_for_type\n",
"from autogen_core.base import AgentId, try_get_known_serializers_for_type\n",
"from autogen_core.components import DefaultSubscription\n",
"\n",
"MESSAGE_TYPE_REGISTRY.add_serializer(try_get_known_serializers_for_type(MyMessage))\n",
"\n",
"worker1 = WorkerAgentRuntime(host_address=\"localhost:50051\")\n",
"worker1.add_message_serializer(try_get_known_serializers_for_type(MyMessage))\n",
"worker1.start()\n",
"await worker1.register(\"worker1\", lambda: MyAgent(\"worker1\"), lambda: [DefaultSubscription()])\n",
"\n",
"worker2 = WorkerAgentRuntime(host_address=\"localhost:50051\")\n",
"worker2.add_message_serializer(try_get_known_serializers_for_type(MyMessage))\n",
"worker2.start()\n",
"await worker2.register(\"worker2\", lambda: MyAgent(\"worker2\"), lambda: [DefaultSubscription()])\n",
"\n",