PMFlow/backend/app/api/model_capability_api/shared.py

30 lines
1.1 KiB
Python

from __future__ import annotations
from typing import Optional
from fastapi import APIRouter, Depends, Query
from fastapi.responses import JSONResponse
from sqlalchemy import and_, select
from sqlalchemy.ext.asyncio import AsyncSession
from ...core.response import ErrorCode, error_response, success_response
from ...database import get_db
from ...dependencies import get_current_user, get_tenant_id
from ...models import Experiment, TruthNode, User
from ...schemas.model_capability import (
CostOptimizationAnalyzeRequest,
ExperimentSyncRequest,
ExperimentSyncReviewRequest,
ExperimentSyncRollbackRequest,
GenerateMigrationTasksRequest,
LifecycleImpactRequest,
ValidateMatchRequest,
)
from ...services.ai_service import get_ai_service
from ...services.cost_optimization_service import CostOptimizationService
from ...services.experiment_sync_service import ExperimentSyncService
from ...services.model_capability_service import ModelCapabilityService
from ...services.model_lifecycle_service import ModelLifecycleService
router = APIRouter(prefix="/model-capability", tags=["model-capability"])