feat: compact durable logs and refine framework routing

This commit is contained in:
CoolBoy
2026-09-04 10:30:21 +08:00
parent 5b1ec4d3eb
commit ff73768537
13 changed files with 621 additions and 44 deletions

View File

@@ -370,6 +370,12 @@ MODEL_UNIQUENESS_ERROR_MARKERS = (
"model already exists",
)
FRAMEWORK_PREREQUISITE_ERROR_MARKERS = (
"必须在非transformers框架验证失败后",
"must fail on a non-transformers framework",
"non-transformers framework verification failed",
)
DEFAULT_CAPACITY_STATE_PATH = Path(".modelhub_state/account_capacity.json")
@@ -396,6 +402,13 @@ def is_model_uniqueness_error(error: ModelHubAPIError) -> bool:
return any(marker in message for marker in MODEL_UNIQUENESS_ERROR_MARKERS)
def is_framework_prerequisite_error(error: ModelHubAPIError) -> bool:
message = str(error).strip().lower()
if isinstance(error.payload, dict):
message = f"{message} {error.payload.get('message') or ''}".lower()
return any(marker in message for marker in FRAMEWORK_PREREQUISITE_ERROR_MARKERS)
class ModelHubClientPool:
def __init__(
self,