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

autocoder/rag/token_limiter.py

@@ -16,7 +16,7 @@ class TokenLimiter:
         full_text_limit: int,
         segment_limit: int,
         buff_limit: int,
-        llm,
+        llm:ByzerLLM,
         disable_segment_reorder: bool,
     ):
         self.count_tokens = count_tokens
@@ -24,6 +24,10 @@ class TokenLimiter:
         self.segment_limit = segment_limit
         self.buff_limit = buff_limit
         self.llm = llm
+        if self.llm.get_sub_client("chunk_model"):
+            self.chunk_llm = self.llm.get_sub_client("chunk_model")
+        else:
+            self.chunk_llm = self.llm
         self.first_round_full_docs = []
         self.second_round_extracted_docs = []
         self.sencond_round_time = 0
@@ -219,10 +223,10 @@ class TokenLimiter:
                 source_code_lines = doc.source_code.split("\n")
                 for idx, line in enumerate(source_code_lines):
                     source_code_with_line_number += f"{idx+1} {line}\n"
-
+                
                 llm = ByzerLLM()
-                llm.setup_default_model_name(self.llm.default_model_name)
                 llm.skip_nontext_check = True
+                llm.setup_default_model_name(self.chunk_llm.default_model_name)
 
                 extracted_info = (
                     self.extract_relevance_range_from_docs_with_conversation.options(

autocoder/utils/__init__.py

@@ -6,24 +6,14 @@ from loguru import logger
 
 def get_last_yaml_file(actions_dir:str)->Optional[str]:
     action_files = [
-        f
-        for f in os.listdir(actions_dir)
-        if f[:3].isdigit() and f.endswith(".yml")
-    ]
-    if not action_files:
-        max_seq = 0
-    else:
-        seqs = [int(f[:3]) for f in action_files]
-        max_seq = max(seqs)
+            f for f in os.listdir(actions_dir) if f[:3].isdigit() and "_" in f and f.endswith(".yml")
+        ]
 
-    new_seq = str(max_seq + 1).zfill(3)
-    prev_files = [f for f in action_files if int(f[:3]) < int(new_seq)]
+    def get_old_seq(name):
+        return int(name.split("_")[0])
 
-    if not prev_files:
-        return None
-    else:
-        prev_file = sorted(prev_files)[-1]  # 取序号最大的文件        
-        return prev_file
+    sorted_action_files = sorted(action_files, key=get_old_seq)
+    return sorted_action_files[-1] if sorted_action_files else None
 
 def open_yaml_file_in_editor(new_file:str):
     try:

autocoder/version.py

@@ -1 +1 @@
-__version__ = "0.1.202"
+__version__ = "0.1.204"