mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-12-30 08:37:20 +00:00
ddba75021a
* "proxy-enabled": disable CORS proxy * "samples-languages": display two languages initially
48 lines
1.6 KiB
Python
Executable File
48 lines
1.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import json
|
|
from pathlib import Path
|
|
import os
|
|
import sys
|
|
import shutil
|
|
|
|
import logging
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
|
|
|
|
sys.path.append(".")
|
|
from rest_api.utils import get_openapi_specs, get_app, get_pipelines # pylint: disable=wrong-import-position
|
|
from haystack import __version__ # pylint: disable=wrong-import-position
|
|
|
|
REST_PATH = Path("./rest_api/rest_api").absolute()
|
|
PIPELINE_PATH = str(REST_PATH / "pipeline" / "pipeline_empty.haystack-pipeline.yml")
|
|
APP_PATH = str(REST_PATH / "application.py")
|
|
DOCS_PATH = Path("./docs") / "_src" / "api" / "openapi"
|
|
|
|
os.environ["PIPELINE_YAML_PATH"] = PIPELINE_PATH
|
|
|
|
logging.info("Loading OpenAPI specs from %s with pipeline at %s", APP_PATH, PIPELINE_PATH)
|
|
|
|
# To initialize the app and the pipelines
|
|
get_app()
|
|
get_pipelines()
|
|
|
|
# Generate the openapi specs
|
|
specs = get_openapi_specs()
|
|
# Add `x-readme` to disable proxy and limit sample languages on documentation (see https://docs.readme.com/main/docs/openapi-extensions)
|
|
specs.update({"x-readme": {"proxy-enabled": False, "samples-languages": ["curl", "python"]}})
|
|
|
|
# Dump the specs into a JSON file
|
|
with open(DOCS_PATH / "openapi.json", "w") as f:
|
|
json.dump(specs, f, indent=4)
|
|
f.write("\n") # We need to add a newline, otherwise there will be a conflict with end-of-file-fixer pre-commit hook
|
|
|
|
# Remove rc versions of the specs from the folder
|
|
for specs_file in os.listdir():
|
|
if os.path.isfile(specs_file) and "rc" in specs_file and Path(specs_file).suffix == ".json":
|
|
os.remove(specs_file)
|
|
|
|
# Add versioned copy
|
|
shutil.copy(DOCS_PATH / "openapi.json", DOCS_PATH / f"openapi-{__version__}.json")
|