o .Ēi@s>UddlZddlZddlZddlmZmZmZddlZzddl m Z ddl m Z m Z m Z mZWneyBdZ eZ Z Z ZYnweddZeeddZeed d Zeed d Zdaed ed<dddZdedeeeffddZdeeefdeeeffddZdeeefdeeeffddZdS)N)AnyDictOptional) Anthropic)APIConnectionErrorAPITimeoutErrorRateLimitErrorAPIStatusError KYC_OCR_MODELzclaude-sonnet-4-5KYC_OCR_MAX_OUTPUT_TOKENS1400KYC_OCR_MAX_RETRIES2KYC_OCR_RETRY_BASE_SLEEPz0.5r _ANTH_CLIENTreturncCs@tdurtStdurtdtd}|stdt|datS)Nz7anthropic package not available (pip install anthropic)ANTHROPIC_API_KEYzANTHROPIC_API_KEY is not setapi_key)rr RuntimeErrorosgetenvrr/var/www/html/llm_ocr_claude.py _get_client1s  rscCsz|pd}|s tdzt|WStyYnw|d}|d}|dkr9||kr9t|||dStd)NzEmpty model output{}rzModel output is not valid JSON)strip ValueErrorjsonloads Exceptionfindrfind)rstartendrrr_extract_json_strictQs     r)bundlecs|dpg}g}|D]}|d}t|tr"|r"||q dddiddiddiddidgdd d fd d |D}ddd|d d idgd d S)z Enforce: { "fields": { "": {candidate_idx:int, text:string, confidence:number, normalized:string } } } We do NOT require all keys here; we fill missing keys in post-processing. fieldskeyobjecttypeintegerstringnumber candidate_idxtext confidence normalizedF)r. propertiesrequiredadditionalPropertiescsi|]}|qSrr).0kfield_value_schemarr sz(_build_output_schema..)r.r7r9)get isinstancestrr append)r* fields_plankeysfr; fields_propsrr<r_build_output_schemaus6   rGc Cs*t}tj|}t|}g}|dpgD]}|d}t|tr-|r-| |qt t dD]]}z|j j ttd|dgdd|did d }d } t|d dpUg} | D]*} t| trp| d dkro| | dpmd 7} qXt| d ddkr| t| dd pd 7} qXt| } | d} t| tsi} |D]}| |}t|tsd d dd d}|| |<z t|dd |d<Wn tyd |d<Ynw|dd }|durd nt||d<|dd}zt|}tdtd||d<Wn tyd|d<Ynw|dd }|dur d nt|}|sdn||d<qd| iWStttfyK}z|t kr?ttd|WYd}~q4td||d}~wty~}z't|dd}|dvrr|t krrttd|WYd}~q4td||d}~wty}ztd||d}~wwdS)Nr+r,ruser)rolecontentformat json_schema)r.schemar)model max_tokensmessages output_config temperaturerrJr.r4gr2r3r5g?r6zllm_ocr retryable failure: status_code)iizllm_ocr APIStatusError: zllm_ocr non-retryable failure: ) rkyc KYCEngine#anthropic_content_items_from_bundlerGr?r@rAr rBrange _MAX_RETRIESrPcreate_DEFAULT_MODEL_MAX_OUTPUT_TOKENSgetattrdictr)intr$floatmaxminrrrtimesleep_RETRY_BASE_SLEEPrr )r*clientcontent_blocks output_schemarDrEr;attemptrespout_textrJblockdatar+vtcccnecoderrrllm_ocrs                    ru)rr)rr"rctypingrrrkyc_policy_engine_llm_v2rU anthropicrrrrr r$rr[r_r\rYr`rer__annotations__rrAr)rGrurrrrs*      "$&T