mirror of
https://github.com/kennethreitz/simplemind.git
synced 2026-06-05 14:50:16 +00:00
Siddhesh Agarwal
44 lines
882 B
Python
44 lines
882 B
Python
from typing import Iterator, List
|
|
|
|
from _context import sm
|
|
from pydantic import BaseModel
|
|
|
|
|
|
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]
|
|
|
|
|
|
def gen_quotes(n: int = 10) -> Iterator[MovieQuote]:
|
|
"""Generate a list of quotes from famous movies."""
|
|
|
|
for q in sm.generate_data(
|
|
llm_provider="openai",
|
|
llm_model="gpt-4o-mini",
|
|
prompt=f"Generate {n} quotes from famous movies",
|
|
response_model=QuotesList,
|
|
).quotes:
|
|
yield q
|
|
|
|
|
|
if __name__ == "__main__":
|
|
for quote in gen_quotes(n=20):
|
|
print(
|
|
f"{quote.charecter.name} from {quote.movie.title} ({quote.movie.year}): {quote.quote!r}"
|
|
)
|