openapi in readthedocs

This commit is contained in:
Xi Yan 2024-11-22 17:22:45 -08:00
parent 72a0dbb15a
commit dbd4b9ecce
4 changed files with 57 additions and 49 deletions

View file

@ -438,6 +438,14 @@ class Generator:
return extra_tags
def _build_operation(self, op: EndpointOperation) -> Operation:
if op.defining_class.__name__ in [
"SyntheticDataGeneration",
"PostTraining",
"BatchInference",
]:
op.defining_class.__name__ = f"(Coming Soon) {op.defining_class.__name__}"
print(op.defining_class.__name__)
doc_string = parse_type(op.func_ref)
doc_params = dict(
(param.name, param.description) for param in doc_string.params.values()