diff --git a/examples/generate_data.py b/examples/generate_data.py new file mode 100644 index 0000000..fb15882 --- /dev/null +++ b/examples/generate_data.py @@ -0,0 +1,29 @@ +from typing import List + +from pydantic import BaseModel + +from _context import sm + + +class Movie(BaseModel): + title: str + year: int + +class MovieCharecter(BaseModel): + name: str + actor: str + +class MovieQuote(BaseModel): + quote: str + movie: Movie + charecter: MovieCharecter + +class QuotesList(BaseModel): + quotes: List[MovieQuote] + theme: str + + +quotes = sm.generate_data(llm_provider="openai", llm_model="gpt-4o-mini", prompt="Generate 20 quotes from famous movies", response_model=QuotesList) + +for quote in quotes.quotes: + print(f"{quote.charecter.name} from {quote.movie.title} ({quote.movie.year}): {quote.quote!r}")