""" Options:
Date: 2025-01-08 03:19:15
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: QueryGenerationApiTypes.*
#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


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 QueryGenerationApiTypes(QueryDb[GenerationApiType], IReturn[QueryResponse[GenerationApiType]]):
    pass