Fix for broken OpenAPI Schema (#7558)

This commit is contained in:
tofarr 2025-03-27 13:28:53 -06:00 committed by GitHub
parent 7aa7eb2399
commit a44cdae36e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -2,13 +2,14 @@ from __future__ import annotations
from enum import Enum
from types import MappingProxyType
from typing import Any, Coroutine, Literal, overload
from typing import Annotated, Any, Coroutine, Literal, overload
from pydantic import (
BaseModel,
Field,
SecretStr,
SerializationInfo,
WithJsonSchema,
field_serializer,
model_validator,
)
@ -58,10 +59,14 @@ class ProviderToken(BaseModel):
PROVIDER_TOKEN_TYPE = MappingProxyType[ProviderType, ProviderToken]
CUSTOM_SECRETS_TYPE = MappingProxyType[str, SecretStr]
PROVIDER_TOKEN_TYPE_WITH_JSON_SCHEMA = Annotated[
PROVIDER_TOKEN_TYPE,
WithJsonSchema({'type': 'object', 'additionalProperties': {'type': 'string'}}),
]
class SecretStore(BaseModel):
provider_tokens: PROVIDER_TOKEN_TYPE = Field(
provider_tokens: PROVIDER_TOKEN_TYPE_WITH_JSON_SCHEMA = Field(
default_factory=lambda: MappingProxyType({})
)