o r\(ihQ@sVddlZddlZddlmZmZmZmZddlmZm Z m Z m Z m Z GdddZ dS)N)DictListAnyOptional)SCENARIO_GENERATION_PROMPTCHARACTER_ROLEPLAY_PROMPTCHARACTER_CONCLUSION_PROMPTSKILL_ANALYSIS_PROMPTget_skill_analysis_templatec@seZdZd2defddZdedefddZd ed eeeffd d Z deeeffd dZ ddZ d3dedede dede eeeeeefff ddZdeeefdeeeeffddZd4deeefdeeeefd ed!edeef d"d#Zdeeefd$eeeefdeeeeffd%d&Zdedeefd'd(Zdedefd)d*Zdedeeeeffd+d,Zdeeeffd-d.Zdefd/d0Zd1S)5 OllamaServicehttp://127.0.0.1:11434base_urlcCs6||_d|_ddddddddddddd|_dS)Nz gemma3:12brinputoutputtotal)preview conversation assessment)r model token_counts)selfr rI/var/www/eduai.edurigo.com/roleplay/production/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_counts/s zOllamaService.get_token_countscCs"|jD] }dddd|j|<qdS)zReset all token countersrrN)r)r operationrrrreset_token_counts3s z OllamaService.reset_token_countsffffff?prompt system_prompt temperature max_tokensc Csz|jd}td|td|j|}|r"d|d|}|j|d||dd}td |d tj||d d }td |j|jdkrWtd|jdWdS||} | dd } | dd| dd| dd| dd| ddd} tdt | td| | | fWStj j y} ztd| td|jWYd} ~ dSd} ~ wtj j y} z!td | t| dr| jdurtd!| 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)r7 num_predict)rr5streamoptionszDEBUG: Payload model: rx)jsontimeoutzDEBUG: Response status: izERROR: 404 - Model 'z>' not found. Available models can be checked with: ollama list)NNresponser2r$rr%r&prompt_eval_duration eval_duration)r$r%r&r@rAzDEBUG: Ollama response length: zDEBUG: Ollama usage stats: z#Error connecting to Ollama server: zMake sure Ollama is running on NzError making Ollama request: zResponse content: z$Unexpected error in Ollama request: )r r(rrequestspost status_coderaise_for_statusr=r'stripr exceptionsConnectionErrorRequestExceptionhasattrr?r Exception) rr5r6r7r8url full_promptpayloadr?result response_textr#errr _make_request8s`         zOllamaService._make_request admin_inputc Cs,z{tj|d|d|dd|dd}d}|j||dd d \}}|s'Wd S|d |z+|d r:|dd }n |drE|dd }|drP|d d}t| WWStj y{}zt d|t d|WYd }~Wd Sd }~wwt y}zt d|WYd }~d Sd }~ww)z-Generate structured scenario from admin inputcategory objectivedetails, skills_to_assess)rTrUrVskillszHYou are a training scenario expert. Always respond with valid JSON only.r3r4r5r6r7r8Nr```json```zJSON parsing error: Response text: z'Error generating scenario with Ollama: ) rformatjoinrRr, startswithendswithr=loadsrFJSONDecodeErrorr(rK)rrSr5r6rPr#rQrrrgenerate_scenariossH        zOllamaService.generate_scenarioFscenarioconversation_history user_message is_conclusionc Cs,z{d}|D]}|ddkrdn|dd}||d|dd 7}q|r't}nt}|j|dd|dd |dd |dd |dd d|dd|dd|dd|dd||d } d} |j| | ddd\} } | rt|d| | ry| WSdWSty} ztd| WYd} ~ dSd} ~ ww)z&AI character responses during roleplayr2speakerlearnerLearner ai_charactername: message personalitygoals backgroundemotional_stateneutralscenario_setupcontext environmentrU constraints) character_namertrurvrwrzr{rUr|rirjzYou 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.g?irZrz:I'm having trouble responding right now. Please try again.z)Error in character roleplay with Ollama: N)rrrar'rRr,rKr()rrhrirjrk history_textturnrlprompt_templater5r6r?r#rQrrrplay_charactersH          zOllamaService.play_characterconversation_turnsc szd}|D]}|ddkrdnd}||d|dd7}qd |d d |d d |ddd|dd|ddd|ddd }|d}t|}tjd||||d} d} |j| | ddd\} } | skWdS|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),zComprehensive skill analysisr2rlrmrnz AI Characterrqrrrsz Category: rTz Objective: rUz Context: ryrzz Success Criteria: success_criteriaz AI Character: rorpz - rvrXrW)rYscenario_contextrskill_analysis_templatez`You are an expert skill assessor. Provide detailed, accurate analysis in valid JSON format only.g333333?i rZNrr[r\r]r^r_c3s|]}|vVqdS)Nr).0keyanalysisrr sz/OllamaService.analyze_skills..skill_analysisoverall_performanceconversation_analysisrecommendationszAnalysis missing required keysz JSON parsing error in analysis: r`z$Error analyzing skills with Ollama: )r r rarbrRr,rcrdrF_sanitize_json_textr=reallr(rf_try_json_fallbacksrK)rrhrconversation_textrrlrrYskill_templater5r6rPr#rQrrranalyze_skillssz            zOllamaService.analyze_skillsc Csfzd}|j||ddd\}}|r|d||WSty2}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.r3irZrz&Error getting completion from Ollama: N)rRr,rKr()rr5r6r?r#rQrrrget_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],[rs: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=rer(rKr_create_fallback_assessment) rrr json_matchextracted_jsonsanitized_jsonrrQskill_analysis_matchoverall_performance_matchconversation_analysis_matchrecommendations_match reconstructedrrrrcsL      z!OllamaService._try_json_fallbackscCsHdddgdgdgdiddddd d gd gd gd dgdgdgddS)zz%DEBUG: Health check response status: modelscSsg|]}|ddqS)rpr2)r')rrrrr sz.OllamaService.health_check..zDEBUG: Available models: zWARNING: Model 'z' not found in available modelszAvailable models: FTzDEBUG: Health check failed: N)r r(rBr'rDr=rrK)rrLr?r model_namesrQrrr health_checks(   zOllamaService.health_checkN)r )r2r3r4)F)__name__ __module__ __qualname__strrintr!rrr,r/r1floattuplerrRrgrboolrrrrrrrrrrrr s 8&;8-6/N0/r )r=rBtypingrrrrconfig.promptsrrrr r r rrrrs