jO2UdZddlZddlmZddlmZmZddlmZddlm Z ddl m Z m Z m Z mZmZddlmZdd lmZmZmZmZdd lmZe eZgd Zeeeeeefed <d ddddZ dddddZ!de"dededzfdZ#GddZ$da%e$dzed<de$fdZ&y)uk PR analysis orchestrator - coordinates the entire Mythos pipeline. Flow: 1. Load PR diff + context 2. Bob Searcher Pass → findings_raw 3. Bob Validator Pass → findings_validated 4. Classifier → assign characters 5. Voice Rewriter (Granite) → add character_dialogue 6. Calculate damage & verdict 7. Generate character dialogues 8. Return EncounterResult N)datetime)AnyAsyncGenerator)get_bob_client) get_logger)CharacterDialogueEncounterResultFindingFindingValidatedVerdict)get_classifier) DiffIndexfind_snippet_line is_grounded parse_diff)get_voice_rewriter))zcharacter_aegis.mdAsecurityaegis)zcharacter_schema.mdSCdatabaseschema)zcharacter_pixel.mdPuxpixel)zcharacter_atlas.mdAT architectureatlas)zcharacter_echo.mdEtestsecho)zcharacter_codex.mdC documentationcodexCHARACTER_PIPELINEZ< )criticalhighmediumlowcrit_hithitgrazewhisperfcategoryreturnc |jdd}t|jdd|jdd|jdxs|jdd||jd xstj|d |jd xstj|d |jd xs|jdd|jdd|jdd|jdd|d S#t$r4}t j dt|t|Yd}~yd}~wwxYw)zOConvert a raw searcher dict into a FindingValidated, returning None on failure.severityr-idF000title explanation descriptiondamager) damage_typer1 file_pathfileunknown line_startline_end code_snippet) r7r9r<r6r=r>r?rBrDrEr3validation_notesnormalize_validated_failed)errorfindingN)getr _SEVERITY_DAMAGE_SEVERITY_DAMAGE_TYPE Exceptionloggerwarningstr)r2r3r6excs :/var/www/html/prparty/backend/app/services/orchestrator.py_normalize_raw_to_validatedrS=s55U+uuT6"%%$m,HmR0H55?H&6&:&:8R&Hm$B(,,XyAeeK(DAEE&),Duu\1-UU:q)~r2  " 33s8SQRVTsD D E *EE ceZdZdZdZdedefdZdee dee fdZ ded e d e d e d e d e e e fdefdZde de de de d e dedee ee e dzffdZded e d e d e d e d e e e fdee dffdZy)PRAnalysisOrchestratorz/Orchestrates the complete PR analysis pipeline.ct|_t|_t |_t jdy)Norchestrator_initialized)rbobr classifierrvoice_rewriterrNinfo)selfs rR__init__zPRAnalysisOrchestrator.__init__\s.!#(*02 ./ total_damager4c|dk\ry|dk\ryy)u Calculate PR verdict based on total damage. Rules: - ≥80 damage: blocked - 50-79 damage: changes_required - <50 damage: approved Pblocked2changes_requiredapproved)r\r_s rR_calculate_verdictz)PRAnalysisOrchestrator._calculate_verdictbs 2  R %r^findingsc Vg}i}|D]=}|j|vrg||j<||jj|?|jDcgc]\}}t|dk\s|}}}t|dk\r}|d}|d} ||d} || d} t | j |d| j jd| d| j jd} |j| tjd t| |Scc}}w) a Generate character dialogues based on finding patterns. For demo, we'll create dialogues when: - Multiple findings from same character (self-reflection) - Related findings from different characters (collaboration) rrCzI've discovered z$. This requires immediate attention!zIndeed! And I've found z+ as well. These issues compound each other.) finding_id character_1 dialogue_1 character_2 dialogue_2dialogues_generated)dialogue_count) character_idappenditemslenrr7r9lowerrNr[) r\rh dialogues by_characterrIcharfindscharacters_with_multiplechar1char2finding1finding2dialogues rR_generate_dialoguesz*PRAnalysisOrchestrator._generate_dialoguesrsQ.0 24  ?G##<757 W112 -- . 5 5g > ?%1$6$6$8$  T5CJ!OD$  $  ' (A -,Q/E,Q/E#E*1-H#E*1-H(#;;!-hnn.B.B.D-EEij!4X^^5I5I5K4LLwx H   X & !y>  3$ s D%1D% pr_numberpr_title pr_authordiff package_json context_filesc Ktjd|t|t|tjd|jj |||d{}|sItjdt |||ddd ggt jj Stjd t| |jj|| d{}|sItjdt |||ddd ggt jj Stjdt||jj|} tjd|jj| d{} td| D} tdd | z } |j!| } tjd| | | tjd|j#| }t |||| | | | |t jj }tjd|| t| | |S777ԭw)a Analyze a PR through the complete pipeline. Args: pr_number: GitHub PR number pr_title: PR title pr_author: PR author username diff: Git diff content package_json: package.json content context_files: Dict of filename -> content Returns: Complete encounter result with voiced findings pr_analysis_start)r diff_lengthcontext_files_countstep_1_searcher_pass)rrrNno_findings_detectedrerd rrrverdictr_ remaining_hprhrwanalysis_timestampstep_2_validator_pass)findings_raw_count)r findings_rawall_findings_filtered_outstep_3_classify_findings)findings_validated_countstep_4_voice_rewritec34K|]}|jywNr=.0r2s rR z4PRAnalysisOrchestrator.analyze_pr..s=188=damage_calculated)r_rrstep_5_generate_dialoguespr_analysis_completerrfindings_countr_)rNr[rurX searcher_passr rutcnow isoformatvalidator_passrYclassify_findingsrZrewrite_findingssummaxrgr)r\rrrrrrrfindings_validatedclassified_findingsvoiced_findingsr_rrrwresults rR analyze_prz!PRAnalysisOrchestrator.analyze_prst.  D #M 2    *+!XX33%'4   KK. /"#!#" #+??#4#>#>#@    +L@Q R#'88#:#:%$;$   " KK3 4"#!#" #+??#4#>#>#@    .M_I` a"oo??@RS  *+ $ 3 3 D DEX YY=_== 1cL01 )),7 %%    /0,,_= !%%$'0::<    "/%   o . 2Zs9A"J $J%BJ 4J5B=J 2J3CJ J J  prompt_file id_prefixr3rr diff_indexNcK jj||||d{\}}g} g} |D]} t| |} | t| j| j | j | j|\} }| s| j|at| j| j|| j }|J|| j k7r;td| j | j z }| j|||zd} | j| | r/tjdt| t| | dd | stjd  gdfStj fd | Dd did{}g}t#| |D]\}}t%|t&r|j)r|n |j*}|jt-|j.|j0|j*|j2|j4|j6|j|j |j |j| tjdt||dfS7e7#t8$rM}tj;dt'|t=|j>gt'|fcYd}~Sd}~wwxYww)a Run searcher + voice rewriter for ONE character end-to-end. Returns (character_id, finished_findings, error_message_or_None). Voice rewrites for that character's findings are awaited in parallel so the per-character pipeline finishes as fast as its slowest finding. Skips the LLM validator pass (which would force us to wait for ALL searchers before any character can finish, breaking the streaming UX) and replaces it with a deterministic grounding filter that drops any finding whose file/line/snippet does not exist in `diff_index`. That kills hallucinations like `frontend/UserDashboard.tsx:123` without adding a single extra LLM call. N)r?rBrDrEindex)snippetr@r hint_liner)rBrD)updatecharacter_grounding_filterr))rrkeptdroppedreasonscharacter_pipeline_no_findings)rrc3VK|] }jj|"ywr)rZrewrite_finding)rfvrrr\s rRrzHPRAnalysisOrchestrator._run_single_character_pipeline..ks,''77 bIs&)return_exceptionsT) r7r9r<r6r=r>r?rBrDrErrcharacter_dialoguecharacter_pipeline_complete)rrrcharacter_pipeline_failed)rrrH error_type) rX_run_character_searcherrSrr?rBrDrErsrr model_copyrNr[ruasynciogatherzip isinstancerPstripr<r r7r9r6r=r>rMrHtype__name__)r\rrr3rrrr_ raw_findings validatedrraw normalizedgroundedreason real_linewidthdialogue_resultsfinishedrrvoicedrQs` ` rR_run_single_character_pipelinez5PRAnalysisOrchestrator._run_single_character_pipelines,j .$(HH$D$DT9h%OA|13I!#G#" -8hG %#.(22)44'00!+!8!8$ $ & NN6* .&33#--$(33  (Y*:O:O-O:#6#69N9N#NOE!+!6!6*3(1E(9 "7"J  ,E" -H 0!-YL#CRL  4!-$R--&-^^'& #' &  ')H #I/? @  H"(C0X^^5E 55 hh$&NN!#!yy$&NN"$,,#%==!#%'__%1+1   . KK-)"8}    4/ /Av N . LL+)#h9--    SX- - .seK*!JJ EJ0K*1"JJC7J K* JJ K'AK"K'K*"K''K*cKtjd|t|t|}tjdt|jt d|j jDtDcgc]}|d } }d|||| dd | D] } d d | id tD cgc]1\} } } }tj|j| | | ||| 3}} } } }g} tj|D]V}|d {\}}}| d||dd |j|d||Dcgc]}|jc}dd X t d|D}t#dd|z }|j%|}t'||||||||j)|t+j,j/ }tjd||t||d|jd y cc}wcc}} } } w7cc}w#tj$r|D]}|j!wxYww)u Run the per-character pipeline and yield events as each character finishes. Event shape (each is a dict with `event` + `data`): - {"event": "started", "data": {pr_number, pr_title, pr_author, characters: [...]}} - {"event": "character_started", "data": {"character_id": id}} - {"event": "character_complete", "data": {"character_id": id, "findings": [...]}} - {"event": "character_error", "data": {"character_id": id, "error": str}} - {"event": "complete", "data": EncounterResult.model_dump()} Failures in one character do NOT abort the others — they emit `character_error` and the remaining characters continue. stream_analysis_start)rr diff_indexedc32K|]}t|ywr)ru)rvs rRrz>PRAnalysisOrchestrator.analyze_pr_streaming..sLqCFLs)files added_linesstarted)rrr characters)eventdatacharacter_startedrr)rrr3rrrrNcharacter_error)rrrHcharacter_complete)rrrhc34K|]}|jywrrrs rRrz>PRAnalysisOrchestrator.analyze_pr_streaming..s:188:rrrrstream_analysis_completercomplete)rNr[rurrrrvaluesr%r create_taskr as_completedextend model_dumpCancelledErrorcancelrrgr rrrr)r\rrrrrrrcall_character_idschar_idrrr3rrtasks all_findingscororhrHr2tr_rrrs rRanalyze_pr_streamingz+PRAnalysisOrchestrator.analyze_pr_streamings,  #D    %  j&&'LJ,B,B,I,I,KLL  ,>>aQqT>>&$&/   ) G,'1  (EW   Ai<   33 +'%!-) 4     ')  ,,U3 6: - h$!2,8%*!##H-1(4=E$FQ\\^$F 6:\:: 1cL01 )),7 %%!..|<'0::<    &|,%  #F,=,=,?@@w?.  $1; %G%%       s[BI! H+I!6H" 7I!?H2H*-H2 H- ! H2-B=I!*H2-H22,II!)r __module__ __qualname____doc__r]intr rglistr rrrPdictr rrtuplerrrrfr^rRrUrUYse90 sw +DM+dCT>U+Zwww w  w  wCH~w wr@.@.@. @.  @.  @.@. sDM3:- .@.DAAAAAA AA  AA  AACH~AA d #AAr^rU _orchestratorc.t tatS)z+Get or create global orchestrator instance.)rrUrfr^rRget_orchestratorr"s.0 r^)'rrrtypingrrapp.clients.bob_clientrapp.logging_configr app.modelsrr r r r app.services.classifierr app.services.diff_groundingrrrrapp.services.voice_rewriterrrrNr%rrrP__annotations__rKrLrrSrUrrrfr^rRrs &1)3 ; H  7DsCc123!#B"RH    8BABAL04 %,30r^