kimi-cli 0.44__py3-none-any.whl → 0.78__py3-none-any.whl

kimi_cli/soul/context.py

@@ -1,10 +1,12 @@
+from __future__ import annotations
+
 import json
 from collections.abc import Sequence
 from pathlib import Path
 
 import aiofiles
 import aiofiles.os
-from kosong.base.message import Message
+from kosong.message import Message
 
 from kimi_cli.soul.message import system
 from kimi_cli.utils.logging import logger
@@ -59,6 +61,10 @@ class Context:
     def n_checkpoints(self) -> int:
         return self._next_checkpoint_id
 
+    @property
+    def file_backend(self) -> Path:
+        return self._file_backend
+
     async def checkpoint(self, add_user_message: bool):
         checkpoint_id = self._next_checkpoint_id
         self._next_checkpoint_id += 1
@@ -90,14 +96,14 @@ class Context:
             logger.error("Checkpoint {checkpoint_id} does not exist", checkpoint_id=checkpoint_id)
             raise ValueError(f"Checkpoint {checkpoint_id} does not exist")
 
-        # rotate the history file
+        # rotate the context file
         rotated_file_path = await next_available_rotation(self._file_backend)
         if rotated_file_path is None:
             logger.error("No available rotation path found")
             raise RuntimeError("No available rotation path found")
-        await aiofiles.os.rename(self._file_backend, rotated_file_path)
+        await aiofiles.os.replace(self._file_backend, rotated_file_path)
         logger.debug(
-            "Rotated history file: {rotated_file_path}", rotated_file_path=rotated_file_path
+            "Rotated context file: {rotated_file_path}", rotated_file_path=rotated_file_path
         )
 
         # restore the context until the specified checkpoint
@@ -125,9 +131,36 @@ class Context:
                     message = Message.model_validate(line_json)
                     self._history.append(message)
 
+    async def clear(self):
+        """
+        Clear the context history.
+        This is almost equivalent to revert_to(0), but without relying on the assumption
+        that the first checkpoint exists.
+        File backend will be rotated.
+
+        Raises:
+            RuntimeError: When no available rotation path is found.
+        """
+
+        logger.debug("Clearing context")
+
+        # rotate the context file
+        rotated_file_path = await next_available_rotation(self._file_backend)
+        if rotated_file_path is None:
+            logger.error("No available rotation path found")
+            raise RuntimeError("No available rotation path found")
+        await aiofiles.os.replace(self._file_backend, rotated_file_path)
+        logger.debug(
+            "Rotated context file: {rotated_file_path}", rotated_file_path=rotated_file_path
+        )
+
+        self._history.clear()
+        self._token_count = 0
+        self._next_checkpoint_id = 0
+
     async def append_message(self, message: Message | Sequence[Message]):
         logger.debug("Appending message(s) to context: {message}", message=message)
-        messages = message if isinstance(message, Sequence) else [message]
+        messages = [message] if isinstance(message, Message) else message
         self._history.extend(messages)
 
         async with aiofiles.open(self._file_backend, "a", encoding="utf-8") as f:

kimi_cli/soul/denwarenji.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from pydantic import BaseModel, Field