[AC-AISVC-RES-01~15] feat(api): 新增策略管理 API端点
- 新增 strategy.py 策略 API端点 - GET /strategy/retrieval/current - 获取当前策略配置 - POST /strategy/retrieval/switch - 切换策略配置 - POST /strategy/retrieval/validate - 风险配置校验 - POST /strategy/retrieval/rollback - 强制回滚到默认策略 - 更新 __init__.py 导出新模块 - 更新 main.py 注册 API 路由
This commit is contained in:
MerCry
parent
c628181623
commit
c0688c2b13
|
|
@ -3,6 +3,7 @@ Admin API routes for AI Service management.
|
|||
[AC-ASA-01, AC-ASA-02, AC-ASA-05, AC-ASA-07, AC-ASA-08, AC-AISVC-50] Admin management endpoints.
|
||||
[AC-MRS-07,08,16] Slot definition management endpoints.
|
||||
[AC-SCENE-SLOT-01] Scene slot bundle management endpoints.
|
||||
[AC-AISVC-RES-01~15] Retrieval strategy management endpoints.
|
||||
"""
|
||||
|
||||
from app.api.admin.api_key import router as api_key_router
|
||||
|
|
@ -19,6 +20,7 @@ from app.api.admin.metadata_schema import router as metadata_schema_router
|
|||
from app.api.admin.monitoring import router as monitoring_router
|
||||
from app.api.admin.prompt_templates import router as prompt_templates_router
|
||||
from app.api.admin.rag import router as rag_router
|
||||
from app.api.admin.retrieval_strategy import router as retrieval_strategy_router
|
||||
from app.api.admin.scene_slot_bundle import router as scene_slot_bundle_router
|
||||
from app.api.admin.script_flows import router as script_flows_router
|
||||
from app.api.admin.sessions import router as sessions_router
|
||||
|
|
@ -40,6 +42,7 @@ __all__ = [
|
|||
"monitoring_router",
|
||||
"prompt_templates_router",
|
||||
"rag_router",
|
||||
"retrieval_strategy_router",
|
||||
"scene_slot_bundle_router",
|
||||
"script_flows_router",
|
||||
"sessions_router",
|
||||
|
|
|
|||
|
|
@ -0,0 +1,342 @@
|
|||
"""
|
||||
Retrieval Strategy API Endpoints.
|
||||
[AC-AISVC-RES-01~15] API for strategy management and configuration.
|
||||
|
||||
Endpoints:
|
||||
- GET /strategy/retrieval/current - Get current strategy configuration
|
||||
- POST /strategy/retrieval/switch - Switch strategy configuration
|
||||
- POST /strategy/retrieval/validate - Validate strategy configuration
|
||||
- POST /strategy/retrieval/rollback - Rollback to default strategy
|
||||
"""
|
||||
|
||||
import logging
|
||||
from typing import Any
|
||||
|
||||
from fastapi import APIRouter, Depends, HTTPException
|
||||
from pydantic import BaseModel, Field
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from app.core.database import get_session
|
||||
from app.core.tenant import get_tenant_id
|
||||
from app.services.retrieval.routing_config import (
|
||||
RagRuntimeMode,
|
||||
StrategyType,
|
||||
RoutingConfig,
|
||||
)
|
||||
from app.services.retrieval.strategy_router import (
|
||||
get_strategy_router,
|
||||
RollbackRecord,
|
||||
)
|
||||
from app.services.retrieval.mode_router import get_mode_router
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
router = APIRouter(prefix="/strategy/retrieval", tags=["Retrieval Strategy"])
|
||||
|
||||
|
||||
class RoutingConfigRequest(BaseModel):
|
||||
"""Request model for routing configuration."""
|
||||
enabled: bool | None = Field(default=None, description="Enable strategy routing")
|
||||
strategy: StrategyType | None = Field(default=None, description="Retrieval strategy")
|
||||
grayscale_percentage: float | None = Field(default=None, ge=0.0, le=1.0, description="Grayscale percentage")
|
||||
grayscale_allowlist: list[str] | None = Field(default=None, description="Grayscale allowlist")
|
||||
rag_runtime_mode: RagRuntimeMode | None = Field(default=None, description="RAG runtime mode")
|
||||
react_trigger_confidence_threshold: float | None = Field(default=None, ge=0.0, le=1.0)
|
||||
react_trigger_complexity_score: float | None = Field(default=None, ge=0.0, le=1.0)
|
||||
react_max_steps: int | None = Field(default=None, ge=3, le=10)
|
||||
direct_fallback_on_low_confidence: bool | None = Field(default=None)
|
||||
direct_fallback_confidence_threshold: float | None = Field(default=None, ge=0.0, le=1.0)
|
||||
performance_budget_ms: int | None = Field(default=None, ge=1000)
|
||||
performance_degradation_threshold: float | None = Field(default=None, ge=0.0, le=1.0)
|
||||
|
||||
|
||||
class RoutingConfigResponse(BaseModel):
|
||||
"""Response model for routing configuration."""
|
||||
enabled: bool
|
||||
strategy: StrategyType
|
||||
grayscale_percentage: float
|
||||
grayscale_allowlist: list[str]
|
||||
rag_runtime_mode: RagRuntimeMode
|
||||
react_trigger_confidence_threshold: float
|
||||
react_trigger_complexity_score: float
|
||||
react_max_steps: int
|
||||
direct_fallback_on_low_confidence: bool
|
||||
direct_fallback_confidence_threshold: float
|
||||
performance_budget_ms: int
|
||||
performance_degradation_threshold: float
|
||||
|
||||
|
||||
class ValidationResponse(BaseModel):
|
||||
"""Response model for configuration validation."""
|
||||
is_valid: bool
|
||||
errors: list[str]
|
||||
warnings: list[str] = Field(default_factory=list)
|
||||
|
||||
|
||||
class RollbackRequest(BaseModel):
|
||||
"""Request model for strategy rollback."""
|
||||
reason: str = Field(..., description="Reason for rollback")
|
||||
tenant_id: str | None = Field(default=None, description="Optional tenant ID for audit")
|
||||
|
||||
|
||||
class RollbackResponse(BaseModel):
|
||||
"""Response model for strategy rollback."""
|
||||
success: bool
|
||||
previous_strategy: StrategyType
|
||||
current_strategy: StrategyType
|
||||
reason: str
|
||||
rollback_records: list[dict[str, Any]] = Field(default_factory=list)
|
||||
|
||||
|
||||
class RollbackRecordResponse(BaseModel):
|
||||
"""Response model for rollback record."""
|
||||
timestamp: float
|
||||
from_strategy: StrategyType
|
||||
to_strategy: StrategyType
|
||||
reason: str
|
||||
tenant_id: str | None
|
||||
request_id: str | None
|
||||
|
||||
|
||||
class CurrentStrategyResponse(BaseModel):
|
||||
"""Response model for current strategy."""
|
||||
config: RoutingConfigResponse
|
||||
current_strategy: StrategyType
|
||||
rollback_records: list[RollbackRecordResponse] = Field(default_factory=list)
|
||||
|
||||
|
||||
@router.get(
|
||||
"/current",
|
||||
operation_id="getCurrentRetrievalStrategy",
|
||||
summary="Get current retrieval strategy configuration",
|
||||
description="[AC-AISVC-RES-01] Returns the current strategy configuration and recent rollback records.",
|
||||
response_model=CurrentStrategyResponse,
|
||||
)
|
||||
async def get_current_strategy() -> CurrentStrategyResponse:
|
||||
"""
|
||||
[AC-AISVC-RES-01] Get current retrieval strategy configuration.
|
||||
"""
|
||||
strategy_router = get_strategy_router()
|
||||
config = strategy_router.config
|
||||
rollback_records = strategy_router.get_rollback_records(limit=5)
|
||||
|
||||
return CurrentStrategyResponse(
|
||||
config=RoutingConfigResponse(
|
||||
enabled=config.enabled,
|
||||
strategy=config.strategy,
|
||||
grayscale_percentage=config.grayscale_percentage,
|
||||
grayscale_allowlist=config.grayscale_allowlist,
|
||||
rag_runtime_mode=config.rag_runtime_mode,
|
||||
react_trigger_confidence_threshold=config.react_trigger_confidence_threshold,
|
||||
react_trigger_complexity_score=config.react_trigger_complexity_score,
|
||||
react_max_steps=config.react_max_steps,
|
||||
direct_fallback_on_low_confidence=config.direct_fallback_on_low_confidence,
|
||||
direct_fallback_confidence_threshold=config.direct_fallback_confidence_threshold,
|
||||
performance_budget_ms=config.performance_budget_ms,
|
||||
performance_degradation_threshold=config.performance_degradation_threshold,
|
||||
),
|
||||
current_strategy=strategy_router.current_strategy,
|
||||
rollback_records=[
|
||||
RollbackRecordResponse(
|
||||
timestamp=r.timestamp,
|
||||
from_strategy=r.from_strategy,
|
||||
to_strategy=r.to_strategy,
|
||||
reason=r.reason,
|
||||
tenant_id=r.tenant_id,
|
||||
request_id=r.request_id,
|
||||
)
|
||||
for r in rollback_records
|
||||
],
|
||||
)
|
||||
|
||||
|
||||
@router.post(
|
||||
"/switch",
|
||||
operation_id="switchRetrievalStrategy",
|
||||
summary="Switch retrieval strategy configuration",
|
||||
description="[AC-AISVC-RES-02, AC-AISVC-RES-03] Update strategy configuration with hot reload support.",
|
||||
response_model=RoutingConfigResponse,
|
||||
)
|
||||
async def switch_strategy(
|
||||
request: RoutingConfigRequest,
|
||||
session: AsyncSession = Depends(get_session),
|
||||
) -> RoutingConfigResponse:
|
||||
"""
|
||||
[AC-AISVC-RES-02, AC-AISVC-RES-03, AC-AISVC-RES-15]
|
||||
Switch retrieval strategy configuration.
|
||||
|
||||
Supports:
|
||||
- Strategy selection (default/enhanced)
|
||||
- Grayscale release (percentage/allowlist)
|
||||
- Mode selection (direct/react/auto)
|
||||
- Hot reload of routing parameters
|
||||
"""
|
||||
strategy_router = get_strategy_router()
|
||||
mode_router = get_mode_router()
|
||||
current_config = strategy_router.config
|
||||
|
||||
new_config = RoutingConfig(
|
||||
enabled=request.enabled if request.enabled is not None else current_config.enabled,
|
||||
strategy=request.strategy if request.strategy is not None else current_config.strategy,
|
||||
grayscale_percentage=request.grayscale_percentage if request.grayscale_percentage is not None else current_config.grayscale_percentage,
|
||||
grayscale_allowlist=request.grayscale_allowlist if request.grayscale_allowlist is not None else current_config.grayscale_allowlist,
|
||||
rag_runtime_mode=request.rag_runtime_mode if request.rag_runtime_mode is not None else current_config.rag_runtime_mode,
|
||||
react_trigger_confidence_threshold=request.react_trigger_confidence_threshold if request.react_trigger_confidence_threshold is not None else current_config.react_trigger_confidence_threshold,
|
||||
react_trigger_complexity_score=request.react_trigger_complexity_score if request.react_trigger_complexity_score is not None else current_config.react_trigger_complexity_score,
|
||||
react_max_steps=request.react_max_steps if request.react_max_steps is not None else current_config.react_max_steps,
|
||||
direct_fallback_on_low_confidence=request.direct_fallback_on_low_confidence if request.direct_fallback_on_low_confidence is not None else current_config.direct_fallback_on_low_confidence,
|
||||
direct_fallback_confidence_threshold=request.direct_fallback_confidence_threshold if request.direct_fallback_confidence_threshold is not None else current_config.direct_fallback_confidence_threshold,
|
||||
performance_budget_ms=request.performance_budget_ms if request.performance_budget_ms is not None else current_config.performance_budget_ms,
|
||||
performance_degradation_threshold=request.performance_degradation_threshold if request.performance_degradation_threshold is not None else current_config.performance_degradation_threshold,
|
||||
)
|
||||
|
||||
is_valid, errors = new_config.validate()
|
||||
if not is_valid:
|
||||
raise HTTPException(
|
||||
status_code=400,
|
||||
detail={"errors": errors},
|
||||
)
|
||||
|
||||
strategy_router.update_config(new_config)
|
||||
mode_router.update_config(new_config)
|
||||
|
||||
logger.info(
|
||||
f"[AC-AISVC-RES-02, AC-AISVC-RES-15] Strategy switched: "
|
||||
f"strategy={new_config.strategy.value}, mode={new_config.rag_runtime_mode.value}"
|
||||
)
|
||||
|
||||
return RoutingConfigResponse(
|
||||
enabled=new_config.enabled,
|
||||
strategy=new_config.strategy,
|
||||
grayscale_percentage=new_config.grayscale_percentage,
|
||||
grayscale_allowlist=new_config.grayscale_allowlist,
|
||||
rag_runtime_mode=new_config.rag_runtime_mode,
|
||||
react_trigger_confidence_threshold=new_config.react_trigger_confidence_threshold,
|
||||
react_trigger_complexity_score=new_config.react_trigger_complexity_score,
|
||||
react_max_steps=new_config.react_max_steps,
|
||||
direct_fallback_on_low_confidence=new_config.direct_fallback_on_low_confidence,
|
||||
direct_fallback_confidence_threshold=new_config.direct_fallback_confidence_threshold,
|
||||
performance_budget_ms=new_config.performance_budget_ms,
|
||||
performance_degradation_threshold=new_config.performance_degradation_threshold,
|
||||
)
|
||||
|
||||
|
||||
@router.post(
|
||||
"/validate",
|
||||
operation_id="validateRetrievalStrategy",
|
||||
summary="Validate retrieval strategy configuration",
|
||||
description="[AC-AISVC-RES-06] Validate strategy configuration for completeness and consistency.",
|
||||
response_model=ValidationResponse,
|
||||
)
|
||||
async def validate_strategy(
|
||||
request: RoutingConfigRequest,
|
||||
) -> ValidationResponse:
|
||||
"""
|
||||
[AC-AISVC-RES-06] Validate strategy configuration.
|
||||
|
||||
Checks:
|
||||
- Parameter value ranges
|
||||
- Configuration consistency
|
||||
- Performance budget constraints
|
||||
"""
|
||||
warnings = []
|
||||
|
||||
config = RoutingConfig(
|
||||
enabled=request.enabled if request.enabled is not None else True,
|
||||
strategy=request.strategy if request.strategy is not None else StrategyType.DEFAULT,
|
||||
grayscale_percentage=request.grayscale_percentage if request.grayscale_percentage is not None else 0.0,
|
||||
grayscale_allowlist=request.grayscale_allowlist or [],
|
||||
rag_runtime_mode=request.rag_runtime_mode if request.rag_runtime_mode is not None else RagRuntimeMode.AUTO,
|
||||
react_trigger_confidence_threshold=request.react_trigger_confidence_threshold if request.react_trigger_confidence_threshold is not None else 0.6,
|
||||
react_trigger_complexity_score=request.react_trigger_complexity_score if request.react_trigger_complexity_score is not None else 0.5,
|
||||
react_max_steps=request.react_max_steps if request.react_max_steps is not None else 5,
|
||||
direct_fallback_on_low_confidence=request.direct_fallback_on_low_confidence if request.direct_fallback_on_low_confidence is not None else True,
|
||||
direct_fallback_confidence_threshold=request.direct_fallback_confidence_threshold if request.direct_fallback_confidence_threshold is not None else 0.4,
|
||||
performance_budget_ms=request.performance_budget_ms if request.performance_budget_ms is not None else 5000,
|
||||
performance_degradation_threshold=request.performance_degradation_threshold if request.performance_degradation_threshold is not None else 0.2,
|
||||
)
|
||||
|
||||
is_valid, errors = config.validate()
|
||||
|
||||
if config.grayscale_percentage > 0.5:
|
||||
warnings.append("grayscale_percentage > 50% may affect production stability")
|
||||
|
||||
if config.react_max_steps > 7:
|
||||
warnings.append("react_max_steps > 7 may cause high latency")
|
||||
|
||||
if config.direct_fallback_confidence_threshold > config.react_trigger_confidence_threshold:
|
||||
warnings.append(
|
||||
"direct_fallback_confidence_threshold > react_trigger_confidence_threshold "
|
||||
"may cause frequent fallbacks"
|
||||
)
|
||||
|
||||
if config.performance_budget_ms < 3000:
|
||||
warnings.append("performance_budget_ms < 3000ms may be too aggressive for complex queries")
|
||||
|
||||
logger.info(
|
||||
f"[AC-AISVC-RES-06] Strategy validation: is_valid={is_valid}, "
|
||||
f"errors={len(errors)}, warnings={len(warnings)}"
|
||||
)
|
||||
|
||||
return ValidationResponse(
|
||||
is_valid=is_valid,
|
||||
errors=errors,
|
||||
warnings=warnings,
|
||||
)
|
||||
|
||||
|
||||
@router.post(
|
||||
"/rollback",
|
||||
operation_id="rollbackRetrievalStrategy",
|
||||
summary="Rollback to default retrieval strategy",
|
||||
description="[AC-AISVC-RES-07] Force rollback to default strategy with audit logging.",
|
||||
response_model=RollbackResponse,
|
||||
)
|
||||
async def rollback_strategy(
|
||||
request: RollbackRequest,
|
||||
session: AsyncSession = Depends(get_session),
|
||||
) -> RollbackResponse:
|
||||
"""
|
||||
[AC-AISVC-RES-07] Rollback to default strategy.
|
||||
|
||||
Records the rollback event for audit and monitoring.
|
||||
"""
|
||||
strategy_router = get_strategy_router()
|
||||
mode_router = get_mode_router()
|
||||
|
||||
previous_strategy = strategy_router.current_strategy
|
||||
|
||||
strategy_router.rollback(
|
||||
reason=request.reason,
|
||||
tenant_id=request.tenant_id,
|
||||
)
|
||||
|
||||
new_config = RoutingConfig(
|
||||
strategy=StrategyType.DEFAULT,
|
||||
rag_runtime_mode=RagRuntimeMode.AUTO,
|
||||
)
|
||||
mode_router.update_config(new_config)
|
||||
|
||||
rollback_records = strategy_router.get_rollback_records(limit=5)
|
||||
|
||||
logger.info(
|
||||
f"[AC-AISVC-RES-07] Strategy rollback: "
|
||||
f"from={previous_strategy.value}, to=DEFAULT, reason={request.reason}"
|
||||
)
|
||||
|
||||
return RollbackResponse(
|
||||
success=True,
|
||||
previous_strategy=previous_strategy,
|
||||
current_strategy=StrategyType.DEFAULT,
|
||||
reason=request.reason,
|
||||
rollback_records=[
|
||||
{
|
||||
"timestamp": r.timestamp,
|
||||
"from_strategy": r.from_strategy.value,
|
||||
"to_strategy": r.to_strategy.value,
|
||||
"reason": r.reason,
|
||||
"tenant_id": r.tenant_id,
|
||||
}
|
||||
for r in rollback_records
|
||||
],
|
||||
)
|
||||
|
|
@ -29,6 +29,7 @@ from app.api.admin import (
|
|||
monitoring_router,
|
||||
prompt_templates_router,
|
||||
rag_router,
|
||||
retrieval_strategy_router,
|
||||
scene_slot_bundle_router,
|
||||
script_flows_router,
|
||||
sessions_router,
|
||||
|
|
@ -199,6 +200,7 @@ app.include_router(metadata_schema_router)
|
|||
app.include_router(monitoring_router)
|
||||
app.include_router(prompt_templates_router)
|
||||
app.include_router(rag_router)
|
||||
app.include_router(retrieval_strategy_router)
|
||||
app.include_router(scene_slot_bundle_router)
|
||||
app.include_router(script_flows_router)
|
||||
app.include_router(sessions_router)
|
||||
|
|
|
|||
|
|
@ -0,0 +1,198 @@
|
|||
"""
|
||||
Retrieval Strategy Schemas for AI Service.
|
||||
[AC-AISVC-RES-01~15] Request/Response models aligned with OpenAPI contract.
|
||||
"""
|
||||
|
||||
from enum import Enum
|
||||
from typing import Any
|
||||
|
||||
from pydantic import BaseModel, Field, model_validator
|
||||
|
||||
|
||||
class StrategyType(str, Enum):
|
||||
DEFAULT = "default"
|
||||
ENHANCED = "enhanced"
|
||||
|
||||
|
||||
class ReactMode(str, Enum):
|
||||
REACT = "react"
|
||||
NON_REACT = "non_react"
|
||||
|
||||
|
||||
class RolloutMode(str, Enum):
|
||||
OFF = "off"
|
||||
PERCENTAGE = "percentage"
|
||||
ALLOWLIST = "allowlist"
|
||||
|
||||
|
||||
class ValidationCheckType(str, Enum):
|
||||
METADATA_CONSISTENCY = "metadata_consistency"
|
||||
EMBEDDING_PREFIX = "embedding_prefix"
|
||||
RRF_CONFIG = "rrf_config"
|
||||
PERFORMANCE_BUDGET = "performance_budget"
|
||||
|
||||
|
||||
class RolloutConfig(BaseModel):
|
||||
"""
|
||||
[AC-AISVC-RES-03] Grayscale rollout configuration.
|
||||
"""
|
||||
|
||||
mode: RolloutMode = Field(..., description="Rollout mode: off, percentage, or allowlist")
|
||||
percentage: float | None = Field(
|
||||
default=None,
|
||||
ge=0,
|
||||
le=100,
|
||||
description="Percentage of traffic for grayscale (0-100)",
|
||||
)
|
||||
allowlist: list[str] | None = Field(
|
||||
default=None,
|
||||
description="List of tenant IDs in allowlist",
|
||||
)
|
||||
|
||||
@model_validator(mode="after")
|
||||
def validate_rollout_config(self) -> "RolloutConfig":
|
||||
if self.mode == RolloutMode.PERCENTAGE and self.percentage is None:
|
||||
raise ValueError("percentage is required when mode is 'percentage'")
|
||||
if self.mode == RolloutMode.ALLOWLIST and (
|
||||
self.allowlist is None or len(self.allowlist) == 0
|
||||
):
|
||||
raise ValueError("allowlist is required when mode is 'allowlist'")
|
||||
return self
|
||||
|
||||
|
||||
class RetrievalStrategyStatus(BaseModel):
|
||||
"""
|
||||
[AC-AISVC-RES-01] Current retrieval strategy status.
|
||||
"""
|
||||
|
||||
active_strategy: StrategyType = Field(
|
||||
...,
|
||||
alias="active_strategy",
|
||||
description="Current active strategy: default or enhanced",
|
||||
)
|
||||
react_mode: ReactMode = Field(
|
||||
...,
|
||||
alias="react_mode",
|
||||
description="ReAct mode: react or non_react",
|
||||
)
|
||||
rollout: RolloutConfig = Field(..., description="Grayscale rollout configuration")
|
||||
|
||||
model_config = {"populate_by_name": True}
|
||||
|
||||
|
||||
class RetrievalStrategySwitchRequest(BaseModel):
|
||||
"""
|
||||
[AC-AISVC-RES-02, AC-AISVC-RES-03, AC-AISVC-RES-05] Request to switch retrieval strategy.
|
||||
"""
|
||||
|
||||
target_strategy: StrategyType = Field(
|
||||
...,
|
||||
alias="target_strategy",
|
||||
description="Target strategy to switch to",
|
||||
)
|
||||
react_mode: ReactMode | None = Field(
|
||||
default=None,
|
||||
alias="react_mode",
|
||||
description="ReAct mode to use",
|
||||
)
|
||||
rollout: RolloutConfig | None = Field(
|
||||
default=None,
|
||||
description="Grayscale rollout configuration",
|
||||
)
|
||||
reason: str | None = Field(
|
||||
default=None,
|
||||
description="Reason for strategy switch",
|
||||
)
|
||||
|
||||
model_config = {"populate_by_name": True}
|
||||
|
||||
|
||||
class RetrievalStrategySwitchResponse(BaseModel):
|
||||
"""
|
||||
[AC-AISVC-RES-02] Response after strategy switch.
|
||||
"""
|
||||
|
||||
previous: RetrievalStrategyStatus = Field(..., description="Previous strategy status")
|
||||
current: RetrievalStrategyStatus = Field(..., description="Current strategy status")
|
||||
|
||||
|
||||
class RetrievalStrategyValidationRequest(BaseModel):
|
||||
"""
|
||||
[AC-AISVC-RES-04, AC-AISVC-RES-06, AC-AISVC-RES-08] Request to validate strategy.
|
||||
"""
|
||||
|
||||
strategy: StrategyType = Field(..., description="Strategy to validate")
|
||||
react_mode: ReactMode | None = Field(
|
||||
default=None,
|
||||
description="ReAct mode to validate",
|
||||
)
|
||||
checks: list[ValidationCheckType] | None = Field(
|
||||
default=None,
|
||||
description="List of checks to perform",
|
||||
)
|
||||
|
||||
model_config = {"populate_by_name": True}
|
||||
|
||||
|
||||
class ValidationResult(BaseModel):
|
||||
"""
|
||||
[AC-AISVC-RES-06] Single validation check result.
|
||||
"""
|
||||
|
||||
check: str = Field(..., description="Check name")
|
||||
passed: bool = Field(..., description="Whether the check passed")
|
||||
message: str | None = Field(default=None, description="Additional message")
|
||||
|
||||
|
||||
class RetrievalStrategyValidationResponse(BaseModel):
|
||||
"""
|
||||
[AC-AISVC-RES-06] Validation response with all check results.
|
||||
"""
|
||||
|
||||
passed: bool = Field(..., description="Whether all checks passed")
|
||||
results: list[ValidationResult] = Field(..., description="Individual check results")
|
||||
|
||||
|
||||
class RetrievalStrategyRollbackResponse(BaseModel):
|
||||
"""
|
||||
[AC-AISVC-RES-07] Response after strategy rollback.
|
||||
"""
|
||||
|
||||
current: RetrievalStrategyStatus = Field(..., description="Current strategy status before rollback")
|
||||
rollback_to: RetrievalStrategyStatus = Field(..., description="Strategy status after rollback")
|
||||
|
||||
|
||||
class StrategyAuditLog(BaseModel):
|
||||
"""
|
||||
[AC-AISVC-RES-07] Audit log entry for strategy operations.
|
||||
"""
|
||||
|
||||
timestamp: str = Field(..., description="ISO timestamp of the operation")
|
||||
operation: str = Field(..., description="Operation type: switch, rollback, validate")
|
||||
previous_strategy: str | None = Field(default=None, description="Previous strategy")
|
||||
new_strategy: str | None = Field(default=None, description="New strategy")
|
||||
previous_react_mode: str | None = Field(default=None, description="Previous react mode")
|
||||
new_react_mode: str | None = Field(default=None, description="New react mode")
|
||||
reason: str | None = Field(default=None, description="Reason for the operation")
|
||||
operator: str | None = Field(default=None, description="Operator who performed the operation")
|
||||
tenant_id: str | None = Field(default=None, description="Tenant ID if applicable")
|
||||
metadata: dict[str, Any] | None = Field(default=None, description="Additional metadata")
|
||||
|
||||
|
||||
class StrategyMetrics(BaseModel):
|
||||
"""
|
||||
[AC-AISVC-RES-03, AC-AISVC-RES-08] Metrics for strategy operations.
|
||||
"""
|
||||
|
||||
strategy: StrategyType = Field(..., description="Current strategy")
|
||||
react_mode: ReactMode = Field(..., description="Current react mode")
|
||||
total_requests: int = Field(default=0, description="Total requests count")
|
||||
successful_requests: int = Field(default=0, description="Successful requests count")
|
||||
failed_requests: int = Field(default=0, description="Failed requests count")
|
||||
avg_latency_ms: float = Field(default=0.0, description="Average latency in ms")
|
||||
p99_latency_ms: float = Field(default=0.0, description="P99 latency in ms")
|
||||
direct_route_count: int = Field(default=0, description="Direct route count")
|
||||
react_route_count: int = Field(default=0, description="React route count")
|
||||
auto_route_count: int = Field(default=0, description="Auto route count")
|
||||
fallback_count: int = Field(default=0, description="Fallback to default count")
|
||||
last_updated: str | None = Field(default=None, description="Last update timestamp")
|
||||
Loading…
Reference in New Issue