U Jje @sddlZddlZddlZddlmZddlmZddlZeeeddZ dddgZ d Z d Z d d Z dddZdddZddZddZddZddZddZddZdS) N)Groq) load_dotenvZ GROQ_API_KEY)Zapi_keyzllama-3.3-70b-versatilezllama-3.1-8b-instantz gemma2-9b-itZ social_v1Z internal_v1c Cshz t|WStk rbz,td|tj}|rFt|WYSWntk r\YnXYnXgS)z6 Strict JSON parser with fallback extraction. z\[.*\])jsonloads ExceptionresearchDOTALLgroup)textmatchr J/var/www/api.dyntland.com/public_html/python_social_comments/llm_client.pysafe_json_parse's  r皙?c Csd}tD]}zRtd|tjjj|d|dg|d}|jdjj }t |}||fWSt k r}zHt | }d|krtd|ntd|d ||}WYqW5d}~XYqXq|dS) Nz[LLM] Trying model: user)rolecontent)modelmessages temperaturerz rate limitz[RATE LIMIT] model=z[LLM ERROR] model=z error=)MODEL_VERSIONSprintclientZchatZ completionscreatechoicesmessagerstriprrstrlower) promptrZ last_errorrresponserparsedeZ error_textr r rexecute_with_fallback:s,  r$c Cshz&t||\}}td|||fWStk rb}ztd|gdfWYSd}~XYnXdS)Nz[LLM SUCCESS] model=z [LLM ERROR] )r$rr)r rr" used_modelr#r r rcall_llmvs r&c Cs>dddt|D}d|d|d|d|d }t|S) N cSs"g|]\}}|dd|qSz. r .0icr r r sz1analyze_social_comments_batch..zc You are a strict JSON generator. TASK: Classify comments in a debate context. DEBATE: Question: z Option A: z Option B: COMMENTS: u* RULES: - Return ONLY valid JSON array - No explanation - No markdown - No extra text - Output MUST match schema exactly - For "side": always output the exact text of Option A or Option B — NEVER output "Neutral" or "A" or "B" - If a comment seems neutral, pick the option it leans toward most OUTPUT FORMAT: [ { "index": 1, "side": "exact text of Option A or Option B", "confidence": 0.0-1.0, "sentiment": "Positive|Negative|Neutral", "emotion": "anger|joy|sadness|fear|surprise|disgust|neutral|frustration|hope|sarcasm" } ] join enumerater&)comments_batchquestionoption_aoption_bformatted_commentsr r r ranalyze_social_comments_batchs  #r8cCs,dddt|D}d|d}t|S)Nr'cSs"g|]\}}|dd|qSr(r r*r r rr.sz2analyze_comments_batch_generic..z You are a strict JSON generator. TASK: Analyze generic social comments. RULES: - Return ONLY valid JSON array - No explanation - No markdown - No extra text - Output MUST match schema exactly COMMENTS: a  OUTPUT FORMAT: [ { "index": 1, "confidence": 0.0-1.0, "sentiment": "Positive|Negative|Neutral", "emotion": "anger|joy|sadness|fear|surprise|disgust|neutral|frustration|hope|sarcasm", "toxicity_score": 0.0-1.0, "language_code": "en" } ] r0r3r7r r r ranalyze_comments_batch_genericsr:cCsLdddt|D}ddd|D}d|d|d|d}t|S) Nr'cSs"g|]\}}|dd|qSr(r r*r r rr.sz/analyze_poll_comments_batch..cSsg|]}d|qS)z- r )r+optr r rr.szf You are a strict JSON generator. TASK: Classify comments based on a poll question. POLL: Question: z Options: r/u RULES: - Return ONLY valid JSON array - No explanation - No markdown - No extra text - Output MUST match schema exactly - For "side": always output the exact text of one of the options above — pick the closest match OUTPUT FORMAT: [ { "index": 1, "side": "exact text of one poll option", "confidence": 0.0-1.0, "sentiment": "Positive|Negative|Neutral", "emotion": "anger|joy|sadness|fear|surprise|disgust|neutral|frustration|hope|sarcasm" } ] r0)r3r4optionsr7Zformatted_optionsr r r ranalyze_poll_comments_batchs  "r=cCs,dddt|D}d|d}t|S)Nr'cSs"g|]\}}|dd|qSr(r r*r r rr.sz3analyze_internal_comments_batch..zT You are a strict JSON generator. TASK: Analyze internal user comments. COMMENTS: a RULES: - Return ONLY valid JSON array - No explanation - No markdown - No extra text - Output MUST match schema exactly FORMAT: [ { "index": 1, "confidence": 0.0-1.0, "sentiment": "Positive|Negative|Neutral", "emotion": "anger|joy|sadness|fear|surprise|disgust|neutral|frustration|hope|sarcasm", "toxicity": 0.0-1.0, "language_code": "en", "impact_score": 0.0-1.0 } ] r0r9r r ranalyze_internal_comments_batchsr>cCstSN)PROMPT_VERSION_SOCIALr r r rget_social_prompt_version0srAcCstSr?)PROMPT_VERSION_INTERNALr r r rget_internal_prompt_version3srC)r)r)osrrZgroqrdotenvrsysgetenvrrr@rBrr$r&r8r:r=r>rArCr r r rs,   < ,'-+