kennethreitz/simplemind
1
0
Fork 0
mirror of https://github.com/kennethreitz/simplemind.git synced 2026-06-05 22:50:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor generate_data.py to use correct conversation setup and formatting

Browse Source
This commit is contained in:
Kenneth Reitz
2024-10-29 12:36:13 -04:00
parent 0fa4b60412
commit dd2b08b4cf
1 changed files with 14 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/generate_data.py
+14 -3
View File
@@ -9,21 +9,32 @@ 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)
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}")
if __name__ == "__main__":
for quote in quotes.quotes:
print(
f"{quote.charecter.name} from {quote.movie.title} ({quote.movie.year}): {quote.quote!r}"
)