mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-06-27 18:50:41 +00:00
75 lines
2.8 KiB
Python
75 lines
2.8 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.
|
|
"""
|
|
|
|
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
|
|
|
|
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}")
|