@iP|dZddlZddlZddlmZmZmZddlmZddl m Z eje Z GddZy)z6Intent encoder for understanding user goals using LLM.N)DictAnyOptional) IntentPrompt) LLMClientceZdZdZd dedeefdZdedee ee ffdZ d e ee fde ee ffd Z d edee ee ffd Z y) IntentEncoderz0Encoder for understanding user intent using LLM.N llm_clientavailable_poi_typescR||_|xsg|_tj|_y)zInitialize intent encoder. Args: llm_client: LLM client instance available_poi_types: List of actual POI types in the dataset N)llmr r INTENT_PROMPTprompt_template)selfr r s I/home/ubuntu/codebase/yexijia/保研/colocation_mvp/llm/intent_encoder.py__init__zIntentEncoder.__init__s&#6#<" +99queryreturnc, tjd|ddd|jrMtjdt|jdt j ||j}n1tjd|j j|}tjd d }|jj|| }tjd t|tjd |j|}|tjd|ddytjdt|j|jrxtjdt|jdg}|j!|}t|jdg}tjd|d|d|jdd}t|jdg} tjd|d| tjd|jdg|S#t"$r$} tj%d| dYd} ~ yd} ~ wwxYw)uParse user query to extract intent. This method implements Stage0 intent understanding: 1. User ambiguous query (e.g., "我想开个烧烤店") 2. LLM analyzes query and extracts structured intent 3. Returns structured intent with business type, pattern preferences, etc. Args: query: User's natural language query Returns: Dictionary containing parsed intent, or None if parsing fails z8IntentEncoder: Starting intent understanding for query: Ndz...z$IntentEncoder: Building prompt with z available POI typesz;IntentEncoder: Building prompt without POI type constraints)rz6IntentEncoder: Calling LLM for intent understanding...z/You are an expert in spatial business analysis.)systemz(IntentEncoder: LLM raw response length: z@IntentEncoder: Extracting structured intent from LLM response...z?IntentEncoder: Failed to extract valid JSON from LLM response: z0IntentEncoder: Extracted intent JSON with keys: z?IntentEncoder: Validating patterns against dataset POI types...pattern_preferencez#IntentEncoder: Pattern validation: u → z valid patternsbusinesszN/Az6IntentEncoder: Successfully parsed intent - Business: z , Patterns: z#IntentEncoder: Preferred patterns: z%IntentEncoder: Error parsing intent: T)exc_info)loggerinfor lenrbuild_intent_promptrformatr generatedebug _extract_jsonwarninglistkeysget_validate_patterns Exceptionerror) rrprompt system_msg raw_response intent_jsonoriginal_countvalidated_countr pattern_countes rparsezIntentEncoder.parsesV-  KKRSXY]Z]S^R__bc d'' B3tG_G_C`Baauvw%99%AYAYZ YZ--4454A KKP QJJ88,,VJ,GL LLCC DUCVW X KKZ [,,\:K"!`amnroras`tuv KKJ4P[P`P`PbKcJde f'' ]^!$[__5I2%N!O"55kB "%koo6JB&O"P A.AQQVWfVggvwx#z59H 0Db IJM KKPQYPZZfgtfuv w KK=kooNbdf>g=hi j   LL@DtL T sD>I&D$I&& J/JJr/cd|vr|Sg}t|j}|dD]}t|ts|Dcgc]}t|ts||vs|}}t |dk\r|j |Yt |dk(shtjd|||d<t |dk(rtjd|Scc}w)zValidate and filter pattern_preference to only include valid POI types. Args: intent_json: Parsed intent JSON Returns: Validated intent JSON with filtered patterns rzSkipping single POI pattern: rzbNo valid patterns found after filtering. Original patterns may have used POI types not in dataset.) setr isinstancer&strrappendrr#r%)rr/valid_patternspoi_setpatternpoi valid_poiss rr)z IntentEncoder._validate_patternsXs { 2 d../"#78Ggt,*1\#JsC4HST[^#J\:!#%%j1ZA% 1$K!W^1$K"a'"#1aBCIgk2H**X.Kk40${2!+.B"CTJ8:K 45 "--A!B9!BA(40Z3=OqcUWW!B9K 45   9##  NN04 5  LL21#6 7 sXD2D DD7-D%DD .D8DDE#D55 E#EE#)N)__name__ __module__ __qualname____doc__rrr&rr:rrr4r)r$rrr r s: :9 :8D> :;3;8DcN#;;z#d38n#c3h#J:#:(4S>*B:rr )rVrHloggingtypingrrrllm.intent_promptr llm.clientr getLoggerrSrr rWrrr]s8< &&*   8 $jjr