o 2hR@sRddlZddlZddlmZmZmZmZddlmZm Z m Z m Z GdddZ dS)N)DictListAnyOptional)SCENARIO_GENERATION_PROMPTCHARACTER_ROLEPLAY_PROMPTSKILL_ANALYSIS_PROMPTget_skill_analysis_templatec@seZdZd0defddZdedefddZd ed eeeffd d Z deeeffd dZ ddZ d1dedede dede eeeeeefff ddZdeeefdeeeeffddZdeeefdeeeefdedeefd d!Zdeeefd"eeeefdeeeeffd#d$Zdedeefd%d&Zdedefd'd(Zdedeeeeffd)d*Zdeeeffd+d,Zdefd-d.Zd/S)2 OllamaServicehttp://127.0.0.1:11434base_urlcCs6||_d|_ddddddddddddd|_dS)Nz gemma3:12brinputoutputtotal)preview conversation assessment)r model token_counts)selfr r[/var/www/eduai.edurigo.com/doc_train/edurigo_ai/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_counts.s zOllamaService.get_token_countscCs"|jD] }dddd|j|<qdS)zReset all token countersrr N)r)r operationrrrreset_token_counts2s 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)r6 num_predict)rr4streamoptionszDEBUG: Payload model: rx)jsontimeoutzDEBUG: Response status: izERROR: 404 - Model 'z>' not found. Available models can be checked with: ollama list)NNresponser1r#rr$r%prompt_eval_duration eval_duration)r#r$r%r?r@zDEBUG: 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) rr4r5r6r7url full_promptpayloadr>result response_textr"errr _make_request7s`         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)rSrTrUskillszHYou are a training scenario expert. Always respond with valid JSON only.r2r3r4r5r6r7Nr```json```zJSON parsing error: Response text: z'Error generating scenario with Ollama: ) rformatjoinrQr+ startswithendswithr<loadsrEJSONDecodeErrorr'rJ)rrRr4r5rOr"rPrrrgenerate_scenariorsH        zOllamaService.generate_scenarioscenarioconversation_history user_messagec Cszod}|D]}|ddkrdn|dd}||d|dd 7}qtj|dd|dd |dd |dd |dd d|dd|dd|dd||d }d}|j||ddd\} } | rh|d| | rm| WSdWSty} ztd| WYd} ~ dSd} ~ ww)z&AI character responses during roleplayr1speakerlearnerLearner ai_charactername: message personalitygoals backgroundemotional_stateneutralscenario_setupcontext environment constraints) character_namerrrsrtrurxryrzrhrizbYou are a roleplay character. Stay in character at all times. Respond naturally and realistically.g?irYrz:I'm having trouble responding right now. Please try again.z)Error in character roleplay with Ollama: N)rr`r&rQr+rJr') rrgrhri history_textturnrjr4r5r>r"rPrrrplay_characters@          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 analysisr1rjrkrlz AI Characterrorprqz Category: rSz Objective: rTz Context: rwrxz Success Criteria: success_criteriaz AI Character: rmrnz - rtrWrV)rXscenario_contextrskill_analysis_templatez`You are an expert skill assessor. Provide detailed, accurate analysis in valid JSON format only.g333333?i rYNrrZr[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 rr`rarQr+rbrcrE_sanitize_json_textr<rdallr're_try_json_fallbacksrJ)rrgrconversation_textr}rjrrXskill_templater4r5rOr"rPrrranalyze_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.r2irYrz&Error getting completion from Ollama: N)rQr+rJr')rr4r5r>r"rPrrrget_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],[rq: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<rdr'rJr_create_fallback_assessment) rrr json_matchextracted_jsonsanitized_jsonrrPskill_analysis_matchoverall_performance_matchconversation_analysis_matchrecommendations_match reconstructedrrrr\sL      z!OllamaService._try_json_fallbackscCsHdddgdgdgdiddddd d gd gd gd dgdgdgddS)zsz.OllamaService.health_check..zDEBUG: Available models: zWARNING: Model 'z' not found in available modelszAvailable models: FTzDEBUG: Health check failed: N)r r'rAr&rCr<rrJ)rrKr>r model_namesrPrrr health_checks(   zOllamaService.health_checkN)r )r1r2r3)__name__ __module__ __qualname__strrintr rrr+r.r0floattuplerrQrfrr~rrrrrboolrrrrrr s 8&;2-6)N0/r ) r<rAtypingrrrrconfig.promptsrrrr r rrrrs