o 2h*@s^ddlZddlZddlmZmZmZmZddlmZddl m Z m Z m Z m Z GdddZdS)N)DictListAnyOptional)Groq)SCENARIO_GENERATION_PROMPTCHARACTER_ROLEPLAY_PROMPTSKILL_ANALYSIS_PROMPTget_skill_analysis_templatec @seZdZddZdefddZdeeeffddZd d Z d eeefde eeeffd d Z deeefde eeefdede efddZ deeefde 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)selfr rY/var/www/eduai.edurigo.com/doc_train/edurigo_ai/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)r/r0r1skillssystemzHYou 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(usagechoicesmessager8strip startswithendswithjsonloadsJSONDecodeErrorr$ Exception)rr.promptresponse response_texterrrgenerate_scenario2sL       zGroqService.generate_scenarioscenarioconversation_history user_messagec Cs&zxd}|D]}|ddkrdn|dd}||d|dd 7}qtj|dd|dd |dd |dd |dd d|dd|dd|dd||d }|jjjjdddd|dg|jddd}|d|j |j dj j WSty} ztd| WYd} ~ dSd} ~ ww) z&AI character responses during roleplayspeakerlearnerLearner ai_charactername: rK personalitygoals backgroundemotional_stateneutralscenario_setupcontext environment constraints) character_namercrdrerfrirjrkrYrZr5zbYou are a roleplay character. Stay in character at all times. Respond naturally and realistically.r6r9g?ir;rrzError in character roleplay: Nz:I'm having trouble responding right now. Please try again.)rrDgetrrFrGrHrr(rIrJrKr8rLrRr$) rrXrYrZ history_textturnr\rSrTrVrrrplay_charactercsH         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 CharacterrarKrbz Category: r/z Objective: r0z Context: rhriz Success Criteria: success_criteriaz AI Character: r_r`z - rer3r2)r4scenario_contextrskill_analysis_templater5z`You are an expert skill assessor. Provide detailed, accurate analysis in valid JSON format only.r6r9g333333?i r;rrr?r@NrArBc3s|]}|vVqdS)Nr).0keyanalysisrr sz-GroqService.analyze_skills..)skill_analysisoverall_performanceconversation_analysisrecommendationszAnalysis missing required keysz JSON parsing error in analysis: rCz\{.*\}z0Successfully extracted JSON using regex fallbackzRegex fallback also failedzError analyzing skills: )r r rDrErrFrGrHrr(rIrJrKr8rLrMrNrOrPallr$rQresearchDOTALLgrouprR)rrXrqconversation_textror\rsr4skill_templaterSrTrUrVr json_matchrrxranalyze_skillss            zGroqService.analyze_skillsrSc Csz&|jjjjdddd|dg|jddd}|d|j|jd jj WSt y@}zt d |WYd }~d Sd }~ww) z)Get a simple text completion from the LLMr5z]You are a helpful assistant. Provide clear, natural responses without any formatting symbols.r6r9r:ir;rrzError getting completion: N) rrFrGrHrr(rIrJrKr8rLrRr$)rrSrTrVrrrget_completions(  zGroqService.get_completionN)__name__ __module__ __qualname__rstrr(rrr+r-rrWrrprrrrrrr s &216/Zr )rrOtypingrrrrgroqrconfig.promptsrrr r r rrrrs