This commit is contained in:
hjlarry 2025-09-12 10:43:15 +08:00
parent 02852ee543
commit 9d75d3d04c
3 changed files with 13 additions and 8 deletions

View File

@ -189,8 +189,8 @@ class ModelProviderCredentialCancelApi(Resource):
provider=provider,
)
return {"result": "success"}
class ModelProviderValidateApi(Resource):
@setup_required
@login_required

View File

@ -190,7 +190,7 @@ class ProviderConfiguration(BaseModel):
if current_quota_configuration.is_valid
else SystemConfigurationStatus.QUOTA_EXCEEDED
)
def get_custom_configuration_status(self) -> Optional[CustomConfigurationStatus]:
"""
Get custom configuration status.
@ -198,8 +198,12 @@ class ProviderConfiguration(BaseModel):
"""
if not self.is_custom_configuration_available():
return CustomConfigurationStatus.NO_CONFIGURE
elif self.custom_configuration.provider.current_credential_status:
return self.custom_configuration.provider.current_credential_status
provider = self.custom_configuration.provider
if provider and hasattr(provider, "current_credential_status"):
status = provider.current_credential_status
if status:
return status
return CustomConfigurationStatus.ACTIVE

View File

@ -40,7 +40,8 @@ class ProviderQuotaType(Enum):
if member.value == value:
return member
raise ValueError(f"No matching enum found for value '{value}'")
class CredentialStatus(StrEnum):
ACTIVE = "active"
CANCELED = "canceled"
@ -71,8 +72,8 @@ class Provider(Base):
last_used: Mapped[Optional[datetime]] = mapped_column(DateTime, nullable=True)
credential_id: Mapped[Optional[str]] = mapped_column(StringUUID, nullable=True)
credential_status: Mapped[Optional[str]] = mapped_column(
String(20), nullable=True, server_default=text("'active'::character varying")
)
String(20), nullable=True, server_default=text("'active'::character varying")
)
quota_type: Mapped[Optional[str]] = mapped_column(
String(40), nullable=True, server_default=text("''::character varying")