langchain/docs/scripts/resolve_local_links.py
2024-05-24 19:44:46 +00:00

27 lines
525 B
Python

import os
import re
import sys
from pathlib import Path
DOCS_DIR = Path(os.path.abspath(__file__)).parents[1]
def update_links(doc_path, docs_link):
with open(DOCS_DIR / doc_path, "r") as f:
content = f.read()
# replace relative links
content = re.sub(r"\]\(\.\/", f"]({docs_link}", content)
frontmatter = """---
custom_edit_url:
---
"""
with open(DOCS_DIR / doc_path, "w") as f:
f.write(frontmatter + content)
if __name__ == "__main__":
update_links(sys.argv[1], sys.argv[2])