auto-coder 0.1.252__py3-none-any.whl → 0.1.254__py3-none-any.whl

{auto_coder-0.1.252.dist-info → auto_coder-0.1.254.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.252
+Version: 0.1.254
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence

{auto_coder-0.1.252.dist-info → auto_coder-0.1.254.dist-info}/RECORD

@@ -1,17 +1,17 @@
 autocoder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/auto_coder.py,sha256=0Bjv8WtImwcObAgBeJ1FZsL6M_wZU04aKMIlLBnCXS8,64568
+autocoder/auto_coder.py,sha256=KTc-twhc4MNabzgNWj8nDZZTD-uMneDABnFSWnt6Ta8,64640
 autocoder/auto_coder_lang.py,sha256=Rtupq6N3_HT7JRhDKdgCBcwRaiAnyCOR_Gsp4jUomrI,3229
 autocoder/auto_coder_rag.py,sha256=DDAmqw36CO6phtdQuN8LYIbIR3YGdoZw5-pG0LjVxMc,29063
 autocoder/auto_coder_rag_client_mcp.py,sha256=WV7j5JUiQge0x4-B7Hp5-pSAFXLbvLpzQMcCovbauIM,6276
 autocoder/auto_coder_rag_mcp.py,sha256=-RrjNwFaS2e5v8XDIrKR-zlUNUE8UBaeOtojffBrvJo,8521
 autocoder/auto_coder_server.py,sha256=XU9b4SBH7zjPPXaTWWHV4_zJm-XYa6njuLQaplYJH_c,20290
 autocoder/benchmark.py,sha256=Ypomkdzd1T3GE6dRICY3Hj547dZ6_inqJbBJIp5QMco,4423
-autocoder/chat_auto_coder.py,sha256=oKVoZr66oHNj43SUlqan0i1qBFIleOTrwFDy9iJRgaU,107441
+autocoder/chat_auto_coder.py,sha256=1QgTy6SiZjN2mkYe5pVvPTdZyedwijDctTUS8tSukVM,107527
 autocoder/chat_auto_coder_lang.py,sha256=OiwjnOqbiWwbaVuOhehEqdUFPOJbhfvQr3sIq3b0xj4,17455
 autocoder/command_args.py,sha256=9aYJ-AmPxP1sQh6ciw04FWHjSn31f2W9afXFwo8wgx4,30441
 autocoder/lang.py,sha256=U6AjVV8Rs1uLyjFCZ8sT6WWuNUxMBqkXXIOs4S120uk,14511
 autocoder/models.py,sha256=uo4mFWb-kdtd-8e2HPfCuXjhkDsE37n1dY6Nt1SU6kk,7625
-autocoder/version.py,sha256=1gGmGBolGnZ8y1dxxyDRAeCnjQZMw6R1feRdZma-dZY,23
+autocoder/version.py,sha256=LLYeCF3eLt_aBS3HsiF4z5DiLpgabod5UhddNdga9mk,23
 autocoder/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/agent/auto_demand_organizer.py,sha256=NWSAEsEk94vT3lGjfo25kKLMwYdPcpy9e-i21txPasQ,6942
 autocoder/agent/auto_filegroup.py,sha256=CW7bqp0FW1GIEMnl-blyAc2UGT7O9Mom0q66ITz1ckM,6635
@@ -25,7 +25,7 @@ autocoder/agent/project_reader.py,sha256=tWLaPoLw1gI6kO_NzivQj28KbobU2ceOLuppHMb
 autocoder/chat/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/common/JupyterClient.py,sha256=O-wi6pXeAEYhAY24kDa0BINrLYvKS6rKyWe98pDClS0,2816
 autocoder/common/ShellClient.py,sha256=fM1q8t_XMSbLBl2zkCNC2J9xuyKN3eXzGm6hHhqL2WY,2286
-autocoder/common/__init__.py,sha256=PjFsST8SG14PiZH4VmMSLGTUz1FqPfcx7GlUmkhT9IQ,11928
+autocoder/common/__init__.py,sha256=6maackdzrYnUPvpgVPl92JdMOnw7X4n3EnEQA9OnLGE,11984
 autocoder/common/anything2images.py,sha256=0ILBbWzY02M-CiWB-vzuomb_J1hVdxRcenAfIrAXq9M,25283
 autocoder/common/anything2img.py,sha256=4TREa-sOA-iargieUy7MpyCYVUE-9Mmq0wJtwomPqnE,7662
 autocoder/common/audio.py,sha256=Kn9nWKQddWnUrAz0a_ZUgjcu4VUU_IcZBigT7n3N3qc,7439
@@ -38,10 +38,10 @@ autocoder/common/code_auto_generate.py,sha256=N_kENWW4eVPKmihpIQ4Q2ivYkugAuCWIPL
 autocoder/common/code_auto_generate_diff.py,sha256=QAm-BVJgGl_wbw0VjY2RZigVCwOBO-l1W-90PrYDdOQ,16741
 autocoder/common/code_auto_generate_editblock.py,sha256=FuVH-sP4_Ys2MJk9SkWgHHLAKp51bKSxTwFzd7voilg,18356
 autocoder/common/code_auto_generate_strict_diff.py,sha256=kieVUx2KuR0BGtQHE0Ll7i8e_3Lb4xfFXHIXDttQACE,15274
-autocoder/common/code_auto_merge.py,sha256=8dtnz61l0B5gNbQmx26TZ4_jD825dsnnWtAFD_zs6es,7335
-autocoder/common/code_auto_merge_diff.py,sha256=yocfe8s3Pz6hTGDUl9wRIewY3NcTize_gEla64lsGT0,15331
-autocoder/common/code_auto_merge_editblock.py,sha256=sp7C0fZJMVcNgI8uWy43CKDk7gGXFTkMB9kbP2VdY8k,17485
-autocoder/common/code_auto_merge_strict_diff.py,sha256=9rm0NJ_n6M3LohEX7xl1Jym0xmm8UEYqj_ZTSO3oSlM,9519
+autocoder/common/code_auto_merge.py,sha256=-ksBjj4ZVcbY_tVH4JLXAMSRtsgaSxrSZ5-MOl9cAgE,7354
+autocoder/common/code_auto_merge_diff.py,sha256=qpEuHJEgX6sWK7EDFEKqcYkyI28wOyM4pytyl8BLohY,15350
+autocoder/common/code_auto_merge_editblock.py,sha256=sxgYMLMACRwJvw-bABkdDHezPelsDFrOCpGuhtT5Dzs,17504
+autocoder/common/code_auto_merge_strict_diff.py,sha256=P0nKNkBrFMybTSZ7kOdA_JixoVmLCZIhAP5q7ILJ9j0,9538
 autocoder/common/code_modification_ranker.py,sha256=qfadP9P-iiidCG2A_MjAf3Ca8cMz7YlnN08D_kH6uFc,6447
 autocoder/common/command_completer.py,sha256=SSeb8MDH0JPvfdyW-S2uaHnui4VBDfSQvQPLbv3ORPA,9314
 autocoder/common/command_generator.py,sha256=-hmbD_AnCa5HxL4BznuEfYAf_l8AxU5fAG5F0sM_fuE,2116
@@ -54,7 +54,7 @@ autocoder/common/image_to_page.py,sha256=O0cNO_vHHUP-fP4GXiVojShmNqkPnZXeIyiY1MR
 autocoder/common/interpreter.py,sha256=62-dIakOunYB4yjmX8SHC0Gdy2h8NtxdgbpdqRZJ5vk,2833
 autocoder/common/llm_rerank.py,sha256=FbvtCzaR661Mt2wn0qsuiEL1Y3puD6jeIJS4zg_e7Bs,3260
 autocoder/common/mcp_hub.py,sha256=2ZyJv3Aiv4Y97UHut49oYhIFcu7ICR-mptDEBSgT3uE,14234
-autocoder/common/mcp_server.py,sha256=QCFa-15kx7rbNsinwdGFFX2y47pww0fVdI-ldKFSSWI,12267
+autocoder/common/mcp_server.py,sha256=1G6e0IbeS_h7CA1vr0dPAnf0o2H1f1A8I4bua8EUtKw,12318
 autocoder/common/mcp_tools.py,sha256=KsLvRrB6pvmebqd-lDaSH6IBJR0AIxWRE-dtCEG_w9k,12485
 autocoder/common/memory_manager.py,sha256=2ZjYG7BPyvbYalZBF6AM_G5e10Qkw_zrqtD4Zd7GSsQ,3663
 autocoder/common/printer.py,sha256=P1WU0QjlfnjqTP5uA55GkHZCpFzRPFkc34DMMandreg,2023
@@ -68,7 +68,7 @@ autocoder/common/text.py,sha256=KGRQq314GHBmY4MWG8ossRoQi1_DTotvhxchpn78c-k,1003
 autocoder/common/types.py,sha256=PXTETrsTvhLE49jqAeUKGySvxBN9pjeyCgRHLDYdd9U,664
 autocoder/common/utils_code_auto_generate.py,sha256=kDW5B_2wRLk7hAls2hewliDacV86lrPz8Jan01BvtCw,3573
 autocoder/common/mcp_servers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/common/mcp_servers/mcp_server_perplexity.py,sha256=jz0LkCgZcqKkNdLZ9swNOu9Besoba4JOyHDedoZnWHo,5546
+autocoder/common/mcp_servers/mcp_server_perplexity.py,sha256=IXTyMpd1CQcBLzVinA-_OIOHoNmbzvuW6pXIadaKHJE,5533
 autocoder/data/tokenizer.json,sha256=7Lb5_DaYlDRvBRH0B0ynXO5c1fOwbQLxujX805-OEh0,7847602
 autocoder/db/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/db/store.py,sha256=tFT66bP2ZKIqZip-uhLkHRSLaaOAUUDZfozJwcqix3c,1908
@@ -80,21 +80,21 @@ autocoder/dispacher/actions/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQ
 autocoder/dispacher/actions/plugins/action_regex_project.py,sha256=ckTbisMlvwMNHQbrt5WB7pBvf2XAhYQYGH8uyYrvGXU,6060
 autocoder/dispacher/actions/plugins/action_translate.py,sha256=nVAtRSQpdGNmZxg1R_9zXG3AuTv3CHf2v7ODgj8u65c,7727
 autocoder/index/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/index/entry.py,sha256=hXSOi-jPgRBKQB55eqWkT95vxvWrbVHDuIMtDkqjNNw,12232
+autocoder/index/entry.py,sha256=1KIGPCtxQN0OdErAco9OmGTd5hB8WJTpWGrxsGLsTcE,12634
 autocoder/index/for_command.py,sha256=BFvljE4t6VaMBGboZAuhUCzVK0EitCy_n5D_7FEnihw,3204
 autocoder/index/index.py,sha256=hVAIyF10N9hxKMWHA_ibYygGRZYJQZfZxRRrhQhrTvk,21225
 autocoder/index/symbols_utils.py,sha256=CjcjUVajmJZB75Ty3a7kMv1BZphrm-tIBAdOJv6uo-0,2037
 autocoder/index/types.py,sha256=a2s_KV5FJlq7jqA2ELSo9E1sjuLwDB-JJYMhSpzBAhU,596
 autocoder/index/filter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/index/filter/normal_filter.py,sha256=APu34iSvWhtlLtWgkj8N3Vo4oW1TegtZQq2bwDX_cs4,8031
-autocoder/index/filter/quick_filter.py,sha256=8R54qtcAAAu0C6TQgJ4WkqkryGOke1mJXrv82vnvMAA,9342
+autocoder/index/filter/quick_filter.py,sha256=Dsm23Z_RrJ_UwCypGUPN1BlKUMibae_9_D8jWD1UDFw,10518
 autocoder/pyproject/__init__.py,sha256=dQ2_7YZ7guybT9BhfxSGn43eLQJGQN2zgeKa6--JlaQ,14403
 autocoder/rag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/rag/api_server.py,sha256=dRbhAZVRAOlZ64Cnxf4_rKb4iJwHnrWS9Zr67IVORw0,7288
-autocoder/rag/doc_filter.py,sha256=P2L9u1zqL_igFE6DY5E2r0BY43TrpORjYb_bq7eKT50,6514
+autocoder/rag/doc_filter.py,sha256=ZCixxUXNBbz6UiGbgXvbDWdn5moLac3HnZEphpasTDc,6579
 autocoder/rag/document_retriever.py,sha256=5oThtxukGuRFF96o3pHKsk306a8diXbhgSrbqyU2BvM,8894
 autocoder/rag/llm_wrapper.py,sha256=sbDxCANiZyWb_ocqNgqu2oy3c2t8orPNRGleEs-Uwl8,2649
-autocoder/rag/long_context_rag.py,sha256=36ibIdFoOOasYbR1C3WmrfvQs3JB0oYSsk-O4YhTh58,25354
+autocoder/rag/long_context_rag.py,sha256=wmNmGsXN8RAFl6e9HaVzlwISXV9D-3bvf2qiaWjRy7w,24646
 autocoder/rag/rag_config.py,sha256=8LwFcTd8OJWWwi1_WY4IzjqgtT6RyE2j4PjxS5cCTDE,802
 autocoder/rag/rag_entry.py,sha256=6TKtErZ0Us9XSV6HgRKXA6yR3SiZGPHpynOKSaR1wgE,2463
 autocoder/rag/raw_rag.py,sha256=yS2Ur6kG0IRjhCj2_VonwxjY_xls_E62jO5Gz5j2nqE,2952
@@ -142,11 +142,11 @@ autocoder/utils/tests.py,sha256=BqphrwyycGAvs-5mhH8pKtMZdObwhFtJ5MC_ZAOiLq8,1340
 autocoder/utils/thread_utils.py,sha256=uXc9HEr1kSTi491lc-BhFUsbc0wb3ikltZNJkL7Ocz4,8195
 autocoder/utils/types.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/utils/auto_coder_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/utils/auto_coder_utils/chat_stream_out.py,sha256=wdunUiqCe6LntrEdxqx2iClwNGXhyL5kB52Y_4qHyNo,11014
+autocoder/utils/auto_coder_utils/chat_stream_out.py,sha256=xWXqICANbDOovH4wcFW1eSI7lB7TjXbk1mSU4bTKEW4,11434
 autocoder/utils/chat_auto_coder_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-auto_coder-0.1.252.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-auto_coder-0.1.252.dist-info/METADATA,sha256=I9oxYdimJoBFwT8BdU5eU6PUhuesvVq-CTZpF5s55GU,2616
-auto_coder-0.1.252.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-auto_coder-0.1.252.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
-auto_coder-0.1.252.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
-auto_coder-0.1.252.dist-info/RECORD,,
+auto_coder-0.1.254.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+auto_coder-0.1.254.dist-info/METADATA,sha256=a7eZ5YDrFy4_RSOLcTBzbdye35TnHukamLS-g3-jNkM,2616
+auto_coder-0.1.254.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+auto_coder-0.1.254.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
+auto_coder-0.1.254.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
+auto_coder-0.1.254.dist-info/RECORD,,

autocoder/auto_coder.py

@@ -1109,8 +1109,7 @@ def main(input_args: Optional[List[str]] = None):
             return
 
         elif raw_args.agent_command == "chat":
-            from autocoder.rag.rag_entry import RAGFactory
-
+            
             memory_dir = os.path.join(args.source_dir, ".auto-coder", "memory")
             os.makedirs(memory_dir, exist_ok=True)
             memory_file = os.path.join(memory_dir, "chat_history.json")
@@ -1335,6 +1334,7 @@ def main(input_args: Optional[List[str]] = None):
             start_time = time.time()    
                         
             if "rag" in args.action:
+                from autocoder.rag.rag_entry import RAGFactory
                 args.enable_rag_search = True
                 args.enable_rag_context = False
                 rag = RAGFactory.get_rag(llm=chat_llm, args=args, path="")
@@ -1348,7 +1348,8 @@ def main(input_args: Optional[List[str]] = None):
                 response = mcp_server.send_request(
                     McpRequest(
                         query=args.query,
-                        model=args.inference_model or args.model
+                        model=args.inference_model or args.model,
+                        product_mode=args.product_mode
                     )
                 )
                 v = [[response.result,None]]

autocoder/chat_auto_coder.py

@@ -49,7 +49,7 @@ from autocoder.common.mcp_server import get_mcp_server, McpRequest, McpInstallRe
 import byzerllm
 from byzerllm.utils import format_str_jinja2
 from autocoder.common.memory_manager import get_global_memory_file_paths 
-from autocoder import models
+from autocoder import models as models_module
 import shlex
 from autocoder.utils.llms import get_single_llm
 import pkg_resources
@@ -1539,11 +1539,12 @@ def mcp(query: str):
         if os.path.exists(temp_yaml):
             os.remove(temp_yaml)
 
-    mcp_server = get_mcp_server()
+    mcp_server = get_mcp_server()    
     response = mcp_server.send_request(
         McpRequest(
             query=query,
-            model=args.inference_model or args.model
+            model=args.inference_model or args.model,
+            product_mode=args.product_mode
         )
     )
 
@@ -1564,19 +1565,19 @@ def mcp(query: str):
         file_path = os.path.join(mcp_dir, f"{timestamp}.md")
         
         # Format response as markdown
-        markdown_content = f"# {printer.get_message_from_key('mcp_response_title')}\n\n{response.result}"
+        markdown_content = response.result
         
         # Save to file
         with open(file_path, "w", encoding="utf-8") as f:
             f.write(markdown_content)
             
-        # Print with markdown formatting
-        printer.print_panel(
-            Markdown(markdown_content),
-            text_options={"justify": "left"},
-            panel_options={
-                "border_style": "green"
-            }
+        console = Console()
+        console.print(
+            Panel(
+                Markdown(markdown_content, justify="left"),
+                title=printer.get_message_from_key('mcp_response_title'),
+                border_style="green"
+            )
         )
 
 
@@ -1741,7 +1742,7 @@ def commit(query: str):
             md5 = hashlib.md5(file_content.encode("utf-8")).hexdigest()
             file_name = os.path.basename(execute_file)
             commit_result = git_utils.commit_changes(
-                ".", f"auto_coder_{file_name}_{md5}"
+                ".", f"auto_coder_{file_name}_{md5}\n{commit_message}"
             )
             git_utils.print_commit_info(commit_result=commit_result)
             if commit_message:
@@ -2173,7 +2174,7 @@ def manage_models(params, query: str):
         printer.print_in_terminal("models_lite_only", style="red")
         return
         
-    models_data = models.load_models()
+    models_data = models_module.load_models()
     subcmd = ""
     if "/list" in query:
         subcmd = "/list"
@@ -2241,7 +2242,7 @@ def manage_models(params, query: str):
             name = args[0]
             try:
                 price = float(args[1])
-                if models.update_model_input_price(name, price):
+                if models_module.update_model_input_price(name, price):
                     printer.print_in_terminal("models_input_price_updated", style="green", name=name, price=price)
                 else:
                     printer.print_in_terminal("models_not_found", style="red", name=name)
@@ -2256,7 +2257,7 @@ def manage_models(params, query: str):
             name = args[0]
             try:
                 price = float(args[1])
-                if models.update_model_output_price(name, price):
+                if models_module.update_model_output_price(name, price):
                     printer.print_in_terminal("models_output_price_updated", style="green", name=name, price=price)
                 else:
                     printer.print_in_terminal("models_not_found", style="red", name=name)
@@ -2271,7 +2272,7 @@ def manage_models(params, query: str):
             name = args[0]
             try:
                 speed = float(args[1])
-                if models.update_model_speed(name, speed):
+                if models_module.update_model_speed(name, speed):
                     printer.print_in_terminal("models_speed_updated", style="green", name=name, speed=speed)
                 else:
                     printer.print_in_terminal("models_not_found", style="red", name=name)
@@ -2286,7 +2287,7 @@ def manage_models(params, query: str):
         if len(args) == 2:
             # Simplified: /models /add <name> <api_key>
             name, api_key = args[0], args[1]            
-            result = models.update_model_with_api_key(name, api_key)
+            result = models_module.update_model_with_api_key(name, api_key)
             if result:
                 printer.print_in_terminal("models_added", style="green", name=name)
             else:
@@ -2328,7 +2329,7 @@ def manage_models(params, query: str):
         }
 
         models_data.append(final_model)
-        models.save_models(models_data)
+        models_module.save_models(models_data)
         printer.print_in_terminal("models_add_model_success", style="green", name=data_dict["name"])
 
     elif subcmd == "/remove":
@@ -2341,7 +2342,7 @@ def manage_models(params, query: str):
         if len(filtered_models) == len(models_data):
             printer.print_in_terminal("models_add_model_remove", style="yellow", name=name)
             return
-        models.save_models(filtered_models)
+        models_module.save_models(filtered_models)
         printer.print_in_terminal("models_add_model_removed", style="green", name=name)
 
     else:

autocoder/common/__init__.py

@@ -359,6 +359,7 @@ class AutoCoderArgs(pydantic.BaseModel):
     product_mode: Optional[str] = "lite"
 
     keep_reasoning_content: Optional[bool] = False
+    keep_only_reasoning_content: Optional[bool] = False
 
     in_code_apply: bool = False
 

autocoder/common/code_auto_merge.py

@@ -177,5 +177,5 @@ class CodeAutoMerge:
 
         self.printer.print_in_terminal("files_merged", total=total)
         if not force_skip_git:
-            commit_result = git_utils.commit_changes(self.args.source_dir, f"auto_coder_{file_name}_{md5}")
+            commit_result = git_utils.commit_changes(self.args.source_dir, f"auto_coder_{file_name}_{md5}\n{self.args.query}")
             git_utils.print_commit_info(commit_result=commit_result)

autocoder/common/code_auto_merge_diff.py

@@ -523,5 +523,5 @@ class CodeAutoMergeDiff:
 
         self.printer.print_in_terminal("files_merged_total", total=total)
         if not force_skip_git:
-            commit_result = git_utils.commit_changes(self.args.source_dir, f"auto_coder_{file_name}_{md5}")
+            commit_result = git_utils.commit_changes(self.args.source_dir, f"auto_coder_{file_name}_{md5}\n{self.args.query}")
             git_utils.print_commit_info(commit_result=commit_result)

autocoder/common/code_auto_merge_editblock.py

@@ -403,7 +403,7 @@ class CodeAutoMergeEditBlock:
             if not force_skip_git:
                 try:
                     commit_result = git_utils.commit_changes(
-                        self.args.source_dir, f"auto_coder_{file_name}_{md5}"
+                        self.args.source_dir, f"auto_coder_{file_name}_{md5}\n{self.args.query}"
                     )
                     git_utils.print_commit_info(commit_result=commit_result)
                 except Exception as e:

autocoder/common/code_auto_merge_strict_diff.py

@@ -233,7 +233,7 @@ class CodeAutoMergeStrictDiff:
                             
         self.printer.print_in_terminal("files_merged_total", total=total)
         if not force_skip_git:
-            commit_result = git_utils.commit_changes(self.args.source_dir, f"auto_coder_{file_name}_{md5}")
+            commit_result = git_utils.commit_changes(self.args.source_dir, f"auto_coder_{file_name}_{md5}\n{self.args.query}")
             git_utils.print_commit_info(commit_result=commit_result)
 
     @byzerllm.prompt(render="jinja2")

autocoder/common/mcp_server.py

@@ -3,7 +3,6 @@ from asyncio import Queue as AsyncQueue
 import threading
 from typing import List, Dict, Any, Optional
 from dataclasses import dataclass
-import byzerllm
 from autocoder.common.mcp_hub import McpHub
 from autocoder.common.mcp_tools import McpExecutor
 from autocoder.common.mcp_hub import MCP_BUILD_IN_SERVERS
@@ -13,12 +12,13 @@ import time
 from pydantic import BaseModel
 import sys
 from loguru import logger
+from autocoder.utils.llms import get_single_llm
 
 @dataclass
 class McpRequest:
     query: str
     model: Optional[str] = None
-
+    product_mode: Optional[str] = None
 
 @dataclass
 class McpInstallRequest:
@@ -279,8 +279,7 @@ class McpServer:
                         await self._response_queue.put(McpResponse(result="", error=f"Failed to refresh MCP servers: {str(e)}"))
 
                 else:
-                    llm = byzerllm.ByzerLLM.from_default_model(
-                        model=request.model)
+                    llm = get_single_llm(request.model,product_mode=request.product_mode)
                     mcp_executor = McpExecutor(hub, llm)
                     conversations = [
                         {"role": "user", "content": request.query}]

autocoder/common/mcp_servers/mcp_server_perplexity.py

@@ -55,7 +55,7 @@ async def handle_list_tools() -> list[types.Tool]:
                             # Commenting out larger models,which have higher risks of timing out,
                             # until Claude Desktop can handle long-running tasks effectively.
                             # "llama-3.1-sonar-large-128k-online",
-                            "llama-3.1-sonar-huge-128k-online",
+                            "sonar-reasoning-pro",
                         ],
                     },
                     "messages": {

autocoder/index/entry.py

@@ -107,7 +107,13 @@ def build_index_and_filter_files(
                 model_name = "unknown(without default model name)"    
             printer.print_in_terminal("quick_filter_start", style="blue", model_name=model_name)
             quick_filter = QuickFilter(index_manager,stats,sources)        
-            final_files = quick_filter.filter(index_manager.read_index(),args.query)
+            quick_filter_result = quick_filter.filter(index_manager.read_index(),args.query)
+            if quick_filter_result.has_error:                
+                raise KeyboardInterrupt(printer.get_message_from_key_with_format("quick_filter_failed",error=quick_filter_result.error_message))
+
+            # Merge quick filter results into final_files
+            final_files.update(quick_filter_result.files)    
+
         
         if not args.skip_filter_index and not args.index_filter_model:
             model_name = getattr(index_manager.llm, 'default_model_name', None)
@@ -115,7 +121,8 @@ def build_index_and_filter_files(
                 model_name = "unknown(without default model name)" 
             printer.print_in_terminal("normal_filter_start", style="blue",model_name=model_name)
             normal_filter = NormalFilter(index_manager,stats,sources)
-            final_files = normal_filter.filter(index_manager.read_index(),args.query)
+            # Merge normal filter results into final_files
+            final_files.update(normal_filter.filter(index_manager.read_index(),args.query))
         
 
     def display_table_and_get_selections(data):

autocoder/index/filter/quick_filter.py

@@ -1,4 +1,5 @@
-from typing import List, Union, Dict, Any
+from typing import List, Union, Dict, Any, Optional
+from pydantic import BaseModel
 from autocoder.utils.auto_coder_utils.chat_stream_out import stream_out
 from autocoder.common.utils_code_auto_generate import stream_chat_with_continue
 from byzerllm.utils.str2model import to_model
@@ -26,6 +27,11 @@ def get_file_path(file_path):
     return file_path
 
 
+class QuickFilterResult(BaseModel):
+    files: Dict[str, TargetFile]
+    has_error: bool
+    error_message: Optional[str] = None
+
 class QuickFilter():
     def __init__(self, index_manager: IndexManager,stats:Dict[str,Any],sources:List[SourceCode]):
         self.index_manager = index_manager
@@ -36,11 +42,7 @@ class QuickFilter():
         self.max_tokens = self.args.index_filter_model_max_input_length
 
 
-    def big_filter(self, index_items: List[IndexItem],) -> Dict[str, TargetFile]:
-        
-        final_files: Dict[str, TargetFile] = {}
-        final_files_lock = threading.Lock()
-        
+    def big_filter(self, index_items: List[IndexItem],) -> QuickFilterResult:
         chunks = []
         current_chunk = []
                 
@@ -70,9 +72,10 @@ class QuickFilter():
                 split_size=len(chunks)
             )    
 
-        def process_chunk(chunk_index: int, chunk: List[IndexItem]) -> None:
+        def process_chunk(chunk_index: int, chunk: List[IndexItem]) -> QuickFilterResult:
             try:
                 model_name = ",".join(get_llm_names(self.index_manager.index_filter_llm))
+                files: Dict[str, TargetFile] = {}
                 
                 if chunk_index == 0:
                     # 第一个chunk使用流式输出
@@ -93,13 +96,16 @@ class QuickFilter():
                     file_number_list = self.quick_filter_files.with_llm(self.index_manager.index_filter_llm).with_return_type(FileNumberList).run(chunk, self.args.query)
                 
                 if file_number_list:
-                    with final_files_lock:
-                        for file_number in file_number_list.file_list:
-                            file_path = get_file_path(chunk[file_number].module_name)
-                            final_files[file_path] = TargetFile(
-                                file_path=chunk[file_number].module_name,
-                                reason=self.printer.get_message_from_key("quick_filter_reason")
-                            )
+                    for file_number in file_number_list.file_list:
+                        file_path = get_file_path(chunk[file_number].module_name)
+                        files[file_path] = TargetFile(
+                            file_path=chunk[file_number].module_name,
+                            reason=self.printer.get_message_from_key("quick_filter_reason")
+                        )
+                return QuickFilterResult(
+                    files=files,
+                    has_error=False
+                )
                             
             except Exception as e:
                 self.printer.print_in_terminal(
@@ -107,18 +113,40 @@ class QuickFilter():
                     style="red",
                     error=str(e)
                 )
+                return QuickFilterResult(
+                    files={},
+                    has_error=True,
+                    error_message=str(e)
+                )
 
+        results: List[QuickFilterResult] = []
         if chunks:
             with ThreadPoolExecutor() as executor:
-                # 提交所有chunks到线程池
+                # 提交所有chunks到线程池并收集结果
                 futures = [executor.submit(process_chunk, i, chunk) 
                           for i, chunk in enumerate(chunks)]
                 
-                # 等待所有任务完成
+                # 等待所有任务完成并收集结果
                 for future in futures:
-                    future.result()
-                
-        return final_files    
+                    results.append(future.result())
+        
+        # 合并所有结果
+        final_files: Dict[str, TargetFile] = {}
+        has_error = False
+        error_messages: List[str] = []
+        
+        for result in results:
+            if result.has_error:
+                has_error = True
+                if result.error_message:
+                    error_messages.append(result.error_message)
+            final_files.update(result.files)
+        
+        return QuickFilterResult(
+            files=final_files,
+            has_error=has_error,
+            error_message="\n".join(error_messages) if error_messages else None
+        )
 
     @byzerllm.prompt()
     def quick_filter_files(self,file_meta_list:List[IndexItem],query:str) -> str:
@@ -161,8 +189,8 @@ class QuickFilter():
         }
         return context    
 
-    def filter(self, index_items: List[IndexItem], query: str) -> Dict[str, TargetFile]:
-        final_files: Dict[str, TargetFile] = {}                
+    def filter(self, index_items: List[IndexItem], query: str) -> QuickFilterResult:
+        final_files: Dict[str, TargetFile] = {}
         start_time = time.monotonic()        
 
         prompt_str = self.quick_filter_files.prompt(index_items,query)            
@@ -217,7 +245,11 @@ class QuickFilter():
                 style="red",
                 error=str(e)
             )
-            return final_files
+            return QuickFilterResult(
+                files=final_files,
+                has_error=True,
+                error_message=str(e)
+            )
         
         if file_number_list:
             for file_number in file_number_list.file_list:
@@ -227,4 +259,7 @@ class QuickFilter():
                 )
         end_time = time.monotonic()            
         self.stats["timings"]["quick_filter"] = end_time - start_time            
-        return final_files
+        return QuickFilterResult(
+            files=final_files,
+            has_error=False
+        )    

autocoder/rag/doc_filter.py

@@ -64,7 +64,7 @@ class DocFilter:
             self.recall_llm = self.llm
         
         self.args = args
-        self.relevant_score = self.args.rag_doc_filter_relevance or 5
+        self.relevant_score = self.args.rag_doc_filter_relevance
         self.on_ray = on_ray
         self.path = path        
 
@@ -140,6 +140,7 @@ class DocFilter:
                     f"  - File: {doc.module_name}\n"
                     f"  - Relevance: {'Relevant' if relevance and relevance.is_relevant else 'Not Relevant'}\n"
                     f"  - Score: {relevance.relevant_score if relevance else 'N/A'}\n"
+                    f"  - Score Threshold: {self.relevant_score}\n"
                     f"  - Raw Response: {v}\n"
                     f"  - Timing:\n"
                     f"    * Total Duration: {task_timing.duration:.2f}s\n"
@@ -148,7 +149,7 @@ class DocFilter:
                 )
                 if (
                     relevance
-                    and relevance.is_relevant
+                    # and relevance.is_relevant
                     and relevance.relevant_score >= self.relevant_score
                 ):
                     relevant_docs.append(

autocoder/rag/long_context_rag.py

@@ -439,7 +439,7 @@ class LongContextRAG:
 
             logger.info(f"Query: {query} only_contexts: {only_contexts}")
             start_time = time.time()
-            relevant_docs: List[FilterDoc] = self._filter_docs(conversations)
+            relevant_docs: List[FilterDoc] = self._filter_docs(conversations)            
             filter_time = time.time() - start_time
 
             # Filter relevant_docs to only include those with is_relevant=True
@@ -449,20 +449,7 @@ class LongContextRAG:
 
             if highly_relevant_docs:
                 relevant_docs = highly_relevant_docs
-                logger.info(f"Found {len(relevant_docs)} highly relevant documents")
-            else:
-                if relevant_docs:
-                    prefix_chunk = FilterDoc(
-                        source_code=SourceCode(
-                            module_name="特殊说明",
-                            source_code="没有找到特别相关的内容，下面的内容是一些不是很相关的文档。在根据后续文档回答问题前，你需要和用户先提前说一下。",
-                        ),
-                        relevance=DocRelevance(False, 0),
-                    )
-                    relevant_docs.insert(0, prefix_chunk)
-                    logger.info(
-                        "No highly relevant documents found. Added a prefix chunk to indicate this."
-                    )
+                logger.info(f"Found {len(relevant_docs)} highly relevant documents")            
 
             logger.info(
                 f"Filter time: {filter_time:.2f} seconds with {len(relevant_docs)} docs"

autocoder/utils/auto_coder_utils/chat_stream_out.py

@@ -169,6 +169,10 @@ def stream_out(
     keep_reasoning_content = True
     if args:
         keep_reasoning_content = args.keep_reasoning_content            
+    
+    keep_only_reasoning_content = False
+    if args:
+        keep_only_reasoning_content = args.keep_only_reasoning_content
 
     lines_buffer = []  # 存储历史行
     current_line = ""  # 当前行
@@ -190,7 +194,10 @@ def stream_out(
                     break
                 last_meta = res[1]                
                 content = res[0]
-                reasoning_content = last_meta.reasoning_content
+
+                reasoning_content = ""
+                if last_meta:
+                    reasoning_content = last_meta.reasoning_content
 
                 if reasoning_content == "" and content == "":
                     continue
@@ -198,21 +205,24 @@ def stream_out(
                 if first_token_time == 0.0:
                     first_token_time = time.time() - first_token_time_start
 
-                if keep_reasoning_content:
-                    # 处理思考内容
-                    if reasoning_content:
-                        if assistant_response == "":  # 首次遇到思考内容时添加开标签
-                            assistant_response = "<thinking>"
-                        assistant_response += reasoning_content
-                        
-                    # 处理正式内容
-                    if content:
-                        # 如果之前有思考内容,需要先关闭thinking标签
-                        if "</thinking>" not in assistant_response and "<thinking>" in assistant_response:
-                            assistant_response += "</thinking>"
+                if keep_only_reasoning_content:
+                    assistant_response += reasoning_content
+                else:    
+                    if keep_reasoning_content:
+                        # 处理思考内容
+                        if reasoning_content:
+                            if assistant_response == "":  # 首次遇到思考内容时添加开标签
+                                assistant_response = "<thinking>"
+                            assistant_response += reasoning_content
+                            
+                        # 处理正式内容
+                        if content:
+                            # 如果之前有思考内容,需要先关闭thinking标签
+                            if "</thinking>" not in assistant_response and "<thinking>" in assistant_response:
+                                assistant_response += "</thinking>"
+                            assistant_response += content
+                    else:
                         assistant_response += content
-                else:
-                    assistant_response += content
 
                 display_delta = reasoning_content if reasoning_content else content
                 
@@ -291,5 +301,7 @@ def stream_out(
                     status=RequestOption.COMPLETED
                 ),
             )
-    last_meta.first_token_time = first_token_time
+
+    if last_meta:
+        last_meta.first_token_time = first_token_time
     return assistant_response, last_meta

autocoder/version.py

@@ -1 +1 @@
-__version__ = "0.1.252"
+__version__ = "0.1.254"