INNER CODE UNIT · Python

BookGeneratorAgent

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

class BookGeneratorAgent(RoutedAgent):
    def __init__(self, api_key: str) -> None:
        super().__init__("Book generator agent")
    
    @message_handler
    async def handle_book_content(self, message: BookSections, ctx: MessageContext) -> None:
        try:
            print(">>> BookGeneratorAgent: Generating story book pdf")
            pdf_path = OUTPUT_DIR / "story_book.pdf"
            generate_pdf_report(
                sections=[s.model_dump() for s in message.sections],
                output_file=str(pdf_path),
                report_title=message.sections[0].title
            )
            print(f"Generated story book at: {pdf_path}")
        except Exception as e:
            print(f"Error in book generation: {str(e)}")

View source record →