INNER CODE UNIT · Python

main

victordibia/designing-multiagent-systems · course/samples/book_generator/autogen_core/app.py:133

async def main(prompt: str) -> None:
    api_key = os.getenv("GEMINI_API_KEY")
    if not api_key:
        raise ValueError("Please set GEMINI_API_KEY environment variable")
    
    runtime = SingleThreadedAgentRuntime()
    
    for agent_class, topic in [
        (StoryGeneratorAgent, "StoryGeneratorAgent"),
        (ImageGeneratorAgent, "ImageGeneratorAgent"),
        (BookGeneratorAgent, "BookGeneratorAgent")
    ]:
        await agent_class.register(
            runtime, type=topic,
            factory=lambda a=agent_class: a(api_key=api_key)
        )
    
    runtime.start()

View source record →