auto-coder-web 0.1.96__py3-none-any.whl → 0.1.97__py3-none-any.whl

auto_coder_web/common_router/chat_list_manager.py

@@ -16,7 +16,7 @@ def _get_chat_list_file_path(project_path: str, name: str) -> str:
     chat_lists_dir = _get_chat_lists_dir(project_path)
     return os.path.join(chat_lists_dir, f"{name}.json")
 
-async def save_chat_list(project_path: str, name: str, messages: List[Dict[str, Any]]) -> None:
+async def save_chat_list(project_path: str, name: str, messages: List[Dict[str, Any]], metadata: dict = None) -> None:
     """
     保存聊天列表到文件
     
@@ -24,14 +24,21 @@ async def save_chat_list(project_path: str, name: str, messages: List[Dict[str,
         project_path: 项目路径
         name: 聊天列表名称
         messages: 聊天消息列表
+        metadata: 聊天元数据
         
     Raises:
         Exception: 如果保存失败
     """
     file_path = _get_chat_list_file_path(project_path, name)
     try:
+        data = {
+            "name": name,
+            "messages": messages
+        }
+        if metadata is not None:
+            data["metadata"] = metadata
         async with aiofiles.open(file_path, 'w') as f:
-            await f.write(json.dumps({"messages": messages}, indent=2, ensure_ascii=False))
+            await f.write(json.dumps(data, indent=2, ensure_ascii=False))
     except Exception as e:
         logger.error(f"Error saving chat list {name}: {str(e)}")
         raise e
@@ -73,18 +80,7 @@ async def get_chat_lists(project_path: str) -> List[str]:
 
 async def get_chat_list(project_path: str, name: str) -> Dict[str, Any]:
     """
-    获取特定聊天列表的内容
-    
-    Args:
-        project_path: 项目路径
-        name: 聊天列表名称
-        
-    Returns:
-        聊天列表内容
-        
-    Raises:
-        FileNotFoundError: 如果聊天列表不存在
-        Exception: 如果读取失败
+    获取特定聊天列表的内容（兼容旧结构）
     """
     file_path = _get_chat_list_file_path(project_path, name)
     if not os.path.exists(file_path):
@@ -93,7 +89,13 @@ async def get_chat_list(project_path: str, name: str) -> Dict[str, Any]:
     try:
         async with aiofiles.open(file_path, 'r') as f:
             content = await f.read()
-            return json.loads(content)
+            data = json.loads(content)
+            # 兼容旧数据结构（只有messages）
+            if "name" not in data:
+                data["name"] = name
+            if "metadata" not in data:
+                data["metadata"] = None
+            return data
     except json.JSONDecodeError as e:
         logger.error(f"Invalid JSON in chat list {name}: {str(e)}")
         raise Exception(f"Invalid JSON in chat list file: {str(e)}")

auto_coder_web/common_router/chat_list_router.py

@@ -33,8 +33,8 @@ router = APIRouter()
 @router.post("/api/chat-lists/save")
 async def save_chat_list_endpoint(chat_list: ChatList, project_path: str = Depends(get_project_path)):
     try:
-        # 调用管理模块保存聊天列表
-        await save_chat_list(project_path, chat_list.name, chat_list.messages)
+        # 调用管理模块保存聊天列表，支持 metadata
+        await save_chat_list(project_path, chat_list.name, chat_list.messages, metadata=chat_list.metadata.dict() if chat_list.metadata else None)
         return {"status": "success", "message": f"Chat list {chat_list.name} saved successfully"}
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))

auto_coder_web/types.py

@@ -24,9 +24,15 @@ class CompletionResponse(BaseModel):
     completions: List[CompletionItem]
 
 
+class ChatMetadata(BaseModel):
+    token_usage: Optional[int] = None  # token 消耗
+    cost: Optional[float] = None       # 费用
+    window_size: Optional[int] = None  # 窗口大小
+
 class ChatList(BaseModel):
     name: str
     messages: List[Dict[str, Any]]
+    metadata: Optional[ChatMetadata] = None  # 新增 metadata 字段
 
 
 class HistoryQuery(BaseModel):

auto_coder_web/version.py

@@ -1 +1 @@
-__version__ = "0.1.96"
+__version__ = "0.1.97"