o sh!@s^ddlZddlZddlmZmZmZmZddlmZddl m Z m Z m Z m Z GdddZdS)N)DictListAnyOptional)Groq)SCENARIO_GENERATION_PROMPTCHARACTER_ROLEPLAY_PROMPTSKILL_ANALYSIS_PROMPTget_skill_analysis_templatec @seZdZddZdeeefdeeeeffddZdeeefde eeefd edeefd d Z deeefd e eeefdeeeeffd dZ dedeefddZ dS) GroqServicecCs,td}|s tdt|d|_d|_dS)N GROQ_API_KEYz/GROQ_API_KEY not found in environment variables)api_keyzllama3-70b-8192)osgetenv ValueErrorrclientmodel)selfr rY/var/www/eduai.edurigo.com/doc_train/edurigo_ai/roleplay/testing/services/groq_service.py__init__ s   zGroqService.__init__ admin_inputreturnc Cszstj|d|d|dd|dd}|jjjjddd d |d g|jd d d }|jdj j }z| dr?|dd}| drJ|dd}t|WWStjys}ztd|td|WYd}~WdSd}~wwty}ztd|WYd}~dSd}~ww)z-Generate structured scenario from admin inputcategory objectivedetails, skills_to_assess)rrrskillssystemzHYou are a training scenario expert. Always respond with valid JSON only.rolecontentuserffffff?imessagesr temperature max_tokensr```jsonN```zJSON parsing error: Response text: zError generating scenario: )rformatjoinrchat completionscreaterchoicesmessager"strip startswithendswithjsonloadsJSONDecodeErrorprint Exception)rrpromptresponse response_texterrrgenerate_scenariosJ       zGroqService.generate_scenarioscenarioconversation_history user_messagec Cszqd}|D]}|ddkrdn|dd}||d|dd 7}qtj|dd|dd |dd |dd |dd d|dd|dd|dd||d }|jjjjdddd|dg|jddd}|jdj j WSt y} zt d| WYd} ~ dSd} ~ ww)z&AI character responses during roleplayspeakerlearnerLearner ai_charactername: r4 personalitygoals backgroundemotional_stateneutralscenario_setupcontext environment constraints) character_namerMrNrOrPrSrTrUrCrDrzbYou are a roleplay character. Stay in character at all times. Respond naturally and realistically.r r#g?ir%rzError in character roleplay: Nz:I'm having trouble responding right now. Please try again.)rr.getrr0r1r2rr3r4r"r5r<r;) rrBrCrD history_textturnrFr=r>r@rrrplay_characterCsF         zGroqService.play_characterconversation_turnsc szd}|D]}|ddkrdnd}||d|dd7}qd |d d |d d |ddd|dd|ddd|ddd }|d}t|}tjd||||d} |jjjjdddd| dg|jddd } | j d!j j } z2| d"r| d#d$} | d%r| d$d&} t| tfd'd(d)Dstd*WWd$SWWStjy} ztd+| td,| WYd$} ~ Wd$Sd$} ~ wwty} ztd-| WYd$} ~ d$Sd$} ~ ww).zComprehensive skill analysisrErFrGrHz AI CharacterrKr4rLz Category: rz Objective: rz Context: rRrSz Success Criteria: success_criteriaz AI Character: rIrJz - rOrr)rscenario_context conversationskill_analysis_templaterz`You are an expert skill assessor. Provide detailed, accurate analysis in valid JSON format only.r r#g333333?i r%rr)r*Nr+r,c3s|]}|vVqdS)Nr).0keyanalysisrr sz-GroqService.analyze_skills..)skill_analysisoverall_performanceconversation_analysisrecommendationszAnalysis missing required keysz JSON parsing error in analysis: r-zError analyzing skills: )r r r.r/rr0r1r2rr3r4r"r5r6r7r8r9allr;r:r<) rrBr[conversation_textrYrFr]rskill_templater=r>r?r@rrbranalyze_skillsosx          zGroqService.analyze_skillsr=c Cstz|jjjjdddd|dg|jddd}|jdjjWSt y9}zt d |WYd }~d Sd }~ww) z)Get a simple text completion from the LLMrz]You are a helpful assistant. Provide clear, natural responses without any formatting symbols.r r#r$ir%rzError getting completion: N) rr0r1r2rr3r4r"r5r<r;)rr=r>r@rrrget_completions&  zGroqService.get_completionN) __name__ __module__ __qualname__rrstrrrrArrZrlrmrrrrr s &2.6,Hr )rr8typingrrrrgroqrconfig.promptsrrr r r rrrrs