j/UdZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z m Z e eZdaeed <daeed <Gd d Zdaedzed <defdZy)u Granite-based analysis client for PR review. Implements the Mythos two-pass pattern: Searcher → Validator. Uses IBM watsonx.ai Granite model instead of Bob Shell for simplified setup. N)Any)get_watsonx_client)settings) get_logger) FindingRawFindingValidated_last_searcher_response_last_searcher_errorc eZdZdZdZ ddededzdedefdZgd Zd ed ed ed ede ee e ff dZ d edede eefde e fdZd ede e de efdZy) BobClientu` Analysis client using Granite (watsonx.ai). Implements the Mythos pattern with Granite: 1. Searcher pass: Analyzes diff → produces findings_raw.json 2. Validator pass: Receives diff + findings_raw → filters noise → findings_validated.json Note: Named "BobClient" for compatibility, but uses Granite internally. ct|_tj|_d|_t jdd|jy)Ngr@analysis_client_initializedgranite)modemodel)rwatsonxrgranite_model_idmodel_idtimeoutloggerinfo)selfs 7/var/www/html/prparty/backend/app/clients/bob_client.py__init__zBobClient.__init__"s7)+  11   1  WNprompt system_prompt max_tokensreturnc jKtjdt||r t|nd|tj} |}|r|d|}|jj |j ||dddd {}tj|z }tjd t|t|d  |S7H#t$r\}tj|z }tjd t|t|jt|d d }~wwxYww)a  Call Granite model for analysis. Args: prompt: User prompt system_prompt: Optional system prompt max_tokens: Maximum tokens to generate Returns: Granite's response text granite_requestr) prompt_lengthsystem_prompt_lengthr g333333?g?2)rrmax_new_tokens temperaturetop_ptop_kNgranite_success)response_lengthelapsed_seconds granite_error)error error_typer.) rrlentimer generate_textrround Exceptionr0strtype__name__) rrrr start_time full_promptresponseelapsedes r _call_granitezBobClient._call_granite)s-"  f+7D]!3!!  YY[ "  K!.tF8< "\\77") 8HiikJ.G KK! #H %gq 1   O#& iikJ.G LL!f7++ %gq 1    s>AD39C C AC D3 C D0AD++D00D3))zcharacter_aegis.mdAsecurity)zcharacter_schema.mdSCdatabase)zcharacter_pixel.mdPux)zcharacter_atlas.mdAT architecture)zcharacter_echo.mdEtests)zcharacter_codex.mdC documentation prompt_filediff id_prefixforced_categorycVKddl}ddlm}|tjjjj}t |dz |z dd5}|j } ddd d|d z} |j| d d{} g} | j} d | vr:| jd d z}| jd|}| ||j} n=d| vr9| jddz}| jd|}| ||j} | jds'|jd| }|r| |jd} | jds'|jd| }|r| d|j} tj | }|j#dg} t%| dD]\}}||d|d<||d< | | fS#1swY{xYw7_#t&$r.}t(j+d|t-|Yd}~| | fSd}~wwxYww)zNRun one character-specific searcher and return (raw_response, findings_dicts).rNPathpromptsrutf-8encodingr%z Return JSON now:ir```json```{z\{}z \}(?=[^}]*$)findings03didcategorycharacter_searcher_parse_failed)rLr0)repathlibrR__file__parentopenreadr?stripfind startswithsearchstartendswithendjsonloadsget enumerater6rwarningr7)rrLrMrNrO_rerR project_rootfprompt_templaterr< raw_findingsjson_strsr>mdatais r_run_character_searcherz!BobClient._run_character_searcherqs@  H~,,33::AA ,*[8# P 'TUffhO '!T$/C#DD++Ft+DD#%  B~~'HH$MM),q0MM%+#Aa=..0("MM%(1,MM%+#Aa=..0&&s+JJuh/' 3H$$S)JJ9'1H::h'D88J3L",2 01&K#w/$ /*  0%%O ' 'E> B NN<'2#a&  B B%%  BsUAH)G&'H) G,H)EG/H)G)$ H)/ H&8!H!H)!H&&H) package_json context_filesc Kddl}tjdt|t||j|j Dcgc]\}}}|j ||||c}}}d{}djdt|j |Da da |D cgc]\} } | D]} | } } } } dd d d d d dddd dtdtffd } g} | D]#} | jtdi| |%|j D cic]\} } t#fd| D}} }tjddt| i|| Scc}}}w7cc} } } w#t$r4}tjdt!|t!|Yd}~d}~wwxYwcc}} ww)u Searcher Pass: run one specialised prompt per character in parallel. Findings arrive pre-classified — no separate classifier step needed. rNsearcher_pass_start) diff_lengthcontext_files_countr%c3K|]D\\}}}\}}d|jddjddjd|Fyw)z=== character_r z.mdz === N)replaceupper).0pf_resps r z*BobClient.searcher_pass..sW. % QIT12::lB/77rBHHJK6RVQW X. sA A r Z< )criticalhighmediumlowcrit_hithitgrazewhisperryr c|jdd}|jdd|jdd|jdxs|jdd||jd xsj|d |jd xsj|d |jd xs|jdd|jdd|jdd|jdd|jddd S)NseverityrrbF000titler explanation descriptiondamager damage_typer file_pathfileunknown line_startr`line_end code_snippetrcgeneral) rbrrrrrrrrrrc)rt)ryr damage_mapdamage_type_maps r normalizez*BobClient.searcher_pass..normalizesuuZ/HeeD&)w+ uu]3Oquu]B7O$%%/IZ^^Hb-I uu]3_7J7J8U^7_UU;/K1553KeeL!4EE*a0 !nb 9EE*i8  rnormalize_failed)r0findingc3BK|]}|jk(sdyw)r`N)rc)rrycats rrz*BobClient.searcher_pass..s>1AJJ#,=Q>sfindings_count)searcher_pass_complete)asynciorrr2gather_CHARACTER_PROMPTSrjoinzipr r dictappendrr6rvr7sum)rrMrr_asynciorLrNrcresultsrr_rall_rawrryr>rcountsrrs ` @@r searcher_passzBobClient.searcher_passs # !D #M 2  (484K4K* * 0 Y  ( (dIx P*   #)++. ),T-D-Dg)N. #  ".5NN{q(XN'7N7NN"$bBrJ '15G\ef  $  &( QA Q :Yq\ :; Q"44  1c >>> >    US]UfU_*  O2 Q1QQPP Q sfA F>E( / F>8E/98F>1E1,F>1 E8F># F8-F>1F>8 F5*F0+F>0F55 F> findings_rawc Ktjdt|ddlm}|t j j j j }|dz dz }t|dd 5}|j}d d d tj|Dcgc]}|jc}d }jd |} | jd|} |j| dd {} d| vr:| jddz} | jd| } | | | j} n| j} tj | }|j#dgDcic]}t%|t&r d|vr|d|}}g}|D]}|j#|j(}||j#dds3|xsij#dxs |j*}|xsij#dd}|j}||d<||d<|j-t/d!i|tjdt|t|t|z |S#1swYxYwcc}w7cc}w#t0$re}tj3dt5|t||Dcgc]}t/d!i|jdd i!ncc}wc}cYd }~Sd }~wwxYww)"a Validator Pass: Filter noise from raw findings using Granite. Args: diff: Original git diff findings_raw: Raw findings from searcher pass Returns: List of validated findings (noise filtered out) validator_pass_start)findings_raw_countrrQrSzbob_validator.mdrTrUrVNr,)indentz{{DIFF}}z{{FINDINGS_RAW}}i rXrYrZr[r_rb validatedTseverity_adjustmentvalidation_notesr rvalidator_pass_complete)findings_validated_countfiltered_count)validator_parse_failed_using_raw_findings)r0 raw_countzvalidator skippedr)rrr2rfrRrgrhrirjrrdumps model_dumprr?rlrkrsrt isinstancerrbrrrr6rvr7)rrMrrRrxvalidator_pathryrz findings_jsonrr< json_startjson_endr|validated_datavvalidation_by_idr_rawdecisionrnotesrr>s rvalidator_passzBobClient.validator_passs  ""<0  !H~,,33::AA % 14FF .# 8 'AffhO ' %1 2Q\\^ 2 !((T: 2MB++Ft+DD- H$%]]959 #== ;#Jx8>>@#>>+!ZZ1N(++J; a&419$    02H# >+//7#x||K'F (B334IJZcllH%^001CRHE>>+D'/D$/4D+,OO$4$ KK)),X"<03x=@   Og ' ' 3E 4  NN;!fl+  &!X1<<>XDWX  sA1L3I7LJ6AL9J :L?A7J6!J 8JB&J6L7J<L J K?.K:$K-,K:4K?5L:K??L)Ni)r9 __module__ __qualname____doc__rr7intr?rtuplelistrrrrrrrrrr r sX%) <<Tz< < <~3&3&3& 3&  3& sDJ  3&jBBBCH~ B j  BHXX:&X   Xrr _bob_clientr c.t tatS)z=Get or create global analysis client instance (uses Granite).)rr rrrget_bob_clientrIsk r)rrrr3typingrapp.clients.watsonx_clientr app.configrapp.logging_configr app.modelsrrr9rr r7__annotations__r r rrrrrrsm  9)3 H  "!ckk^ !% Y $ r