o #qhbB@sndZddlmZmZmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlZGd d d ZdS) zA Updated agent.py to prioritize user query over job profile data )ListDictAny)datetime) YouTubeMCP)GoogleCoursesMCP) CourseraMCP)KhanAcademyMCPNc @seZdZddeedededefddZd efd d Zd eeeeffd d Z d eefddZ ded eefddZ ddZ ddZ dS) CareerAgentyoutubeN user_skills user_query platform_namejob_profile_datacCs:||_||_||_||_||_||_||_ dSN) r r rrget_mcp_instancemcpidentify_skill_gaps skill_gapsextract_target_role_from_query target_role)selfr r rrrM/var/www/eduai.edurigo.com/doc_train/edurigo_ai/my_career/production/agent.py__init__s  zCareerAgent.__init__returncCs^|jsdS|j}gd}|D]}t||}|r,|d}tdd|}|SqdS)zDExtract target role from user query for career transition scenarios.)ztransition to\s+([^,.!?]+)zbecome\s+a?\s*([^,.!?]+)zmove to\s+([^,.!?]+)zswitch to\s+([^,.!?]+)zcareer change to\s+([^,.!?]+)zwant to be\s+a?\s*([^,.!?]+)z\s+ )r lowerresearchgroupstripsub)r query_lowertransition_patternspatternmatchrrrrrs  z*CareerAgent.extract_target_role_from_querycCs|j|j|j|j|j|jS)z@Fetch courses from the appropriate MCP with user query priority.)r fetch_coursesr rrr rrrrrr)3szCareerAgent.fetch_coursesc s"g}|}|ri||}dd|jD}|D]Q}|d}||vr@||ddt||dd|ddd d d d q||}t|d d}|dd} || krh||d|d|| ddd d q|jrt |jt r|jdg} | r|sdd|jD}| D]O} | dd} t| d d} || } | s|| dd| ddd| dddd qt| d d}|| kr|| d| dd|| dddd q|s|jD]!}t|d d}|dkr||d|d|ddddd qddddd|j fdd d!|S)"zLIdentify skill gaps prioritizing user query intent over current job profile.cSi|] }|dd|qS skill_namergetr.0skillrrr Iz3CareerAgent.identify_skill_gaps..r-skill_idtarget_rrequired_leveltransition_skillcriticaluser_query_target)r5r- current_level target_levelgap_typeprioritysource skill_typertransition_upgradehighjob_profile_skillscSr+r,r.r0rrrr3or4rmissingcurrent_role_skillmediumcurrent_job_profilecurrent_role_upgradegeneral_improvementlowr )r:rCrGrKcs|dddS)Nr?rGr8r/xpriority_orderrrsz1CareerAgent.identify_skill_gaps..)key) rget_target_role_skillsr rappendr/lenintr isinstancedictsort)rgapsr target_skillsuser_skill_map target_skillskill_name_lower current_skillr<r7rD job_skilljob_skill_name user_skillr2 skill_levelrrQrr=s               zCareerAgent.identify_skill_gapsrc s&|ddddddddddddddddd dd d dgd ddd ddd d dddddddgddddddddddd ddddgddddd ddd dddddddgd}|D]\}}|vs|tfdd|Dr|Sqgddddddddddd dgS)zFGet required skills for target role based on common role requirements.zSales Managementr8)r-r7zTeam Leadershipz Customer Relationship Management Communication NegotiationzStrategic PlanningrLzPerformance ManagementzProject ManagementzAgile MethodologyzRisk Management LeadershipzDigital MarketingzBrand ManagementzMarketing Strategy AnalyticszProduct ManagementzUser Experience Designz Data Analysis)z sales managerzproject managerzmarketing managerzproduct managerc3s|]}|vVqdSrrr1wordtarget_role_lowerrr sz5CareerAgent.get_target_role_skills.. ManagementzStrategic Thinking)ritemsanysplit)rrrole_skills_maproleskillsrrlrrUsJ ""z"CareerAgent.get_target_role_skillscCsL|jdkr tS|jdkrtS|jdkrtStd|j)z| d?| d@|r|nddAt|}| r| |dB<||q|jdCdDd9dE|S)FzBRecommend courses with user query priority and meaningful reasons.job_profile_namersnippettitle description thumbnailsNrCurlrGdefaultcSg|] }|ddkr|qS)r?r:rNr1gaprrr r4z1CareerAgent.recommend_courses..cSr)r?rCrNrrrrrr4r-r:cSsg|] }|ddvr|qS)r?)r:rCrNrrrrrr4rLcSsg|]}|dqS)r-rrrrrrcSsg|]}|ddqS)r?rGrNrrrrrrz(Essential for your career transition to z$Recommended for your target role as z/Recommended to enhance your professional skillsz&Addresses critical transition skills: z, zCovers essential skills: zHelps improve skills in: zCourse content: z...z. <c3s|] }|vVqdSrrrjr{rrrn2sz0CareerAgent.recommend_courses.. c3s |] }|vrdVqdS)rNrrjrrrrn>sidvideoIdcourseIdzauto-generated courseNameplatform instructor channelTitledurationz Self-paced matchScoredreasonscorerisOnlineCourseTz https://www.youtube.com/watch?v=typevideo lastUpdatedaddressed_gapsgap_prioritiestarget_role_alignmenttransition_focused thumbnailcSs|dS)NrrrOrrrrS^sz/CareerAgent.recommend_courses..)rTreverse)rrr/rnowyearrrrVjoinrqrrr summinrroundboolr[)rcoursesrecommended_coursesrprimary_context current_yearcourserzr|r} thumbnail_urlrr critical_gaps high_gapsrgap_nametop_gaps reason_partsr match_score query_words title_matchesvideo_idcourse_recommendationrrrrecommend_coursess                  "              zCareerAgent.recommend_courses)r N)__name__ __module__ __qualname__rrstrrrrr)rrUrrrrrrr s   f3 r )__doc__typingrrrr youtube_mcprgoogle_courses_mcpr coursera_mcprkhan_academy_mcpr r r rrrrs