eKGj1dZddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl Z ddl Z ddlmZej eZde de fd Zd e de fd ZGd d Zy)z Brain module for Kimi Copilot. Maintains session context, decides when to invoke the side model, calls the side model, and writes valuable corrections to RAG. N)datetime)Path)Any) HintPanelvaluereturnczt|ts|Stjd}d}|j ||S)z5Replace ${VAR:default} placeholders in config values.z\$\{([^}:]+)(?::([^}]*))?\}c|jd}|jdxsd}tjj||S)N)grouposenvironget)matchvar_namedefaults 0/Users/admin/workspace/jyq/kimi-copilot/brain.pyreplz_resolve_env..repl"s6;;q>++a.&Bzz~~h00) isinstancestrrecompilesub)rpatternrs r _resolve_envrs8 eS ! jj78G1 ;;tU ##robjct|tr-|jDcic]\}}|t|c}}St|tr|Dcgc] }t|c}St |Scc}}wcc}wN)rdictitems_deep_resolve_envlistr)rkvitems rr$r$*sg#t47IIK@DAq$Q''@@#t478D!$'88  A8s A2A8c eZdZ dPdedefdZdedefdZdZdefd Z e d edefd Z e d ed e ede fd Ze d ed e ede fdZdQdede fdZdRdede fdZd efdZd efdZd efdZdZd edefdZde efdZde efdZdZdedzfdZdZdZd efdZd ededzfdZd ede efd Z d efd!Z!d"Z"d#Z#de efd$Z$d%efd&Z%d'Z&d(efd)Z'd%efd*Z(d ededzfd+Z)d,ed-efd.Z*d,ed/e efd0Z+d(efd1Z,d2e ede efd3Z-e dSd4ed5ed6ede fd7Z.dTd8ed9e/de fd:Z0d8efd;Z1d4ed5ede fd<Z2d=ede fd>Z3d?ede efd@Z4e dAe5de efdBZ6de7efdCZ8dDe ede efdEZ9dFede efdGZ:d=edHe edIe ede efdJZ;d=edHe ededzfdKZy)UBrainN config_pathpanelc|j||_g|_d|_d|_d|_d|_|xs t|_|xsd|_ |j|_ |jr tjd|jg|_t|_d|_d|_d|_d|_d|_d|_d|_d|_d|_|j5y)Nr rct|dS)Nr )end)print)msgs rz Brain.__init__..Bs uSb?QrzDetected project: %sF) _load_configconfighistory pending_inputpending_output round_countlast_trigger_roundrr,notify_callback_detect_projectprojectloggerinfo recent_issuessetstuck_alerts_shownlast_trigger_timelast_scan_timefrustration_scorelast_frustration_timepending_frustration_summaryfrustration_trigger_timelast_frustration_summary_timelast_frustration_check_timefrustration_notified_setup_logging)selfr+r,r;s r__init__zBrain.__init__3s '' 4 #%  "#)ik  /R3Q++- << KK. =*,,/E!$!"#%("+0((+%-0*+.($)! rpathrct|dd5}tj|}dddt|S#1swYtSxYw)Nrutf-8encoding)openyaml safe_loadr$)rMrOfraws rr4zBrain._load_config]sD $g . $!..#C $ %% $ %%s 8A c|jjdijdd}|jjdijdd}tj}|jrytj |d}|j tjd|jtt|jtj|j|y) NlogginglevelINFOfilezkimi-copilot.logrRrSz'%(asctime)s [%(levelname)s] %(message)s) r5rr[ getLoggerhandlers FileHandler setFormatter FormattersetLevelgetattrupperr] addHandler)rMr\log_pathroothandlers rrLzBrain._setup_loggingbs  2.227FC;;??9b155f>PQ  " == %%hA   G H  ggu{{}gllCD  rcftjjddj}|r|S|jjddj}|r|Stj j dd}|jdDcgc]}|s| }}|r|dSycc}w)zDetermine the current project for scoping knowledge. Priority: 1. COPILOT_PROJECT environment variable 2. config.yaml ``project`` field 3. The current working directory name COPILOT_PROJECTr r=\/)rrrstripr5getcwdreplacesplit)rMr=cwdppartss rr<zBrain._detect_projectqs**..!2B7==? N++//)R0668 Niik!!$,IIcN0qa00 9 1s B. B.textcPtjd}|jd|S)zERemove ANSI escape sequences including CSI and OSC (e.g. hyperlinks).zC\x1B(?:\][^\x07\x1b]*(?:\x07|\x1b\\)|[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])r )rrr)rw ansi_escapes r strip_ansizBrain.strip_ansis*jj  r4((rpatternscZ|j}|D]}|j|vsyy)NTFlower)rwr{loweredrs r _matches_anyzBrain._matches_anys2**, G}}') rc|j}|D]^}|j}d} |j||}|dk(r.|dkDr+||dz }|js|dvr|t|z}Iyy)uMatch patterns at a word boundary on the left side. This avoids firing on embedded substrings like "可能性" while still allowing Chinese phrases such as "我可能错了" to trigger. rTror z_-F)r~findisalnumlen)rwr{rrneedlestartidxprevs r_matches_word_boundaryzBrain._matches_word_boundarys**, G]]_FEll651"97"37+D||~ #c&k 1  rcooldown_secondsc|tjj}||jz |kry||_y)z+Throttle side-model triggers to avoid spam.FT)rnow timestamprCrMrrs r_should_triggerzBrain._should_triggers8lln&&( '' '*: :!$rc|tjj}||jz |kry||_y)z:Throttle real-time scans to avoid spamming the side model.FT)rrrrDrs r _should_scanzBrain._should_scans8lln&&( $$ $'7 7!rc|j|}|xj|z c_d|jvsd|jvr|jj}|rhtj d||j j d|d|j|j||j|dd|_yy) z*Feed a chunk of user input into the brain.  zUser input: %ruserrolecontentsourcer N) rzr7rpr>debugr6append_prune_history_check_frustration_check_trigger)rMrwcleanlines rfeed_user_inputzBrain.feed_user_inputs% e# 4%% %1C1C)C%%++-D -t4 ##V$EF##%''-##D#8!#D *Drc|j|}|xj|z c_d|jvsd|jvrtjd|j}|d|_|ddD]}}|j }|st j d||jjd|d|j|j||j|d yy) z+Feed a chunk of Kimi output into the brain.rrz[\r\n]+roNzKimi output: %rkimirr) rzr8rrsrpr>rr6rr_check_resolutionr)rMrwrlinesrs rfeed_kimi_outputzBrain.feed_kimi_outputs% u$ 4&& &$$2E2E*EHHZ)<)<=E"')D cr  =zz|LL!2D9LL''D(IJ'')**40''V'< =+Frc|j|j}|rXtjd||jj d|d|j |j|dyy)z:Feed explicit tool/command output (e.g. failure messages).zTool output: %rtoolrrN)rzrpr>rr6rrr)rMrwrs rfeed_tool_outputzBrain.feed_tool_outputsf%++-  LL*E 2 LL  E B C    !   f  5 rcR|jjdijdd}td|jD}||kDrZ|jrM|jj d}|t |jddz}||kDr|jrKyyyy) z?Keep history bounded so memory and request size do not explode.copilotmax_history_charsi@ c3RK|]}t|jdd!ywrr Nrr.0ms r z'Brain._prune_history..s D!Ci,-D%'rrr N)r5rsumr6popr)rM max_charstotalremoveds rrzBrain._prune_historysKKOOIr2667JFS Dt||DDiDLLll&&q)G SY34 4EiDLLLrrc>|jjdijdi}|jjdijdd}|jddr{|dk(rv|jjdijdg}|j||r8|j|r&tj d ||j y |jd drz|d vrv|jjdijd g}|j||r8|j|r&tj d||j y |jddr{|dk(rv|jjdijdg}|j||r8|j|r&tj d||j y |jdd}|dkDrx|dk(rs|xjdz c_|j|jz |k\rB|j|r0tj d|j|j y |jy )z(Decide whether to invoke the side model.rtriggerstrigger_cooldown_seconds@ user_invokeTrinvoke_keywordszTriggered by user invoke: %rNcommand_failure)rrfailure_patternsz Triggered by failure pattern: %runcertainty_keywordsruncertainty_patternsz$Triggered by uncertainty keyword: %rperiodic_roundsrr z"Triggered periodically at round %d) r5rrrr>r?_invoke_copilotrr9r:_scan_and_trigger) rMrwrrcooldownrrrperiodics rrzBrain._check_triggers:;;??9b155j"E;;??9b1556PRUV << t ,61A"kkooi<@@ARTVWO  7''1KK >E((* <<)4 0V?O5O#{{y"=AABTVXY   '78''1KK BDI((* <<. 5&F:J#';;??9b#A#E#EF\^`#a **41EF''1KK FM((*<< 115 a>+ &k39 *Q!" * *CN%.ABkkooi4889LcR 8;DLL8IK8W{lm4]a]i]i! GCv;Dv~" OOTc)nE F  G * *s DDc\|jjdijdd}tjj |r(t |dd5}|j }dddnd}d d g}|jjd ijd i}|jd d}t|j|kDr|j| dn |j}|D]-}|d} | dk(rd} n| dk(rd} |j| |dd /|S#1swYxYw)z9Build a short context for the lightweight real-time scan.rscanzprompts/copilot_scan.txtrQrRrSNzqYou are a helpful coding assistant. Review the conversation and output JSON with has_issue and risk_score (0-10).rrrreal_time_scanrrrrrrrr) rM scan_pathrXrrscan_cfgrrr1rs r_build_scan_messageszBrain._build_scan_messages5s-KKOOIr266v?YZ 77>>) $iw7 *1!" * *`  &.AB;;??9b1556FKll#6; 8;DLL8IK8W{lm4]a]i]i! GCv;Dv~" OOTc)nE F  G' * *s D""D+c |jsy|j|_|j}|jj dij dd}|jj dij dd}t d|D}t||kDr!tjdt||y||kDrtjd ||y|jj d i}|j d ||jj dij d d |jj dij ddd}tjdt| tj|j ddd|j ddd||jj dij dd}|js:tjd|j|j |j"ddy|j% |j)} | dddd } |j1| y#tj&$r }tjd|Yd}~yd}~wwxYw#t*t,t(j.f$r }tjd!|Yd}~yd}~wwxYw)"z*Call the side model and handle the result.Nrmax_messages_hardxmax_request_charsc3RK|]}t|jdd!ywrrrs rrz(Brain._invoke_copilot..Z G3quuY34Grz5Skipping side model call: too many messages (%d > %d)z;Skipping side model call: request too large (%d > %d chars)modelname temperature皙? max_tokensrrrrz#Calling side model with %d messagesbase_urlapplication/jsonBearer api_keyr z Content-Type Authorizationtimeoutheadersjsonrz%Side model request failed: %s %s - %szSide model request failed: %schoicesrmessagerz'Failed to parse side model response: %s)r6r9r:rr5rrrr>warningr?requestspostokerror status_codereasonrwraise_for_statusRequestExceptionrKeyError IndexErrorJSONDecodeError_handle_copilot_result rMr max_messagesr request_sizecfgpayloadrespedatars rrzBrain._invoke_copilotNs|| "&"2"2''){{y"599:MsS  KKOOIr:>>?RTZ[GhGG x=< ' NNGH    + + NNM!  kkoogr*WWV_ ;;??9b9==mSQ++//)R8<<\4P    93x=I == #$6'.swwy"/E.F%G  26::9cJD77 ;$$KKIIdsO    ! ! #  99;D9oa(3I>G ##G,((  LL8! <  *d&:&:;  LLBA F  s7 B-I:9I: J0:J- J((J-0K.K))K.c |jsy|j}|jjdijdd}|jjdijdd}t d|D}t ||kDr!t jdt ||y||kDrt jd ||y|jjd i}|jjdijd i}|jd ||jjdijd d|jddd}t jdt | tj|jddd|jddd||jdd}|js:t jd|j|j|jddy|j! |j%} | dd d!d"} |j-| } | st jd$| dd%y| S#tj"$r } t jd| Yd} ~ yd} ~ wwxYw#t&t(t$j*f$r } t jd#| Yd} ~ yd} ~ wwxYw)&z0Call the side model for a lightweight risk scan.Nrrrrrc3RK|]}t|jdd!ywrrrs rrz%Brain._invoke_scan..rrz4Skipping real-time scan: too many messages (%d > %d)z:Skipping real-time scan: request too large (%d > %d chars)rrrrrrirz6Calling side model for real-time scan with %d messagesrrrrr rrrz)Real-time scan request failed: %s %s - %srz!Real-time scan request failed: %srrrrz!Failed to parse scan response: %sz"Scan did not return valid JSON: %r)r6rr5rrrr>rr?rrrrrrrwrrrrrr _extract_json) rMrr rr r rr rrrrresults r _invoke_scanzBrain._invoke_scans||,,.{{y"599:MsS  KKOOIr:>>?RTZ[GhGG x=< ' NNFH    + + NNL!  kkoogr*;;??9b1556FKWWV_ ;;??9b9==mSQ",,|S9    LcRZm\ == #$6'.swwy"/E.F%G Y3D77 ?$$KKIIdsO    ! ! #  99;D9oa(3I>G ##G, NN?# O !((  LL I>1J4>J1J,,J14K2K--K2c|jjdijdi}|jddsy|jdd}|j|sy|j}|sy|jdd}|jd d }|jd d }tj d ||||s||krytj d|j y)zGLightweight qwen scan after each turn; triggers deep analysis if risky.rrenabledFNr@ has_issue risk_scorer thresholdz6Scan result: has_issue=%s, risk_score=%s, threshold=%szScan triggered deep analysis)r5rrrr>r?r)rMrrrrrrs rrzBrain._scan_and_triggers;;??9b1556FK||Iu- << 2C8  * ""$ JJ{E2 ZZ a0 LLa0  D     Z)3  23 rcd|jjdijdi}|jdd}tjj }||j z |kDr>|j dkDr tjd|j d|_d|_ yy) zHReset frustration score if too much time has passed without new signals.rfrustration_detection decay_secondsi,rz&Frustration score decayed from %d to 0FN) r5rrrrrFrEr>rrK)rMr decayrs r_decay_frustrationzBrain._decay_frustrationskkooi,001H"M-lln&&( ++ +e 3%%) EtG]G]^%&D "(-D % 4rc|jjdijdi}|jddsy|jtjj }|jdd}||j z |kry||_|j|}|sy|jdd }t|ttfr|d kry|xjt|z c_ ||_ tjd ||j|jd d }|j|k\r3d |_||_|j"sd |_|j%yyy)zRUse qwen to semantically detect user frustration and mark for later summarization.rr rFNrg@scorerz*Frustration check score=%s, accumulated=%drT)r5rr#rrrrJ_invoke_frustration_checkrintfloatrErFr>r?rGrHrK_notify_frustration_detected)rMrwr rrrr%rs rrzBrain._check_frustrationsPkkooi,001H"Mwwy%(  !lln&&(77-s3 11 1H < +.(//5  7A&%#u.%1*  #e*,%(" 8   " " GGK+  ! !Y ./3D ,,/D ),,,0)113- /rc $|j|}|jjdijdd}|jjdijdd}td|D}t ||kDs||kDry|jjdijdi}|jjd i}|jd ||jjdijd d |jd dd} t j |jddd|jddd||jdd} | js:tjd| j| j| jddy| j | j} | dddd} |j'| } | stj)d!| dd"y| S#t j$r } tjd| Yd} ~ yd} ~ wwxYw#t t"tj$f$r } tjd | Yd} ~ yd} ~ wwxYw)#z6Call qwen to judge whether the user sounds frustrated.rrrrrc3RK|]}t|jdd!ywrrrs rrz2Brain._invoke_frustration_check..2rrNr rrrrrrrrrrr rrrz,Frustration check request failed: %s %s - %srz$Frustration check request failed: %srrrrz.Failed to parse frustration check response: %sz/Frustration check did not return valid JSON: %rr)!_build_frustration_check_messagesr5rrrrrrr>rrrrwrrrrrrrr)rMrwrr rr r  model_cfgr rrrrrs rr'zBrain._invoke_frustration_check,sD99$?{{y"599:MsS  KKOOIr:>>?RTZ[GhGG x=< '<:K+Kkkooi,001H"MKKOOGR0 ]]6* ;;??9b9==mSQ'',4   == j)$6'.y}}Y/K.L%M 2.D77 B$$KKIIdsO    ! ! #  99;D9oa(3I>G ##G, NNLgVZWZm \ ((  LL? C *d&:&:;  LLI1 M s7)BH=HII.I  IJ/J  Jc |jjdijdd}tjj |r(t |dd5}|j }dddnd}d d g}t|jd kDr|jd dn |j}|D]-}|d }|dk(rd}n|dk(rd}|j||dd /|jdd|d |S#1swYxYw)zBuild a short prompt asking qwen to judge user frustration. Includes the most recent conversation turns so qwen can understand context and distinguish a normal correction from real frustration. r frustrationzprompts/copilot_frustration.txtrQrRrSNzzJudge whether the user input shows frustration or dissatisfaction. Output JSON with is_frustrated (bool) and score (0-10).rrr&irrrrrru【当前输入】r) rMrw prompt_pathrXrrrr1rs rr/z'Brain._build_frustration_check_messagescs kkooi488 <  77>>+ &k39 *Q!" * *J  &.AB.1$,,.?!.Cbc*! GCv;Dv~" OOTc)nE F  G 6H4OPQ' * *s C::DcX|jsy|jjdijdi}|jddsytjj }|jdd}||j z |kDr&tjd|jy|jd d }||jz |kry|jd g}|sy|j||r&tjd |jyy) zRIf the user was frustrated and Kimi now seems to have solved it, summarize to RAG.Nrr rFsummary_window_secondsiXz-Frustration summary window expired, resettingsummary_cooldown_seconds<resolution_patternszDResolution signal detected after frustration, triggering RAG summary) rGr5rrrrrHr>r?_reset_frustration_staterIr_invoke_frustration_rag_summary)rMrwr rwindowrr{s rrzBrain._check_resolutions// kkooi,001H"Mwwy%( lln&&(137 .. . 7 KKG H  ) ) + 775r: 33 3h > 770"5    T8 , KK^ _  0 0 2 -rc<d|_d|_d|_d|_y)NrFr3)rErGrHrK)rMs rr9zBrain._reset_frustration_states"!"+0((+%$)!rc F|j}|jjdijdd}|jjdijdd}td|D}t ||kDr1t j dt |||jy||kDr(t j d |||jy|jjd i}|jd ||jjdijd d |jjdijddd}t jdt | tj|jddd|jddd||jjdijdd}|jsJt jd|j|j|jdd|jy|j! |j%} | dddd } |j-| y#tj"$r0}t jd||jYd}~yd}~wwxYw#t&t(t$j*f$r0}t jd!||jYd}~yd}~wwxYw)"zGCall the side model to summarize a frustration-resolution arc into RAG.rrrrrc3RK|]}t|jdd!ywrrrs rrz8Brain._invoke_frustration_rag_summary..rrz=Skipping frustration RAG summary: too many messages (%d > %d)NzCSkipping frustration RAG summary: request too large (%d > %d chars)rrrrrrrz?Calling side model for frustration RAG summary with %d messagesrrrrr rrrz2Frustration RAG summary request failed: %s %s - %srz*Frustration RAG summary request failed: %srrrrz4Failed to parse frustration RAG summary response: %s)_build_frustration_rag_messagesr5rrrr>rr9r?rrrrrrrwrrrrrr_handle_frustration_rag_resultr s rr:z%Brain._invoke_frustration_rag_summarys779{{y"599:MsS  KKOOIr:>>?RTZ[GhGG x=< ' NNOH    ) ) +  + + NNU!   ) ) + kkoogr*WWV_ ;;??9b9==mSQ++//)R8<<\4P    UWZ[cWde == #$6'.swwy"/E.F%G  26::9cJD77 H$$KKIIdsO  --/  ! ! #  99;D9oa(3I>G ++G4((  LLEq I  ) ) +  *d&:&:;  LLOQR S  ) ) +  s7 B=J J K K&K  KL 0&LL c|jjdijdd}tjj |r(t |dd5}|j }dddnd}d d g}|jD]-}|d }|d k(rd }n|dk(rd}|j||dd /|S#1swYMxYw)z4Build messages summarizing the full frustration arc.rfrustration_ragz#prompts/copilot_frustration_rag.txtrQrRrSNzySummarize the following coding conversation into a structured RAG entry. Output JSON with worth_rag and rag_entry fields.rrrrrrrr) r5rrrOrrUrr6r)rMr3rXrrr1rs rr?z%Brain._build_frustration_rag_messagesskkooi488 D  77>>+ &k39 *Q!" * *C  &.AB<< GCv;Dv~" OOTc)nE F  G! * *s B::Crctjj|_|j |j |}|st jd|ddy|jdst jdy|jdi}|st jdy|jdd }t jd ||j||j|y) z9Parse the side model output and write to RAG if valuable.z5Frustration RAG summary did not return valid JSON: %rNr worth_ragz1Frustration RAG summary deemed not worth storing. rag_entryz@Frustration RAG summary returned worth_rag=true but no rag_entrytitle未命名知识z*Frustration RAG summary produced entry: %s) rrrrIr9rr>rrr? _write_rag_notify_frustration_rag_write)rMrrrErFs rr@z$Brain._handle_frustration_rag_results-5\\^-E-E-G* %%'##G, NNRT[\`]`Ta b zz+& KKK L JJ{B/  NN] ^  g'89 @%H  " **51rcLgd}|jdj|y)zBLet the user know we'll auto-summarize once the issue is resolved.)r u<[Copilot] 😤 检测到你似乎对当前问题不太满意ubKimi 解决后，我会自动把这段经验提炼成知识库条目，避免下次再走弯路。 [Copilot]r rN)r;join)rMrs rr*z"Brain._notify_frustration_detecteds"  TYYu-.rrFctjjd}ddd|d|ddg}|jdj |y) z=Confirm that a frustration-resolution arc was written to RAG.%Y-%m-%d %H:%M:%Sr u6[Copilot] ✅ 已自动总结踩坑经验并写入 RAG时间: 标题: rKrNrrstrftimer;rLrMrFrrs rrIz#Brain._notify_frustration_rag_write#sYlln%%&9:  Dse ug      TYYu-.rc|j|}|stjd|ddy|jdd}|jdd}|jdd}|jd d }|jd r"|jd ijd gng}|stj d ytj d||||||d}|j j ||jj|||||jj||\} } | r(tj d||j|| |j|r|j|| |j|||jdr'|jd i} | r|j| yyy)z*Parse the side model output and act on it.z(Side model did not return valid JSON: %rNrrFsummaryr suggestionseveritymediumrEtagszSide model found no issue.zSide model found issue [%s]: %s)rUrVrYrWzRepeated mistake detected: %srD)rr>rrr?r@rr, add_issue is_repeated_notify_repeated_detect_stuck_state_handle_stuck_state _notify_userrH) rMrrrrUrVrWrY issue_recordr[relatedrEs rrzBrain._handle_copilot_result0s##G, NNEwtPS} U JJ{E2 **Y+ZZ b1 ::j(3>Djj>Uvzz+r*..vr:[] KK4 5  5xI$  !!,/ Wj$A#zz55dGD W  KK7 A  ! !'7 3  # #L 1  $ $\7 ; ':. ::k " ;3I * #rc|j}|jdrC|jd}|jjdr|ddj} tj|S#tj $rYnwxYwt jd|t j}|r> tj|jdS#tj $rYywxYwy)z.Best-effort JSON extraction from model output.z````rNz\{.*\}r) rp startswithr~rloadsrrsearchDOTALLr)rMrwrs rrzBrain._extract_json^szz| ??5 !::c?Dzz|&&v.ABx~~' ::d# ###    )T2995  zz%++a.11''  s$&A;;BB<#C C65C6rUrVcddd|g}|r|jd||jddg|jdj|y)z2Emit a notice to the user in the wrapped terminal.r u/[Copilot] ⚠️ 旁路模型发现潜在问题u摘要: u建议: rKrNrextendr;rL)rMrUrVrs rr_zBrain._notify_userssVF(SZR[H\]  LL8J<0 1 k2&' TYYu-.rracvddd|dt|ddddg}|jdj|y ) z8Warn the user when a similar issue has been seen before.r u$[Copilot] 🔁 检测到重复踩坑当前问题: u历史相关提示: u 条u9运行 python3 copilot_panel.py --open 打开提示面板rKrN)rr;rL)rMrUrars rr\zBrain._notify_repeated{sM  2WI &"3w<. 5 G    TYYu-.rctjjd}ddd|d|ddg}|jdj |y) z7Emit a confirmation when a new entry is written to RAG.rNr u![Copilot] ✅ 知识已写入 RAGrOrPrKrNrQrSs r_notify_rag_writezBrain._notify_rag_writesYlln%%&9:  /se ug      TYYu-.rrYct|xsg}|jr$d|j}||vr|j||jj dij dgD]}||vs|j||S)z2Add project and default scope tags to a RAG entry.project:ragdefault_scopes)r%r=rr5r)rMrYenriched project_tagscopes r _enrich_tagszBrain._enrich_tagss # <<$T\\N3K(* ,[[__UB/334DbI 'EH$& 'rabrc|jdxsdj}|jdxsdj}|r-|r+tjd||j }||k\ry|jdxsdj}|jdxsdj}|r-|r+tjd||j }||k\ryy)z4Return True if two RAG entries look like duplicates.rFr NTquestionF)rr~difflibSequenceMatcherratio)rxryra_titleb_titler~ a_question b_questions r_is_similar_ragzBrain._is_similar_rags55>'R..055>'R..0 w++D'7CIIKE !eeJ'-2446 eeJ'-2446 *++D*jIOOQE !rrElookbackc|jj| dD]+}|jddk7r|j||s+yy)zDCheck whether a similar RAG entry already exists in the local panel.NtyperrTF)r,recordsrr)rMrErrecords r_has_duplicate_in_panelzBrain._has_duplicate_in_panelsPjj(((4 Fzz&!U*##Iv6   rc |jjdi}|jd}|stjdy|jdd}|jdd}|jd d}|jd d}|j |jd g}|j |rtj d |yd |d|d|ddj|} || d} tj d| tj|| d} | jtj d| j|jj||||||j|y#tj$r } tj!d| Yd} ~ yd} ~ wwxYw)z3Write a structured entry to the RAG knowledge base.rrurlz'RAG URL not configured, skipping write.NrFrGr{r wrong_approachcorrect_approachrYz Skipping duplicate RAG entry: %su 问题：u 错误方案： 正确方案：u 标签：, )rFrzWriting to RAG: %srrrzRAG write OK: %s)rFr{wrongcorrectrYzRAG write failed: %s)r5rr>rrwrr?rLrrrrr,add_ragrorr) rMrErag_cfgrrFr{rrrYrr rrs rrHzBrain._write_rags++//%,kk%  NND E  g'89==R0 .3-- 2B7  vr!:;  ' ' 2 KK:E B z"#W%%Y' $( *  "g6 (%0 4==7B?D  ! ! # KK*DIIK 8 JJ  !    " "5 )(( 4 LL/ 3 3 4sA=FG!F<<Gc:td|jdgD}td|jdgD}|r|r||zry|jddj}|jddj}|r |r ||vs||vryy)z@Return True if two issues look like the same underlying problem.c3BK|]}|s|jywr!r}rts rrz*Brain._is_similar_issue..?1QQWWY?rYc3BK|]}|s|jywr!r}rs rrz*Brain._is_similar_issue..rrTrUr F)rArr~)rMrxrya_tagsb_tags a_summary b_summarys r_is_similar_issuezBrain._is_similar_issues?fb(9???fb(9?? f&6/EE)R(..0 EE)R(..0  Y(>)yBXr current_issuec&|jjdijdi}|jddsy|jdd}d}t|jd d D]}|j ||r|dz }||k\S||k\S) z?Check whether Kimi has failed multiple times on similar issues.rstuck_detectionrTFconsecutive_similar_failuresr r Nro)r5rreversedr@r)rMr stuck_cfgr similar_countpasts rr]zBrain._detect_stuck_statesKKOOIr2667H"M }}Y-MM"@!D  T//45 D%%mT:"  ))  ))rqueryc|jjdijdd}|sgSd|id|id|ig}|D]M} tj||d}|j |j }|j |cSgS#tj$r }tjd |Yd }~d }~wt$r }tjd |Yd }~d }~wwxYw) z@Query the RAG knowledge base. Tries a few common payload shapes.rr query_urlr qrr{r.rzRAG query attempt failed: %sNzRAG query parse failed: %s) r5rrrrr_normalize_rag_resultsrr>r Exception)rMrr candidatesr rrrs r _query_ragzBrain._query_ragsKKOOE2.22;C I%L e    " G }}YWbI%%'yy{22488  ,,  ;Q?  91= s%ABC*#B>> C* C%%C*rct|tr|gSt|trng}|D]e}t|tr|j|%t|ts6dD]+}||vs||s|jt||eg|St|tr1dD] }||vst j ||cSt|gSt|gS)zIBest-effort extraction of text snippets from various RAG response shapes.)rrwanswersnippetdocument)resultsr documentschunksrresponser)rrr%rr"r*r)rsnippetsr(keys rrzBrain._normalize_rag_results s dC 6M dD !H "dC(OOD)d+S"$;49$OOCS N;!"  "O dD !b C$; 77S BB CI; D {rct}|jr|jd|j|jj dij dgD]}|j||S)z6Tags that are allowed for the current project context.rqrrrs)rAr=addr5r)rMscopesrvs r _scope_tagszBrain._scope_tags8sd << JJ$,,0 1[[__UB/334DbI E JJu   rrc|j}|s|Sg}|D]L}ttjd|}|s|j |6||zs<|j |N|S)zHDrop RAG snippets that belong to a different project and are not common.z(?:project|scope):[\w-]+)rrArfindallr)rMrallowedfilteredrtags_in_snippets r_filter_rag_snippets_by_scopez#Brain._filter_rag_snippets_by_scopeAst""$O )G! 6@O#((( )rissuec |jj|jdg|jddd|j}g}|D]}|jddk(rK|j d|jd dd |jd dd |jd db|jddk(sw|j d|jddd|jdd|S)z2Use the local panel history as fallback knowledge.rYrUr r)limitallowed_scopesrrru 标题：rFu 问题：r{rrr 摘要：u 建议：rV)r, find_relatedrrr)rMrrarrQs r_query_local_knowledgezBrain._query_local_knowledgeTs**)) IIfb ! IIi $++- *   AuuV}%gr 234 !j" 567&&'ee,>&C%DF v')i 456 !lB 78: rrelated_issuesknowledge_snippetsc |jjdijdd}tjj |r(t |dd5}|j }dddnd}dd g}|jD]-}|d } | d k(rd } n| d k(rd} |j| |dd /dd|jddd|jddddj|jdgddg} t|dddD]a\} } | j| d| jdd| jds>| jd| jddc|r@| jddgt|dddD]\} } | j| d| |jdd j| d |S#1swYvxYw)!Nrstuckzprompts/copilot_stuck.txtrQrRrSrrrrrrrrru#=== 当前反复出现的问题 ===rrUr u 建议：rVu 标签：rrYu,=== 本次会话中相似的历史问题 ===r z. u 建议：u=== 相关知识 ===rr) r5rrrOrrUrr6rrL enumeraterk)rMrrrrrXrrr1rrvrrrs r_build_stuck_messageszBrain._build_stuck_messagesks kkooi488 0  77>>+ &k39 *Q!" * *CN%.AB<< GCv;Dv~" OOTc)nE F  G 2 )))R89 : )),;< = -"3"3FB"?@A B  :  $N23$7; KJC LLC5599Y#;"<= >yy& |EIIlB,G+HIJ K  LL"45 6 )*r?rrrrrrrrwrrrrrrr)rMrrr rag_snippetslocal_snippetsrrr r rrrrs r_invoke_stuck_analysiszBrain._invoke_stuck_analysiss8 !!)R0 <<||nJug6Eu- 99,G 44]C)N:--m^M_`kkoogr*WWV_ ;;??9b9==mSQ++//)R8<<\4P    LcRZm\ == #$6'.swwy"/E.F%G  26::9cJD77 ?$$KKIIdsO    ! ! #  99;D9oa(3I>G !!'**((  LL>Jrelated_panel_issuesc |jdd}|jdd}||jvry|jj||jddDcgc]}|j ||r|}}|j ||}|sy|jdstjdy|jdd}|jd d} |jd g} | stjd y|jj||| | |jd g |j||ycc}w)z:Produce an intervention hint when Kimi is stuck in a loop.rUr NProis_stuckz4Side model does not consider the issue a stuck loop. root_causeintervention_promptsourcesz2Stuck analysis returned empty intervention prompt.rY)rUrrrrY) rr~rBrr@rrr>r?rr,add_intervention _notify_stuck) rMrrrU alert_keyrsession_relatedrrrrs rr^zBrain._handle_stuck_statesL##Ir2MMOCR( // /  ##I. $11#26 %%mU;   ,,]OL zz*% KKN O ZZ b1 $jj)>C**Y+" NNO P  ##! 3""62. $  7J/7 s!E rcddd|g}|r|jd||jgd|jdj|y)zNWarn the user that Kimi appears to be stuck and an intervention hint is ready.r u)[Copilot] 🚨 Kimi 似乎陷入了循环rmu根因: )u9运行 python3 copilot_panel.py --open 查看干预建议rKr rNrj)rMrUrrs rrzBrain._notify_stucksW  7WI &   LL8J<0 1 cd TYYu-.r)z config.yamlNN)r)r)g?)2)?__name__ __module__ __qualname__rrrNr"r4rLr< staticmethodrzr%boolrrr)rrrrrrrrrrrrr#rr'r/rr9r:r?r@r*rIrrr_r\rorwrr(rrHrr]rrrrArrrrrr^rrrr*r*2s) (((T&&& !, ) ) ) )3$s)SDI$0UT $C $=S= 6S65+!3+!+!Zd(d4j2C-JGdTkGR> .$4s$4L5c5dTk5ncd4j>3c38* B5Hd02c20 / /3 /,+c,+\#$+*/C/S/ / /d4j / /s / c tCy 4DUd$d)4D)4V 4 D T **$*&S 2ST#Y.SXd3iDI&DT#Y...T .!I . d .`6+6+T 6+  6+p$0$0TRVZ$0L /S /c /rr*)__doc__r|rr[rrstringrpathlibrtypingrrrV copilot_panelrr_rr>rr$r*rrrrsv   #   8 $ $ $ $33Q/Q/r