^iJSrSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSKJ r J r J r J r J r SSKJr SSKJr SSKJr SSKJr "S S 5rg) u} Augment CLI 服务适配器 这个模块提供了与 Augment CLI 的集成，使其能够像 OpenAI API 一样被调用。 N)DictListAnyOptionalUnionPath) StageDetector)InstructionBuilder)GuidelineManagerc\rSrSrSrSFS\\S\\S\S\\4Sjjr S r S r SGS \\4S jjr S \S\4Sjr SHS\S\\S\\\\\44S\S\S\S\\\\4S\\S\\\44SjjrSIS\\\\4S\S\S\\S\S\\\44 SjjrS\4SjrS\S\S\S\\S\S\4 S jrSJS\S\\S\S\S\S\S\\\\4S\4S!jjrS"\S\\4S#jrS"\S\4S$jrS%\S&\S\\4S'jrS\S\S\\4S(jrSKS\S\\4S)jjrS*\S+\SS4S,jrS-\SS4S.jr SLS/\S\S\\S\4S0jjr!S1\S-\S\\\44S2jr"SKS3\S\4S4jjr#S/\S\$\\\44S5jr%SGS1\S3\S6\S7\\S\\\44 S8jjr&SGS1\S3\S6\S7\\4S9jjr'S:\S\4S;jr(S<\S\4S=jr)S<\S\\4S>jr*S?\S7\S@\SA\SS4 SBjr+SGS/\S3\S7\\S\\\44SCjjr,SGS/\S3\S7\\4SDjjr-SEr.g)MAugmentServiceuAugment CLI 服务适配器N node_pathuser_idverbosemodelc U=(d [RRSS5Ul[ [ 5R R R R R UlURS- UlURR5(d[SUR35eX l SUl SUl U=(d SUlSUlSUlSUlSUlX0lSUlS SKnUR-[.5UlUR0R3S 5 g) uF初始化 Augment 服务 Args: node_path: Node.js 可执行文件路径，如果不提供则使用默认路径 user_id: 用户ID，用于获取用户的Augment Token verbose: 是否显示详细输出（默认 False，只显示关键信息） model: 模型ID（可选），如果不提供则使用默认模型 NODE_PATHz%$HOME/.nvm/versions/node/v22.13.1/binge10uge10 目录不存在: augmentz Augment CLINFruAugment 服务初始化完成)osenvirongetrr __file__parent project_rootge10_direxists Exceptionr provider_type provider_namerbase_url log_callbacktimeout_recoveredtimeout_warningrtask_idlogging getLogger__name__loggerinfo)selfrrrrkwargsr)s C/Users/yu22x/Desktop/ge/ctf/app/services/ai/core/augment_service.py__init__AugmentService.__init__s# bjjnn  3' !N1188??FFMM))F2 }}##%%4T]]ODE E '*'i  !"'#   ''1 89c[RR5nURSUR SS53US'[ R "SS/SSUSS9nURS :wa[S UR35e[S URR535 g![a [S 5e[ Ra [S 5e[an[S[U535eSnAff=f)u验证 Augment CLI 是否可用:PATHauggiez --versionT )capture_outputtextenvtimeoutruAugment CLI 不可用: u✅ Augment CLI 版本: u9未找到 auggie 命令，请确保 Augment CLI 已安装uAugment CLI 验证超时uAugment CLI 验证失败: N)rrcopyrr subprocessrun returncoder!stderrprintstdoutstripFileNotFoundErrorTimeoutExpiredstr)r.r<resultes r0_verify_augment_cli"AugmentService._verify_augment_cliUs C**//#C!^^,Acggfb.A-BCCK ^^;'# F  A%"9&-- IJJ ,V]]-@-@-B,CD E  YWX X(( 867 7 C8QAB B CsB*B--7AugmentService.generate_ctf_challenge..saffhFWFWr3keyc6UR5R$rurvrys r0r{r|sUVU[U[U]UfUfr3success)statusresponse output_diru❌ 生成CTF题目失败: error)rmessagerr)app.routes.generator.utilsrfrgrr set_stage_namesr,r-lenrClistvaluesrangewarning_setup_prompt_from_databaserr rHresolve_current_rules_file_setup_guideline_for_difficulty_build_instructionr( _call_augmentiterdiris_dirnamemax_find_latest_output_dirr!)r.r\r]r^r_r`rarbr(rfrg task_statusrhriidb_prompt_loaded rules_dirdefault_rules_file instructionroutput_base_dirrz matching_dirs latest_dirrchallenge_dirslatest_challenge_dirrJ error_msgs r0generate_ctf_challenge%AugmentService.generate_ctf_challenges4l [$"5"9"9'2"FK"-//-"DK#.??>1#EL"%55kB ((+H[IYHZZd)efK0@/AASTXYdYkYkYmTnSopx|}(1,+.{+;L:F7.kAV[\hVi?jViQR9 Vi?j O < KK,,}\NJ]-^_ ++,XY'%, $?? Yatyz #bjno!MMJ6@ %.1J%J"%,,../23E3M3M3O/PD,A$BZBZA[\dhi88D)0022367I7Q7Q7S3T0 B4C[C[B\]eij `imn11(U`rBKLK - 8 9 &M +  - #L))+;)`H"&--+"="H"))++0?0G0G0I%I0I1'(xxz&'6=6G&'0IM%I$%(⚠️ 不在 Flask 应用上下文中，无法访问数据库TrjFu⚠️ 方向配置不存在: beginnereasymediumhard)u入门u简单u中等u困难Nu阶段\s*\d+\s*末r uC⚠️ 检测到旧的 Prompt 格式（包含'阶段X末'），第 u 次重新编译...)dbu@✅ 重新编译成功，已清理旧格式并保存到数据库u:⚠️ 重新编译后仍包含旧格式，继续重试...u)⚠️ 重新编译后仍未找到 Promptu⚠️ 重新编译失败u⚠️ 警告：重新编译 uJ 次后仍包含旧格式，可能 stage 配置中包含'阶段X末'内容u2✅ 已从数据库加载 Prompt 模板（方向: u , 难度: u）u#⚠️ 数据库中未找到方向 u 难度 u# 的 Prompt，尝试重新编译...u0✅ 重新编译成功，已加载 Prompt 模板u 的 Promptu&✅ 已加载 Prompt 模板，长度:  字符u&❌ 从数据库加载 Prompt 失败: )flaskrapp.models.database.modelsr$app.services.prompt.compiler_servicerrCqueryrget_compiled_promptsresearchcompile_and_save_promptsrsessionrefreshreplace_placeholdersr_compiled_promptr!rH traceback print_exc)r.rar_r\r]r^rrrcategorydifficulty_mapdifficulty_keycompiled_promptscompiled_prompt old_promptrold_format_pattern max_retries retry_countrrJrs r0r*AugmentService._setup_prompt_from_databaseVs6(l  - A R"$$V^bc%++//  "ON$F-= %:"  ii 2J??KR]D]1$K_`k_llAIMN-EEhOOA **84,4+H+H+J(+0R)9)IJ#%99-?#L#L %(hqu v % %(bko p!$MVZ[! 9$G3ii 2J??KR]D]6+-")) *J99J>c rU(a[R"XgXX4U5$[R"XX4U5$)u7构建 Augment 指令（委托给 InstructionBuilder）)r build_dynamic_instructionbuild_simple_instruction)r.r\r]r^r_r`rarbs r0r!AugmentService._build_instructionsA %??5Vh  &>>5>P r3linec.[R"U5$)u>检测输出行中的阶段信息（委托给 StageDetector）)r detect_stage)r.rs r0 _detect_stageAugmentService._detect_stages))$//r3c^SSKnUR(agUR"ST5(dUR"ST5(agTR5(dgUR"ST5(agUR"ST5(agUR"S T5(agUR"S T5(agUR"S T5(agS T;agS T;agST;dST;ag[ U4SjS55(agST;dST;ag/SQn[ U4SjU55(agST;a[ TR55S:aggg)uv判断是否应该打印这一行（非详细模式下） Args: line: 输出行 Returns: 是否应该打印 rNTu\[已输出 \d+ 行\]u\[已处理 \d+ 行\]Fz\[90mz ^\s{4,}[^\s]z2^\s+(command|cwd|wait|max_wait_seconds|path|type):z ^\s+\d+\s+u(?:##\s*)?阶段\s*\d+[：:]u🔧 Tool call:u📋 Tool result:u✅ Command completedu✅ Successfullyc3,># UH oT;v M g7fru).0markerrs r0 4AugmentService._should_print_line..,so,n&~,n)❌u⚠️ErrorrERRORWarningWARNINGz File savedz Saved file) BuildingzSuccessfully builtzSuccessfully taggedzCreating networkzCreating containerzStarting containerzContainer startedStoppingRemovingc3,># UH oT;v M g7frur)rkeywordrs r0rr?s>o7$oru🤖d)rrrrEanyr)r.rrdocker_keywordss ` r0_should_print_line!AugmentService._should_print_lines[  << 99-t 4 4 BZ\`8a8azz|| 99Xt $ $ 99_d + + 99JD Q Q 99]D ) ) 994d ; ;  $ $ & #d *.@D.H o,no o o 4 <4#7   >o> > > T>4::< 3&r3stage output_linescSRU5nSnUR(aUSSKJnJn U UR UR05nUR S05nU(a X;aUXS.nSSS5 [R"XUS9$!,(df  N#=f![ a(n URRSU 35 Sn A NSSn A ff=f) u\从输出中提取阶段关键信息（委托给 StageDetector，支持动态阶段配置）r7Nrrerh)idru获取阶段配置失败: ) stage_config) joinr(rrfrgrr!r,rNr extract_stage_info) r.rr output_textrrfrgrrhrJs r0_extract_stage_info"AugmentService._extract_stage_infoLsggl+  << D[$"5"9"9$,,"KK"-//-"DK"u';#($/$6( %//Q]^^%_ D !!$>qc"BCC Ds5 B AB2B  BB B C*C  CcSSKJn SSKnURU- SU3- nURU- S- nUR SSS9 UR 5(dUR RSU35 gUS- nUR 5(Ga-UR5Vs/sHoR5(dMUPM n nU (a[U S S 9n U Rn Xk- n U R 5(a8SSK n U RR5RS 5nU S U3n Xk- n UR"[!U 5[!U 55 UR R#S U 35 UR$"[!U55 UR R#SU35 [!U 5$US- R 5nUS- R 5nU(dU(GabSSK n U RR5RS 5nUS3n U(aUS- R)SS9nSSKnUR,"SUUR.5nU(aNUR1S5R35nUR4"SSU5nUR4"SS U5nUS U3n Xk- n U R 5(a8XkS [7U RR5R9553- n UR"[!U5[!U 55 UR R#SU 35 [!U 5$UR RS5 UR;U5$s snf![&a)nUR RSU35 SnAGNSnAff=f![&a)nUR RSU35 SnAGN4SnAff=f)u将临时工作目录移动到 output/ 目录 Args: category_id: 方向ID task_id: 任务ID Returns: 移动后的输出目录路径，如果失败则返回 None rrNtemp_rsTparentsexist_oku临时目录不存在: c6UR5R$rurvrys r0r{5AugmentService._move_temp_to_output..sIZIZr3r}z %Y%m%d_%H%M%S_u,已将题目目录从临时目录移动到: u已清理临时目录: u清理临时目录失败: writeup.mddocker _challengeutf-8encodingz)^#\s+([^#\n]+?)(?:\s*-\s*[Ww]riteup)?\s*$r z[^\w\s-]r7z\s+u#从 writeup 提取题目名失败: u已将临时目录移动到: uE临时目录中未找到题目文件，尝试查找最新输出目录)pathlibr shutilrmkdirr r,rrrrrdatetimenowstrftimemoverHr-rmtreer! read_textrr MULTILINEgrouprEsubint timestampr)r.rar(r rtemp_dirrtemp_output_dirrzrrchallenge_dir_name target_dirrr%rJ has_writeup has_dockerwriteup_contentr title_matchchallenge_names r0_move_temp_to_output#AugmentService._move_temp_to_outputds !==;.5 1BB--+5@dT:  KK  "9( D E#X-  ! ! # #)8)@)@)BQ)BAhhja)BNQ'*>?Z'[$%9%>%>"-A $$&&# ( 1 1 5 5 7 @ @ QI,>+?q )L&!0!EJ C 45s:G   #OPZ|!\]JMM#h-0KK$$'>xj%IJ:& ,.668 )113 *  ))--/88II$-;j!9  S'/,'>&I&ISZ&I&[O"$)),XZikmkwkw"xK")4):):1)=)C)C)E)+ R)P)+^)L09{!N;K-L*)=J  "",1EQs8K\K\K`K`KbKlKlKnGoFp/qq KKH s: 7 KK  .s8I8Ir3r}N)rr rr rrrrH)r.rar category_output_dirrzsubdirsrs r0r&AugmentService._find_latest_output_dirst !#mmk9HD  % % ' '"5"="="?N"?Q88:q"?GN .IJ :& Os BBsourcetargetc$SSKnUR5(dgUR5(dURSSS9 UR5HnX$R- nUR 5(aOUR5(aUR XE5 MNUR"[U5[U55 MuUR5(a8UR5(aUR5 OUR"U5 UR"[U5[U55 M UR5(a0[UR55(dUR5 ggg![a gf=f)u递归合并目录内容（移动源目录内容到目标目录） Args: source: 源目录路径 target: 目标目录路径 rNTr )rr rrrr_merge_directoryrrHis_fileunlinkrrrmdirOSError)r.r7r8ritem target_items r0r:AugmentService._merge_directorys" }} }} LLL 5NN$D 99,K{{}}%%''))$<KKD 3{+;<%%''"**,,#**, k2 CIs;'78'%, }}s6>>+;'<'< (=   sUS?3- nOlSSK%nURDRM5ROS@5S[R"5SURQ5RRSSA3nUS>US?3- nURSBUSCUS,35 SSDK*J+nJ,n U(a0U UU;a0UU'[5U5UUSE'SUUSF'SSS5 OSSGK*J-n [5U5USE'SUSF'UR "5nSSK.n [^R`"U[^Rb[^RdSU [5U5SSSH9n!/n"Sn#SIn$SSJK3J4n%J5n& U%"[5U5SK5n'SLn(SIn$U!RlRo5n)U)(dU!Rq5bGOM7U'RsU)SS 9 U"RuU)5 URwU)5(a%[U)SSSM9 U RlRy5 SRCU"5n*[zR|"U*5(Ga[SNSS 9 U'RSO5 [zR"5n+U+S:a[U+5Hjn,U,U$:dM SRCU"5n-[USP5(aURU,U"5OSn.[zR"U,5n/U'RU,U/SQ5 Ml U$S:a_SRCU"5n-[USP5(aURU$U"5OSn.[zR"U$5n/U'RU$U/SQ5 Sn(GOqURU)5n0U0b[SRU0SSU$3SS 9 U0bU0U$:wa[STU$SUU03SS 9 U(aTSSDK*J+nJ,n U UR'U05n1U1R'SV05n2U2(a[zR"U25 SSS5 SRCU"5n-[zR"U$U0U-URURSW9n$[zR"U$5n3U'RU$U3SX5 U#SY- n#UR(a+U#SZ-S:Xa"S[U#S\3n4[U4SS 9 U'RU45 O/UR(dU#SZ-S:XaU'RS[U#S\35 GMHU!RS]S^9 U((d<SRCU"5n-[zR"U$U-URURS`9 UR "5U- n5U'RSaU5SbSc35 URRSdU5SbSc35 SSS5 SRCU"5n6U!RS:wGaSgU!R3n7URRU75 SLn8Sn9U"U5Rn:ShU:;a+U:RSh5n;[3U:5U;SY-:aU:U;SY-n9U"GHn)SiU);dM [zR"U)[5UR55n[RR [RRCU(aSoOSp35 [SrU?(aSoOSp35 O U8(d/UR(aURSSsU75 [U75eStU!RSu3nA[SvUA35 SUlWUAUlXUR(aURSSwU7Sx35 UR(aURSSyU65 [Sz[3U65S{35 U6S|[5;a2U R 5(aU R5 [S}U 3SS 9 S[5;a?U(a7UR 5(a!US- nBURUB5 [SSS 9 $$$$![an[S US 3SS 9 SnAG NgSnAff=f![a)n[SUS3SS 9 U RUSS9 SnAG NSnAff=f!,(df  GNf=f!,(df  GN=f![^Ra U!R5 [S_5ef=f![aJnU'RSe[5U53U5 URRSe[5U53SSf9 eSnAff=f!,(df  GN=f! GN=f![an[S~U3SS 9 SnAGNSnAff=f![an[SU3SS 9 SnA$SnAff=f![^Ra@ S_n7[SU735 UR(aURSSsU75 [U75e[aFnS[5U53n7[SU735 UR(aURSSsU75 eSnAff=f!S|[5;a2U R 5(aU R5 [S}U 3SS 9 O$![an[S~U3SS 9 SnAOSnAff=fS[5;a?U(a7UR 5(a!US- nBURUB5 [SSS 9 ffff![an[SU3SS 9 SnAfSnAff=f=f)u调用 Augment CLI 执行指令 Args: instruction: 用户指令 category_id: 方向ID，用于设置工作目录 task_id: 任务ID（可选，用于创建独立工作目录避免多任务冲突） Returns: Augment 的响应文本 u4使用共享工作目录（与 API 模式一致）: u ，task_id=uF未提供 task_id，使用共享工作目录（与 API 模式一致）Tr z CLAUDE.mdNrzCLAUDE.md.backup_uN📌 已临时重命名工作目录下的 CLAUDE.md，避免规则文件冲突rju#⚠️ 重命名 CLAUDE.md 失败: u$，可能会造成规则文件冲突riz.augment_rules_.mdrz --- rru9📜 已将 Prompt 和用户指令合并到临时文件: u=⚠️ 未找到编译后的 Prompt，仅使用用户指令: u!⚠️ 创建合并文件失败: u!，回退到仅使用指令文件r5r6r7get_user_augment_tokenconvert_token_to_augment_formatAUGMENT_SESSION_AUTH✅ 已设置用户  的Augment Tokenu🔑 Token (前50字符): 2z...rCrDu📂 用户缓存目录: u'❌ Token 转换失败，原始 Token: u⚠️ 警告: 用户 u, 未设置Token，将使用系统默认Tokenu>⚠️ 警告: 未指定用户ID，将使用系统默认Tokenu9✅ 环境变量 AUGMENT_SESSION_AUTH 已设置 (长度: )u0❌ 环境变量 AUGMENT_SESSION_AUTH 未设置!r8-if-p--workspace-rootrz--modelu🤖 使用模型: u📂 工作目录: --augment-cache-diru执行 Augment CLI:  u工作目录: u合并的规则文件: rlogsraugment_.txt %Y%m%d%H%M%Su日志文件: z (task_id: relog_file log_position)generation_statusrDrBr;r<cwdbufsizeuniversal_newlines)UnifiedLogWriterLogLevelAugmentFendrku9✅ 检测到生成完成标记 [CTF_GENERATION_COMPLETE]u3检测到生成完成标记，所有阶段已完成ru完成u🔍 检测到阶段标记: u, 当前阶段: u🔄 阶段切换: z -> rh) current_stage new_stagerextract_info_funcr,u开始r ru [已处理 u 行]Xr=.Augment CLI 执行超时（超过 10 分钟）)rsrrur,uAugment 执行完成，耗时: .2f 秒uAugment 执行时间: uAugment 执行异常: )exc_info&Augment CLI 执行失败，返回码: rzoutput/rzexp.pyru<⚠️ 虽然AI超时，但检测到题目文件已生成: z - writeup.md: u✅rz - exp.py: z - docker/: ruAI执行超时（返回码: u$），但题目文件已完整生成u⚠️ ru（但文件已生成）ru)✅ Augment 执行完成，输出长度: rtemp_rules_fileu!✅ 已清理临时规则文件: u!⚠️ 清理临时文件失败: claude_md_backupu✅ 已恢复 CLAUDE.md 文件u&⚠️ 恢复 CLAUDE.md 文件失败: u❌ Augment CLI 执行异常: )[rr,r-rrr timer$renamerCr!rgetpidhasattrr write_textrr>rrr&app.services.ai.helpers.augment_helperrTrUrrrHrrextendr)r*r+rrrr uuidrruuid4hexrrfrgrhsysr?PopenPIPESTDOUT"app.services.ai.core.log_formatterrnrorDreadlinepoll write_rawappendrrkr detect_completion write_successget_total_stagesrrget_stage_name write_stagerrhandle_stage_transitionr write_infowaitrGkillfinalize_stage write_errorrrApartsindexdetect_output_dirpathbasenamer%r&r'localsr<)Cr.rrar( workspace_dirclaude_md_in_workspacer~rrJ unique_suffixr} rules_contentcombined_contentr<user_cache_dirrTrU token_strconverted_tokencmdr)r,rr log_dirrfr unique_idrfrgrh start_timerprocessr line_countrsrnro log_writergeneration_completeroutput_text_so_farri stage_numr stage_info stage_namestage_detectedrrh stage_msg progress_msg elapsed_timersroutput_dir_foundcategory_id_from_workspaceworkspace_parts ge10_index detected_dirr*has_expr+dir_name warning_msg original_namesC r0rAugmentService._call_augment2s 3  KK  STaSbbmnumvw x KK   h iD48"/!< ! ( ( * * ,1B3tyy{CSBT/UU  q&--.>?fost 99;-qTYY[4-?)@(AB 'OM?#*NN Ft/00T5J5J $ 5 5 &3_K }#M **+;g*NQRaQbckop **;*IUVeUfgost T P**//#C!^^,Acggfb.A-BCCK"N||z24<<@ &Ei&PO&6E23 4T\\NBST\`a :?3B;O:PPST\`a*.):):=M)MRWX\XdXdWePf)f&,,TD,I 9.9IJRVW G RUSUGWWZ[cgh3DLL>Amnvz{V^bc&,QRUVYZpVqRrQsstu~BCHQUV s?2245"C (=(=(?$@ CzzdjjI5 Itzz23+DJJ<8E ' (=(=(?'@A N 13~3FGH &&x0F KK.sxx}o> ? KK.8 9 KK1/2I2I2K1LM N  $6lG MM4M ("xy%=='00446??OPPQRTR[R[R]Q^^_`d`j`j`l`p`pqsrs`t_uv "x {$%?? KK. +gYaH I X$&9979+G4?B8}'0$++D1 22488!$Bd;JJ,,..0WW\-B*(::;MNN!$]fjk&445jk,9+I+I+KL+a/16|1DI'0='@68ggl6K jqrvyNkOkOT5M5MiYe5fUY 5B5Q5QR[5\ (2(>(>y*V^(_ 2E$1A#524'',2GKjqrvyNkOkO1I1I-Ye1fUYJ1>1M1Mm1\J$.$:$:=*V^$_37/!*.););D)A)5!$@@PP`an`o"px|})5.M:Y!$7 d>JZ"[cgh ' k%42E2I2I'SU2VK2=//-QS2TK'2(5(E(Ek(R &5+-'',*?K,9,Q,Q.;*8,7262J2J'+{{ -M)6(D(D](SI&22=)XV#a <bckopp q* F 5aS8YZbf g  & &{W & E Ft%_B&5_B&44Z '(XYYZ$!**-CCF8+LaPKK%%(>s1vh&GRV%W a<;RN K9!=TJJ K P>qcB$O PC(( 'HI D $ %  !!!Wi8I& & 4SVH=I D $ %  !!!Wi8   K$0_5K5K5M5M#**,=o=NOW[\ K9!=TJ K P%16FK[KbKbKdKd$1K$?M$++M::4HLe6F1 P>qcB$O Ps%t At/t/5My6%u%,By6xD v<Dv</Au71C!v<v "B v<0Ay6 Ax% y6/y6 "y63Cy6;D>y6:Ax-;A y t,t''t,/ u"9uu"% u4/y67 v v< 0v99v<< xAx  xx x"y6%x*'y6- y7 y  y y3 y..y36A|A|||5A}5 }?( }:55:}??5A  5 2 -(5-225 challenge_idcU(a$[RRU5(d[SU35e[ U5nUS- nUR SS9 [ SU35 [U5[U5S.$)uN为题目创建独立的工作空间（直接使用题目的output_dir） Args: challenge_id: 题目ID output_dir: 原始输出目录路径(ge10/{category_id}/output/xxx) Returns: 包含工作空间信息的字典: { 'workspace_dir': 工作空间目录路径, 'session_dir': 会话目录路径 } u题目输出目录不存在: rETrau*✅ 使用题目目录作为工作空间: )r session_dir)rrr rr rrCrH)r.rrrrs r0create_isolated_workspace(AugmentService.create_isolated_workspaces !;!; !r3session_cache_dirrassistant_messagecSSKnSSKJnJn US- US3- nUR5(d.SSKn U R [ 5n U RSU35 g[USSS 9n UR"U 5n SSS5 S UUS UR5R53S SS U0S./S.0n W SRU 5 URUR5R5RSS5U S'[USSS 9n UR "XSSS9 SSS5 [#S[%U S5S3SS9 g!,(df  N=f!,(df  N:=f![&an[#SU3SS9 SnAgSnAff=f)u+手动保存对话到会话文件 因为使用 -p (print mode) 时，Augment CLI 不会自动保存会话历史， 所以我们需要手动更新会话文件。 Args: session_cache_dir: 会话缓存目录 session_id: 会话ID user_message: 用户消息 assistant_message: AI回复 rN)rtimezonerFz.jsonu&会话文件不存在，无法保存: rrrexchangezmanual-r rX)rtype text_node)request_message response_text request_id request_nodes chatHistoryz+00:00ZmodifiedwFr) ensure_asciiindentu)✅ 已保存对话到会话文件（共 u 条）Trju⚠️ 保存会话失败: )jsonrrr r)r*r+ropenloadrr%rutc isoformatreplacedumprCrr!)r.rrrrr&rr session_filer)r,f session_data new_exchangerJs r0_save_conversation_to_session,AugmentService._save_conversation_to_sessionAs$ /(:5:,e8LL ""$$ &&x0F NNCL>R S # BlC':a#yy| ; '3%6$+HLLN,D,D,F+G"H#$$% )<*& L"  ' . .| <(0||HLL'A'K'K'M'U'UV^`c'dL $lC':a ,aH; =c,}B]>^=__fgos t=;:6;:  B 04D A BsI E$'E:BE$ E"E$ E E$ E!E$$ F. FFc  [R"SSSSS9nURU5 URnSSS5 [ U5nUR 5(dUR U- n[UR55n[RR5nURSURSS 53US'UR(aES S KJn Jn U "UR5n U (a$U "U 5n XS '[%S URS3SS9 US- n U R'SSS9 [%SU 3SS9 U(aSSUSWSSU/n[%SU3SS9 OSSSWSSU/n[%SSS9 UR)S[U 5/5 [%SSR+U53SS9 [%SU3SS9 [%S SS9 S SKn[ S!5nUR'SS"9 UR.(aUS#UR.S$3- nOlS SKnUR,R35R5S%5S&[R6"5S&UR95R:SS'3nUS#US$3- n[<R<"5n[>R@"U[>RB[>RDSUUS SS(9n/n[GUSSS)9nURHHVnURKU5 URMU5(a [%US SS*9 URU5 URO5 MX URQS+S,9 SSS5 [<R<"5U- n[%S-5 [%S.US/S035 [%S 5 S R+U5nURRS :wa[US1URR35eURWU5nUnU(dURYU5nU(dU (aU S2- nUR[5(alS SK.nUR][US3- 55nU(aD[_U[R`RbS49n [ U 5Rdn[%S5U3SS9 U(aU (aURgU UUUS69 UUS7.[Rh"U5 $!,(df  GN=f!,(df  GN=f! $=f![>Rja WRm5 [US85e[Tan![US9[U!535eSn!A!ff=f![Rh"W5 f! f=f=f):u调用 Augment CLI 继续对话 Args: instruction: 用户指令 workspace_dir: 工作空间目录 session_id: 会话ID（可选，如果提供则恢复特定会话） Returns: 包含响应文本和会话ID的字典 r#rRFrmodesuffixdeleterNr5r6r7rrSrVrWrXTrjrEr u📂 会话目录: r8--resumer[r\r]u📌 恢复会话: --continueu📌 继续最近的会话r^u(🚀 执行 Augment CLI (继续对话): r_rrrr`raaugment_continue_rcrdrrerirrqrvrwrqu⏱️ 执行时间: ryrzr|rF*.jsonr}u📌 检测到会话ID: rrrr)rrrxr)7tempfileNamedTemporaryFilewriterr rrrHrrrr>rrrrrTrUrCrrrrr(rrrrrrrr?rrrr'rDrrrkrrAr!rrr globrrgetmtimestemr1r<rGr)"r.rrrr.instruction_filer workspace_absr<rTrUrrrrrrrfrrrrrlog_frrrscleaned_outputextracted_session_id sessions_dirr@ session_fileslatest_sessionrJs" r0r%AugmentService._call_augment_continues7" ( (   GGK  vv  \ !-0N!--//!%!2!2^!C 6 6 89M**//#C!^^,Acggfb.A-BCCK||z24<<@ &Ei&PO2A./0>OPX\]!/1C C   # #D4 # @ '(9':;4 H  +& +J<8E +& 2$? JJ-s3D/EF G UUUUc ## [R"SSSSS9nURU5 URnSSS5 [ U5nUR 5(dUR U- n[UR55n[RR5nURSURSS 53US'UR(a-S S KJn Jn U "UR5n U (a U "U 5n XS 'US - n U R%SSS9 U(a SSUSWSSU/nO SSSWSSU/nUR'S[U 5/5 [(R*"U[(R,[(R.SUUSSS9n/nUR0H)nUR3U5 UR5U5nUv M+ UR7SS9 UR8S :wa[;SUR835eS R=U5nUR?U5nUnU(dURAU5nU(dU (ayU S- nURC5(a_S SK"nURE[US- 55nU(a7[GU[RHRJS9n[ U5RLnU(aU (aUROU UUUS9 [RT"U5 g!,(df  GN=f![(RPa WRS5 [;S 5e[:an[;S![U535eSnAff=f! g=f![RT"W5 f! f=f=f7f)"u调用 Augment CLI 继续对话 - 流式输出版本 Args: instruction: 用户指令 workspace_dir: 工作空间目录 session_id: 会话ID（可选） Yields: 输出的文本块 r#rRFrr4Nr5r6r7rrSrVrETr r8r8r[r\r]r9r^r rirvrwr|rFr;r}r<rxr)+r=r>r?rr rrrHrrrr>rrrrrTrUrrr?rrrrDrrrrAr!rrrr r@rrrArBr1rGrr<)r.rrrr.rCrrDr<rTrUrrrrr full_outputr cleaned_chunkrsrFrGrHr@rIrJrJs r0r,AugmentService._call_augment_continue_stream;sa" ( (   GGK  vv  k !-0N!--//!%!2!2^!C 6 6 89M**//#C!^^,Acggfb.A-BCCK||z24<<@ &Ei&PO2A./!/1C C   # #D4 # @ +&  +&  JJ-s3D/EF G!&&!!((!#' GK""4( $ 8 8 > ## ' LLL %!!Q&"HI[I[H\ ]^^WW[)F!77?N$. ''+'K'KF'S$',=0:= &&(($(IIc,2I.J$KM$),]@P@P)Q/3N/C/H/H,$(922&73!,&4 3  *+e  T(( N LLNLM M C8QAB B C    *+ svN K9N J L "M"8N 9 LN  8MMMM)"M&$N )N +NN NN  N )rrr$rr%r,rrrr#r"r(r&r'rr)NNFNru)r7rNN)rgffffff?NF)r7rN)r)rN)/r+ __module__ __qualname____firstlineno____doc__rrHr$boolr1rKrPrUrZrrrrrfloatrrrrrrrrr/rr r:rOrrrtuplerrrrrrr1rr__static_attributes__rr3r0rrs %9:(3-9:#9:`d9:u}BvC9:vC89 H H 0 0c 0(#% .2!%FFcFS$sCx.() F  F  FFDcN+F#F c3hFV $( 1 tCH~&1 1  1 SM 1  1  c3h1 fT#T@@@ @ c @  @ @P#% .2c     DcN+ ,0#0(3-0XsXtXt__D_Xd^_0Z9Z9cZ9hsmZ9x38C=&)t)T)d)V2H2H2HhSPSP3SPQYZ]Q^SPjmSPr  c  s  tTWY\T\~  Dp8tp8RUp8dSU4RUCV=WD%) @ @ @  @ SM @ c3h @ N%) 333 3 SM 3j$)C)C)Vchsm"@B@B@B @B  @B  @BL%) vvvSM v c3h vx%) EEESM EEr3r)rSrr?r=r&rrrtypingrrrrrrr stage_detectorr instruction_builderr guideline_managerr rrr3r0r\s?   33)3/iir3