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

autocoder/events/event_manager.py

@@ -62,7 +62,7 @@ class EventManager:
         self.event_store.append_event(event)
         return event
     
-    def write_completion(self, content: Union[Dict[str, Any], Any]) -> Event:
+    def write_completion(self, content: Union[Dict[str, Any], Any], metadata: Dict[str, Any] = {}) -> Event:
         """
         Write a completion event.
         
@@ -74,11 +74,11 @@ class EventManager:
         """
         if not isinstance(content, dict):
             content = content.to_dict()
-        event = Event(event_type=EventType.COMPLETION, content=content)
+        event = Event(event_type=EventType.COMPLETION, content=content, metadata=metadata)
         self.event_store.append_event(event)
         return event
     
-    def write_error(self, content: Union[Dict[str, Any], Any]) -> Event:
+    def write_error(self, content: Union[Dict[str, Any], Any], metadata: Dict[str, Any] = {}) -> Event:
         """
         Write an error event.
         
@@ -90,7 +90,7 @@ class EventManager:
         """
         if not isinstance(content, dict):
             content = content.to_dict()
-        event = Event(event_type=EventType.ERROR, content=content)
+        event = Event(event_type=EventType.ERROR, content=content, metadata=metadata)
         self.event_store.append_event(event)
         return event
     

autocoder/events/event_types.py

@@ -68,6 +68,7 @@ class EventMetadata(BaseModel):
     stream_out_type: str = field(default="") # 标记，比如当前这个流式是用于什么场景的，比如用于产生命令的还是啥
     is_streaming: bool = field(default=False) # 是否是流式输出
     output: str = field(default="") # result or delta, 在流式里，我们也可能会输出 ResultContent 类型
+    path: str = field(default="") # 唯一路径，比如 /agent/edit/tool/call
 
     def to_dict(self) -> Dict[str, Any]:
         """Convert event metadata to dictionary for serialization"""

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/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.335"
+__version__ = "0.1.336"