j ddlmZmZddlmZmZddlmZ ddlZddl m Z GddejZ y#e $rdZdZ Y!wxYw))AnyOptional) base_metric score_result)MetricComputationErrorN)vadercfeZdZdZ d dededeeffd Zdedede jfd Z xZ S) Sentimenta A metric that analyzes the sentiment of text using NLTK's VADER sentiment analyzer. Returns sentiment scores for positive, neutral, negative, and compound sentiment. The compound score is a normalized score between -1.0 (extremely negative) and 1.0 (extremely positive). Args: name: The name of the metric. Defaults to "sentiment_metric". track: Whether to track the metric. Defaults to True. project_name: Optional project name to track the metric in for the cases when there are no parent span/trace to inherit project name from. Example: >>> from opik.evaluation.metrics import Sentiment >>> sentiment_metric = Sentiment() >>> result = sentiment_metric.score("I love this product! It's amazing.") >>> print(result.value) # Compound score (e.g., 0.8802) >>> print(result.metadata) # All sentiment scores nametrack project_namect||||tt t d tj |_y#t$r1tjdtj |_YywxYw)N)r r r z`nltk` library is required for sentiment analysis. Install via `pip install nltk` and then download the vader_lexicon: `python -m nltk.downloader vader_lexicon`. vader_lexicon) super__init__nltkr ImportErrorSentimentIntensityAnalyzer _analyzer LookupErrordownload)selfr r r __class__s /Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/evaluation/metrics/heuristics/sentiment.pyrzSentiment.__init__$s %  <5==   @"==?DN @ MM/ *"==?DN @sA7B?Boutputignored_kwargsreturnc |js td|jj|}|d}|dk\rd}n |dkrd}nd}t j ||j d|d |d d | S) a Analyze the sentiment of the provided text. Args: output: The text to analyze for sentiment. **ignored_kwargs: Additional keyword arguments that are ignored. Returns: score_result.ScoreResult: A ScoreResult object with: - value: The compound sentiment score (-1.0 to 1.0) - name: The metric name - reason: A brief explanation of the sentiment analysis - metadata: Dictionary containing all sentiment scores (pos, neu, neg, compound) Raises: MetricComputationError: If the input text is empty. z+Empty text provided for sentiment analysis.compoundg?positivegnegativeneutralzText sentiment analysis: z (compound score: z.4f))valuer reasonmetadata)striprrpolarity_scoresr ScoreResultr )rrrsentiment_scorescompound_scoresentiment_categorys rscorezSentiment.score>s$||~()VW W>>99&A)*5 T !!+  u $!+ !* '' ./A.BBTUcdgThhij%   )sentiment_metricTN) __name__ __module__ __qualname____doc__strboolrrrrr)r- __classcell__)rs@rr r s\.'&* @@@sm @4$ C$ 3$ <;S;S$ r.r )typingrropik.evaluation.metricsrropik.exceptionsrrnltk.sentimentrr BaseMetricr r.rr=sH =2$ T &&T  D Es 8 AA