Fix the pyopenapi generator avoid potential circular imports

This commit is contained in:
Ashwin Bharambe 2024-11-18 23:37:52 -08:00
parent d463d68e1e
commit 8ed79ad0f3
6 changed files with 5678 additions and 32467 deletions

View file

@ -33,10 +33,8 @@ schema_utils.json_schema_type = json_schema_type
# this line needs to be here to ensure json_schema_type has been altered before
# the imports use the annotation
from llama_stack.distribution.stack import ( # noqa: E402
LLAMA_STACK_API_VERSION,
LlamaStack,
)
from llama_stack.apis.version import LLAMA_STACK_API_VERSION # noqa: E402
from llama_stack.distribution.stack import LlamaStack # noqa: E402
def main(output_dir: str):

View file

@ -204,7 +204,7 @@ class ContentBuilder:
if self.schema_transformer:
schema_transformer: Callable[[SchemaOrRef], SchemaOrRef] = (
self.schema_transformer
) # type: ignore
)
schema = schema_transformer(schema)
if not examples:

View file

@ -12,7 +12,7 @@ import uuid
from dataclasses import dataclass
from typing import Any, Callable, Dict, Iterable, Iterator, List, Optional, Tuple, Union
from llama_stack.distribution.stack import LLAMA_STACK_API_VERSION
from llama_stack.apis.version import LLAMA_STACK_API_VERSION
from termcolor import colored