Example Project

Description

We created an example project that uses Dandy to generate a book using most of the examples in this tutorial.

For more information and the full context of the provided example, please check out the Example Project on GitHub to view and understand the code structure in more detail.

Code Snippets

Main

example/main.py

import logging
import os

from book.intelligence.workflow import BookWorkflow
from dandy.cache import SqliteCache
from dandy.core.exceptions import DandyException
from dandy.recorder import Recorder

if __name__ == '__main__':
    try:
        # if os.getenv('DEBUG'):
        #     SqliteCache.clear('example')

        logging.basicConfig(
            level=logging.INFO,
            format='%(asctime)s - %(levelname)s - %(message)s',
            datefmt='%Y-%m-%d %H:%M:%S',
        )

        Recorder.start_recording('book_generation')

        logging.info('Creating a Book')

        dandy_book = BookWorkflow.process(
            user_input="""It's 2035 and the dandy intelligence python library has changed the world for ever. In an effort
            to build awareness the people behind the library have created a working version of their lovable mascot the 
            dandy robot. He has now been turned on and given the directive to let the world know about the dandy library
            and given full agency to do it ... what could go wrong!""",
        )

        dandy_book.to_markdown_file()

        logging.info('Complete')

    except DandyException:
        raise DandyException

    finally:
        Recorder.stop_recording('book_generation')
        Recorder.to_html_file('book_generation')

Intel

example/book/intelligence/intel.py

from typing import Union

from dandy.intel import BaseIntel
from example.book.enums import BookTheme
from example.book.intelligence.chapter.intel import ChaptersIntel

from example.book.intelligence.character.intel import CharactersIntel
from example.book.intelligence.plot.intel import PlotIntel
from example.book.intelligence.world.intel import WorldIntel


class BookStartIntel(BaseIntel):
    title: str
    overview: str


class BookIntel(BaseIntel):
    user_input: str
    theme: BookTheme
    start: Union[BookStartIntel, None] = None
    characters: Union[CharactersIntel, None] = None
    plot: Union[PlotIntel, None] = None
    world: Union[WorldIntel, None] = None
    chapters: Union[ChaptersIntel, None] = None

Prompt

example/book/intelligence/prompts.py

from __future__ import annotations

from typing_extensions import TYPE_CHECKING

from dandy.llm import Prompt
from example.book.intelligence.character.prompts import characters_intel_prompt
from example.book.intelligence.plot.prompts import plot_intel_prompt

if TYPE_CHECKING:
    from example.book.intelligence.intel import BookIntel


def book_intel_prompt(book_intel: BookIntel) -> Prompt:
    prompt = Prompt()

    if book_intel.start:
        prompt.title(f'Title: {book_intel.start.title}')
        prompt.heading(f'Overview: {book_intel.start.overview}')
        prompt.line_break()

    if book_intel.world:

        prompt.heading('World:')
        prompt.sub_heading(f'Name: {book_intel.world.name}')
        prompt.text(label='Description', text=book_intel.world.description)

        prompt.line_break()

        prompt.heading('Locations:')
        for location_intel in book_intel.world.locations:
            prompt.line_break()

            prompt.sub_heading(f'Name: {location_intel.name}')
            prompt.text(label='Description', text=location_intel.description)

        prompt.line_break()

    if book_intel.characters:
        prompt.prompt(characters_intel_prompt(book_intel.characters))

        prompt.line_break()

    if book_intel.plot:
        prompt.prompt(plot_intel_prompt(book_intel.plot))

    return prompt

LLMBot

example/book/intelligence/bots/book_start_llm_bot.py

from dandy.cache import cache_to_sqlite
from dandy.llm import BaseLlmBot, Prompt
from example.book.intelligence.intel import BookStartIntel


class BookStartLlmBot(BaseLlmBot):
    instructions_prompt = (
        Prompt()
        .text('You are a book starting bot. You will be given an idea by the user.')
        .text('you will generate a book title and overview.')
    )

    @classmethod
    @cache_to_sqlite('example')
    def process(
            cls,
            user_input: str,
    ) -> BookStartIntel:

        return cls.process_prompt_to_intel(
            prompt=Prompt(user_input),
            intel_class=BookStartIntel
        )

Workflow

example/book/intelligence/workflow.py

from __future__ import annotations

import logging

from typing_extensions import TYPE_CHECKING

from dandy.workflow import BaseWorkflow
from example.book.intelligence.bots import BookStartLlmBot
from example.book.intelligence.chapter.workflow import ChaptersWorkflow
from example.book.intelligence.character.workflow import CharactersWorkflow
from example.book.intelligence.intel import BookIntel
from example.book.intelligence.maps import BookThemeLlmMap
from example.book.intelligence.plot.workflow import PlotWorkflow
from example.book.intelligence.world.bot import WorldLlmBot
from example.book.models import Book, Chapter

if TYPE_CHECKING:
    pass


class BookWorkflow(BaseWorkflow):
    @classmethod
    def process(
            cls,
            user_input: str,
    ) -> Book:
        book_intel = BookIntel(
            user_input=user_input,
            theme=BookThemeLlmMap.process(user_input)[0]
        )

        logging.info('Working on book title and overview')
        book_intel.start = BookStartLlmBot.process(user_input)

        logging.info('Creating "those" characters')
        book_intel.characters = CharactersWorkflow.process(book_intel)

        logging.info('Forging a world')
        book_intel.world = WorldLlmBot.process(book_intel)

        logging.info('Setting up the plot')
        book_intel.plot = PlotWorkflow.process(book_intel)

        logging.info('Writing the chapters for all to read')
        book_intel.chapters = ChaptersWorkflow.process(
            book_intel=book_intel,
            chapter_count=4
        )

        new_book = Book(
            title=book_intel.start.title,
            author='Dandy McAuthor',
            theme=book_intel.theme,
            overview=book_intel.start.overview,
            chapters=[
                Chapter(title=chapter_intel.title, content=chapter_intel.content)
                for chapter_intel in book_intel.chapters
            ]
        )

        print(new_book.model_dump_json(indent=4))

        print(book_intel.model_dump_json(indent=4))

        return new_book