o 5&h:@sRddlZddlZddlmZmZmZmZddlmZm Z m Z m Z GdddZ dS)N)DictListAnyOptional)SCENARIO_GENERATION_PROMPTCHARACTER_ROLEPLAY_PROMPTSKILL_ANALYSIS_PROMPTget_skill_analysis_templatec@sPeZdZd*defddZdedefddZd ed eeeffd d Z deeeffd dZ ddZ d+dedede dede eeeeeefff ddZdeeefdeeeeffddZdeeefdeeeefdedeefd d!Zdeeefd"eeeefdeeeeffd#d$Zdedeefd%d&Zdefd'd(Zd)S), OllamaServicehttp://127.0.0.1:11434base_urlcCs6||_d|_ddddddddddddd|_dS)Nz gemma3:12brinputoutputtotal)preview conversation assessment)r model token_counts)selfr rX/var/www/eduai.edurigo.com/doc_train/edurigo_ai/roleplay/prod/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)rcopy)rrrrget_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)r5 num_predict)rr3streamoptionszDEBUG: Payload model: rx)jsontimeoutzDEBUG: Response status: izERROR: 404 - Model 'z>' not found. Available models can be checked with: ollama list)NNresponser0r#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) rr3r4r5r6url 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)rRrSrTskillszHYou are a training scenario expert. Always respond with valid JSON only.r1r2r3r4r5r6Nr```json```zJSON parsing error: Response text: z'Error generating scenario with Ollama: ) rformatjoinrPr+ startswithendswithr;loadsrDJSONDecodeErrorr'rI)rrQr3r4rNr"rOrrrgenerate_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 roleplayr0speakerlearnerLearner ai_charactername: message personalitygoals backgroundemotional_stateneutralscenario_setupcontext environment constraints) character_namerqrrrsrtrwrxryrgrhzbYou are a roleplay character. Stay in character at all times. Respond naturally and realistically.g?irXrz:I'm having trouble responding right now. Please try again.z)Error in character roleplay with Ollama: N)rr_r&rPr+rIr') rrfrgrh history_textturnrir3r4r=r"rOrrrplay_characters@          zOllamaService.play_characterconversation_turnsc sLz d}|D]}|ddkrdnd}||d|dd7}qd |d d |d d |ddd|dd|ddd|ddd }|d}t|}tjd||||d} d} |j| | ddd\} } | slWdS|d| zB| d r| d!d} n | d"r| d#d} | d"r| dd$} | } t | t fd%d&d'Dst d(WWdSWWSt j y } zHt d)| t d*| d+dl}|d,| |j}|rzt |t d-WWYd} ~ WSt j yt d.YnwWYd} ~ WdSd} ~ wwty%} zt d/| WYd} ~ dSd} ~ ww)0zComprehensive skill analysisr0rirjrkz AI Characterrnrorpz Category: rRz Objective: rSz Context: rvrwz Success Criteria: success_criteriaz AI Character: rlrmz - rsrVrU)rWscenario_contextrskill_analysis_templatez`You are an expert skill assessor. Provide detailed, accurate analysis in valid JSON format only.g333333?i rXNrrYrZr[r\r]c3s|]}|vVqdS)Nr).0keyanalysisrr sz/OllamaService.analyze_skills..)skill_analysisoverall_performanceconversation_analysisrecommendationszAnalysis missing required keysz JSON parsing error in analysis: r^rz\{.*\}z0Successfully extracted JSON using regex fallbackzRegex fallback also failedz$Error analyzing skills with Ollama: )r rr_r`rPr+rarbrDr;rcallr'rdresearchDOTALLgrouprI)rrfr~conversation_textr|rirrWskill_templater3r4rNr"rOr json_matchrrranalyze_skillss            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.r1irXrz&Error getting completion from Ollama: N)rPr+rIr')rr3r4r=r"rOrrrget_completions   zOllamaService.get_completionc CszU|jd}td|tj|dd}td|j|jdkrS|dg}dd |D}td ||j|vrPtd |jd td |WdSWdSWdStyo}ztd|WYd}~dSd}~ww)z$Check if Ollama service is availablez /api/tagsz"DEBUG: Checking Ollama health at: )r<z%DEBUG: Health check response status: modelscSsg|]}|ddqS)rmr0)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'r@r&rBr;rrI)rrJr=r model_namesrOrrr health_check2s(   zOllamaService.health_checkN)r )r0r1r2)__name__ __module__ __qualname__strrintr rrr+r-r/floattuplerrPrerr}rrboolrrrrrr s 8&;2-6)Tr ) r;r@typingrrrrconfig.promptsrrrr r rrrrs