auto-coder 0.1.326__py3-none-any.whl → 0.1.327__py3-none-any.whl

{auto_coder-0.1.326.dist-info → auto_coder-0.1.327.dist-info}/METADATA

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

{auto_coder-0.1.326.dist-info → auto_coder-0.1.327.dist-info}/RECORD

@@ -14,7 +14,7 @@ autocoder/command_parser.py,sha256=fx1g9E6GaM273lGTcJqaFQ-hoksS_Ik2glBMnVltPCE,1
 autocoder/lang.py,sha256=U6AjVV8Rs1uLyjFCZ8sT6WWuNUxMBqkXXIOs4S120uk,14511
 autocoder/models.py,sha256=AyoZ-Pzy0oyYUmWCxOIRiOImsqboSfRET7LO9-UOuxI,11172
 autocoder/run_context.py,sha256=IUfSO6_gp2Wt1blFWAmOpN0b0nDrTTk4LmtCYUBIoro,1643
-autocoder/version.py,sha256=YGdEBe_w43bQfdo8_Mx3DWCOSWhBc0wXQua_nK1cUuo,23
+autocoder/version.py,sha256=miEiKBcN07ycv510z1gvZ9J70UiJ9CPTI0zhuZXmYxg,23
 autocoder/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/agent/auto_demand_organizer.py,sha256=URAq0gSEiHeV_W4zwhOI_83kHz0Ryfj1gcfh5jwCv_w,6501
 autocoder/agent/auto_filegroup.py,sha256=pBsAkBcpFTff-9L5OwI8xhf2xPKpl-aZwz-skF2B6dc,6296
@@ -83,7 +83,7 @@ autocoder/common/search.py,sha256=245iPFgWhMldoUK3CqCP89ltaxZiNPK73evoG6Fp1h8,16
 autocoder/common/search_replace.py,sha256=GphFkc57Hb673CAwmbiocqTbw8vrV7TrZxtOhD0332g,22147
 autocoder/common/shells.py,sha256=elminFpNosnV0hsEUcsugDxlGO8NfH96uah-8bkaBvA,19929
 autocoder/common/stats_panel.py,sha256=wGl9O45pjVVDxhNumLv4_NfLYSlUP_18Tw4hcJSjw50,4596
-autocoder/common/stream_out_type.py,sha256=XWONDrdNsWQTSzloVJSwbKD_Tacxe9mX_ozXdl-5VIc,379
+autocoder/common/stream_out_type.py,sha256=PTkrY4_QEAFBPTcc_VUzb2dLL8XnRtV7MypNFGMjrHs,443
 autocoder/common/sys_prompt.py,sha256=JlexfjZt554faqbgkCmzOJqYUzDHfbnxly5ugFfHfEE,26403
 autocoder/common/text.py,sha256=KGRQq314GHBmY4MWG8ossRoQi1_DTotvhxchpn78c-k,1003
 autocoder/common/token_cost_caculate.py,sha256=MSWJtl7YpQSUt-gFQoqUcJMblyPqHXe2ZioiZOFkV80,10085
@@ -117,7 +117,7 @@ autocoder/dispacher/actions/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQ
 autocoder/dispacher/actions/plugins/action_regex_project.py,sha256=Vp07ANh0LlHoVPLM9ovTiIKv1gsUuCF57Gje4XiFVDA,8000
 autocoder/dispacher/actions/plugins/action_translate.py,sha256=GEn7dZA22jy5WyzINomjmzzB795p2Olg-CJla97lRF8,7744
 autocoder/events/__init__.py,sha256=1x_juwr9Ows2RADDa2LyI4QlmPxOVOXZeLO1cht-slM,1443
-autocoder/events/event_content.py,sha256=0unVmskGrfvnsniLvNXobco9KG-r6cv_C-gXK3L5zXQ,11557
+autocoder/events/event_content.py,sha256=g55Z55s-ud9IzYp6u7BnLSsHHAAg34w1jrHo-C1w68A,12280
 autocoder/events/event_manager.py,sha256=WsEog4diXPsidCcN-Ycm18iPbX21PBibkcmjee-aRcQ,11739
 autocoder/events/event_manager_singleton.py,sha256=8aBp97fgbas0KYrpHbljTg2n4RcWd85SOgQITu47ZWA,8809
 autocoder/events/event_store.py,sha256=y6tT3P-o3yhDptrKi-UmqI_ZBNg7v21FriI3f7lo_ME,12709
@@ -127,7 +127,7 @@ autocoder/helper/project_creator.py,sha256=RITjVfZ_Y5zytSHTNYiwxfn2Eqfvf2Ywe13e6
 autocoder/index/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/index/entry.py,sha256=a8KhkNd6tKuD6W9fh9ohdri2r4UYeZWswi14pVwpr-Q,15740
 autocoder/index/for_command.py,sha256=BFvljE4t6VaMBGboZAuhUCzVK0EitCy_n5D_7FEnihw,3204
-autocoder/index/index.py,sha256=Y72N_w6tdI-Lv3GhgGDeCKxKfSzhxpdVuZpSAdJTsDU,30831
+autocoder/index/index.py,sha256=-_GtT67ERxqdI4ajDyWr2cTsordpG_G9SYSwbTxDAcA,32491
 autocoder/index/symbols_utils.py,sha256=_EP7E_qWXxluAxq3FGZLlLfdrfwx3FmxCdulI8VGuac,2244
 autocoder/index/types.py,sha256=a2s_KV5FJlq7jqA2ELSo9E1sjuLwDB-JJYMhSpzBAhU,596
 autocoder/index/filter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -226,9 +226,9 @@ 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=xuBeWD0YOckqRo8JB1WkVIMOYH6c24m7JfV4svBfPDo,15113
 autocoder/utils/chat_auto_coder_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-auto_coder-0.1.326.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-auto_coder-0.1.326.dist-info/METADATA,sha256=aHhdCGHnZfnbnfXPguIXzw8gTa4xtOPdRMrvr3pjgbs,2747
-auto_coder-0.1.326.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-auto_coder-0.1.326.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
-auto_coder-0.1.326.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
-auto_coder-0.1.326.dist-info/RECORD,,
+auto_coder-0.1.327.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+auto_coder-0.1.327.dist-info/METADATA,sha256=xGfc33ogoMBS7t9utBeDgZs3YW_7yRnJWaOb0tZuXX0,2747
+auto_coder-0.1.327.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+auto_coder-0.1.327.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
+auto_coder-0.1.327.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
+auto_coder-0.1.327.dist-info/RECORD,,

autocoder/common/stream_out_type.py

@@ -15,3 +15,6 @@ class CodeRankStreamOutType(Enum):
 
 class LintStreamOutType(Enum):
     LINT = "lint"
+
+class IndexStreamOutType(Enum):
+    INDEX_BUILD = "index_build"

autocoder/events/event_content.py

@@ -119,6 +119,35 @@ class ResultTokenStatContent(BaseModel):
         """转换为JSON字符串"""
         return self.model_dump_json()
 
+class IndexBuildStartContent(BaseModel):
+    file_number:int = 0
+    total_files:int = 0
+
+    def to_dict(self) -> Dict[str, Any]:
+        """转换为字典"""
+        return self.model_dump()
+    
+    def to_json(self) -> str:
+        """转换为JSON字符串"""
+        return self.model_dump_json()
+
+class IndexBuildEndContent(BaseModel):
+    updated_files:int = 0
+    removed_files:int = 0
+    input_tokens:int = 0
+    output_tokens:int = 0
+    input_cost:float = 0.0
+    output_cost:float = 0.0
+
+    def to_dict(self) -> Dict[str, Any]:
+        """转换为字典"""
+        return self.model_dump()
+    
+    def to_json(self) -> str:
+        """转换为JSON字符串"""
+        return self.model_dump_json()
+    
+
 
 class ResultCommandPrepareStatContent(BaseModel):
     command:str = ""

autocoder/index/index.py

@@ -28,7 +28,9 @@ from autocoder.index.types import (
 from autocoder.common.global_cancel import global_cancel
 from autocoder.utils.llms import get_llm_names
 from autocoder.rag.token_counter import count_tokens
-
+from autocoder.common.stream_out_type import IndexStreamOutType
+from autocoder.events.event_manager_singleton import get_event_manager
+from autocoder.events import event_content as EventContentCreator
 
 class IndexManager:
     def __init__(
@@ -190,7 +192,7 @@ class IndexManager:
         {{ code }}
 
         ## 输出
-        """    
+        """
 
     @byzerllm.prompt()
     def jsp_get_all_file_symbols(self, path: str, code: str) -> str:
@@ -224,11 +226,11 @@ class IndexManager:
         url : "admWorkingDay!updateAdmWorkingDayList.action",  中，那么对应的为 <ajax method="post" url="admWorkingDay!updateAdmWorkingDayList.action">
         10. 导入语句中需要包含超连接，比如 <a href="admRiskLimits!queryAdmRiskLimitsById.action?admRiskLimits.id=${fn:escapeXml(item.id)}">${fn:escapeXml(item.classification)}， 对应的为 <a href="admRiskLimits!queryAdmRiskLimitsById.action"}
         11. 导入语句中需要包含一些js函数里的值，比如 window.onbeforeunload = function()  {
-	var actionType = document.ap_ActionPlanForm.action.value;
-	if(typeof(window.opener.reloadApCount)!="undefined"&&(actionType&&actionType!='Discard')){
-		}
+        var actionType = document.ap_ActionPlanForm.action.value;
+        if(typeof(window.opener.reloadApCount)!="undefined"&&(actionType&&actionType!='Discard')){
+                }
 }  对应为 <window.onbeforeunload url="from 表单名字为ap_ActionPlanForm的action的值">
-        
+
         下面是一段示例：
 
         ## 输入
@@ -363,14 +365,14 @@ class IndexManager:
             else:
                 meta_holder = byzerllm.MetaHolder()
                 ext = os.path.splitext(file_path)[1].lower()
-                ## 需要有特殊prompt
+                # 需要有特殊prompt
                 if ext == ".jsp":
                     symbols = self.jsp_get_all_file_symbols.with_llm(
                         self.index_llm).with_meta(meta_holder).run(source.module_name, source_code)
                 else:
                     symbols = self.get_all_file_symbols.with_llm(
                         self.index_llm).with_meta(meta_holder).run(source.module_name, source_code)
-                
+
                 time.sleep(self.anti_quota_limit)
 
                 if meta_holder.get_meta():
@@ -478,7 +480,8 @@ class IndexManager:
 
         # 删除被排除的文件
         try:
-            exclude_patterns = self.parse_exclude_files(self.args.exclude_files)            
+            exclude_patterns = self.parse_exclude_files(
+                self.args.exclude_files)
             for file_path in index_data:
                 if self.filter_exclude_files(file_path, exclude_patterns):
                     keys_to_remove.append(file_path)
@@ -546,6 +549,19 @@ class IndexManager:
                 num_files=num_files
             )
 
+            get_event_manager(self.args.event_file).write_result(
+                EventContentCreator.create_result(
+                    content=EventContentCreator.IndexBuildStartContent(
+                        file_number=num_files,
+                        total_files=total_files
+                    ).to_dict(),
+                    metadata={
+                        "stream_out_type": IndexStreamOutType.INDEX_BUILD.value,
+                        "action_file": self.args.file
+                    }
+                )
+            )
+
             futures = [
                 executor.submit(self.build_index_for_single_source, source)
                 for source in wait_to_build_files
@@ -592,6 +608,24 @@ class IndexManager:
             )
             print("")
 
+            # 记录索引构建完成事件                            
+            get_event_manager(self.args.event_file).write_result(
+                EventContentCreator.create_result(
+                    content=EventContentCreator.IndexBuildEndContent(
+                        updated_files=len(updated_sources),
+                        removed_files=len(keys_to_remove),
+                        input_tokens=total_input_tokens,
+                        output_tokens=total_output_tokens,
+                        input_cost=total_input_cost,
+                        output_cost=total_output_cost
+                    ).to_dict(),
+                    metadata={}
+                    ).to_dict(),
+                    metadata={
+                        "stream_out_type": IndexStreamOutType.INDEX_BUILD.value,
+                        "action_file": self.args.file
+                    }
+                )            
         return index_data
 
     def read_index_as_str(self):

autocoder/version.py

@@ -1 +1 @@
-__version__ = "0.1.326"
+__version__ = "0.1.327"