""" Options:
Date: 2025-01-08 03:22:59
Version: 8.53
Tip: To override a DTO option, remove "#" prefix before updating
BaseUrl: https://blazordiffusion.com

#GlobalNamespace: 
#AddServiceStackTypes: True
#AddResponseStatus: False
#AddImplicitVersion: 
#AddDescriptionAsComments: True
IncludeTypes: QueryGenerationApiProviders.*
#ExcludeTypes: 
#DefaultImports: datetime,decimal,marshmallow.fields:*,servicestack:*,typing:*,dataclasses:dataclass/field,dataclasses_json:dataclass_json/LetterCase/Undefined/config,enum:Enum/IntEnum
#DataClass: 
#DataClassJson: 
"""

import datetime
import decimal
from marshmallow.fields import *
from servicestack import *
from typing import *
from dataclasses import dataclass, field
from dataclasses_json import dataclass_json, LetterCase, Undefined, config
from enum import Enum, IntEnum


@dataclass_json(letter_case=LetterCase.CAMEL, undefined=Undefined.EXCLUDE)
@dataclass
class ApiProviderBase:
    id: int = 0
    name: Optional[str] = None
    api_key_var: Optional[str] = None
    api_key: Optional[str] = None
    api_key_header: Optional[str] = None
    api_base_url: Optional[str] = None
    heartbeat_url: Optional[str] = None
    concurrency: int = 0
    priority: int = 0
    enabled: bool = False
    offline_date: Optional[datetime.datetime] = None
    created_date: datetime.datetime = datetime.datetime(1, 1, 1)


@dataclass_json(letter_case=LetterCase.CAMEL, undefined=Undefined.EXCLUDE)
@dataclass
class AiProviderBase(ApiProviderBase):
    models: List[str] = field(default_factory=list)


class AiServiceProvider(str, Enum):
    REPLICATE = 'Replicate'
    COMFY = 'Comfy'
    OPEN_AI = 'OpenAi'


@dataclass_json(letter_case=LetterCase.CAMEL, undefined=Undefined.EXCLUDE)
@dataclass
class ApiTypeBase:
    id: int = 0
    api_base_url: Optional[str] = None
    api_key_header: Optional[str] = None
    name: Optional[str] = None
    website: Optional[str] = None
    icon: Optional[str] = None
    api_models: Dict[str, str] = field(default_factory=dict)
    provider: Optional[AiServiceProvider] = None


@dataclass_json(letter_case=LetterCase.CAMEL, undefined=Undefined.EXCLUDE)
@dataclass
class GenerationApiType(ApiTypeBase):
    pass


@dataclass_json(letter_case=LetterCase.CAMEL, undefined=Undefined.EXCLUDE)
@dataclass
class GenerationApiProvider(AiProviderBase):
    # @References(typeof(GenerationApiType))
    generation_api_type_id: int = 0

    type: Optional[GenerationApiType] = None


@dataclass_json(letter_case=LetterCase.CAMEL, undefined=Undefined.EXCLUDE)
@dataclass
class QueryGenerationApiProviders(QueryDb[GenerationApiProvider], IReturn[QueryResponse[GenerationApiProvider]]):
    id: Optional[int] = None
    name: Optional[str] = None