File size: 1,350 Bytes
3d92de3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
from autogen_ext.runtimes.grpc import GrpcWorkerAgentRuntimeHost
from agent import Agent
from creator import Creator
from autogen_ext.runtimes.grpc import GrpcWorkerAgentRuntime
from autogen_core import AgentId
import messages
import asyncio
#Change how many agents you want to create dynamically
HOW_MANY_AGENTS = 2
async def create_and_message(worker, creator_id, i: int):
try:
result = await worker.send_message(messages.Message(content=f"agent{i}.py"), creator_id)
with open(f"output/idea{i}.md", "w") as f:
f.write(result.content)
except Exception as e:
print(f"Failed to run worker {i} due to exception: {e}")
async def main():
host = GrpcWorkerAgentRuntimeHost(address="localhost:50051")
host.start()
worker = GrpcWorkerAgentRuntime(host_address="localhost:50051")
await worker.start()
result = await Creator.register(worker, "Creator", lambda: Creator("Creator"))
creator_id = AgentId("Creator", "default")
coroutines = [create_and_message(worker, creator_id, i) for i in range(1, HOW_MANY_AGENTS+1)]
await asyncio.gather(*coroutines)
try:
await worker.stop()
await host.stop()
except Exception as e:
print(e)
if __name__ == "__main__":
asyncio.run(main())
print("All agents have been created and ran successfully.")
|