mirror of
https://github.com/kennethreitz/simplemind.git
synced 2026-06-05 14:50:16 +00:00
Add script to generate movie quotes from famous movies
This commit is contained in:
@@ -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}")
|
||||
Reference in New Issue
Block a user