forked from phoenix-oss/llama-stack-mirror
5547ef953c
# What does this PR do? An API spec must talk about Error handling. This was a pretty glaring omission so far. This PR begins to address it by adding a set of standard error responses we can attach to all our API calls. At a future point, we can add specific error types where necessary (although we should not hurry to do that; it is best done very late.) ## Test Plan Checked that Stainless SDK generation succeeds.
77 lines
3 KiB
Python
77 lines
3 KiB
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the terms described in the LICENSE file in
|
|
# the root directory of this source tree.
|
|
|
|
import dataclasses
|
|
from dataclasses import dataclass
|
|
from http import HTTPStatus
|
|
from typing import Callable, ClassVar, Dict, List, Optional, Tuple, Union
|
|
|
|
from .specification import (
|
|
Info,
|
|
SecurityScheme,
|
|
SecuritySchemeAPI,
|
|
SecuritySchemeHTTP,
|
|
SecuritySchemeOpenIDConnect,
|
|
Server,
|
|
)
|
|
|
|
HTTPStatusCode = Union[HTTPStatus, int, str]
|
|
|
|
|
|
@dataclass
|
|
class Options:
|
|
"""
|
|
:param server: Base URL for the API endpoint.
|
|
:param info: Meta-information for the endpoint specification.
|
|
:param version: OpenAPI specification version as a tuple of major, minor, revision.
|
|
:param default_security_scheme: Security scheme to apply to endpoints, unless overridden on a per-endpoint basis.
|
|
:param extra_types: Extra types in addition to those found in operation signatures. Use a dictionary to group related types.
|
|
:param use_examples: Whether to emit examples for operations.
|
|
:param success_responses: Associates operation response types with HTTP status codes.
|
|
:param error_responses: Associates error response types with HTTP status codes.
|
|
:param error_wrapper: True if errors are encapsulated in an error object wrapper.
|
|
:param property_description_fun: Custom transformation function to apply to class property documentation strings.
|
|
:param captions: User-defined captions for sections such as "Operations" or "Types", and (if applicable) groups of extra types.
|
|
:param include_standard_error_responses: Whether to include standard error responses (400, 429, 500, 503) in all operations.
|
|
"""
|
|
|
|
server: Server
|
|
info: Info
|
|
version: Tuple[int, int, int] = (3, 1, 0)
|
|
default_security_scheme: Optional[SecurityScheme] = None
|
|
extra_types: Union[List[type], Dict[str, List[type]], None] = None
|
|
use_examples: bool = True
|
|
success_responses: Dict[type, HTTPStatusCode] = dataclasses.field(
|
|
default_factory=dict
|
|
)
|
|
error_responses: Dict[type, HTTPStatusCode] = dataclasses.field(
|
|
default_factory=dict
|
|
)
|
|
error_wrapper: bool = False
|
|
property_description_fun: Optional[Callable[[type, str, str], str]] = None
|
|
captions: Optional[Dict[str, str]] = None
|
|
include_standard_error_responses: bool = True
|
|
|
|
default_captions: ClassVar[Dict[str, str]] = {
|
|
"Operations": "Operations",
|
|
"Types": "Types",
|
|
"Events": "Events",
|
|
"AdditionalTypes": "Additional types",
|
|
}
|
|
|
|
def map(self, id: str) -> str:
|
|
"Maps a language-neutral placeholder string to language-dependent text."
|
|
|
|
if self.captions is not None:
|
|
caption = self.captions.get(id)
|
|
if caption is not None:
|
|
return caption
|
|
|
|
caption = self.__class__.default_captions.get(id)
|
|
if caption is not None:
|
|
return caption
|
|
|
|
raise KeyError(f"no caption found for ID: {id}")
|