ipdZddlZddlZddlmZmZej eZdZ dZ gdZ GddZ y) uU core/workers/lead_qualification_worker.py Story 5.07 — LeadQualificationWorker: Qualification Script Runner Uses Gemini Flash to generate 3 qualifying questions based on extracted entities from an IntentSignal, then writes them to Redis for AIVA's next utterance. No SQLite. No real I/O in isolation — all dependencies are injected. N)AnyOptionalzgemini-2.0-flashz A potential customer called about: {service} in {location}. Generate 3 concise qualifying questions AIVA should ask next. Focus on: urgency, budget range, specific requirements. Return JSON: {{"questions": ["Q1", "Q2", "Q3"]}} )z!When would you need this done by?z#Do you have a budget range in mind?z8Are there any specific requirements I should know about?cbeZdZdZeZd dZdedefdZdede e fdZ d ed e ddfd Z defd Z y)LeadQualificationWorkerz Worker that generates 3 qualifying questions via Gemini Flash and writes them to Redis for AIVA's next conversational turn. Nc ||_||_y)N)_gemini_redis)self gemini_client redis_clients ?/mnt/e/genesis-system/core/workers/lead_qualification_worker.py__init__z LeadQualificationWorker.__init__+s$ " entitiesreturncx|jdd}|jdd}tj||S)z0Build the Gemini prompt from extracted entities.servicezgeneral serviceslocationz your area)rr)getQUALIFICATION_PROMPTformat)r rrrs r _build_promptz%LeadQualificationWorker._build_prompt/s9,,y*<=<< K8#**7X*NNr response_textc tj|}|jdg}t|trt |dk\r|ddSy#tj ttf$rYywxYw)z Parse Gemini response JSON and extract questions list. Returns list of 3 questions or None on parse failure. questionsN) jsonloadsr isinstancelistlenJSONDecodeErrorAttributeError TypeError)r rdatars r _parse_questionsz(LeadQualificationWorker._parse_questions5sh  ::m,Db1I)T*s9~/B !}$$$ni@  sA A A.-A. session_idrc|jtjdyd|}|jj|dt j |tj d|y)z2Write qualification questions to Redis state hash.Nu(No Redis client — skipping state writez aiva:state:qualification_questionsz-Wrote qualification_questions to Redis key=%s)r loggerwarninghsetrdumpsdebug)r r'rkeys r _write_to_redisz'LeadQualificationWorker._write_to_redisCsW ;;  NNE F J<( 7I9NO DcJrcK|jxsi}|j}d}|jH |j|}|jj |}|j |j }|+tjd|t|j}|j||d|iS#t$r"}tjd|d}Yd}~id}~wwxYww)z 1. Build prompt from intent.extracted_entities 2. Call Gemini Flash for 3 qualifying questions 3. Write questions to Redis aiva:state:{session_id} 4. Return {"questions": ["Q1", "Q2", "Q3"]} Nu4Gemini call failed (%s) — using fallback questionsz5Using fallback qualification questions for session=%sr)extracted_entitiesr'rrgenerate_contentr&text Exceptionr*r+infor FALLBACK_QUESTIONSr0)r intentrr'rpromptresponseexcs r executezLeadQualificationWorker.executeLs,,2&&   << # !++H5<<88@ 11(--@   KKOQ[ \T445I Z3Y'' !UWZ[  !s0+C&AB85AC&8 C#CC&C##C&)NN)__name__ __module__ __qualname____doc__r7rdictstrrrr r&r0r<rr rr#sk ,#OdOsO c htn K#K$K4K(t(rr) r@rloggingtypingrr getLoggerr=r* GEMINI_MODELrr7rrCrr rHsH     8 $! G(G(r