o ;iS@s4UddlZddlZddlZddlZddlZddlZddlZddlmZm Z m Z m Z ej ej dedZeddZeddZd d d d Zeeefed <ddddZeeefed<dedefddZdBdedededefddZde edefddZd ede ejfd!d"Zded#e ejdefd$d%Zd&ed'ed(ed ed)ed*ed+edefd,d-Z . .  dCd/eee fd ed0ed*ede e ed)e edeee ffd1d2Ze d3krd4d5d6d7Z!d8d9id8d:id8d;igZ"e#d<ee!d=d.d.gd>Z$e#ej%e$d?d@e#dAee!d=d.de"d>Z&e#ej%e&d?d@dSdS)DN)DictListAnyOptional)levelskill_assessment_processorOLLAMA_API_URLz#http://localhost:11434/api/generate OLLAMA_MODELz gemma3:12bBeginner IntermediateAdvanced)DIFFICULTY_LEVEL_MAPS1S2S3)r r r SKILL_ID_MAPtextreturncCstdd|dd}|d}|dkr|Sd}t||d|D]#\}}|dkr0|d7}q#|d krF|d8}|dkrF|||dSq#||dS) z>Strip markdown fences and pull the first complete JSON object.z ```(?:json)?z```{rNr })resubreplacestripfind enumerate)rstartdepthichr%Q/var/www/eduai.edurigo.com/skill_assessment/testing/skill_assessment_processor.py _extract_jsons   r'rpromptsystemretriesc Cst||ddddddddd }td }td |d D]H}z%td |d|d dtjt|dd}|| ddWStj j yb}z|}t d|d|WYd}~qd}~ww|)NFjsong?i ig?g?) temperaturenum_ctx num_predicttop_prepeat_penalty)modelr(r)streamformatoptionszNo attempts mader rzOllama call attempt /z ...ih)r+timeoutresponserzOllama attempt z failed: )r RuntimeErrorrangeloggerinforequestspostrraise_for_statusr+get exceptionsRequestExceptionwarning)r(r)r*payloadlast_excattemptrespexcr%r%r& _call_ollama,s2"rHpreviously_generated_questionscCs\|sdSg}t|dD]\}}|dd}|r$|d|d|q |s)dSd|S)z Build a human-readable summary of previously generated questions so the LLM knows exactly which questions to avoid. u=None — this is the first attempt for this difficulty level.r question_namerz z.  )r r?rappendjoin)rIlinesr#qq_textr%r%r&#_build_previously_generated_summaryHs rQ company_namecCs|}|sgSd}g}t|}|t|d|dtjtjdd|tjd}tjdd|tjd}|rX||krXt|}|t|d|dtj|S)z Build a list of compiled regex patterns that match the company name and its common variations (with/without corporate suffixes, partial tokens, abbreviations, etc.). z\s*(?:Pvt\.?\s*Ltd\.?|Private\s+Limited|Ltd\.?|Limited|Inc\.?|Incorporated|LLC|LLP|Corp\.?|Corporation|Co\.?\s*Ltd\.?|Co\.?|Pte\.?\s*Ltd\.?|GmbH|AG|S\.?A\.?|NV|PLC)z\s*?zh\s+(?:Pvt|Private|Ltd|Limited|Inc|Incorporated|LLC|LLP|Corp|Corporation|Co|Pte|GmbH|AG|SA|NV|PLC)[\s.]*$rflags)rrescaperLcompile IGNORECASErlower)rRname_CORP_SUFFIXESpatterns escaped_full core_name escaped_corer%r%r&_build_company_patterns^s8  r`r\cCs|r|s|S|}|D]}tjd|jdd|tjd}|d|}q tdd|}tdd|}|rH|drH|d|d d }|S) a Remove any company-name reference from *text*. Strategy (in order): 1. If company name appears as the start of a sentence followed by more content (e.g. "Ascent Cyber Solutions uses Python for parsing log files."), remove that entire leading clause/sentence. 2. Otherwise, replace the company name inline, leaving the surrounding text intact. 3. Clean up leftover whitespace / punctuation artefacts. z(?:^|(?<=\. ))z[^.?!]*[.?!]\s*rrTz\s{2,} z ^[,;:\s]+rr N)rrpatternrXrislowerupper)rr\cleanedpatr%r%r&_sanitize_company_referencess  rgdata skill_gap designationdifficulty_labelrEskill_idc Cs|dg}t|dkrtdt|dt|}gd} g} t|D]\} } d| d| d<|| d <| d } | d vrE| | | d <| d g}t|d kr`td| ddt|d| d dkrdd|D}t|dkrtd| ddt|dt|D] \}}|dk|d<q| d dkrdd|D}t|dkrtd| ddt|dd|dd<d|dd<t|t|D]\}}t d||d<| ddq| dd}t ||| d<|D]}|d d}t |||d <q|| dkrt d| dd!|d"| d|| d <||||d#g| d$<| dd s6td| dd%| | q$td&d'|d(d)}| |d<d|d*<| d+d,|d'|d-|d'tjd(d.| d/|d0|d1||S)2a Enforce structure for exactly 3 questions at a single difficulty level. - Type 2 = single-correct MCQ (exactly 1 isCorrect: true) - Type 3 = multi-correct MCQ (at least 2 isCorrect: true) - Each question must have exactly 4 options. questionsrz"Expected exactly 3 questions, got z+. Ensure the LLM returns the correct count.)rrrQr questionId difficulty questionType)rrr4z" must have exactly 4 options, got .rcSg|] }|dr|qS isCorrectr?.0or%r%r& z+_validate_and_repair_3q..z (Type 2) had zA correct options; defaulting correct answer to the second option.rvcSrtrurwrxr%r%r&r{r|z (Type 3) had only zO correct option(s); defaulting correct answers to the first and second options.TrAoptionIdF questionTextr optionTextz=: Stripped company-name reference from questionText. Before: z After: )skillId skillName skillLevelrEskillsz has an empty questionText.z [^a-z0-9]-NtotalQuestions assessmentIdz assessment--aassessmentTitlera Assessment for )r?len ValueErrorr`r r:rBrandomshufflechr setdefaultrgr;rrLrrrYuuiduuid4hex)rhrirjrRrkrErlrmcompany_patternsdefault_type_patternrepairedidxrOqtoptscorrectoioptoriginal_q_text original_optsafe_idr%r%r&_validate_and_repair_3qs                  ,rr user_detailsquestions_difficulty_levelc Csl|durg}|dp d}|dpd}|dpd}|p&d} |dur1t|d }t|d |} td |d |d |d| dt| t|} d} t j dd} d gd|d|d|d|d|d| d|d| d| d|d|d|d|d |d!|d"| d#|d$|d%|d&|d!|d'| d#|d$|d%|d(|d!|d"| d#|d$|d%|d)}zt || }Wn&ty0}ztd*|d+|d,d-d.gd/WYd}~Sd}~wwt|}zt|}Wn.tjyj}z td0|d1|dd2d3|d,d-d.gd/WYd}~Sd}~wwz t|||| ||| d4}Wn&ty}ztd5|d6|d,d-d.gd/WYd}~Sd}~wwtd7|d8d9|d:d;|d ||S)", "difficulty": "z:", "questionType": 2, "skills": [{"skillId": "z", "skillName": "z", "skillLevel": "z", "attempt": a}], "options": [ {"optionId": "A", "optionText": "