| from dataclasses import dataclass |
| from typing import TYPE_CHECKING |
|
|
| from .compressor import ContextCompressor |
| from .token_counter import TokenCounter |
|
|
| if TYPE_CHECKING: |
| from astrbot.core.provider.provider import Provider |
|
|
|
|
| @dataclass |
| class ContextConfig: |
| """Context configuration class.""" |
|
|
| max_context_tokens: int = 0 |
| """Maximum number of context tokens. <= 0 means no limit.""" |
| enforce_max_turns: int = -1 |
| """Maximum number of conversation turns to keep. -1 means no limit. Executed before compression.""" |
| truncate_turns: int = 1 |
| """Number of conversation turns to discard at once when truncation is triggered. |
| Two processes will use this value: |
| |
| 1. Enforce max turns truncation. |
| 2. Truncation by turns compression strategy. |
| """ |
| llm_compress_instruction: str | None = None |
| """Instruction prompt for LLM-based compression.""" |
| llm_compress_keep_recent: int = 0 |
| """Number of recent messages to keep during LLM-based compression.""" |
| llm_compress_provider: "Provider | None" = None |
| """LLM provider used for compression tasks. If None, truncation strategy is used.""" |
| custom_token_counter: TokenCounter | None = None |
| """Custom token counting method. If None, the default method is used.""" |
| custom_compressor: ContextCompressor | None = None |
| """Custom context compression method. If None, the default method is used.""" |
|
|
