INNER CODE UNIT · Python

BookSections

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

class BookSections(BaseModel):
    sections: List[Section]

class StoryRequest(BaseModel):
    prompt: str

@type_subscription(topic_type="StoryGeneratorAgent")
class StoryGeneratorAgent(RoutedAgent):
    def __init__(self, api_key: str) -> None:
        super().__init__("Story generator agent")
        self.client = genai.Client(api_key=api_key)

    @message_handler
    async def handle_story_request(self, message: StoryRequest, ctx: MessageContext) -> None:
        try:
            print(f">>> StoryGeneratorAgent: {message.prompt}")
            response = self.client.models.generate_content(
                model="gemini-2.0-flash-exp",

View source record →