Teknium
e83816a4d1
For each issue anchor present in BASE 63279301bc non-test .py and absent on HEAD, the BASE comment/docstring block was re-attached at the HEAD location of the code it explained (matched by the distinctive code line / enclosing def). Sentences already covered by an existing HEAD comment were deduped; the issue number always survives. Insert-only: no code lines changed.
91 lines
3.4 KiB
Python
91 lines
3.4 KiB
Python
"""Typed failure-reason codes for bot turns and relay replies.
|
|
|
|
A closed vocabulary of machine-readable reason codes carried ALONGSIDE the
|
|
free-text ``error`` fields (additive — old consumers keep working). Platform-side
|
|
codes are assigned by the transport/relay layer; agent-side codes are derived
|
|
from raw agent/provider error text via ``classify_agent_error``. Classifier
|
|
precedence is the order of ``_RULES``: auth outranks quota by design — real
|
|
provider 401 bodies (e.g. Anthropic) say "invalid, blocked or out of funds".
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import re
|
|
|
|
# platform-side
|
|
RUNTIME_OFFLINE = "runtime_offline"
|
|
QUEUED_EXPIRED = "queued_expired"
|
|
DELIVERY_TIMEOUT = "delivery_timeout"
|
|
AGENT_BLOCKED = "agent_blocked"
|
|
CANCELLED = "cancelled"
|
|
|
|
# agent-side
|
|
PROVIDER_AUTH_OR_ACCESS = "provider_auth_or_access"
|
|
PROVIDER_QUOTA_LIMIT = "provider_quota_limit"
|
|
PROVIDER_RATE_LIMIT = "provider_rate_limit"
|
|
PROVIDER_SERVER_ERROR = "provider_server_error"
|
|
CONTEXT_OVERFLOW = "context_overflow"
|
|
MISSING_CONFIG = "missing_config"
|
|
MODEL_UNAVAILABLE = "model_unavailable"
|
|
UNKNOWN = "unknown"
|
|
|
|
ALL_REASONS = frozenset({
|
|
RUNTIME_OFFLINE, QUEUED_EXPIRED, DELIVERY_TIMEOUT, AGENT_BLOCKED, CANCELLED,
|
|
PROVIDER_AUTH_OR_ACCESS, PROVIDER_QUOTA_LIMIT, PROVIDER_RATE_LIMIT,
|
|
PROVIDER_SERVER_ERROR, CONTEXT_OVERFLOW, MISSING_CONFIG, MODEL_UNAVAILABLE, UNKNOWN,
|
|
})
|
|
|
|
#: Reasons a supervisor may retry automatically without human intervention.
|
|
AUTO_RETRYABLE = frozenset({RUNTIME_OFFLINE, DELIVERY_TIMEOUT, PROVIDER_RATE_LIMIT, PROVIDER_SERVER_ERROR})
|
|
|
|
|
|
def is_auto_retryable(reason: str) -> bool:
|
|
return reason in AUTO_RETRYABLE
|
|
|
|
|
|
# Retry session policy: a retried bot turn NEVER mints a fresh session. Transient
|
|
# classes resume as-is; context_overflow runs context compression (the one
|
|
# sanctioned context mutation) on the same session first; everything else
|
|
# (auth/quota/config/model/unknown) is never auto-retried — it can't be fixed by
|
|
# a retry and only burns quota.
|
|
# See #93091.
|
|
RETRY_RESUME = "resume"
|
|
RETRY_COMPRESS_THEN_RESUME = "compress_then_resume"
|
|
RETRY_NONE = "none"
|
|
|
|
|
|
def retry_action(reason: str) -> str:
|
|
"""Map a failure reason to the bot-turn retry action (see policy above)."""
|
|
if reason in AUTO_RETRYABLE:
|
|
return RETRY_RESUME
|
|
if reason == CONTEXT_OVERFLOW:
|
|
return RETRY_COMPRESS_THEN_RESUME
|
|
return RETRY_NONE
|
|
|
|
|
|
_STATUS = r"(?:error code:?\s*|status(?:\s*code)?:?\s*|http\s*)"
|
|
|
|
# Ordered (pattern, code) — first match wins.
|
|
_RULES: tuple[tuple[re.Pattern[str], str], ...] = tuple(
|
|
(re.compile(pat, re.IGNORECASE), code)
|
|
for pat, code in (
|
|
(rf"authentication_error|invalid api key|{_STATUS}(?:401|403)\b", PROVIDER_AUTH_OR_ACCESS),
|
|
(rf"{_STATUS}402\b|out of funds|quota|balance", PROVIDER_QUOTA_LIMIT),
|
|
(rf"{_STATUS}429\b|rate.?limit", PROVIDER_RATE_LIMIT),
|
|
(rf"{_STATUS}5\d{{2}}\b|server error|overloaded", PROVIDER_SERVER_ERROR),
|
|
(r"context length|context_overflow|maximum context", CONTEXT_OVERFLOW),
|
|
(r"no llm provider configured|missing config|no access token", MISSING_CONFIG),
|
|
(r"model .*(not found|does not exist)|model_not_found", MODEL_UNAVAILABLE),
|
|
)
|
|
)
|
|
|
|
|
|
def classify_agent_error(text: str) -> str:
|
|
"""Map raw agent/provider error text to a closed reason code (``unknown`` when unmatched/empty)."""
|
|
raw = str(text or "")
|
|
if raw.strip():
|
|
for pattern, code in _RULES:
|
|
if pattern.search(raw):
|
|
return code
|
|
return UNKNOWN
|