mirror of
https://github.com/tiennm99/DocsGPT.git
synced 2026-10-04 12:13:05 +00:00
arc53-machine
d70644743b
SAGEMAKER_REGION, SAGEMAKER_ACCESS_KEY and SAGEMAKER_SECRET_KEY survive only as a fallback for the S3_* credentials. They carry Field(deprecated=...) now, so any read emits a DeprecationWarning naming the replacement and the generated reference shows the notice. The S3 store is the one sanctioned reader; it silences that warning locally because it already logs its own operator-facing one when the fallback is actually used. DEFAULT_MAX_HISTORY was referenced nowhere. RETRIEVERS_ENABLED was read by no code at all, while two docs pages described it as an enforced allow-list; both the setting and those claims are removed.
92 lines
4.1 KiB
Python
92 lines
4.1 KiB
Python
"""Agent runtime: default tools, context management, workflows and artifacts."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Optional
|
|
|
|
from pydantic import Field
|
|
|
|
from docsgpt.core.settings._shared import SettingsGroup
|
|
|
|
|
|
class AgentSettings(SettingsGroup):
|
|
"""What an agent may do per turn and how its context is kept within budget."""
|
|
|
|
AGENT_NAME: str = Field(default="classic", description="Default agent type for agentless chats.")
|
|
DEFAULT_AGENT_LIMITS: dict[str, int] = Field(
|
|
default={"token_limit": 50000, "request_limit": 500},
|
|
description="Per-agent default quotas: tokens and requests.",
|
|
)
|
|
DEFAULT_CHAT_TOOLS: list[str] = Field(
|
|
default=["memory", "read_webpage", "scheduler"],
|
|
description=(
|
|
"Config-free tools on by default in agentless chats. scheduler is dual-registered in "
|
|
"BUILTIN_AGENT_TOOLS so one synthetic id resolves via defaults or the agent picker. Add "
|
|
"code_executor and artifact_generator once a sandbox runner is configured; both execute through "
|
|
"it and would fail on every call without one."
|
|
),
|
|
)
|
|
ENABLE_TOOL_PREFETCH: bool = Field(default=True, description="Pre-fetch retrieval before the agent's first turn.")
|
|
TOOL_RESULT_MAX_TOKENS: int = Field(
|
|
default=20000,
|
|
ge=0,
|
|
description="Cap on one tool result entering the LLM context (0 disables); journal and DB keep it whole.",
|
|
)
|
|
|
|
# Conversation compression.
|
|
ENABLE_CONVERSATION_COMPRESSION: bool = Field(
|
|
default=True, description="Compress long conversations once they approach the context window."
|
|
)
|
|
COMPRESSION_THRESHOLD_PERCENTAGE: float = Field(
|
|
default=0.8, gt=0, le=1, description="Fraction of the context window at which compression triggers."
|
|
)
|
|
COMPRESSION_MODEL_OVERRIDE: Optional[str] = Field(
|
|
default=None, description="Use a different model for compression; unset reuses the answer model."
|
|
)
|
|
COMPRESSION_PROMPT_VERSION: str = Field(default="v1.0", description="Tracks compression prompt iterations.")
|
|
COMPRESSION_MAX_HISTORY_POINTS: int = Field(
|
|
default=3, description="Keep only the last N compression points to prevent DB bloat."
|
|
)
|
|
COMPRESSION_RECENT_FIELD_MAX_TOKENS: int = Field(
|
|
default=8000, ge=0, description="Per-field cap on the verbatim tail kept after a compression point (0 disables)."
|
|
)
|
|
|
|
# Workflows.
|
|
WORKFLOW_NODE_NATIVE_MAX_FILES: int = Field(
|
|
default=5,
|
|
description=(
|
|
"Files per node passed natively to the LLM; past the cap they are extracted to text or dropped, to "
|
|
"bound context and cost. Re-uses SANDBOX_MAX_INPUT_BYTES per file."
|
|
),
|
|
)
|
|
WORKFLOW_NODE_EXTRACT_MAX_FILES: int = Field(
|
|
default=5,
|
|
description=(
|
|
"Documents per node extracted via the parsing worker. Each issues a separate blocking parse; past "
|
|
"the cap they are skipped with a truncation note."
|
|
),
|
|
)
|
|
WORKFLOW_NODE_EXTRACT_BUDGET_SECONDS: int = Field(
|
|
default=900,
|
|
description=(
|
|
"Wall clock one node may spend on blocking parses, shared across all of them. Without it a node "
|
|
"could serialize WORKFLOW_NODE_EXTRACT_MAX_FILES full windows on a web threadpool slot."
|
|
),
|
|
)
|
|
WORKFLOW_RUN_STALE_SECONDS: int = Field(
|
|
default=3600,
|
|
description=(
|
|
"A run row is pre-created as running; a disconnect or crash can strand it there. The beat reaper "
|
|
"fails runs still running past this. Generous so a long run is never cut off."
|
|
),
|
|
)
|
|
|
|
# Per-user artifact quotas, enforced at persistence time. 0 or negative disables a quota.
|
|
ARTIFACT_MAX_BYTES: int = Field(
|
|
default=50 * 1024 * 1024, description="Cap on a single stored artifact version's bytes (0 disables)."
|
|
)
|
|
ARTIFACT_MAX_COUNT_PER_USER: int = Field(default=5000, description="Cap on artifacts a user may own (0 disables).")
|
|
ARTIFACT_MAX_TOTAL_BYTES_PER_USER: int = Field(
|
|
default=5 * 1024 * 1024 * 1024, description="Cap on a user's total stored artifact bytes (0 disables)."
|
|
)
|