openapi in readthedocs

docs/openapi_generator/generate.py

@@ -52,12 +52,11 @@ def main(output_dir: str):
         Options(
             server=Server(url="http://any-hosted-llama-stack.com"),
             info=Info(
-                title="[DRAFT] Llama Stack Specification",
+                title="Llama Stack Specification",
                 version=LLAMA_STACK_API_VERSION,
-                description="""This is the specification of the llama stack that provides
+                description="""This is the specification of the Llama Stack that provides
                 a set of endpoints and their corresponding interfaces that are tailored to
-                best leverage Llama Models. The specification is still in draft and subject to change.
-                Generated at """
+                best leverage Llama Models. Generated at """
                 + now,
             ),
         ),

docs/openapi_generator/pyopenapi/generator.py

@@ -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()