diff --git a/examples/sentiment_analysis.py b/examples/sentiment_analysis.py new file mode 100644 index 0000000..94ae229 --- /dev/null +++ b/examples/sentiment_analysis.py @@ -0,0 +1,19 @@ +from _context import sm + +from pydantic import BaseModel +from typing import Literal + + +class SentimentAnalysis(BaseModel): + sentiment: Literal["positive", "negative", "neutral"] + confidence: float + + +print( + sm.generate_data( + prompt="Analyze the sentiment of the following text:\n\n'The product arrived late and was broken. Worst purchase ever!'", + llm_provider="openai", + llm_model="gpt-4o", + response_model=SentimentAnalysis, + ) +)