ju dZddlmZddlmZmZddlmZddlm Z ddl m Z  ddl m Z Gd d eZy#e$rdZ YwxYw) zVADER sentiment metric wrapper.) annotations)AnyOptional) BaseMetric) ScoreResult)MetricComputationError)SentimentIntensityAnalyzerNcHeZdZdZ d dfd ZddZxZS)VADERSentimenta Compute the VADER compound sentiment for a piece of text. References: - Hutto & Gilbert, "VADER: A Parsimonious Rule-based Model for Sentiment Analysis of Social Media Text" (ICWSM 2014) https://ojs.aaai.org/index.php/ICWSM/article/view/14550 - VADER Sentiment GitHub repository (official implementation) https://github.com/cjhutto/vaderSentiment Args: name: Display name for the metric result. Defaults to ``"vader_sentiment_metric"``. track: Whether to automatically track metric results. Defaults to ``True``. project_name: Optional tracking project name. Defaults to ``None``. analyzer: Optional pre-initialised ``SentimentIntensityAnalyzer`` or compatible callable. Example: >>> from opik.evaluation.metrics import VADERSentiment >>> metric = VADERSentiment() >>> result = metric.score("I absolutely love this experience!") # doctest: +SKIP >>> round(result.value, 2) # doctest: +SKIP 0.94 c~t|||||||_yt t dt|_y)N)nametrack project_namezyVADER sentiment metric requires the optional 'nltk' package. Install via `pip install nltk` or provide a custom analyzer.)super__init__ _analyzerr ImportError)selfr rranalyzer __class__s /Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/evaluation/metrics/heuristics/vader_sentiment.pyrzVADERSentiment.__init__,sM d%lK  %DN+2!H89DNc |r|js td|jj|}t |j dd}|dzdz }t ||jd|dd|i S) NzText is empty (VADERSentiment).compoundgg?g@z#VADER compound score (normalized): z.4fvader)valuer reasonmetadata)striprrpolarity_scoresfloatgetrr )routputignored_kwargsscoresr normalizeds rscorezVADERSentiment.scoreAs|V\\^()JK K//7J45n+ 8C8HIv&   r)vader_sentiment_metricTNN) r strrboolrz Optional[str]rz Optional[Any]returnNone)r#r)r$rr+r)__name__ __module__ __qualname____doc__rr' __classcell__)rs@rr r sP8-&*"& :::$ :  :  :*  rr )r0 __future__rtypingrr#opik.evaluation.metrics.base_metricr$opik.evaluation.metrics.score_resultropik.exceptionsrnltk.sentimentr rr rrr9sB%" :<2&9 < Z< &!%&s6AA