auto-coder 0.1.334__py3-none-any.whl → 0.1.336__py3-none-any.whl

autocoder/memory/active_context_manager.py

@@ -99,37 +99,10 @@ class ActiveContextManager:
         if self._is_initialized:
             return
         self.source_dir = source_dir
-        # 设置日志目录和文件
-        log_dir = os.path.join(source_dir, ".auto-coder", "active-context")
-        os.makedirs(log_dir, exist_ok=True)
-        log_file = os.path.join(log_dir, "active.log")
-
-        # 配置全局日志输出到文件，不输出到控制台
-        global_logger.configure(
-            handlers=[
-                # 移除控制台输出，只保留文件输出
-                # 文件 Handler
-                {
-                    "sink": log_file,
-                    "level": "INFO",
-                    "rotation": "10 MB",
-                    "retention": "1 week",
-                    "format": "{time:YYYY-MM-DD HH:mm:ss} | {level} | {name} | {message}",
-                    "filter": lambda record: record["extra"].get("name") in ["DirectoryMapper", "ActiveContextManager","ActivePackage","AsyncProcessor"]
-                },
-                # 控制台 Handler
-                {
-                    "sink": sys.stdout,
-                    "level": "INFO",
-                    "format": "{time:YYYY-MM-DD HH:mm:ss} | {name} | {message}",
-                    "filter": lambda record: record["extra"].get("name") not in ["DirectoryMapper", "ActiveContextManager","ActivePackage","AsyncProcessor","TokenCostCalculator"]
-                }
-            ]
-        )
-
+        
         # 创建专用的logger实例
         self.logger = global_logger.bind(name="ActiveContextManager")
-        self.logger.info(f"初始化 ActiveContextManager，日志输出到 {log_file}")
+    
 
         self.llm = llm
         self.directory_mapper = DirectoryMapper()

autocoder/models.py

@@ -17,7 +17,8 @@ default_models_list = [
         "is_reasoning": True,
         "input_price": 0.0,  # 单位:M/百万 input tokens
         "output_price": 0.0,  # 单位:M/百万 output tokens
-        "average_speed": 0.0  # 单位:秒/请求
+        "average_speed": 0.0,  # 单位:秒/请求
+        "max_output_tokens": 8096
     },    
     {
         "name": "deepseek_chat",
@@ -29,7 +30,8 @@ default_models_list = [
         "is_reasoning": False,
         "input_price": 0.0,
         "output_price": 0.0,
-        "average_speed": 0.0
+        "average_speed": 0.0,
+        "max_output_tokens": 8096
     },
     {
         "name":"o1",
@@ -302,6 +304,8 @@ def update_model(name: str, model_data: Dict) -> Dict:
             - is_reasoning: 是否为推理模型
             - input_price: 输入价格
             - output_price: 输出价格
+            - max_output_tokens: 最大输出tokens
+            - average_speed: 平均速度
             
     Returns:
         Dict: 更新后的模型信息，如果未找到则返回None
@@ -327,6 +331,10 @@ def update_model(name: str, model_data: Dict) -> Dict:
                 model["input_price"] = float(model_data["input_price"])
             if "output_price" in model_data:
                 model["output_price"] = float(model_data["output_price"])
+            if "max_output_tokens" in model_data:
+                model["max_output_tokens"] = int(model_data["max_output_tokens"])
+            if "average_speed" in model_data:
+                model["average_speed"] = float(model_data["average_speed"])
             
             # 保存更新后的模型
             models[i] = model

autocoder/shadows/shadow_manager.py

@@ -84,7 +84,7 @@ class ShadowManager:
         abs_path = os.path.abspath(path)
         
         # 检查路径是否在源目录内
-        if not abs_path.startswith(self.source_dir):
+        if not abs_path.startswith(self.source_dir):            
             raise ValueError(f"路径 {path} 不在源目录 {self.source_dir} 内")
         
         # 获取相对于source_dir的相对路径

autocoder/utils/llms.py

@@ -69,7 +69,8 @@ def get_single_llm(model_names: str, product_mode: str):
                         "saas.base_url": model_info["base_url"],
                         "saas.api_key": model_info["api_key"],
                         "saas.model": model_info["model_name"],
-                        "saas.is_reasoning": model_info["is_reasoning"]
+                        "saas.is_reasoning": model_info["is_reasoning"],
+                        "saas.max_output_tokens": model_info.get("max_output_tokens", 8096)
                     }
                 )
                 return target_llm
@@ -87,7 +88,8 @@ def get_single_llm(model_names: str, product_mode: str):
                     "saas.base_url": model_info["base_url"],
                     "saas.api_key": model_info["api_key"],
                     "saas.model": model_info["model_name"],
-                    "saas.is_reasoning": model_info["is_reasoning"]
+                    "saas.is_reasoning": model_info["is_reasoning"],
+                    "saas.max_output_tokens": model_info.get("max_output_tokens", 8096)
                 }
             )
             return target_llm

autocoder/version.py

@@ -1 +1 @@
-__version__ = "0.1.334"
+__version__ = "0.1.336"