auto-coder 0.1.364__py3-none-any.whl → 0.1.365__py3-none-any.whl

{auto_coder-0.1.364.dist-info → auto_coder-0.1.365.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.364
+Version: 0.1.365
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Programming Language :: Python :: 3.10

{auto_coder-0.1.364.dist-info → auto_coder-0.1.365.dist-info}/RECORD

@@ -1,5 +1,5 @@
 autocoder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/auto_coder.py,sha256=dElVs1kOxEArM2T5nfIzDJwT6-_YEq_5Sc3WPTPcBBk,67178
+autocoder/auto_coder.py,sha256=7602L3tG0JErNxh8vkLAmGUgv2c-DGPzPCkmWIQt9bs,69757
 autocoder/auto_coder_lang.py,sha256=Rtupq6N3_HT7JRhDKdgCBcwRaiAnyCOR_Gsp4jUomrI,3229
 autocoder/auto_coder_rag.py,sha256=ru5o86IaKylyVRlVORmnrdf3Q1To2eWi2KLdT9FMW0k,37580
 autocoder/auto_coder_rag_client_mcp.py,sha256=QRxUbjc6A8UmDMQ8lXgZkjgqtq3lgKYeatJbDY6rSo0,6270
@@ -14,7 +14,7 @@ autocoder/command_parser.py,sha256=fx1g9E6GaM273lGTcJqaFQ-hoksS_Ik2glBMnVltPCE,1
 autocoder/lang.py,sha256=PFtATuOhHRnfpqHQkXr6p4C893JvpsgwTMif3l-GEi0,14321
 autocoder/models.py,sha256=Gu50IATQtZtgEir1PpCfwgH6o4ygw6XqqbQRj3lx5dU,13798
 autocoder/run_context.py,sha256=IUfSO6_gp2Wt1blFWAmOpN0b0nDrTTk4LmtCYUBIoro,1643
-autocoder/version.py,sha256=VL9hmmJg1RyRRRaJG6Hcv2loEzXKwXbfJgcVK9hH6Us,23
+autocoder/version.py,sha256=IcO5L1bOdV3qLFshmlEIP9ycK_uaklkhdNmRNqLG91g,23
 autocoder/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/agent/agentic_edit.py,sha256=XsfePZ-t6M-uBSdG1VLZXk1goqXk2HPeJ_A8IYyBuWQ,58896
 autocoder/agent/agentic_edit_types.py,sha256=oFcDd_cxJ2yH9Ed1uTpD3BipudgoIEWDMPb5pAkq4gI,3288
@@ -78,7 +78,7 @@ autocoder/commands/auto_web.py,sha256=K0Gv7lil5UqmExr_sAJNOcwNxw_q1vhvre1jjQ7tA3
 autocoder/commands/tools.py,sha256=IX_zx5mWAvQDED7wUHTqNtrCmLNo9ztFV1aZ6AflY4o,34292
 autocoder/common/JupyterClient.py,sha256=O-wi6pXeAEYhAY24kDa0BINrLYvKS6rKyWe98pDClS0,2816
 autocoder/common/ShellClient.py,sha256=fM1q8t_XMSbLBl2zkCNC2J9xuyKN3eXzGm6hHhqL2WY,2286
-autocoder/common/__init__.py,sha256=LFR-2IrVK0NtHeefzCB8iqLb-6SuWJRPKQec6bS37_0,14734
+autocoder/common/__init__.py,sha256=qZWLTeEjimIGDUvbAq_QV1o5jgMiD1k4TQcoU0-wOo8,14887
 autocoder/common/action_yml_file_manager.py,sha256=DdF5P1R_B_chCnnqoA2IgogakWLZk_nItiJZUfX0_Wo,17857
 autocoder/common/anything2images.py,sha256=0ILBbWzY02M-CiWB-vzuomb_J1hVdxRcenAfIrAXq9M,25283
 autocoder/common/anything2img.py,sha256=iZQmg8srXlD7N5uGl5b_ONKJMBjYoW8kPmokkG6ISF0,10118
@@ -141,7 +141,7 @@ autocoder/common/test_run_cmd.py,sha256=0piPrNnxTPS8vJRnsVH6-lgB5zeLaXSRY5pPH13H
 autocoder/common/text.py,sha256=KGRQq314GHBmY4MWG8ossRoQi1_DTotvhxchpn78c-k,1003
 autocoder/common/token_cost_caculate.py,sha256=MSWJtl7YpQSUt-gFQoqUcJMblyPqHXe2ZioiZOFkV80,10085
 autocoder/common/types.py,sha256=Cw_4RH-rGmAgQE-Ck69maMAMqlPCDA4Yj37QmuUY0mQ,713
-autocoder/common/utils_code_auto_generate.py,sha256=Tvb3Mqxedj0Jhi0faKyWvILuNCty9fZi341fyIs33tA,4550
+autocoder/common/utils_code_auto_generate.py,sha256=zol3E5VQgriGzpR0p51yYnx8unthyurYM83QIudkxeI,4592
 autocoder/common/conversations/__init__.py,sha256=xGZeOFrDsgg2fkPK1zmvYBDhAyX66FtgOcZaxhYKJXU,1338
 autocoder/common/conversations/compatibility.py,sha256=WuBXB4-dw5X9LUMsB16VWbihvRZQ1tT99m6zuBwDfqE,9606
 autocoder/common/conversations/conversation_manager.py,sha256=ZhuhfSdOTncqgy3nHPoEU7Cg0dCsSl-VPcvLbUlL2Tk,18295
@@ -151,8 +151,9 @@ autocoder/common/directory_cache/cache.py,sha256=Jknygb_U6DkF04_SX04IwsOcQdd-2QQ
 autocoder/common/directory_cache/test_cache.py,sha256=0iQkHaZQPhZBwSS6dwK_je93QMLbYGY0BYrTSt45Cao,6610
 autocoder/common/file_checkpoint/__init__.py,sha256=qwoM0tIU-IMr-zGVCMN8yZtmz0NWpRe027l8z8sCjjk,847
 autocoder/common/file_checkpoint/backup.py,sha256=JO26vOG9k7d8b5jgT24PdccSrTuPqKghp1nz5cmjSiE,8813
+autocoder/common/file_checkpoint/conversation_checkpoint.py,sha256=SFSTjA0fF5rsHlYdLQ-Dr9dfDl5JihndhjeqhN3OuMY,6322
 autocoder/common/file_checkpoint/examples.py,sha256=HTik8E0ddvKjEPGwzizWJBHIP9URrWRyRUOKSjYRUG8,6272
-autocoder/common/file_checkpoint/manager.py,sha256=DvPzXvU4LVNb8ZVbEIWJawevX6fIm4VgVmV1g7TrsS4,14774
+autocoder/common/file_checkpoint/manager.py,sha256=mkEPURbF109qnCwXOikNmJI8INhdfCwBW1v6sRB7ubY,23757
 autocoder/common/file_checkpoint/models.py,sha256=dcZL2QGnklsa_BV_QY81fH-H5hYfhelXrH6GSrubMZo,4730
 autocoder/common/file_checkpoint/store.py,sha256=dgQe-1O_gPJ3QU6tHihGRp0G2jgD2IDxQ-w9zM6Yq54,12920
 autocoder/common/file_checkpoint/test_backup.py,sha256=Z9Y2RyGqxwKPNc7nW-2jtsMAYzqt0qZGzLoq3pn2zCI,8930
@@ -186,9 +187,9 @@ autocoder/common/v2/code_editblock_manager.py,sha256=DMwJw-FAM6VyaBQV3p4xespHpgZ
 autocoder/common/v2/code_manager.py,sha256=C403bS-f6urixwitlKHcml-J03hci-UyNwHJOqBiY6Q,9182
 autocoder/common/v2/code_strict_diff_manager.py,sha256=Bys7tFAq4G03R1zUZuxrszBTvP4QB96jIw2y5BDLyRM,9424
 autocoder/common/v2/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/common/v2/agent/agentic_edit.py,sha256=hM2IPCV-Z3ajEsmVUGcFlD01544CloMehyxHAx320Qk,106932
+autocoder/common/v2/agent/agentic_edit.py,sha256=Fx_T6YBWPa4tlTLY8dbecsd1n5a93dGE7b4gygmK56I,115043
 autocoder/common/v2/agent/agentic_edit_conversation.py,sha256=pFgWPWHKhZ4J9EcFmIdiGsrSolTZuYcH1qkgKdD8nwk,7726
-autocoder/common/v2/agent/agentic_edit_types.py,sha256=ftFTXz-KF96GZvhnYRAxiiYOWfotwU_d9Lb8la2I47U,4779
+autocoder/common/v2/agent/agentic_edit_types.py,sha256=nEcZc2MOZ_fQLaJX-YDha_x9Iim22ao4tykYM2iIy4k,4908
 autocoder/common/v2/agent/agentic_tool_display.py,sha256=-a-JTQLc4q03E_rdIILKMI0B6DHN-5gcGlrqq-mBYK4,7239
 autocoder/common/v2/agent/ignore_utils.py,sha256=gnUchRzKMLbUm_jvnKL-r-K9MWKPtt-6iiuzijY7Es0,1717
 autocoder/common/v2/agent/agentic_edit_tools/__init__.py,sha256=RbPZZcZg_VnGssL577GxSyFrYrxQ_LopJ4G_-mY3z_Q,1337
@@ -197,16 +198,16 @@ autocoder/common/v2/agent/agentic_edit_tools/attempt_completion_tool_resolver.py
 autocoder/common/v2/agent/agentic_edit_tools/base_tool_resolver.py,sha256=Zid2m1uZd-2wVFGc_n_KAViXZyNjbdLSpI5n7ut1RUQ,1036
 autocoder/common/v2/agent/agentic_edit_tools/execute_command_tool_resolver.py,sha256=sX00xzczfmyW6yPG3nMm0xO8p-WARQTiD4jcoUiTxsg,3844
 autocoder/common/v2/agent/agentic_edit_tools/list_code_definition_names_tool_resolver.py,sha256=8QoMsADUDWliqiDt_dpguz31403syB8eeW0Pcw-qfb8,3842
-autocoder/common/v2/agent/agentic_edit_tools/list_files_tool_resolver.py,sha256=FvpEyBwAfrYvnxcQ_gt7vTmfgoFoR7qr-r1OPx13Xks,5475
+autocoder/common/v2/agent/agentic_edit_tools/list_files_tool_resolver.py,sha256=1tJX9RYRU0zRjKZMzFlZzKm-4R32CzRnZ0UQJj4pxAk,8074
 autocoder/common/v2/agent/agentic_edit_tools/list_package_info_tool_resolver.py,sha256=dIdV12VuczHpHuHgx2B1j_3BZYc9PL0jfHCuBk9ryk8,2005
 autocoder/common/v2/agent/agentic_edit_tools/plan_mode_respond_tool_resolver.py,sha256=lGT4_QYJK6Fa9f6HVSGo0cSsGK7qCsDYgJGUowNxPzk,1499
-autocoder/common/v2/agent/agentic_edit_tools/read_file_tool_resolver.py,sha256=yO7ZORGsKpPOiwbFamWzC_UtUKc7_7AchZz2R6I8HP0,2850
-autocoder/common/v2/agent/agentic_edit_tools/replace_in_file_tool_resolver.py,sha256=iJ9OocL1pHCWpwXMRl6MtNMwdcdsXI3YfluwPqn2dsM,12274
-autocoder/common/v2/agent/agentic_edit_tools/search_files_tool_resolver.py,sha256=ACvygitMVOroQ8I-07v9fIvX9Xm9bO1HrIW--Xv9qKw,6894
+autocoder/common/v2/agent/agentic_edit_tools/read_file_tool_resolver.py,sha256=6JhjM3VXV3BGelh1dNcdr-M5FoVPoqLkls1-y8ND8_c,6721
+autocoder/common/v2/agent/agentic_edit_tools/replace_in_file_tool_resolver.py,sha256=J-GOlaLMwEu83v9VuHIstTpRR_vgPMR9wP_AzoK173s,18009
+autocoder/common/v2/agent/agentic_edit_tools/search_files_tool_resolver.py,sha256=VtDRDFJ2qzACy2jpD1ZNhywgCT3_iYUmGoUhHHmMX3o,9581
 autocoder/common/v2/agent/agentic_edit_tools/test_search_files_tool_resolver.py,sha256=9eBo3WLkrr77iNotwIwVmH1ZL3UY0JQgLpdAIc9wTTM,6127
 autocoder/common/v2/agent/agentic_edit_tools/test_write_to_file_tool_resolver.py,sha256=ZWRPsJny_My4UMzovrB8J2_x5N0rEW-xx3DVI-kDRFI,15870
 autocoder/common/v2/agent/agentic_edit_tools/use_mcp_tool_resolver.py,sha256=wM2Xy4bcnD0TSLEmcM8rvvyyWenN5_KQnJMO6hJ8lTE,1716
-autocoder/common/v2/agent/agentic_edit_tools/write_to_file_tool_resolver.py,sha256=hRCFosXaf76Di3aJp_pEhTrXMfHRy5DdWeoNrelq55g,10129
+autocoder/common/v2/agent/agentic_edit_tools/write_to_file_tool_resolver.py,sha256=ISJ2jNz0Dfv1W7P2A_mYSM0vPdN0yW1w7Qqt6QjTk98,11170
 autocoder/compilers/__init__.py,sha256=C0HOms70QA747XD0uZEMmGtRFcIPenohyqECNStv0Bw,1647
 autocoder/compilers/base_compiler.py,sha256=dsTzMO4H_RoqWfE-SntIk2B52hWuvSlWVLtkdCbHgGs,3244
 autocoder/compilers/compiler_config_api.py,sha256=QRSwWm_EX7jSeZ3dtQqM9HI__x5aZ7U0c3fHIW_2N48,13839
@@ -353,9 +354,9 @@ autocoder/utils/types.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/utils/auto_coder_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/utils/auto_coder_utils/chat_stream_out.py,sha256=t902pKxQ5xM7zgIHiAOsTPLwxhE6VuvXAqPy751S7fg,14096
 autocoder/utils/chat_auto_coder_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-auto_coder-0.1.364.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-auto_coder-0.1.364.dist-info/METADATA,sha256=G52msypzQ4D-qqJYxzrJC3r8J3LleJ2lpj5BYpjyk6o,2775
-auto_coder-0.1.364.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-auto_coder-0.1.364.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
-auto_coder-0.1.364.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
-auto_coder-0.1.364.dist-info/RECORD,,
+auto_coder-0.1.365.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+auto_coder-0.1.365.dist-info/METADATA,sha256=vo07pJAu6M2IMTWjRq9caGUXRwr88NMs05TIHDGUJKs,2775
+auto_coder-0.1.365.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+auto_coder-0.1.365.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
+auto_coder-0.1.365.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
+auto_coder-0.1.365.dist-info/RECORD,,

autocoder/auto_coder.py

@@ -376,8 +376,8 @@ def main(input_args: Optional[List[str]] = None):
                 llm.setup_sub_client("inference_model", inference_model)                 
 
             if args.index_filter_model:
-                model_info = models_module.get_model_by_name(args.index_filter_model)
-                model_name = args.index_filter_model
+                model_name = args.index_filter_model.strip()
+                model_info = models_module.get_model_by_name(model_name)                
                 index_filter_model = byzerllm.SimpleByzerLLM(default_model_name=model_name)
                 index_filter_model.deploy(
                     model_path="",
@@ -393,6 +393,41 @@ def main(input_args: Optional[List[str]] = None):
                 )
                 llm.setup_sub_client("index_filter_model", index_filter_model)            
 
+            if args.context_prune_model:
+                model_name = args.context_prune_model.strip()
+                model_info = models_module.get_model_by_name(model_name)                
+                context_prune_model = byzerllm.SimpleByzerLLM(default_model_name=model_name)
+                context_prune_model.deploy(
+                    model_path="",
+                    pretrained_model_type=model_info["model_type"],
+                    udf_name=model_name,
+                    infer_params={
+                        "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.max_output_tokens": model_info.get("max_output_tokens", 8096)
+                    }
+                )
+                llm.setup_sub_client("context_prune_model", context_prune_model)
+
+            if args.conversation_prune_model:
+                model_name = args.conversation_prune_model.strip()
+                model_info = models_module.get_model_by_name(model_name)                
+                conversation_prune_model = byzerllm.SimpleByzerLLM(default_model_name=model_name)
+                conversation_prune_model.deploy(
+                    model_path="",
+                    pretrained_model_type=model_info["model_type"],
+                    udf_name=model_name,
+                    infer_params={
+                        "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.max_output_tokens": model_info.get("max_output_tokens", 8096)
+                    }
+                )
+                llm.setup_sub_client("conversation_prune_model", conversation_prune_model)
 
         if args.product_mode == "pro":
             if args.code_model:
@@ -437,6 +472,15 @@ def main(input_args: Optional[List[str]] = None):
                 index_filter_model.setup_default_model_name(args.index_filter_model)
                 llm.setup_sub_client("index_filter_model", index_filter_model)
             
+            if args.context_prune_model:
+                context_prune_model = byzerllm.ByzerLLM()
+                context_prune_model.setup_default_model_name(args.context_prune_model)
+                llm.setup_sub_client("context_prune_model", context_prune_model)
+
+            if args.conversation_prune_model:
+                conversation_prune_model = byzerllm.ByzerLLM()
+                conversation_prune_model.setup_default_model_name(args.conversation_prune_model)
+                llm.setup_sub_client("conversation_prune_model", conversation_prune_model)
 
         if get_run_context().mode != RunMode.WEB and args.human_as_model:
 

autocoder/common/__init__.py

@@ -254,6 +254,8 @@ class AutoCoderArgs(pydantic.BaseModel):
     emb_model: Optional[str] = ""
     code_model: Optional[str] = ""
     generate_rerank_model: Optional[str] = ""
+    context_prune_model: Optional[str] = ""
+    conversation_prune_model: Optional[str] = ""
     inference_model: Optional[str] = ""
     system_prompt: Optional[str] = ""
     planner_model: Optional[str] = ""
@@ -399,6 +401,7 @@ class AutoCoderArgs(pydantic.BaseModel):
 
     context_prune_strategy: Optional[str] = "extract"
     context_prune: Optional[bool] = True
+    context_prune_safe_zone_tokens: Optional[int] = 32*1024
     context_prune_sliding_window_size: Optional[int] = 1000
     context_prune_sliding_window_overlap: Optional[int] = 100
 

autocoder/common/file_checkpoint/conversation_checkpoint.py

@@ -0,0 +1,182 @@
+import os
+import json
+import time
+import uuid
+from typing import List, Dict, Any, Optional
+from pydantic import BaseModel
+from loguru import logger
+
+
+class ConversationCheckpoint(BaseModel):
+    """对话检查点，用于保存特定时刻的对话状态"""
+    
+    checkpoint_id: str  # 检查点ID，与变更组ID对应
+    timestamp: float  # 创建时间戳
+    conversations: List[Dict[str, Any]]  # 对话历史
+    metadata: Optional[Dict[str, Any]] = None  # 元数据，可包含额外信息
+
+
+class ConversationCheckpointStore:
+    """对话检查点存储管理器"""
+    
+    def __init__(self, store_dir: Optional[str] = None, max_history: int = 50):
+        """
+        初始化对话检查点存储
+        
+        Args:
+            store_dir: 存储目录，默认为用户主目录下的.autocoder/conversation_checkpoints
+            max_history: 最大保存的历史版本数量
+        """
+        if store_dir is None:
+            home_dir = os.path.expanduser("~")
+            store_dir = os.path.join(home_dir, ".autocoder", "conversation_checkpoints")
+        
+        self.store_dir = os.path.abspath(store_dir)
+        self.max_history = max_history
+        
+        # 确保存储目录存在
+        os.makedirs(self.store_dir, exist_ok=True)
+        logger.info(f"对话检查点存储目录: {self.store_dir}")
+    
+    def save_checkpoint(self, checkpoint: ConversationCheckpoint) -> str:
+        """
+        保存对话检查点
+        
+        Args:
+            checkpoint: 对话检查点对象
+            
+        Returns:
+            str: 检查点ID
+        """
+        # 确保检查点有ID
+        if not checkpoint.checkpoint_id:
+            checkpoint.checkpoint_id = str(uuid.uuid4())
+        
+        # 确保时间戳存在
+        if not checkpoint.timestamp:
+            checkpoint.timestamp = time.time()
+        
+        # 构建文件路径
+        file_path = os.path.join(self.store_dir, f"{checkpoint.checkpoint_id}.json")
+        
+        # 保存为JSON文件
+        try:
+            with open(file_path, 'w', encoding='utf-8') as f:
+                f.write(checkpoint.json(ensure_ascii=False, indent=2))
+            
+            logger.info(f"已保存对话检查点: {checkpoint.checkpoint_id}")
+            
+            # 检查并清理过期的检查点
+            self._cleanup_old_checkpoints()
+            
+            return checkpoint.checkpoint_id
+        except Exception as e:
+            logger.error(f"保存对话检查点失败: {str(e)}")
+            return ""
+    
+    def get_checkpoint(self, checkpoint_id: str) -> Optional[ConversationCheckpoint]:
+        """
+        获取指定ID的对话检查点
+        
+        Args:
+            checkpoint_id: 检查点ID
+            
+        Returns:
+            Optional[ConversationCheckpoint]: 对话检查点对象，如果不存在则返回None
+        """
+        file_path = os.path.join(self.store_dir, f"{checkpoint_id}.json")
+        
+        if not os.path.exists(file_path):
+            logger.warning(f"对话检查点不存在: {checkpoint_id}")
+            return None
+        
+        try:
+            with open(file_path, 'r', encoding='utf-8') as f:
+                data = json.load(f)
+            
+            return ConversationCheckpoint(**data)
+        except Exception as e:
+            logger.error(f"读取对话检查点失败: {str(e)}")
+            return None
+    
+    def get_latest_checkpoint(self) -> Optional[ConversationCheckpoint]:
+        """
+        获取最新的对话检查点
+        
+        Returns:
+            Optional[ConversationCheckpoint]: 最新的对话检查点对象，如果不存在则返回None
+        """
+        checkpoints = self._get_all_checkpoints()
+        
+        if not checkpoints:
+            return None
+        
+        # 按时间戳降序排序
+        checkpoints.sort(key=lambda x: x.timestamp, reverse=True)
+        
+        return checkpoints[0] if checkpoints else None
+    
+    def delete_checkpoint(self, checkpoint_id: str) -> bool:
+        """
+        删除指定的对话检查点
+        
+        Args:
+            checkpoint_id: 检查点ID
+            
+        Returns:
+            bool: 是否成功删除
+        """
+        file_path = os.path.join(self.store_dir, f"{checkpoint_id}.json")
+        
+        if not os.path.exists(file_path):
+            logger.warning(f"要删除的对话检查点不存在: {checkpoint_id}")
+            return False
+        
+        try:
+            os.remove(file_path)
+            logger.info(f"已删除对话检查点: {checkpoint_id}")
+            return True
+        except Exception as e:
+            logger.error(f"删除对话检查点失败: {str(e)}")
+            return False
+    
+    def _get_all_checkpoints(self) -> List[ConversationCheckpoint]:
+        """
+        获取所有对话检查点
+        
+        Returns:
+            List[ConversationCheckpoint]: 对话检查点列表
+        """
+        checkpoints = []
+        
+        try:
+            for filename in os.listdir(self.store_dir):
+                if filename.endswith('.json'):
+                    file_path = os.path.join(self.store_dir, filename)
+                    
+                    try:
+                        with open(file_path, 'r', encoding='utf-8') as f:
+                            data = json.load(f)
+                        
+                        checkpoint = ConversationCheckpoint(**data)
+                        checkpoints.append(checkpoint)
+                    except Exception as e:
+                        logger.error(f"读取对话检查点文件失败 {filename}: {str(e)}")
+        except Exception as e:
+            logger.error(f"获取所有对话检查点失败: {str(e)}")
+        
+        return checkpoints
+    
+    def _cleanup_old_checkpoints(self):
+        """清理过期的检查点，保持历史记录数量在限制范围内"""
+        checkpoints = self._get_all_checkpoints()
+        
+        if len(checkpoints) <= self.max_history:
+            return
+        
+        # 按时间戳降序排序
+        checkpoints.sort(key=lambda x: x.timestamp, reverse=True)
+        
+        # 删除超出限制的旧检查点
+        for checkpoint in checkpoints[self.max_history:]:
+            self.delete_checkpoint(checkpoint.checkpoint_id)

autocoder/common/file_checkpoint/manager.py

@@ -16,6 +16,7 @@ from autocoder.common.file_checkpoint.models import (
 )
 from autocoder.common.file_checkpoint.backup import FileBackupManager
 from autocoder.common.file_checkpoint.store import FileChangeStore
+from autocoder.common.file_checkpoint.conversation_checkpoint import ConversationCheckpointStore,ConversationCheckpoint
 
 logger = logging.getLogger(__name__)
 
@@ -24,7 +25,8 @@ class FileChangeManager:
     """文件变更管理器，提供高层次的API接口"""
     
     def __init__(self, project_dir: str, backup_dir: Optional[str] = None, 
-                 store_dir: Optional[str] = None, max_history: int = 50):
+                 store_dir: Optional[str] = None, max_history: int = 50,
+                 conversation_store_dir: Optional[str] = None):
         """
         初始化文件变更管理器
         
@@ -33,10 +35,61 @@ class FileChangeManager:
             backup_dir: 备份文件存储目录，默认为用户主目录下的.autocoder/backups
             store_dir: 变更记录存储目录，默认为用户主目录下的.autocoder/changes
             max_history: 最大保存的历史版本数量
+            conversation_store_dir: 对话检查点存储目录
         """
         self.project_dir = os.path.abspath(project_dir)
         self.backup_manager = FileBackupManager(backup_dir)
         self.change_store = FileChangeStore(store_dir, max_history)
+        
+        # 初始化对话检查点存储
+        if conversation_store_dir is None and store_dir is not None:
+            # 默认在变更记录存储目录的同级目录创建 conversation_checkpoints 目录
+            parent_dir = os.path.dirname(store_dir)
+            conversation_store_dir = os.path.join(parent_dir, "conversation_checkpoints")
+        
+        try:            
+            self.conversation_store = ConversationCheckpointStore(conversation_store_dir, max_history)
+            logger.info(f"对话检查点存储初始化成功: {conversation_store_dir}")
+        except ImportError as e:
+            logger.warning(f"对话检查点存储初始化失败: {str(e)}")
+            self.conversation_store = None
+    
+    def apply_changes_with_conversation(self, changes: Dict[str, FileChange], 
+                                   conversations: List[Dict[str, Any]],
+                                   change_group_id: Optional[str] = None,
+                                   metadata: Optional[Dict[str, Any]] = None) -> ApplyResult:
+        """
+        应用文件变更并保存对话状态
+        
+        Args:
+            changes: 文件变更字典，格式为 {file_path: FileChange}
+            conversations: 当前对话历史
+            change_group_id: 变更组ID，用于将相关变更归为一组
+            metadata: 元数据，可包含额外信息
+            
+        Returns:
+            ApplyResult: 应用结果对象
+        """
+        # 应用文件变更
+        result = self.apply_changes(changes, change_group_id)
+        
+        if result.success and self.conversation_store is not None:
+            try:                
+                
+                # 创建并保存对话检查点
+                checkpoint_id = change_group_id or result.change_ids[0] if result.change_ids else str(uuid.uuid4())
+                checkpoint = ConversationCheckpoint(
+                    checkpoint_id=checkpoint_id,
+                    timestamp=time.time(),
+                    conversations=conversations,
+                    metadata=metadata
+                )
+                self.conversation_store.save_checkpoint(checkpoint)
+                logger.info(f"已保存对话检查点: {checkpoint_id}")
+            except Exception as e:
+                logger.error(f"保存对话检查点失败: {str(e)}")
+        
+        return result
     
     def apply_changes(self, changes: Dict[str, FileChange], change_group_id: Optional[str] = None) -> ApplyResult:
         """
@@ -152,6 +205,27 @@ class FileChangeManager:
         
         return diff_results
     
+    def undo_last_change_with_conversation(self) -> Tuple[UndoResult, Optional[ConversationCheckpoint]]:
+        """
+        撤销最近的一次变更并恢复对话状态
+        
+        Returns:
+            Tuple[UndoResult, Optional[ConversationCheckpoint]]: 撤销结果和恢复的对话检查点
+        """
+        # 获取最近的变更记录
+        latest_changes = self.change_store.get_latest_changes(limit=1)
+        if not latest_changes:
+            return UndoResult(success=False, errors={"general": "没有找到最近的变更记录"}), None
+        
+        latest_change = latest_changes[0]
+        
+        # 如果最近的变更属于一个组，撤销整个组
+        if latest_change.group_id:
+            return self.undo_change_group_with_conversation(latest_change.group_id)
+        else:
+            # 否则只撤销这一个变更
+            return self.undo_change_with_conversation(latest_change.change_id)
+    
     def undo_last_change(self) -> UndoResult:
         """
         撤销最近的一次变更
@@ -173,6 +247,39 @@ class FileChangeManager:
             # 否则只撤销这一个变更
             return self.undo_change(latest_change.change_id)
     
+    def undo_change_with_conversation(self, change_id: str) -> Tuple[UndoResult, Optional[ConversationCheckpoint]]:
+        """
+        撤销指定的变更并恢复对话状态
+        
+        Args:
+            change_id: 变更记录ID
+            
+        Returns:
+            Tuple[UndoResult, Optional[ConversationCheckpoint]]: 撤销结果和恢复的对话检查点
+        """
+        # 获取变更记录
+        change_record = self.change_store.get_change(change_id)
+        if change_record is None:
+            return UndoResult(success=False, errors={"general": f"变更记录 {change_id} 不存在"}), None
+        
+        # 获取关联的对话检查点
+        checkpoint = None
+        checkpoint_id = change_record.group_id or change_id
+        if self.conversation_store is not None:
+            try:
+                checkpoint = self.conversation_store.get_checkpoint(checkpoint_id)
+                if checkpoint:
+                    logger.info(f"找到关联的对话检查点: {checkpoint_id}")
+                else:
+                    logger.info(f"未找到关联的对话检查点: {checkpoint_id}")
+            except Exception as e:
+                logger.error(f"获取对话检查点失败: {str(e)}")
+        
+        # 撤销文件变更
+        undo_result = self.undo_change(change_id)
+        
+        return undo_result, checkpoint
+    
     def undo_change(self, change_id: str) -> UndoResult:
         """
         撤销指定的变更
@@ -234,6 +341,38 @@ class FileChangeManager:
         
         return result
             
+    def undo_change_group_with_conversation(self, group_id: str) -> Tuple[UndoResult, Optional[ConversationCheckpoint]]:
+        """
+        撤销指定组的所有变更并恢复对话状态
+        
+        Args:
+            group_id: 变更组ID
+            
+        Returns:
+            Tuple[UndoResult, Optional[ConversationCheckpoint]]: 撤销结果和恢复的对话检查点
+        """
+        # 获取组内的所有变更记录
+        changes = self.change_store.get_changes_by_group(group_id)
+        if not changes:
+            return UndoResult(success=False, errors={"general": f"变更组 {group_id} 不存在或为空"}), None
+        
+        # 获取关联的对话检查点
+        checkpoint = None
+        if self.conversation_store is not None:
+            try:
+                checkpoint = self.conversation_store.get_checkpoint(group_id)
+                if checkpoint:
+                    logger.info(f"找到关联的对话检查点: {group_id}")
+                else:
+                    logger.info(f"未找到关联的对话检查点: {group_id}")
+            except Exception as e:
+                logger.error(f"获取对话检查点失败: {str(e)}")
+        
+        # 撤销文件变更
+        undo_result = self.undo_change_group(group_id)
+        
+        return undo_result, checkpoint
+    
     def undo_change_group(self, group_id: str) -> UndoResult:
         """
         撤销指定组的所有变更
@@ -265,6 +404,39 @@ class FileChangeManager:
         
         return result
     
+    def undo_to_version_with_conversation(self, version_id: str) -> Tuple[UndoResult, Optional[ConversationCheckpoint]]:
+        """
+        撤销到指定的历史版本并恢复对话状态
+        
+        Args:
+            version_id: 目标版本ID（变更记录ID）
+            
+        Returns:
+            Tuple[UndoResult, Optional[ConversationCheckpoint]]: 撤销结果和恢复的对话检查点
+        """
+        # 获取目标版本的变更记录
+        target_change = self.change_store.get_change(version_id)
+        if target_change is None:
+            return UndoResult(success=False, errors={"general": f"变更记录 {version_id} 不存在"}), None
+        
+        # 获取关联的对话检查点
+        checkpoint = None
+        checkpoint_id = target_change.group_id or version_id
+        if self.conversation_store is not None:
+            try:
+                checkpoint = self.conversation_store.get_checkpoint(checkpoint_id)
+                if checkpoint:
+                    logger.info(f"找到关联的对话检查点: {checkpoint_id}")
+                else:
+                    logger.info(f"未找到关联的对话检查点: {checkpoint_id}")
+            except Exception as e:
+                logger.error(f"获取对话检查点失败: {str(e)}")
+        
+        # 撤销文件变更
+        undo_result = self.undo_to_version(version_id)
+        
+        return undo_result, checkpoint
+    
     def undo_to_version(self, version_id: str) -> UndoResult:
         """
         撤销到指定的历史版本
@@ -359,6 +531,41 @@ class FileChangeManager:
         """
         return self.change_store.get_change_groups(limit)
     
+    def get_available_checkpoints(self, limit: int = 10) -> List[Dict[str, Any]]:
+        """
+        获取可用的检查点列表，包含对话状态信息
+        
+        Args:
+            limit: 返回的检查点数量限制
+            
+        Returns:
+            List[Dict[str, Any]]: 检查点信息列表
+        """
+        # 获取变更组列表
+        change_groups = self.get_change_groups(limit)
+        
+        # 构建检查点信息
+        checkpoints = []
+        for group_id, timestamp, count in change_groups:
+            has_conversation = False
+            
+            # 检查是否有对话检查点
+            if self.conversation_store is not None:
+                try:
+                    checkpoint = self.conversation_store.get_checkpoint(group_id)
+                    has_conversation = checkpoint is not None
+                except Exception as e:
+                    logger.error(f"获取对话检查点失败: {str(e)}")
+            
+            checkpoints.append({
+                "id": group_id,
+                "timestamp": timestamp,
+                "changes_count": count,
+                "has_conversation": has_conversation
+            })
+        
+        return checkpoints
+    
     def get_diff_text(self, old_content: str, new_content: str) -> str:
         """
         获取两个文本内容的差异文本

autocoder/common/utils_code_auto_generate.py

@@ -89,7 +89,8 @@ def stream_chat_with_continue(
             content = res[0]
             current_content += content 
             if current_metadata is None:
-                current_metadata = res[1]            
+                current_metadata = res[1]      
+                metadatas[count] = res[1]      
             else:
                 metadatas[count] = res[1]                
                 current_metadata.finish_reason = res[1].finish_reason