o 2&i\+@sbddlZddlZddlmZmZmZmZddlmZddl m Z m Z m Z m Z mZGdddZdS)N)DictListAnyOptional)Groq)SCENARIO_GENERATION_PROMPTCHARACTER_ROLEPLAY_PROMPTCHARACTER_CONCLUSION_PROMPTSKILL_ANALYSIS_PROMPTget_skill_analysis_templatec @seZdZddZdefddZdeeeffddZd d Z d eeefde eeeffd d Z ddeeefde eeefdede de ef ddZdeeefde eeefde eeeffddZdede efddZdS) GroqServicecCsRtd}|s tdt|d|_d|_ddddddddddddd|_dS)N GROQ_API_KEYz/GROQ_API_KEY not found in environment variables)api_keyzopenai/gpt-oss-20brinputoutputtotal)preview conversation assessment)osgetenv ValueErrorrclientmodel token_counts)selfrrD/var/www/eduai.edurigo.com/roleplay/testing/services/groq_service.py__init__s      zGroqService.__init__operation_typec Cs|rHt|drJt|drL|j}|j}|j}|j|d|7<|j|d|7<|j|d|7<td|d|d|d |d Sd Sd Sd S) z*Update token counts for specific operation prompt_tokenscompletion_tokensrrrzGROQ Token count for z: Input=z , Output=z, Total=N)hasattrr!r" total_tokensrprint)rr usage_stats input_tokens output_tokensr$rrr_update_token_counts$ zGroqService._update_token_countreturncCs |jS)z+Get current token counts for all operations)rcopy)rrrrget_token_counts*s zGroqService.get_token_countscCs"|jD] }dddd|j|<qdS)zReset all token countersrrN)r)r operationrrrreset_token_counts.s zGroqService.reset_token_counts admin_inputc Cs*zztj|d|d|dd|dd}|jjjjddd d |d g|jd d d }|d|j |j dj j }z|drF|dd}|drQ|dd}t|WWStjyz}ztd|td|WYd}~WdSd}~wwty}ztd|WYd}~dSd}~ww)z-Generate structured scenario from admin inputcategory objectivedetails, skills_to_assess)r0r1r2skillssystemzHYou are a training scenario expert. Always respond with valid JSON only.rolecontentuserffffff?imessagesr temperature max_tokensrr```jsonN```zJSON parsing error: Response text: zError generating scenario: )rformatjoinrchat completionscreaterr)usagechoicesmessager9strip startswithendswithjsonloadsJSONDecodeErrorr% Exception)rr/promptresponse response_texterrrgenerate_scenario3sL       zGroqService.generate_scenarioFscenarioconversation_history user_message is_conclusionc Cs>zd}|D]}|ddkrdn|dd}||d|dd 7}q|r't}nt}|j|dd|dd |dd |dd |dd d|dd|dd|dd|dd||d } |jjjjdddd| dg|jddd} | d| j | j dj j WSty} ztd| WYd} ~ d Sd} ~ ww)!z&AI character responses during roleplayspeakerlearnerLearner ai_charactername: rL personalitygoals backgroundemotional_stateneutralscenario_setupcontext environmentr1 constraints) character_namererfrgrhrkrlr1rmrZr[r6zYou are a roleplay character. Stay in character at all times. Respond naturally, realistically, and politely. Maintain a courteous and professional tone throughout the conversation. Guide conversations toward professional conclusions when appropriate.r7r:g?ir<rrzError in character roleplay: Nz:I'm having trouble responding right now. Please try again.)r rrEgetrrGrHrIrr)rJrKrLr9rMrSr%) rrYrZr[r\ history_textturnr^prompt_templaterTrUrWrrrplay_characterdsP          zGroqService.play_characterconversation_turnsc sjzd}|D]}|ddkrdnd}||d|dd7}qd |d d |d d |ddd|dd|ddd|ddd }|d}t|}tjd||||d} |jjjjdddd| dg|jddd } | d!| j | j d"j j } zB| d#r| d$d%} n | d&r| d'd%} | d&r| d%d(} | } t| tfd)d*d+Dstd,WWd%SWWStjy} zJtd-| td.| d"d%l} | d/| | j}|r zt|td0WWYd%} ~ WStjy td1YnwWYd%} ~ Wd%Sd%} ~ wwty4} ztd2| WYd%} ~ d%Sd%} ~ ww)3zComprehensive skill analysisr]r^r_r`z AI CharacterrcrLrdz Category: r0z Objective: r1z Context: rjrkz Success Criteria: success_criteriaz AI Character: rarbz - rgr4r3)r5scenario_contextrskill_analysis_templater6z`You are an expert skill assessor. Provide detailed, accurate analysis in valid JSON format only.r7r:g333333?i r<rrr@rANrBrCc3s|]}|vVqdS)Nr).0keyanalysisrr sz-GroqService.analyze_skills..)skill_analysisoverall_performanceconversation_analysisrecommendationszAnalysis missing required keysz JSON parsing error in analysis: rDz\{.*\}z0Successfully extracted JSON using regex fallbackzRegex fallback also failedzError analyzing skills: )r r rErFrrGrHrIrr)rJrKrLr9rMrNrOrPrQallr%rRresearchDOTALLgrouprS)rrYrtconversation_textrqr^rvr5skill_templaterTrUrVrWr json_matchrr{ranalyze_skillss            zGroqService.analyze_skillsrTc Csz&|jjjjdddd|dg|jddd}|d|j|jd jj WSt y@}zt d |WYd }~d Sd }~ww) z)Get a simple text completion from the LLMr6z]You are a helpful assistant. Provide clear, natural responses without any formatting symbols.r7r:r;ir<rrzError getting completion: N) rrGrHrIrr)rJrKrLr9rMrSr%)rrTrUrWrrrget_completions(  zGroqService.get_completionN)F)__name__ __module__ __qualname__rstrr)rrr,r.rrXrboolrsrrrrrrr s &8165Zr )rrPtypingrrrrgroqrconfig.promptsrrr r r r rrrrs