unitysvc-services 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

unitysvc_services/models/base.py

@@ -206,16 +206,38 @@ class UpstreamStatusEnum(StrEnum):
     deprecated = "deprecated"
 
 
+class ProviderStatusEnum(StrEnum):
+    """Provider status enum."""
+
+    active = "active"
+    disabled = "disabled"
+    incomplete = "incomplete"  # Provider information is incomplete
+
+
+class SellerStatusEnum(StrEnum):
+    """Seller status enum."""
+
+    active = "active"
+    disabled = "disabled"
+    incomplete = "incomplete"  # Seller information is incomplete
+
+
 class Document(BaseModel):
     model_config = ConfigDict(extra="forbid")
 
     # fields that will be stored in backend database
     #
     title: str = Field(min_length=5, max_length=255, description="Document title")
-    description: str | None = Field(default=None, max_length=500, description="Document description")
+    description: str | None = Field(
+        default=None, max_length=500, description="Document description"
+    )
     mime_type: MimeTypeEnum = Field(description="Document MIME type")
-    version: str | None = Field(default=None, max_length=50, description="Document version")
-    category: DocumentCategoryEnum = Field(description="Document category for organization and filtering")
+    version: str | None = Field(
+        default=None, max_length=50, description="Document version"
+    )
+    category: DocumentCategoryEnum = Field(
+        description="Document category for organization and filtering"
+    )
     meta: dict[str, Any] | None = Field(
         default=None,
         description="JSON containing operation stats",
@@ -249,8 +271,12 @@ class RateLimit(BaseModel):
     window: TimeWindowEnum = Field(description="Time window for the limit")
 
     # Optional additional info
-    description: str | None = Field(default=None, max_length=255, description="Human-readable description")
-    burst_limit: int | None = Field(default=None, description="Short-term burst allowance")
+    description: str | None = Field(
+        default=None, max_length=255, description="Human-readable description"
+    )
+    burst_limit: int | None = Field(
+        default=None, description="Short-term burst allowance"
+    )
 
     # Status
     is_active: bool = Field(default=True, description="Whether rate limit is active")
@@ -260,57 +286,105 @@ class ServiceConstraints(BaseModel):
     model_config = ConfigDict(extra="forbid")
 
     # Usage Quotas & Billing
-    monthly_quota: int | None = Field(default=None, description="Monthly usage quota (requests, tokens, etc.)")
-    daily_quota: int | None = Field(default=None, description="Daily usage quota (requests, tokens, etc.)")
-    quota_unit: RateLimitUnitEnum | None = Field(default=None, description="Unit for quota limits")
-    quota_reset_cycle: QuotaResetCycleEnum | None = Field(default=None, description="How often quotas reset")
-    overage_policy: OveragePolicyEnum | None = Field(default=None, description="What happens when quota is exceeded")
+    monthly_quota: int | None = Field(
+        default=None, description="Monthly usage quota (requests, tokens, etc.)"
+    )
+    daily_quota: int | None = Field(
+        default=None, description="Daily usage quota (requests, tokens, etc.)"
+    )
+    quota_unit: RateLimitUnitEnum | None = Field(
+        default=None, description="Unit for quota limits"
+    )
+    quota_reset_cycle: QuotaResetCycleEnum | None = Field(
+        default=None, description="How often quotas reset"
+    )
+    overage_policy: OveragePolicyEnum | None = Field(
+        default=None, description="What happens when quota is exceeded"
+    )
 
     # Authentication & Security
-    auth_methods: list[AuthMethodEnum] | None = Field(default=None, description="Supported authentication methods")
-    ip_whitelist_required: bool | None = Field(default=None, description="Whether IP whitelisting is required")
-    tls_version_min: str | None = Field(default=None, description="Minimum TLS version required")
+    auth_methods: list[AuthMethodEnum] | None = Field(
+        default=None, description="Supported authentication methods"
+    )
+    ip_whitelist_required: bool | None = Field(
+        default=None, description="Whether IP whitelisting is required"
+    )
+    tls_version_min: str | None = Field(
+        default=None, description="Minimum TLS version required"
+    )
 
     # Request/Response Constraints
-    max_request_size_bytes: int | None = Field(default=None, description="Maximum request payload size in bytes")
-    max_response_size_bytes: int | None = Field(default=None, description="Maximum response payload size in bytes")
-    timeout_seconds: int | None = Field(default=None, description="Request timeout in seconds")
-    max_batch_size: int | None = Field(default=None, description="Maximum number of items in batch requests")
+    max_request_size_bytes: int | None = Field(
+        default=None, description="Maximum request payload size in bytes"
+    )
+    max_response_size_bytes: int | None = Field(
+        default=None, description="Maximum response payload size in bytes"
+    )
+    timeout_seconds: int | None = Field(
+        default=None, description="Request timeout in seconds"
+    )
+    max_batch_size: int | None = Field(
+        default=None, description="Maximum number of items in batch requests"
+    )
 
     # Content & Model Restrictions
     content_filters: list[ContentFilterEnum] | None = Field(
         default=None, description="Active content filtering policies"
     )
-    input_languages: list[str] | None = Field(default=None, description="Supported input languages (ISO 639-1 codes)")
-    output_languages: list[str] | None = Field(default=None, description="Supported output languages (ISO 639-1 codes)")
-    max_context_length: int | None = Field(default=None, description="Maximum context length in tokens")
+    input_languages: list[str] | None = Field(
+        default=None, description="Supported input languages (ISO 639-1 codes)"
+    )
+    output_languages: list[str] | None = Field(
+        default=None, description="Supported output languages (ISO 639-1 codes)"
+    )
+    max_context_length: int | None = Field(
+        default=None, description="Maximum context length in tokens"
+    )
     region_restrictions: list[str] | None = Field(
         default=None, description="Geographic restrictions (ISO country codes)"
     )
 
     # Availability & SLA
-    uptime_sla_percent: float | None = Field(default=None, description="Uptime SLA percentage (e.g., 99.9)")
-    response_time_sla_ms: int | None = Field(default=None, description="Response time SLA in milliseconds")
-    maintenance_windows: list[str] | None = Field(default=None, description="Scheduled maintenance windows")
+    uptime_sla_percent: float | None = Field(
+        default=None, description="Uptime SLA percentage (e.g., 99.9)"
+    )
+    response_time_sla_ms: int | None = Field(
+        default=None, description="Response time SLA in milliseconds"
+    )
+    maintenance_windows: list[str] | None = Field(
+        default=None, description="Scheduled maintenance windows"
+    )
 
     # Concurrency & Connection Limits
-    max_concurrent_requests: int | None = Field(default=None, description="Maximum concurrent requests allowed")
-    connection_timeout_seconds: int | None = Field(default=None, description="Connection timeout in seconds")
-    max_connections_per_ip: int | None = Field(default=None, description="Maximum connections per IP address")
+    max_concurrent_requests: int | None = Field(
+        default=None, description="Maximum concurrent requests allowed"
+    )
+    connection_timeout_seconds: int | None = Field(
+        default=None, description="Connection timeout in seconds"
+    )
+    max_connections_per_ip: int | None = Field(
+        default=None, description="Maximum connections per IP address"
+    )
 
 
 class AccessInterface(BaseModel):
     model_config = ConfigDict(extra="allow")
 
-    access_method: AccessMethodEnum = Field(default=AccessMethodEnum.http, description="Type of access method")
+    access_method: AccessMethodEnum = Field(
+        default=AccessMethodEnum.http, description="Type of access method"
+    )
 
     api_endpoint: str = Field(max_length=500, description="API endpoint URL")
 
-    api_key: str | None = Field(default=None, max_length=2000, description="API key if required")
+    api_key: str | None = Field(
+        default=None, max_length=2000, description="API key if required"
+    )
 
     name: str | None = Field(default=None, max_length=100, description="Interface name")
 
-    description: str | None = Field(default=None, max_length=500, description="Interface description")
+    description: str | None = Field(
+        default=None, max_length=500, description="Interface description"
+    )
 
     request_transformer: dict[RequestTransformEnum, dict[str, Any]] | None = Field(
         default=None, description="Request transformation configuration"
@@ -324,9 +398,13 @@ class AccessInterface(BaseModel):
         default=None,
         description="Rate limit",
     )
-    constraint: ServiceConstraints | None = Field(default=None, description="Service constraints and conditions")
+    constraint: ServiceConstraints | None = Field(
+        default=None, description="Service constraints and conditions"
+    )
     is_active: bool = Field(default=True, description="Whether interface is active")
-    is_primary: bool = Field(default=False, description="Whether this is the primary interface")
+    is_primary: bool = Field(
+        default=False, description="Whether this is the primary interface"
+    )
     sort_order: int = Field(default=0, description="Display order")
 
 
@@ -334,9 +412,13 @@ class Pricing(BaseModel):
     model_config = ConfigDict(extra="forbid")
 
     # Pricing tier name (Basic, Pro, Enterprise, etc.)
-    name: str | None = Field(default=None, description="Pricing tier name (e.g., Basic, Pro, Enterprise)")
+    name: str | None = Field(
+        default=None, description="Pricing tier name (e.g., Basic, Pro, Enterprise)"
+    )
 
-    description: str | None = Field(default=None, description="Pricing model description")
+    description: str | None = Field(
+        default=None, description="Pricing model description"
+    )
 
     # Currency and description
     currency: str | None = Field(default=None, description="Currency code (e.g., USD)")
@@ -349,4 +431,6 @@ class Pricing(BaseModel):
     )
 
     # Optional reference to upstream pricing
-    reference: str | None = Field(default=None, description="Reference URL to upstream pricing")
+    reference: str | None = Field(
+        default=None, description="Reference URL to upstream pricing"
+    )

unitysvc_services/models/listing_v1.py

@@ -17,7 +17,9 @@ class ListingV1(BaseModel):
     #
     # fields for business data collection and maintenance
     #
-    schema_version: str = Field(default="listing_v1", description="Schema identifier", alias="schema")
+    schema_version: str = Field(
+        default="listing_v1", description="Schema identifier", alias="schema"
+    )
     time_created: datetime
 
     #
@@ -30,8 +32,8 @@ class ListingV1(BaseModel):
         ),
     )
 
-    seller_name: str = Field(
-        description="Name of the seller offering this service listing"
+    seller_name: str | None = Field(
+        default=None, description="Name of the seller offering this service listing"
     )
 
     # unique name for each provider, usually following upstream naming convention
@@ -48,7 +50,9 @@ class ListingV1(BaseModel):
     #  - code_examples
     # multiple access interfaces can be provided, for example, if the service
     # is available through multiple interfaces or service groups
-    user_access_interfaces: list[AccessInterface] = Field(description="Dictionary of user access interfaces")
+    user_access_interfaces: list[AccessInterface] = Field(
+        description="Dictionary of user access interfaces"
+    )
 
     #
     # how upstream charges for their services, which can include

unitysvc_services/models/provider_v1.py

@@ -3,7 +3,7 @@ from typing import Any
 
 from pydantic import BaseModel, ConfigDict, EmailStr, Field, HttpUrl
 
-from unitysvc_services.models.base import AccessInterface, Document
+from unitysvc_services.models.base import AccessInterface, Document, ProviderStatusEnum
 
 
 class ProviderV1(BaseModel):
@@ -12,13 +12,17 @@ class ProviderV1(BaseModel):
     #
     # fields for business data collection and maintenance
     #
-    schema_version: str = Field(default="provider_v1", description="Schema identifier", alias="schema")
+    schema_version: str = Field(
+        default="provider_v1", description="Schema identifier", alias="schema"
+    )
     time_created: datetime
     # how to automatically populate service data, if available
     services_populator: dict[str, Any] | None = None
     # parameters for accessing service provider, which typically
     # include "api_endpoint" and "api_key"
-    provider_access_info: AccessInterface = Field(description="Dictionary of upstream access interface")
+    provider_access_info: AccessInterface = Field(
+        description="Dictionary of upstream access interface"
+    )
     #
     # fields that will be stored in backend database
     #
@@ -51,3 +55,9 @@ class ProviderV1(BaseModel):
     homepage: HttpUrl
     contact_email: EmailStr
     secondary_contact_email: EmailStr | None = None
+
+    # Status field to track provider state
+    status: ProviderStatusEnum = Field(
+        default=ProviderStatusEnum.active,
+        description="Provider status: active, disabled, or incomplete",
+    )

unitysvc_services/models/seller_v1.py

@@ -2,7 +2,7 @@ from datetime import datetime
 
 from pydantic import BaseModel, ConfigDict, EmailStr, Field, HttpUrl
 
-from unitysvc_services.models.base import Document, SellerTypeEnum
+from unitysvc_services.models.base import Document, SellerStatusEnum, SellerTypeEnum
 
 
 class SellerV1(BaseModel):
@@ -17,7 +17,9 @@ class SellerV1(BaseModel):
     #
     # fields for business data collection and maintenance
     #
-    schema_version: str = Field(default="seller_v1", description="Schema identifier", alias="schema")
+    schema_version: str = Field(
+        default="seller_v1", description="Schema identifier", alias="schema"
+    )
     time_created: datetime
 
     #
@@ -47,7 +49,9 @@ class SellerV1(BaseModel):
     # Contact information
     contact_email: EmailStr = Field(description="Primary contact email for the seller")
 
-    secondary_contact_email: EmailStr | None = Field(default=None, description="Secondary contact email")
+    secondary_contact_email: EmailStr | None = Field(
+        default=None, description="Secondary contact email"
+    )
 
     # Account manager
     account_manager: str | None = Field(
@@ -98,10 +102,10 @@ class SellerV1(BaseModel):
     # fields for business operation purposes
     #
 
-    # Status flags - these would typically be set by the backend
-    is_active: bool = Field(
-        default=True,
-        description="Whether the seller is active on the marketplace",
+    # Status field to track seller state
+    status: SellerStatusEnum = Field(
+        default=SellerStatusEnum.active,
+        description="Seller status: active, disabled, or incomplete",
     )
 
     is_verified: bool = Field(