o uii@sZddlZddlZddlmZmZmZmZddlmZm Z m Z m Z m Z m Z GdddZdS)N)DictListAnyOptional)SCENARIO_GENERATION_PROMPTCHARACTER_ROLEPLAY_PROMPTCHARACTER_CONCLUSION_PROMPT CHARACTER_GUIDED_ROLEPLAY_PROMPTSKILL_ANALYSIS_PROMPTget_skill_analysis_templatec@seZdZd6defddZdedefddZd ed eeeffd d Z deeeffd dZ ddZ d7dedede dede eeeeeefff ddZdeeefdeeeeffddZdeeeefdeeeefdefdd Zd8d#eeefdeeeefd$ed%edeeeefdeef d&d'Zd9d#eeefd(eeeefd)edeeeeffd*d+Zdedeefd,d-Zdedefd.d/Zdedeeeeffd0d1Zdeeeffd2d3Zdefd4d5Zd"S): OllamaServicehttp://127.0.0.1:11434base_urlcCs6||_d|_ddddddddddddd|_dS)Nz gemma3:12brinputoutputtotal)preview conversation assessment)rmodel token_counts)selfrrF/var/www/eduai.edurigo.com/roleplay/testing/services/ollama_service.py__init__s     zOllamaService.__init__textreturncCstdt|dS)uMEstimate token count for text (rough approximation: 1 token ≈ 4 characters))maxlen)rrrrr_estimate_tokensszOllamaService._estimate_tokensoperation_type usage_statsc Cs|rQ|dd}|dd}|d||}|dkr||}|j|d|7<|j|d|7<|j|d|7<td |d |d |d |dStd |ddS)zDUpdate token counts for specific operation using actual Ollama statsprompt_eval_countr eval_counttotal_durationirrrzOLLAMA Token count for z: Input=z , Output=z, Total=z%OLLAMA: No usage stats available for z, skipping token countN)getrprint)rr#r$ input_tokens output_tokens total_tokensrrr_update_token_counts  $z!OllamaService._update_token_countcCs |jS)z+Get current token counts for all operations)rcopyrrrrget_token_counts0s zOllamaService.get_token_countscCs"|jD] }dddd|j|<qdS)zReset all token countersrrN)r)r operationrrrreset_token_counts4s z OllamaService.reset_token_countsffffff?prompt system_prompt temperature max_tokensc sxz|jd}td|td|j|}|r#d|d|}|j|d||dd}td |d tjd d 4Id H}|j||dId H} Wd Id Hn 1Id Hs\wYtd| j| jdkrztd|jdWdS| | } | dd } | dd| dd| dd| dd| ddd} tdt | td| | | fWStj y} ztd| td |jWYd } ~ dSd } ~ wtj y} ztd!| WYd } ~ dSd } ~ wtjy!} ztd!| | jd urtd"| jjWYd } ~ dSd } ~ wty;} ztd#| WYd } ~ dSd } ~ ww)$zEMake a request to Ollama API and return response text and usage statsz /api/generatez!DEBUG: Making Ollama request to: zDEBUG: Using model: zSystem: z User: F)r8 num_predict)rr6streamoptionszDEBUG: Payload model: rg^@timeoutN)jsonzDEBUG: Response status: izERROR: 404 - Model 'z>' not found. Available models can be checked with: ollama list)NNresponser3r%rr&r'prompt_eval_duration eval_duration)r%r&r'rArBzDEBUG: Ollama response length: zDEBUG: Ollama usage stats: z#Error connecting to Ollama server: zMake sure Ollama is running on zError making Ollama request: zResponse content: z$Unexpected error in Ollama request: )rr)rhttpx AsyncClientpost status_coderaise_for_statusr?r(stripr! ConnectError RequestErrorHTTPStatusErrorr@r Exception)rr6r7r8r9url full_promptpayloadclientr@result response_textr$errr _make_request9sn  (        zOllamaService._make_request admin_inputc szd}|dD]\}}||d|d7}q d}|d}|rz?OllamaService._analyze_question_progression..rr[r3Nrgc3$|] }t|dkr|vVqdSrgNr!r~keywordconversation_textrr "z>OllamaService._analyze_question_progression..u ✓ ADDRESSEDu⚠ NOT YET ADDRESSEDz Question rW - c3rrrrrrrrrz **NEXT FOCUS**: Question rX)rmlowerrkr(splitanyappend) rrYryprogression_linesrvrw question_textkeywords addressedstatusrrr_analyze_question_progressions& &" z+OllamaService._analyze_question_progressionFNscenario user_message is_conclusionc s|zd}|D]}|ddkrdn|dd}||d|dd 7}qtd d |Dd } |d d} |dd} |dd} |dd} |dd}|rt}|dd|dd|dd|dd|ddd|dd|dd|dd||d }n|rt}d}t|d D]\}}||d |d!dd 7}q|||}id"|ddd|ddd|ddd|ddd|dddd|ddd|ddd#|d#dd$|d%iddd|ddd&|d'|d(|d)|d | d*| d| | || d+}nit}id"|ddd|ddd|ddd|ddd|dddd|ddd|ddd#|d#dd$|d%iddd|ddd(|d)|d | d*| d| d| d|d,| i}|jd5i|}d-}|j||d.d/d0Id1H\}}|r| d2||r|WSd3WSt y}zt d4|WYd1}~d3Sd1}~ww)6z&AI character responses during roleplayr3speakerlearnerLearner ai_characternamerWr|rXcSsg|] }|ddkr|qS)rrr})r~trrrrsz0OllamaService.play_character..r roleplay_namezRoleplay Sessionr_ Characterr^ difficultymediumdurationi, personalitygoals backgroundemotional_stateneutralscenario_setupcontext environment constraints) character_namerrrrrrrryrrZr[rr]additional_inforVguided_questionsquestion_progress_analysisryrcharacter_role)rr current_turnra"You are a world-class roleplay AI. You MUST stay strictly in character as a human-like counterpart (client, customer, etc.). You NEVER break character. You MUST follow the provided 'MANDATORY GUIDED QUESTIONS' if they exist. Respond professionally, naturally, and concisely (1-3 sentences).g?ircNrz:I'm having trouble responding right now. Please try again.z)Error in character roleplay with Ollama: r) r!r(rr rkrrrlrTr-rLr))rrryrrrY history_textrrrrrr^rrprompt_template prompt_kwargsrurvrwquestion_progressr6r7r@r$rSrrrplay_characters                                        zOllamaService.play_characterconversation_turnsengagement_metadatac szd}|D]}|ddkrdnd}||d|dd7}qd |d d |d d |ddd|dd|ddd|ddd }|d}t|} tjd||||| d} d} |j| | dddIdH\} } | spWdS|d| zG| d r| d!d} n | d"r| d#d} | d"r| dd$} | } | | } t | t fd%d&d'Dst d(WWdSWWSt jy}zt d)|t d*| || WYd}~WSd}~wwty}zt d+|WYd}~dSd}~ww),z:Comprehensive skill analysis with engagement-aware scoringr3rrrz AI CharacterrWr|rXz Category: r\z Objective: r]z Context: rrz Success Criteria: success_criteriaz AI Character: rrrrrar`)rbscenario_contextrrskill_analysis_templatezYou are an expert skill assessor. Provide STRICT, ACCURATE analysis in valid JSON format only. Do NOT inflate scores. Follow the engagement-based score ceilings exactly.g333333?i rcNrrdrerfrgrhc3s|]}|vVqdS)Nr)r~rsanalysisrrrsz/OllamaService.analyze_skills..skill_analysisoverall_performanceconversation_analysisrecommendationszAnalysis missing required keysz JSON parsing error in analysis: riz$Error analyzing skills with Ollama: )r r rlrmrTr-rnrorH_sanitize_json_textr?rpallr)rq_try_json_fallbacksrL)rrrrrrrrrbskill_templater6r7rRr$rSrrranalyze_skillsCs~           zOllamaService.analyze_skillsc snzd}|j||dddIdH\}}|r|d||WSty6}ztd|WYd}~dSd}~ww)z(Get a simple text completion from Ollamaz]You are a helpful assistant. Provide clear, natural responses without any formatting symbols.r4ircNrz&Error getting completion from Ollama: )rTr-rLr))rr6r7r@r$rSrrrget_completions" zOllamaService.get_completionc CsDz|}ddl}|dd|}|dd|}|dddd}|dd }|d d |}|d d |}|dd|}|d}t|D];\}}d|vr~|dr~|dd}t|dkr~|d}|dddkr~|dd|}|dd|||<qCd |}|WSt y} zt d| |WYd} ~ Sd} ~ ww)z6Sanitize JSON text to fix common issues before parsingrNz (\w+)n's\bz\1n'tz (\w+)'s\bz\1's"z, "'").replace('z ,(\s*[}\]])z\1z}\s*{z},{z]\s*\[z],[rX:rz+(?}z td|WYd}~nd}~wwzctd|d||j}|d||j} |d ||j} |d ||j} t || | | grd } | |d 7} | | d 7} | | d 7} | | 7} | d 7} || } t| }td|WSWnty}z td|WYd}~nd}~wwtd| S)z3Try multiple fallback methods to extract valid JSONrNz\{.*\}zBSuccessfully extracted JSON using regex fallback with sanitizationz)Regex fallback with sanitization failed: z+Attempting to reconstruct JSON structure...z$"skill_analysis":\s*\{.*?\}(?=,\s*")z)"overall_performance":\s*\{.*?\}(?=,\s*")z+"conversation_analysis":\s*\{.*?\}(?=,\s*")z"recommendations":\s*\{.*?\}{,}z-Successfully reconstructed JSON from sectionszJSON reconstruction failed: zDAll JSON parsing attempts failed, creating minimal fallback response) rsearchDOTALLgrouprr?rpr)rLr_create_fallback_assessment) rrr json_matchextracted_jsonsanitized_jsonrrSskill_analysis_matchoverall_performance_matchconversation_analysis_matchrecommendations_match reconstructedrrrrsL      z!OllamaService._try_json_fallbackscCsHdddgdgdgdiddddd d gd gd gd dgdgdgddS)z.zDEBUG: Available models: zWARNING: Model 'z' not found in available modelszAvailable models: FTzDEBUG: Health check failed: ) rr)rCrDr(rFr?rrL)rrMrPr@r model_namesrSrrr health_check$s. (  zOllamaService.health_check)r )r3r4r5)FN)r3)__name__ __module__ __qualname__strrintr"rrr-r0r2floattuplerrTrxrrboolrrrrrrrrrrrr s  8&A.s