o wh5@sddlZddlZddlmZmZmZmZddlmZddl m Z ddl Z ddl m Z ddlmZddlmZejejdeeZeGd d d ZeGd d d ZeGd ddZGdddZGdddZGdddZddZedkredSdS)N)DictListOptionalTuple) dataclass)SequenceMatcher) ChatOllama)PromptTemplate) HumanMessage)levelc@s"eZdZUdZeed<eed<dS) Departmentz%Data class for department informationidnameN)__name__ __module__ __qualname____doc__int__annotations__strrrI/var/www/eduai.edurigo.com/doc_train/edurigo_ai/TNA/production/tna_dep.pyr s  r c@s&eZdZUdZeed<eeed<dS) TNARequestzData class for TNA request user_promptorganization_departmentsN)rrrrrrrr rrrrrs rc@s:eZdZUdZeed<eed<eed<eed<eed<dS) TNAResponsezData class for TNA response department_iddepartment_nameinferred_departmentsimilarity_scorematchedN) rrrrrrrfloatboolrrrrrs  rc@sDeZdZdZddefddZdedefdd Zd edefd d Zd S)DepartmentInferenceEnginez&Handles department inference using LLM gemma3:12b model_namec CsNzt|d|_td|WdSty&}z td|d}~ww)zInitialize the LLM model)modelzInitialized LLM with model: zFailed to initialize LLM: N)rllmloggerinfo Exceptionerror)selfr%errr__init__'s z"DepartmentInferenceEngine.__init__rreturnc Cstdgdd}z%|j|d}|jt|dg}|j}||}t d||WSt yG}zt d|WYd}~d Sd}~ww) z Infer the most relevant department for the given user prompt Args: user_prompt: User's business goal or challenge Returns: Inferred department name ra^ You are an expert business analyst specializing in organizational structure and training needs. Based on the following business goal or challenge, identify the SINGLE most relevant department. User Goal/Challenge: {user_prompt} Common departments include: - Sales - Marketing - HR (Human Resources) - Operations - Finance - Product Management - Development/Engineering - Customer Service - Quality Assurance - Research and Development Instructions: 1. Analyze the user's goal/challenge carefully 2. Identify which department would be primarily responsible for addressing this goal 3. Return ONLY the department name (e.g., "Sales", "Marketing", "HR") 4. Use general department names, not specific subdivisions 5. If the goal spans multiple departments, choose the PRIMARY one Department: )input_variablestemplate)r)contentzInferred department: zError in department inference: NUnknown) r formatr'invoker r2strip_clean_department_namer(r)r*r+)r,rprompt_templateformatted_promptresponse inferred_deptr-rrrinfer_department0s     z*DepartmentInferenceEngine.infer_department dept_namecCsD|dddd}|dddd}|dd}|d}|S) z#Clean and normalize department namez Department:r zTeam:Team rz "')replacesplitr6)r,r=rrrr7ks  z0DepartmentInferenceEngine._clean_department_nameN)r$)rrrrrr.r<r7rrrrr#$s  ;r#c@s@eZdZdZedededefddZededefdd Zd S) SimilarityMatcherzPHandles string similarity matching between inferred and organization departmentsstr1str2r/c Cst|}t|}td||}||krdnd}d}||vs$||vr&d}t|}t|}|rE|rEt||t||} nd} t |||| } | S)z Calculate similarity between two strings using multiple methods Args: str1: First string str2: Second string Returns: Similarity score between 0 and 1 Ng?g?) rC_normalize_stringrratiosetrBlen intersectionunionmax) rDrE str1_norm str2_normseq_similarity exact_matchsubstring_matchwords1words2word_similarity final_scorerrrcalculate_similarity|s   z&SimilarityMatcher.calculate_similarityscCsl|}tdd|}tdd|}dddddddd d d d d d d d }|D] \}}|||}q'|S) zNormalize string for comparisonz[^a-zA-Z0-9\s]r>z\s+ hritrdzcustomer supportproduct development developer)zhuman resourceszhuman resourcezinformation technologyzresearch and developmentzresearch & developmentzcustomer servicez customer carezproduct managementzproduct managerzsoftware developmentzsoftware engineering engineering programmer programming)lowerresubitemsrAr6)rX replacementsoldnewrrrrGs*z#SimilarityMatcher._normalize_stringN) rrrr staticmethodrr!rWrGrrrrrCys 'rCc@steZdZdZddedefddZded efd d Z ded e fd d Z dede e d efddZded efddZdS) TNASystemz:Main TNA System class that orchestrates the entire processr$Q?r%similarity_thresholdcCs,t||_t|_||_td|dS)z Initialize TNA System Args: model_name: LLM model name similarity_threshold: Minimum similarity score for department matching z2TNA System initialized with similarity threshold: N)r#inference_enginerCsimilarity_matcherrmr(r))r,r%rmrrrr.s zTNASystem.__init__ request_datar/c CszB||}|j|j}|||j}|d|jkr-t|dj|dj ||ddd}n tdd||ddd}t d||WSt ye}zt d |tddd d ddWYd }~Sd }~ww) a Process TNA request and return department mapping result Args: request_data: Dictionary containing user_prompt and organization_departments Returns: TNAResponse with department mapping result r departmentT)rrrrr rNullFzTNA processing completed: zError processing TNA request: ErrorrFN)_parse_requestrnr<r_find_best_matchrrmrr rr(r)r*r+)r,rp tna_requestr best_matchr:r-rrrprocess_requests@ zTNASystem.process_requestcCs"dd|dD}t|d|dS)z)Parse request data into TNARequest objectcSs g|] }t|d|ddqS)r rr r)r ).0deptrrr sz,TNASystem._parse_request..rrrr)r)r,rp departmentsrrrrt szTNASystem._parse_requestr;org_departmentsc Cszddd}|D]}|j||j}||dkr||d<||d<qtd|d|dr/|djndd |dd d |S) zAFind the best matching department from organization's departmentsNrF)rqrrrqzBest match for 'z': Nonez (Score: z.4f))rorWrr(r))r,r;rrwr{rrrrrus 6zTNASystem._find_best_match json_datac Cszt|}||}d|j|j|jt|jd|j|j dWStj yC}zt d|dd|ddd WYd }~Sd }~wt ye}zt d |dt|ddd WYd }~Sd }~ww) z Process JSON request and return JSON response Args: json_data: JSON string with request data Returns: Dictionary with response data T)successrrrrr threshold_usedzJSON decode error: FzInvalid JSON format: rrr)rr+rrNzProcessing error: )jsonloadsrxrrrroundrr rmJSONDecodeErrorr(r+r*r)r,rrpr:r-rrrprocess_json_request)s8    zTNASystem.process_json_requestN)r$rl)rrrrrr!r.rrrxrrtrr rurrrrrrks3 rkc Cstdd}dddddddd d dd d dd dddddgd}td|t|}tdtd|dtd|dtd|dtd|dtd|dtd|dtd |d!d"S)#zExample usage of the TNA Systemrl)rmz%I Want to improve my technical skills OperationsryHRFinancezProduct Managementsales Developerr}zProcessing TNA request...z TNA Results:z Success: rzDepartment ID: rzDepartment Name: rzInferred Department: rzSimilarity Score: rz Matched: r zThreshold Used: rN)rkprintrrdumps) tna_systemsample_requestresultrrrmainSs(  r__main__)rloggingtypingrrrr dataclassesrdifflibrrdlangchain_ollamarlangchain.promptsr langchain.schemar basicConfigINFO getLoggerrr(r rrr#rCrkrrrrrs2      UL!