auto-coder 0.1.264__py3-none-any.whl → 0.1.265__py3-none-any.whl

autocoder/common/stats_panel.py

@@ -0,0 +1,126 @@
+
+
+
+
+
+
+
+
+from rich.console import Console
+from rich.panel import Panel
+from rich.columns import Columns
+from rich.text import Text
+import math
+
+class StatsPanel:
+    def __init__(self, console: Console = None):
+        self.console = console if console else Console()
+
+    def _format_speed_bar(self, speed: float) -> Text:
+        """生成速度可视化进度条（保持原30-60区间）"""
+        if speed < 30:
+            color = "red"
+            level = "低"
+        elif 30 <= speed < 60:
+            color = "yellow"
+            level = "中"
+        else:
+            color = "green"
+            level = "高"
+
+        bar_length = min(int(speed), 100)
+        bar = Text("▮" * bar_length, style=color)
+        bar.append(f" {speed:.1f} tokens/s ({level})", style="bold white")
+        return bar
+
+    def _format_progress_bar(self, value: int, max_value: int, label: str, color: str) -> Text:
+        """生成通用进度条"""
+        progress = min(value / max_value, 1.0)
+        bar_length = int(progress * 20)
+        bar = Text("▮" * bar_length, style=color)
+        bar.append(f" {value} ({label})", style="bold white")
+        return bar
+
+    def generate(
+         self,
+         model_names: str,
+         duration: float,
+         sampling_count: int,
+         input_tokens: int,
+         output_tokens: int,
+         input_cost: float,
+         output_cost: float,
+         speed: float,
+     ) -> None:
+         """新版紧凑布局"""
+         # 复合标题（带图标和关键数据）
+         title = Text.assemble(
+             "📊 ", ("代码生成统计", "bold cyan underline"),
+             " │ ⚡", (f"{speed:.1f}t/s ", "bold green"),
+             "│ 💰", (f"${input_cost + output_cost:.4f}", "bold yellow")
+         )
+
+         # 处理耗时颜色逻辑（新增15-30-60区间）
+         duration_color = "green"
+         if 15 <= duration < 30:
+             duration_color = "yellow"
+         elif duration >= 30:
+             duration_color = "red"
+
+         # 处理成本颜色逻辑（新增0.5-1区间）
+         def get_cost_color(cost: float) -> str:
+             if cost < 0.5: return "green"
+             elif 0.5 <= cost < 1: return "yellow"
+             else: return "red"
+
+         # 紧凑网格布局
+         grid = [
+             Panel(
+                 Text.assemble(
+                     ("🤖 模型: ", "bold"), model_names + "\n",
+                     self._format_mini_progress(duration, 60.0, duration_color),  # 耗时max=60
+                     (" ⏱", duration_color), f" {duration:.1f}s │ ",
+                     self._format_mini_progress(sampling_count, 100, "blue"),
+                     (" 🔢", "blue"), f" {sampling_count}\n",
+                     ("📥", "green"), " ", 
+                     self._format_mini_progress(input_tokens, 65536.0, "green"),  # token分母改为65536
+                     f" {input_tokens} ({input_tokens/65536*100:.2f}%) │ ",  # 新增百分比显示
+                     ("📤", "bright_green"), " ", 
+                     self._format_mini_progress(output_tokens, 65536.0, "bright_green"),
+                     f" {output_tokens} ({output_tokens/65536*100:.2f}%)"  # 新增百分比显示
+                 ),
+                 border_style="cyan",
+                 padding=(0, 2)
+             ),
+             Panel(
+                 Text.assemble(
+                     ("💵 成本: ", "bold"), 
+                     self._format_mini_progress(input_cost, 1.0, get_cost_color(input_cost)),  # 成本max=1
+                     (" IN", get_cost_color(input_cost)), f" {input_cost:.3f}\n",
+                     ("💸 ", "bold"), 
+                     self._format_mini_progress(output_cost, 1.0, get_cost_color(output_cost)),
+                     (" OUT", get_cost_color(output_cost)), f" {output_cost:.3f}\n",
+                     self._format_speed_bar(speed)
+                 ),
+                 border_style="yellow",
+                 padding=(0, 1)
+             )
+         ]
+
+         # 组合布局
+         main_panel = Panel(
+             Columns(grid, equal=True, expand=True),
+             title=title,
+             border_style="bright_blue",
+             padding=(1, 2)
+         )
+
+         self.console.print(main_panel)
+    
+
+    def _format_mini_progress(self, value: float, max_value: float, color: str) -> Text:
+        """紧凑型进度条（支持浮点数）"""
+        progress = min(value / max_value, 1.0)
+        filled = "▮" * int(progress * 10)
+        empty = "▯" * (10 - len(filled))
+        return Text(filled + empty, style=color)

autocoder/dispacher/actions/action.py

@@ -86,7 +86,7 @@ class ActionTSProject(BaseAction):
                 max_iter=self.args.image_max_iter,
             )
             html_code = ""
-            with open(html_path, "r") as f:
+            with open(html_path, "r",encoding="utf-8") as f:
                 html_code = f.read()
             
             source_code_list.sources.append(SourceCode( 
@@ -190,9 +190,7 @@ class ActionTSProject(BaseAction):
                         conversations=generate_result.conversations[0],
                         model=self.llm.default_model_name,
                     )
-
-                with open(args.target_file, "w") as file:
-                    file.write(content)
+                
 
 
 class ActionPyScriptProject(BaseAction):
@@ -300,11 +298,7 @@ class ActionPyScriptProject(BaseAction):
                     instruction=self.args.query,
                     conversations=generate_result.conversations[0],
                     model=self.llm.default_model_name,
-                )
-
-            end_time = time.time()            
-            with open(self.args.target_file, "w") as file:
-                file.write(content)
+                )                        
 
 
 class ActionPyProject(BaseAction):
@@ -435,9 +429,7 @@ class ActionPyProject(BaseAction):
                     instruction=self.args.query,
                     conversations=generate_result.conversations[0],
                     model=self.llm.default_model_name,
-                )            
-            with open(args.target_file, "w") as file:
-                file.write(content)
+                )                        
 
 
 class ActionSuffixProject(BaseAction):
@@ -551,9 +543,7 @@ class ActionSuffixProject(BaseAction):
                 instruction=self.args.query,
                 conversations=merge_result.conversations[0],
                 model=self.llm.default_model_name,
-            )
-            with open(args.target_file, "w") as file:
-                file.write(content)
+            )            
         else:
             content = generate_result.contents[0]
 
@@ -563,7 +553,5 @@ class ActionSuffixProject(BaseAction):
                 conversations=generate_result.conversations[0],
                 model=self.llm.default_model_name,
             )
-
-            with open(args.target_file, "w") as file:
-                file.write(content)
+            
 

autocoder/dispacher/actions/copilot.py

@@ -343,7 +343,7 @@ class ActionCopilot:
             logger.info(
                 "model is not specified and we will generate prompt to the target file"
             )
-            with open(args.target_file, "w") as f:
+            with open(args.target_file, "w",encoding="utf-8") as f:
                 f.write(q)
             return True
 
@@ -379,7 +379,7 @@ class ActionCopilot:
         logger.info(result)
 
         # 将结果写入文件
-        with open(args.target_file, "w") as f:
+        with open(args.target_file, "w",encoding="utf-8") as f:
             f.write("=================CONVERSATION==================\n\n")
             for conversation in conversations:
                 f.write(f"{conversation['role']}: {conversation['content']}\n")

autocoder/dispacher/actions/plugins/action_regex_project.py

@@ -146,6 +146,4 @@ class ActionRegexProject:
                     conversations=generate_result.conversations[0],
                     model=self.llm.default_model_name,
                 )
-
-            with open(args.target_file, "w") as file:
-                file.write(content)
+            

autocoder/dispacher/actions/plugins/action_translate.py

@@ -209,6 +209,6 @@ class ActionTranslate:
                 new_filename = f"{filename}{new_file_mark}{extension}"
 
             logger.info(f"Writing to {new_filename}...")
-            with open(new_filename, "w") as file:
+            with open(new_filename, "w",encoding="utf-8") as file:
                 file.write(readme.content)
         return True

autocoder/index/index.py

@@ -342,7 +342,7 @@ class IndexManager:
 
     def build_index(self):
         if os.path.exists(self.index_file):
-            with open(self.index_file, "r") as file:
+            with open(self.index_file, "r",encoding="utf-8") as file:
                 index_data = json.load(file)
         else:
             index_data = {}
@@ -433,13 +433,13 @@ class IndexManager:
                     index_data[module_name] = result
                     updated_sources.append(module_name)
                     if len(updated_sources) > 5:
-                        with open(self.index_file, "w") as file:
+                        with open(self.index_file, "w",encoding="utf-8") as file:
                             json.dump(index_data, file, ensure_ascii=False, indent=2)
                         updated_sources = []
         
         # 如果 updated_sources 或 keys_to_remove 有值，则保存索引文件
         if updated_sources or keys_to_remove:
-            with open(self.index_file, "w") as file:
+            with open(self.index_file, "w",encoding="utf-8") as file:
                 json.dump(index_data, file, ensure_ascii=False, indent=2)
 
             print("")    
@@ -461,14 +461,14 @@ class IndexManager:
         if not os.path.exists(self.index_file):
             return []
 
-        with open(self.index_file, "r") as file:
+        with open(self.index_file, "r",encoding="utf-8") as file:
             return file.read()
 
     def read_index(self) -> List[IndexItem]:
         if not os.path.exists(self.index_file):
             return []
 
-        with open(self.index_file, "r") as file:
+        with open(self.index_file, "r",encoding="utf-8") as file:
             index_data = json.load(file)
 
         index_items = []

autocoder/models.py

@@ -97,7 +97,7 @@ def load_models() -> List[Dict]:
         if model.get("api_key_path",""):           
             api_key_file = os.path.join(api_key_dir, model["api_key_path"])
             if os.path.exists(api_key_file):
-                with open(api_key_file, "r") as f:
+                with open(api_key_file, "r",encoding="utf-8") as f:
                     model["api_key"] = f.read()                   
     return target_models
 
@@ -269,7 +269,7 @@ def update_model_with_api_key(name: str, api_key: str) -> Dict:
         api_key_dir = os.path.expanduser("~/.auto-coder/keys")
         os.makedirs(api_key_dir, exist_ok=True)
         api_key_file = os.path.join(api_key_dir, api_key_path)
-        with open(api_key_file, "w") as f:
+        with open(api_key_file, "w",encoding="utf-8") as f:
             f.write(api_key.strip())
         
         # 如果是新模型，添加到模型列表中

autocoder/pyproject/__init__.py

@@ -174,7 +174,8 @@ class PyProject:
         return False
 
     def output(self):
-        return open(self.target_file, "r").read()
+        with open(self.target_file, "r",encoding="utf-8") as file:
+            return file.read()
 
     def is_python_file(self, file_path):
         return file_path.endswith(".py")
@@ -182,9 +183,8 @@ class PyProject:
     def read_file_content(self, file_path):
         if self.args.auto_merge == "strict_diff":
             result = []
-            with open(file_path, "r") as file:
-                for line_number, line in enumerate(file, start=1):
-                    result.append(f"{line_number}:{line}")
+            for line_number, line in FileUtils.read_file_with_line_numbers(file_path,line_number_start=1):
+                result.append(f"{line_number}:{line}")
             return "\n".join(result)
 
         return FileUtils.read_file(file_path)
@@ -357,7 +357,7 @@ class PyProject:
             self.clone_repository()
 
         if self.target_file:
-            with open(self.target_file, "w") as file:
+            with open(self.target_file, "w",encoding="utf-8") as file:
 
                 for code in self.get_rest_source_codes():
                     self.sources.append(code)                    

autocoder/rag/cache/byzer_storage_cache.py

@@ -126,7 +126,7 @@ class ByzerStorageCache(BaseCacheManager):
         """Load cache from file"""
         if os.path.exists(self.cache_file):
             try:
-                with open(self.cache_file, "r") as f:
+                with open(self.cache_file, "r",encoding="utf-8") as f:
                     lines = f.readlines()
                     cache = {}
                     for line in lines:
@@ -147,7 +147,7 @@ class ByzerStorageCache(BaseCacheManager):
 
         if not fcntl:
             try:
-                with open(cache_file, "w") as f:
+                with open(cache_file, "w",encoding="utf-8") as f:
                     for data in self.cache.values():
                         json.dump(data, f, ensure_ascii=False)
                         f.write("\n")
@@ -155,12 +155,12 @@ class ByzerStorageCache(BaseCacheManager):
                 logger.error(f"Error writing cache file: {str(e)}")
         else:
             lock_file = cache_file + ".lock"
-            with open(lock_file, "w") as lockf:
+            with open(lock_file, "w",encoding="utf-8") as lockf:
                 try:
                     # 获取文件锁
                     fcntl.flock(lockf, fcntl.LOCK_EX | fcntl.LOCK_NB)
                     # 写入缓存文件
-                    with open(cache_file, "w") as f:
+                    with open(cache_file, "w",encoding="utf-8") as f:
                         for data in self.cache.values():
                             json.dump(data, f, ensure_ascii=False)
                             f.write("\n")

autocoder/rag/cache/file_monitor_cache.py

@@ -106,11 +106,11 @@ class AutoCoderRAGDocListener(BaseCacheManager):
         gitignore_path = os.path.join(self.path, ".gitignore")
 
         if os.path.exists(serveignore_path):
-            with open(serveignore_path, "r") as ignore_file:
+            with open(serveignore_path, "r",encoding="utf-8") as ignore_file:
                 patterns = ignore_file.readlines()
                 return [pattern.strip() for pattern in patterns]
         elif os.path.exists(gitignore_path):
-            with open(gitignore_path, "r") as ignore_file:
+            with open(gitignore_path, "r",encoding="utf-8") as ignore_file:
                 patterns = ignore_file.readlines()
                 return [pattern.strip() for pattern in patterns]
         return []

autocoder/rag/cache/simple_cache.py

@@ -160,7 +160,7 @@ class AutoCoderRAGAsyncUpdateQueue(BaseCacheManager):
 
         cache = {}
         if os.path.exists(cache_file):
-            with open(cache_file, "r") as f:
+            with open(cache_file, "r",encoding="utf-8") as f:
                 for line in f:
                     data = json.loads(line)
                     cache[data["file_path"]] = data
@@ -171,7 +171,7 @@ class AutoCoderRAGAsyncUpdateQueue(BaseCacheManager):
         cache_file = os.path.join(cache_dir, "cache.jsonl")
 
         if not fcntl:
-            with open(cache_file, "w") as f:
+            with open(cache_file, "w",encoding="utf-8") as f:
                 for data in self.cache.values():
                     try:
                         json.dump(data, f, ensure_ascii=False)
@@ -181,12 +181,12 @@ class AutoCoderRAGAsyncUpdateQueue(BaseCacheManager):
                             f"Failed to write {data['file_path']} to .cache/cache.jsonl: {e}")
         else:
             lock_file = cache_file + ".lock"
-            with open(lock_file, "w") as lockf:
+            with open(lock_file, "w",encoding="utf-8") as lockf:
                 try:
                     # 获取文件锁
                     fcntl.flock(lockf, fcntl.LOCK_EX | fcntl.LOCK_NB)
                     # 写入缓存文件
-                    with open(cache_file, "w") as f:
+                    with open(cache_file, "w",encoding="utf-8") as f:
                         for data in self.cache.values():
                             try:
                                 json.dump(data, f, ensure_ascii=False)

autocoder/rag/long_context_rag.py

@@ -245,10 +245,10 @@ class LongContextRAG:
         gitignore_path = os.path.join(self.path, ".gitignore")
 
         if os.path.exists(serveignore_path):
-            with open(serveignore_path, "r") as ignore_file:
+            with open(serveignore_path, "r",encoding="utf-8") as ignore_file:
                 return pathspec.PathSpec.from_lines("gitwildmatch", ignore_file)
         elif os.path.exists(gitignore_path):
-            with open(gitignore_path, "r") as ignore_file:
+            with open(gitignore_path, "r",encoding="utf-8") as ignore_file:
                 return pathspec.PathSpec.from_lines("gitwildmatch", ignore_file)
         return None
 

autocoder/regexproject/__init__.py

@@ -78,7 +78,8 @@ class RegexProject:
             raise ValueError("Invalid project_type format. Expected 'regex//<pattern>'")
 
     def output(self):
-        return open(self.target_file, "r").read()
+        with open(self.target_file, "r",encoding="utf-8") as file:
+            return file.read()
 
     def is_regex_match(self, file_path):
         return re.search(self.regex_pattern, file_path) is not None
@@ -231,7 +232,7 @@ class RegexProject:
             self.clone_repository()
 
         if self.target_file:
-            with open(self.target_file, "w") as file:
+            with open(self.target_file, "w",encoding="utf-8") as file:
                 for code in self.get_source_codes():
                     self.sources.append(code)
                     file.write(f"##File: {code.module_name}\n")

autocoder/suffixproject/__init__.py

@@ -114,7 +114,8 @@ class SuffixProject:
         return False
 
     def output(self):
-        return open(self.target_file, "r").read()
+        with open(self.target_file, "r",encoding="utf-8") as file:
+            return file.read()
 
     def is_suffix_file(self, file_path):
         return any([file_path.endswith(suffix) for suffix in self.suffixs])
@@ -273,7 +274,7 @@ class SuffixProject:
             self.clone_repository()
 
         if self.target_file:
-            with open(self.target_file, "w") as file:
+            with open(self.target_file, "w",encoding="utf-8") as file:
                 for code in self.get_source_codes():
                     self.sources.append(code)
                     file.write(f"##File: {code.module_name}\n")

autocoder/tsproject/__init__.py

@@ -106,7 +106,8 @@ class TSProject:
         return False
 
     def output(self):
-        return open(self.target_file, "r").read()
+        with open(self.target_file, "r",encoding="utf-8") as file:
+            return file.read()
 
     def read_file_content(self, file_path):
         return FileUtils.read_file(file_path)        
@@ -308,7 +309,7 @@ class TSProject:
             self.clone_repository()
 
         if self.target_file:
-            with open(self.target_file, "w") as file:
+            with open(self.target_file, "w",encoding="utf-8") as file:
                 for code in self.get_rest_source_codes():
                     self.sources.append(code)
                     file.write(f"##File: {code.module_name}\n")

autocoder/utils/conversation_store.py

@@ -34,7 +34,7 @@ def load_code_model_conversation_from_store(args: AutoCoderArgs):
         return []
 
     conversations = []
-    with open(conversation_file, "r") as f:
+    with open(conversation_file, "r",encoding="utf-8") as f:
         for line in f:
             conversations.append(json.loads(line))
 

autocoder/utils/operate_config_api.py

@@ -15,7 +15,7 @@ import hashlib
 def convert_yaml_to_config(yaml_file: str):
 
     args = AutoCoderArgs()
-    with open(yaml_file, "r") as f:
+    with open(yaml_file, "r",encoding="utf-8") as f:
         config = yaml.safe_load(f)
         config = load_include_files(config, yaml_file)
         for key, value in config.items():
@@ -75,7 +75,7 @@ def get_llm_friendly_package_docs(memory,
                                         if return_paths:
                                             docs.append(file_path)
                                         else:
-                                            with open(file_path, "r") as f:
+                                            with open(file_path, "r",encoding="utf-8") as f:
                                                 docs.append(f.read())
 
     return docs
@@ -130,7 +130,7 @@ def get_llm(memory, model:Optional[str]=None):
             # 临时保存yaml文件，然后读取yaml文件，转换为args
             temp_yaml = os.path.join("actions", f"{uuid.uuid4()}.yml")
             try:
-                with open(temp_yaml, "w") as f:
+                with open(temp_yaml, "w",encoding="utf-8") as f:
                     f.write(convert_yaml_config_to_str(
                         yaml_config=yaml_config))
                 args = convert_yaml_to_config(temp_yaml)

autocoder/version.py

@@ -1 +1 @@
-__version__ = "0.1.264"
+__version__ = "0.1.265"