@iP|SrSSKrSSKrSSKJrJrJr SSKJr SSK J r \R"\ 5r "SS5rg)z6Intent encoder for understanding user goals using LLM.N)DictAnyOptional) IntentPrompt) LLMClientc\rSrSrSrSS\S\\4SjjrS\ S\\ \ \ 44S jr S \ \ \ 4S\ \ \ 44S jr S \ S\\ \ \ 44S jrSrg) IntentEncoder z0Encoder for understanding user intent using LLM.N llm_clientavailable_poi_typescZXlU=(d /Ul[RUlg)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 E/home/ubuntu/codebase/yexijia/保研/iCoLoc/src/llm/intent_encoder.py__init__IntentEncoder.__init__s$#6#<" +99queryreturnc0[RSUSSS35 UR(aM[RS[UR5S35 [R "XR5nO.[RS5 UR RUS9n[RS 5 S nURRX#S 9n[RS [U535 [RS 5 URU5nUc[RSUSS35 g[RS[UR5535 UR(ax[RS5 [URS/55nUR!U5n[URS/55n[RSUSUS35 URSS5n[URS/55n [RSUSU 35 [RSURS/535 U$!["a!n [R%SU 3SS9 Sn A gSn A ff=f)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_promptrformatrgeneratedebug _extract_jsonwarninglistkeysget_validate_patterns Exceptionerror) rrprompt system_msg raw_response intent_jsonoriginal_countvalidated_countr pattern_countes rparseIntentEncoder.parsesR-  KKRSXY]Z]S^R__bc d'' B3tG_G_C`Baauvw%99%AYAYZ YZ--4454A KKP QJJ88,,V,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** J4JJr1cSU;aU$/n[UR5nUSHn[U[5(dMUVs/sH#n[U[5(dMXS;dM!UPM% nn[ U5S:aUR U5 Ml[ U5S:XdM}[RSU35 M X!S'[ U5S:Xa[RS5 U$s snf)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(strr!appendrr%r')rr1valid_patternspoi_setpatternpoi valid_poiss rr+ IntentEncoder._validate_patternsXs { 2 d../"#78Ggt,,*1\#JsC4H#S^#J\:!#%%j1ZA% 1$KW^1$K"a'"#1aBCgk2H**X.Kk400${2!+.B"CTJJ8:K 45 "--A!B9!BA(400Z3=O=OqcUWW!B9K 45   9##  NN04 5  LL21#6 7 sXD%1D% D%!D%>2D%1$D%D%;D D% D%%F9E F#FF)r rr)N)__name__ __module__ __qualname____firstlineno____doc__rrr(rr=rrr6r+r&__static_attributes__rrr r s: :9 :8D> :;3;8DcN#;;z#d38n#c3h#J:#:(4S>*B:rr )r\rLloggingtypingrrrllm.intent_promptr llm.clientr getLoggerrXrr r^rrrds6< &&*   8 $jjr