Add script to generate movie quotes from famous movies

This commit is contained in:
2024-10-29 06:46:55 -04:00
parent f9b1472cf4
commit 03204aa9a2
+29
View File
@@ -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}")