auto-coder 0.1.202__py3-none-any.whl → 0.1.204__py3-none-any.whl

{auto_coder-0.1.202.dist-info → auto_coder-0.1.204.dist-info}/METADATA

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

{auto_coder-0.1.202.dist-info → auto_coder-0.1.204.dist-info}/RECORD

@@ -1,18 +1,18 @@
 autocoder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/auto_coder.py,sha256=AX6CBb8rk4KeHTi-VPiWQRN8YHAcYAq5lekuQvlPM38,39037
+autocoder/auto_coder.py,sha256=IP4aSmuZh3HaIzLIThTKHF0PB38eTPUU1GDdfuJBam4,38971
 autocoder/auto_coder_lang.py,sha256=Rtupq6N3_HT7JRhDKdgCBcwRaiAnyCOR_Gsp4jUomrI,3229
-autocoder/auto_coder_rag.py,sha256=vRc3O0qSyRjiuBKD6Bfh3oMVchnO3VHoAeOz5j_jjlY,16578
+autocoder/auto_coder_rag.py,sha256=0JAQNHbrCiWJMFipr1ZIe9sTBNxSlHJj1T6tfeRt_Zc,17622
 autocoder/auto_coder_server.py,sha256=XU9b4SBH7zjPPXaTWWHV4_zJm-XYa6njuLQaplYJH_c,20290
-autocoder/chat_auto_coder.py,sha256=kgcD4HKKmvDd2CI048TqZwdx2hrQRbZKYPO10qyrbA8,82105
-autocoder/chat_auto_coder_lang.py,sha256=QYtu5gWEQmWKVovR_qUZ8plySZarNFX_Onk-1vN9IiA,8524
-autocoder/command_args.py,sha256=MBsVjZpADu5u7CjY_1F8fGUW-9dVGyNDpzWyrDIMxz8,29890
+autocoder/chat_auto_coder.py,sha256=GXyCrX7GTYHYTCg2-62LYuO9Vt7Y5DdgfpCoTCQbYfE,86488
+autocoder/chat_auto_coder_lang.py,sha256=zU9VRY-l80fZnLJ0Op8A3wq27UhQHh9WcpSYU4SmnqU,8708
+autocoder/command_args.py,sha256=nmVD3xgLKKNt3fLJ4wVNUdJBtBOkfAdZ2ZmaRMlS7qg,29891
 autocoder/lang.py,sha256=Ajng6m7towmx-cvQfEHPFp43iEfddPvr8ju5GH4H8qA,13819
-autocoder/version.py,sha256=KbKAfOQ1mRxioaKsb1VhjDtdk7RG8UIteq2BswiztKI,24
+autocoder/version.py,sha256=SlTxvgnTF5p3a0pdXcWVuAco21_eHUa0QV_CmfrzNEY,24
 autocoder/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/agent/auto_tool.py,sha256=DBzip-P_T6ZtT2eHexPcusmKYD0h7ufzp7TLwXAY10E,11554
 autocoder/agent/coder.py,sha256=x6bdJwDuETGg9ebQnYlUWCxCtQcDGg73LtI6McpWslQ,72034
 autocoder/agent/designer.py,sha256=EpRbzO58Xym3GrnppIT1Z8ZFAlnNfgzHbIzZ3PX-Yv8,27037
-autocoder/agent/planner.py,sha256=vwCY0SiPoKFQZCZnTVlOIBnVzkgIu1uk-xMC0d6_TNw,9048
+autocoder/agent/planner.py,sha256=wvvuX9o6yhWlGm9t7fIT0wiChV5NkXBrYElxKiAjldU,9127
 autocoder/agent/project_reader.py,sha256=tWLaPoLw1gI6kO_NzivQj28KbobU2ceOLuppHMbfGl8,18234
 autocoder/chat/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/common/JupyterClient.py,sha256=O-wi6pXeAEYhAY24kDa0BINrLYvKS6rKyWe98pDClS0,2816
@@ -22,10 +22,10 @@ autocoder/common/anything2images.py,sha256=0ILBbWzY02M-CiWB-vzuomb_J1hVdxRcenAfI
 autocoder/common/audio.py,sha256=Kn9nWKQddWnUrAz0a_ZUgjcu4VUU_IcZBigT7n3N3qc,7439
 autocoder/common/cleaner.py,sha256=NU72i8C6o9m0vXExab7nao5bstBUsfJFcj11cXa9l4U,1089
 autocoder/common/code_auto_execute.py,sha256=4KXGmiGObr_B1d6tzV9dwS6MifCSc3Gm4j2d6ildBXQ,6867
-autocoder/common/code_auto_generate.py,sha256=mZkua-qGa-j6FCFeh-FSVsDnWU6ix51FaY7HBf7lYTg,8081
-autocoder/common/code_auto_generate_diff.py,sha256=By7ai1EO_5IgyVWDn9iIvJ658VudauYIusUiwcVP4aE,13197
-autocoder/common/code_auto_generate_editblock.py,sha256=MIhtoPnVzbxP-VanpruyKXX5OO5jiWEDlENkzhChPiE,15826
-autocoder/common/code_auto_generate_strict_diff.py,sha256=kqGRCLsIsfwTDMPWo5EVifJ3lWgWxcF1y9FHeensNJY,11807
+autocoder/common/code_auto_generate.py,sha256=bREGPj2yNQ1oy35ivHRHQSXAdmIsPaW4HQ9BkO7Ytco,8099
+autocoder/common/code_auto_generate_diff.py,sha256=sItR67cGIY0QXOgJcK28HCHhhloFpAALIDC7zz5DpBM,13215
+autocoder/common/code_auto_generate_editblock.py,sha256=X-F13eEjM5Gl778FdyFQo-Khvvr3mOdbROxA7f6bK-Y,15844
+autocoder/common/code_auto_generate_strict_diff.py,sha256=APSAbH6cKDQwAOLkhoNNvenY1s7fehZxc5_YBKkJghg,11825
 autocoder/common/code_auto_merge.py,sha256=3PB2fQtqgA0wPQSdwFqoOHfNT-mORgGfOXjfRGflikQ,5141
 autocoder/common/code_auto_merge_diff.py,sha256=b5INrBJJ2EI0uj5AIiMobCS5S9xevCvdescs3nzCPhY,12594
 autocoder/common/code_auto_merge_editblock.py,sha256=gIM90aZLKLHft4OzfUuAB6dWx7ntbnrqzZiJuOVukJQ,13774
@@ -34,14 +34,14 @@ autocoder/common/command_completer.py,sha256=pmPpdfGhpNV3v4u3KadL1BEG5H0QRO6eGv9
 autocoder/common/command_generator.py,sha256=v4LmU7sO-P7jEZIXCWHUC6P-vT7AvBi_x_PTwCqBAE8,1323
 autocoder/common/command_templates.py,sha256=3G-pCNbL6iHbnkG6v1JZpbIK3Mc9d373_RYGmCcDPMY,8548
 autocoder/common/const.py,sha256=eTjhjh4Aj4CUzviJ81jaf3Y5cwqsLATySn2wJxaS6RQ,2911
-autocoder/common/git_utils.py,sha256=UdsbUC8SpQdWt2VZ-6px6GQBWRvSJotD4IuDM4J7NBk,6598
+autocoder/common/git_utils.py,sha256=btK45sxvfm4tX3fBRNUPRZoGQuZuOEQrWSAwLy1yoLw,23095
 autocoder/common/image_to_page.py,sha256=O0cNO_vHHUP-fP4GXiVojShmNqkPnZXeIyiY1MRLpKg,13936
 autocoder/common/interpreter.py,sha256=62-dIakOunYB4yjmX8SHC0Gdy2h8NtxdgbpdqRZJ5vk,2833
 autocoder/common/llm_rerank.py,sha256=FbvtCzaR661Mt2wn0qsuiEL1Y3puD6jeIJS4zg_e7Bs,3260
 autocoder/common/screenshots.py,sha256=_gA-z1HxGjPShBrtgkdideq58MG6rqFB2qMUJKjrycs,3769
 autocoder/common/search.py,sha256=_ZX03ph89rDPGMY1OrfqaDfxsDR-flh6YEHixherjwM,16616
 autocoder/common/search_replace.py,sha256=GphFkc57Hb673CAwmbiocqTbw8vrV7TrZxtOhD0332g,22147
-autocoder/common/sys_prompt.py,sha256=Ul-lZQBoc2rwydccvKm0UsGtxQtUJfS81K1WJcd5Oao,26389
+autocoder/common/sys_prompt.py,sha256=JlexfjZt554faqbgkCmzOJqYUzDHfbnxly5ugFfHfEE,26403
 autocoder/common/text.py,sha256=KGRQq314GHBmY4MWG8ossRoQi1_DTotvhxchpn78c-k,1003
 autocoder/common/types.py,sha256=_kl8q2oCAZe-HqU9LjQgSqVHAw4ojMojlF58-bp2xiE,354
 autocoder/db/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -60,10 +60,10 @@ autocoder/index/symbols_utils.py,sha256=CjcjUVajmJZB75Ty3a7kMv1BZphrm-tIBAdOJv6u
 autocoder/pyproject/__init__.py,sha256=RrCY_gyUHK7IFhjOvNRs_ckMsO6Mw7pRymWEjzpr2Rw,14354
 autocoder/rag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/rag/api_server.py,sha256=dRbhAZVRAOlZ64Cnxf4_rKb4iJwHnrWS9Zr67IVORw0,7288
-autocoder/rag/doc_filter.py,sha256=Ha0Yae_G_hF72YzvrO7NoDZcG18K4hRcqGAEqfrIwAs,9330
+autocoder/rag/doc_filter.py,sha256=H0g4s-Q8gJDUbYk9b1qkEyl7RgNzLbRF4UWELVKPDwk,9865
 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=Y0bpO0mNOBGUtQ8WBOSbrjFLKxRUcZnD-Dax1zU7q-I,24436
+autocoder/rag/long_context_rag.py,sha256=nrARGBJkqA8S5aiEE3jDjVdB_M2n3FDix6hmCygGGBA,24626
 autocoder/rag/rag_config.py,sha256=8LwFcTd8OJWWwi1_WY4IzjqgtT6RyE2j4PjxS5cCTDE,802
 autocoder/rag/rag_entry.py,sha256=V1RJ8RGqM30DNPmzymv64rZjNRGWn6kfc8sRy_LECg0,2451
 autocoder/rag/raw_rag.py,sha256=yS2Ur6kG0IRjhCj2_VonwxjY_xls_E62jO5Gz5j2nqE,2952
@@ -72,7 +72,7 @@ autocoder/rag/simple_directory_reader.py,sha256=LkKreCkNdEOoL4fNhc3_hDoyyWTQUte4
 autocoder/rag/simple_rag.py,sha256=I902EUqOK1WM0Y2WFd7RzDJYofElvTZNLVCBtX5A9rc,14885
 autocoder/rag/token_checker.py,sha256=jc76x6KWmvVxds6W8juZfQGaoErudc2HenG3sNQfSLs,2819
 autocoder/rag/token_counter.py,sha256=LReZEYXmWriDI3KYjUvK0E4Gn4MeDJX6RPCfJCmluGY,2110
-autocoder/rag/token_limiter.py,sha256=50sq3pid6up8MBc8jESEqeZFzXN7z8-DxcwTPRHuB9Y,10935
+autocoder/rag/token_limiter.py,sha256=dtIxCtHswZ2ut-XKbx8_SiWyv-xqnR1WAIcmh6f8Ktw,11137
 autocoder/rag/types.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/rag/utils.py,sha256=MOMy0l2_YQ8foWnQQSvYmYAwghAWJ8_wVrdZh5DSaTg,4822
 autocoder/rag/variable_holder.py,sha256=PFvBjFcR7-fNDD4Vcsc8CpH2Te057vcpwJMxtrfUgKI,75
@@ -93,7 +93,7 @@ autocoder/regex_project/__init__.py,sha256=EBZeCL5ORyD_9_5u_UuG4s7XtpXOu0y1sWDmx
 autocoder/regexproject/__init__.py,sha256=lHTpHfYkguCMtczXoH4bMr-IMNZQtXIjmtSvjtfX0Ro,9674
 autocoder/suffixproject/__init__.py,sha256=bEPW9AyGSQ8kNzrgKEXyRikrUCUEuJ6b6vCLzO8Ja6g,11017
 autocoder/tsproject/__init__.py,sha256=avSnMA3uSdZmv5MxTilDPFLRFmtfzFr1NPhHaRMFhX4,11625
-autocoder/utils/__init__.py,sha256=O3n6cpsgkIbbMuwmBHSQ1dls_IBD7_7YKFFaeKNo_tc,1193
+autocoder/utils/__init__.py,sha256=KtcGElFNBgZPF7dEL8zF9JpXkCAjoyDrzaREJBhJrcs,994
 autocoder/utils/_markitdown.py,sha256=RU88qn4eZfYIy0GDrPxlI8oYXIypbi63VRJjdlnE0VU,47431
 autocoder/utils/coder.py,sha256=rK8e0svQBe0NOP26dIGToUXgha_hUDgxlWoC_p_r7oc,5698
 autocoder/utils/conversation_store.py,sha256=sz-hhY7sttPAUOAQU6Pze-5zJc3j0_Emj22dM_0l5ro,1161
@@ -106,9 +106,9 @@ autocoder/utils/request_event_queue.py,sha256=r3lo5qGsB1dIjzVQ05dnr0z_9Z3zOkBdP1
 autocoder/utils/request_queue.py,sha256=nwp6PMtgTCiuwJI24p8OLNZjUiprC-TsefQrhMI-yPE,3889
 autocoder/utils/rest.py,sha256=HawagAap3wMIDROGhY1730zSZrJR_EycODAA5qOj83c,8807
 autocoder/utils/tests.py,sha256=BqphrwyycGAvs-5mhH8pKtMZdObwhFtJ5MC_ZAOiLq8,1340
-auto_coder-0.1.202.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-auto_coder-0.1.202.dist-info/METADATA,sha256=JU3TFfBNRhy5U-nqB2vFy2gLKLQs-E89lUX3B0MrNj8,2575
-auto_coder-0.1.202.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-auto_coder-0.1.202.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
-auto_coder-0.1.202.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
-auto_coder-0.1.202.dist-info/RECORD,,
+auto_coder-0.1.204.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+auto_coder-0.1.204.dist-info/METADATA,sha256=DDQpyvBKzzC3fplWR0n7JCSamGxTdad-0WFMUEnjbuw,2575
+auto_coder-0.1.204.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+auto_coder-0.1.204.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
+auto_coder-0.1.204.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
+auto_coder-0.1.204.dist-info/RECORD,,

autocoder/agent/planner.py

@@ -102,18 +102,20 @@ def get_tools(args: AutoCoderArgs, llm: byzerllm.ByzerLLM):
             return
 
         action_files = [
-            f
-            for f in os.listdir(actions_dir)
-            if f[:3].isdigit() and f.endswith(".yml")
+            f for f in os.listdir(actions_dir) if f[:3].isdigit() and "_" in f and f.endswith(".yml")
         ]
+
+        def get_old_seq(name):
+            return name.split("_")[0]
+
         if not action_files:
             max_seq = 0
         else:
-            seqs = [int(f[:3]) for f in action_files]
+            seqs = [int(get_old_seq(f)) for f in action_files]
             max_seq = max(seqs)
 
-        new_seq = str(max_seq + 1).zfill(3)
-        prev_files = [f for f in action_files if int(f[:3]) < int(new_seq)]
+        new_seq = str(max_seq + 1).zfill(12)
+        prev_files = [f for f in action_files if int(get_old_seq(f)) < int(new_seq)]
 
         if not prev_files:
             new_file = os.path.join(actions_dir, f"{new_seq}_{yaml_file_name}.yml")

autocoder/auto_coder.py

@@ -194,16 +194,20 @@ def main(input_args: Optional[List[str]] = None):
             return
 
         action_files = [
-            f for f in os.listdir(actions_dir) if f[:3].isdigit() and f.endswith(".yml")
+            f for f in os.listdir(actions_dir) if f[:3].isdigit() and "_" in f and f.endswith(".yml")
         ]
+
+        def get_old_seq(name):
+            return name.split("_")[0]
+
         if not action_files:
             max_seq = 0
         else:
-            seqs = [int(f[:3]) for f in action_files]
+            seqs = [int(get_old_seq(f)) for f in action_files]
             max_seq = max(seqs)
 
-        new_seq = str(max_seq + 1).zfill(3)
-        prev_files = [f for f in action_files if int(f[:3]) < int(new_seq)]
+        new_seq = str(max_seq + 1).zfill(12)
+        prev_files = [f for f in action_files if int(get_old_seq(f)) < int(new_seq)]
 
         if raw_args.from_yaml:
             # If --from_yaml is specified, copy content from the matching YAML file
@@ -489,7 +493,7 @@ def main(input_args: Optional[List[str]] = None):
         from autocoder.index.for_command import index_query_command
 
         index_query_command(args, llm)
-        return
+        return   
 
     if raw_args.command == "agent":
         if raw_args.agent_command == "planner":
@@ -763,7 +767,7 @@ def main(input_args: Optional[List[str]] = None):
 
             loaded_conversations = pre_conversations + \
                 chat_history["ask_conversation"]
-                        
+
             if args.human_as_model:
                 console = Console()
 
@@ -821,8 +825,8 @@ def main(input_args: Optional[List[str]] = None):
                             border_style="blue",
                             expand=False,
                         )
-                    )                                    
-                if args.request_id:                    
+                    )
+                if args.request_id:
                     request_queue.add_request(
                         args.request_id,
                         RequestValue(
@@ -836,7 +840,7 @@ def main(input_args: Optional[List[str]] = None):
                         ),
                     )
                     return {}
-                                
+
                 lines = []
                 while True:
                     line = prompt(FormattedText(
@@ -875,12 +879,12 @@ def main(input_args: Optional[List[str]] = None):
 
                 return {}
 
-            if args.enable_rag_search or args.enable_rag_context:                
+            if args.enable_rag_search or args.enable_rag_context:
                 rag = RAGFactory.get_rag(llm=chat_llm, args=args, path="")
                 response = rag.stream_chat_oai(
                     conversations=loaded_conversations)[0]
                 v = ([item, None] for item in response)
-            else:                
+            else:
                 v = chat_llm.stream_chat_oai(
                     conversations=loaded_conversations, delta_mode=True
                 )
@@ -896,7 +900,7 @@ def main(input_args: Optional[List[str]] = None):
                     for res in v:
                         markdown_content += res[0]
                         assistant_response += res[0]
-                        if args.request_id:                            
+                        if args.request_id:
                             request_queue.add_request(
                                 args.request_id,
                                 RequestValue(

autocoder/auto_coder_rag.py

@@ -286,6 +286,24 @@ def main(input_args: Optional[List[str]] = None):
         help="Whether to return responses without contexts. only works when pro plugin is installed",
     )
 
+    serve_parser.add_argument(
+        "--recall_model",
+        default="",
+        help="The model used for recall documents",
+    )
+
+    serve_parser.add_argument(
+        "--chunk_model",
+        default="",
+        help="The model used for chunk documents",
+    )
+
+    serve_parser.add_argument(
+        "--qa_model",
+        default="",
+        help="The model used for question answering",
+    )
+
     # Tools command
     tools_parser = subparsers.add_parser("tools", help="Various tools")
     tools_subparsers = tools_parser.add_subparsers(dest="tool", help="Available tools")
@@ -334,6 +352,22 @@ def main(input_args: Optional[List[str]] = None):
         llm = byzerllm.ByzerLLM()
         llm.setup_default_model_name(args.model)
 
+        # Setup sub models if specified
+        if args.recall_model:
+            recall_model = byzerllm.ByzerLLM()
+            recall_model.setup_default_model_name(args.recall_model)
+            llm.setup_sub_client("recall_model", recall_model)
+
+        if args.chunk_model:
+            chunk_model = byzerllm.ByzerLLM()
+            chunk_model.setup_default_model_name(args.chunk_model)
+            llm.setup_sub_client("chunk_model", chunk_model)
+
+        if args.qa_model:
+            qa_model = byzerllm.ByzerLLM()
+            qa_model.setup_default_model_name(args.qa_model)
+            llm.setup_sub_client("qa_model", qa_model)
+
         # 当启用hybrid_index时,检查必要的组件
         if auto_coder_args.enable_hybrid_index:                           
             if not llm.is_model_exist("emb"):