llama-stack-mirror/scripts/openapi_generator/state.py
Sébastien Han e79a03b697
chore: chop fastapi_generator into its module
Decoupled the large script with distinct files and purpose.

Signed-off-by: Sébastien Han <seb@redhat.com>
2025-11-14 09:56:45 +01:00

23 lines
788 B
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.
"""
Shared state for the OpenAPI generator module.
"""
from typing import Any
from llama_stack.apis.datatypes import Api
# Global list to store dynamic models created during endpoint generation
_dynamic_models: list[Any] = []
# Cache for protocol methods to avoid repeated lookups
_protocol_methods_cache: dict[Api, dict[str, Any]] | None = None
# Global dict to store extra body field information by endpoint
# Key: (path, method) tuple, Value: list of (param_name, param_type, description) tuples
_extra_body_fields: dict[tuple[str, str], list[tuple[str, type, str | None]]] = {}