Spaces:
Sleeping
Sleeping
Update app/services/routing.py
Browse files- app/services/routing.py +9 -78
app/services/routing.py
CHANGED
|
@@ -1,86 +1,17 @@
|
|
| 1 |
# app/services/routing.py
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
from app.configs import CLASSIFIER_URI, SEGMENTER_URI
|
| 6 |
|
|
|
|
| 7 |
|
| 8 |
router = APIRouter()
|
| 9 |
|
| 10 |
-
# =========================
|
| 11 |
-
# Model Registry (for A/B testing)
|
| 12 |
-
# =========================
|
| 13 |
-
# Maps version names to actual model URIs
|
| 14 |
-
MODEL_REGISTRY = {
|
| 15 |
-
"default": CLASSIFIER_URI,
|
| 16 |
-
"v1": CLASSIFIER_URI,
|
| 17 |
-
"hf_savedmodel": "hf_savedmodel",
|
| 18 |
-
# يمكنك إضافة إصدارات أخرى هنا
|
| 19 |
-
# "v2": "hf_savedmodel_v2",
|
| 20 |
-
}
|
| 21 |
-
|
| 22 |
-
# A/B Testing configuration
|
| 23 |
-
AB_TESTING_ENABLED = False
|
| 24 |
-
AB_TESTING_RATIO = 0.0 # 0.0 = no traffic to new model
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
def get_ab_status() -> Dict[str, Any]:
|
| 28 |
-
"""Get current A/B testing status"""
|
| 29 |
-
return {
|
| 30 |
-
"enabled": AB_TESTING_ENABLED,
|
| 31 |
-
"ratio": AB_TESTING_RATIO,
|
| 32 |
-
"default_version": CLASSIFIER_URI,
|
| 33 |
-
"available_versions": list(MODEL_REGISTRY.keys()),
|
| 34 |
-
}
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
def resolve_classifier_version(version: Optional[str]) -> str:
|
| 38 |
-
"""
|
| 39 |
-
Resolve a version name to actual model URI.
|
| 40 |
-
|
| 41 |
-
Args:
|
| 42 |
-
version: Version name (e.g., "v1", "default") or None
|
| 43 |
-
|
| 44 |
-
Returns:
|
| 45 |
-
Actual model URI
|
| 46 |
-
"""
|
| 47 |
-
if version is None or version == "":
|
| 48 |
-
return CLASSIFIER_URI
|
| 49 |
-
|
| 50 |
-
resolved = MODEL_REGISTRY.get(version)
|
| 51 |
-
if resolved is None:
|
| 52 |
-
# Fallback to default if version not found
|
| 53 |
-
return CLASSIFIER_URI
|
| 54 |
-
|
| 55 |
-
return resolved
|
| 56 |
-
|
| 57 |
-
|
| 58 |
-
# =========================
|
| 59 |
-
# Optional: A/B Testing Router Endpoints
|
| 60 |
-
# =========================
|
| 61 |
-
@router.get("/api/models/versions")
|
| 62 |
-
async def list_model_versions():
|
| 63 |
-
"""List all available model versions"""
|
| 64 |
-
return {
|
| 65 |
-
"versions": list(MODEL_REGISTRY.keys()),
|
| 66 |
-
"default": CLASSIFIER_URI,
|
| 67 |
-
"registry": MODEL_REGISTRY,
|
| 68 |
-
}
|
| 69 |
-
|
| 70 |
|
| 71 |
-
@router.
|
| 72 |
-
async def
|
| 73 |
-
"""
|
| 74 |
-
global AB_TESTING_ENABLED, AB_TESTING_RATIO
|
| 75 |
-
|
| 76 |
-
if not 0.0 <= ratio <= 1.0:
|
| 77 |
-
return {"error": "Ratio must be between 0.0 and 1.0"}
|
| 78 |
-
|
| 79 |
-
AB_TESTING_ENABLED = enabled
|
| 80 |
-
AB_TESTING_RATIO = ratio
|
| 81 |
-
|
| 82 |
return {
|
| 83 |
-
"
|
| 84 |
-
"
|
| 85 |
-
"
|
| 86 |
}
|
|
|
|
| 1 |
# app/services/routing.py
|
| 2 |
+
# ✅ This file is now simplified - no A/B testing or model versioning needed
|
| 3 |
+
# Kept for future extensibility
|
|
|
|
|
|
|
| 4 |
|
| 5 |
+
from fastapi import APIRouter
|
| 6 |
|
| 7 |
router = APIRouter()
|
| 8 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
|
| 10 |
+
@router.get("/api/models/info")
|
| 11 |
+
async def models_info():
|
| 12 |
+
"""Get basic model information"""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
return {
|
| 14 |
+
"classifier": "hf_savedmodel",
|
| 15 |
+
"segmenter": "hf_savedmodel",
|
| 16 |
+
"status": "production",
|
| 17 |
}
|