o fh! @sddlZddlmZddlmZddlmZddlmZm Z m Z m Z ddl m Z ddlmZddlZejejdeeZGd d d eZdd ed eedededee ee ff ddZddZedkrkedSdS)N)Ollama)PromptTemplate)JsonOutputParser)ListDictAnyTuple) dataclass)Enum)levelc@seZdZdZdZdZdZdS) SkillLevelN)__name__ __module__ __qualname__BEGINNER INTERMEDIATEADVANCEDEXPERTrrJ/var/www/eduai.edurigo.com/doc_train/edurigo_ai/TNA/production/tna_step.pyr s r gemma3:12bpromptavailable_departments model_name temperaturereturnc st||d}d|}d|d|d|d|d }ztd|d |d d d ||}|}|d r@|dd }|drK|d d}|}t |} g} | d|r_|dnd} | |vrwt d| d|ru|dnd} | dgD]zqt fdddDst dWq}d dd} | ddtDvrt d dd!d"} nd} td#}d$|krd%ksnt d&|d'td$td%|}| d(| || d)d*d+Wq}ttfy}ztd,d |WYd }~q}d }~wwd-| vr td.| d-td/t| d0| d1| WStjyV}ztd2|td3|gWYd }~Sd }~wtyr}ztd4|gWYd }~Sd }~ww)5a Extract required skills from a business prompt using Ollama LLM Args: prompt (str): Business objective or challenge description available_departments (List[str]): List of valid department names from your database model_name (str): Ollama model to use (default: "gemma3:12b") temperature (float): Model temperature for consistency (default: 0) Returns: List[Dict]: List of skills with name, required_level, weight, and department Example: departments = ["Sales", "Marketing", "Customer Service", "IT", "HR", "Finance"] skills = extract_skills_from_prompt("I want to improve my sales revenue", departments) )modelrz, z You are a Training Needs Analysis expert. Analyze the following business objective and identify the specific skills needed to achieve it. Business Objective: zB IMPORTANT: You must choose the department from this exact list: aR - Choose only ONE department that is most relevant to this objective - Use the exact name from the list (case-sensitive) - If multiple departments could be involved, pick the PRIMARY one most responsible for this objective Return a JSON response with the following structure: { "department": "ONE department from the provided list: as", "skills": [ { "name": "Skill name", "required_level": "Beginner|Intermediate|Advanced|Expert", "weight": 0.0-1.0, "description": "Why this skill is needed" } ], "reasoning": "Brief explanation of why you chose this department and these skills" } Rules: 1. Department MUST be exactly one of: aN 2. Focus on 3-6 most critical skills for the chosen department 3. Weight represents business importance (1.0 = most critical) 4. Required_level should match the complexity of the objective 5. Be specific with skill names (avoid generic terms) 6. Return ONLY valid JSON, no additional text Example for "improve customer satisfaction" with departments [Sales, Marketing, Customer Service]: { "department": "Customer Service", "skills": [ {"name": "Active Listening", "required_level": "Expert", "weight": 0.9, "description": "Essential for understanding customer concerns"}, {"name": "Problem Resolution", "required_level": "Advanced", "weight": 0.8, "description": "Needed to solve customer issues effectively"} ], "reasoning": "Customer Service is the primary department responsible for customer satisfaction" } zAnalyzing prompt with z: N2z...z```jsonz``` departmentrGeneralz!LLM returned invalid department 'z', using first availableskillsc3s|]}|vVqdS)Nr).0keyskillrr vsz-extract_skills_from_prompt..)namerequired_levelweightzSkipping incomplete skill: r- _cSsg|]}|jqSr)r,)r'r rrr |sz.extract_skills_from_prompt..zInvalid skill level 'z', defaulting to INTERMEDIATE Intermediater.gg?zInvalid weight z, clamping to [0.0, 1.0]r, description)r,r-r.r$r3zError processing skill reasoningzLLM Analysis: zSuccessfully extracted z skills for z departmentzJSON parsing error: zRaw LLM response: zError extracting skills: )rjoinloggerinfoinvokestrip startswithendswithjsonloadsgetwarningallupperreplacer floatmaxminappend ValueErrorKeyErrorerrorlenJSONDecodeError Exception)rrrrllm dept_list system_promptresponsecleaned_responseresultr&r$ level_str skill_levelr.err)rextract_skills_from_prompts   +             rWcCsgd}gd}|D]l}tddtd|tdt||}|rrtd|ddtd t|d td t|d D]-\}}t|d |dtd|dtd|dtd|dtqCq tdq dS)zB Test the skills extraction function with various prompts )Sales MarketingzCustomer ServiceITHRFinance Operations)z"I want to improve my sales revenuez+We need to enhance customer service qualityz0How can we increase our marketing effectiveness?z&Our IT systems need better performancez+Improve team productivity and collaboration z<============================================================zPROMPT: z Department: rr$zSkills needed (z):z(----------------------------------------r z. r,z Level: r-z Weight: r.z Reason: r3zNo skills extractedN)printrWrK enumerate) departments test_promptsrr&ir*rrrmains(  rd__main__)rr)r=langchain_community.llmsrlangchain_core.promptsrlangchain_core.output_parsersrtypingrrrr dataclassesr enumr logging basicConfigINFO getLoggerrr7r strrDrWrdrrrrs"      0$