auto-coder 0.1.326__tar.gz → 0.1.327__tar.gz

{auto_coder-0.1.326 → auto_coder-0.1.327}/PKG-INFO

@@ -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 → auto_coder-0.1.327}/src/auto_coder.egg-info/PKG-INFO

@@ -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 → auto_coder-0.1.327}/src/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"

{auto_coder-0.1.326 → auto_coder-0.1.327}/src/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 = ""

{auto_coder-0.1.326 → auto_coder-0.1.327}/src/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):

auto_coder-0.1.327/src/autocoder/version.py

@@ -0,0 +1 @@
+__version__ = "0.1.327"

auto_coder-0.1.326/src/autocoder/version.py

@@ -1 +0,0 @@
-__version__ = "0.1.326"