github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-20 05:43:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

docs: fix api ref link autogeneration (#20766)

Browse Source
This commit is contained in:
Bagatur 2024-04-22 17:36:41 -07:00 committed by GitHub
parent c807f0a6dd
commit be51cd3bc9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
docs/scripts/generate_api_reference_links.py
View File

@ -13,10 +13,10 @@ logger = logging.getLogger(__name__)
_BASE_URL = "https://api.python.langchain.com/en/latest/"
# Regular expression to match Python code blocks
code_block_re = re.compile(r"^(```python\n)(.*?)(```\n)", re.DOTALL | re.MULTILINE)
code_block_re = re.compile(r"^(```\s?python\n)(.*?)(```)", re.DOTALL | re.MULTILINE)
# Regular expression to match langchain import lines
_IMPORT_RE = re.compile(
r"from\s+(langchain(?:_\w+)?\.\w+(?:\.\w+)*?)\s+import\s+"
r"from\s+(langchain(?:_\w+)?(?:\.\w+)*?)\s+import\s+"
r"((?:\w+(?:,\s*)?)*" # Match zero or more words separated by a comma+optional ws
r"(?:\s*\(.*?\))?)", # Match optional parentheses block
re.DOTALL, # Match newlines as well
@ -64,7 +64,6 @@ def main():
global_imports = {}
for file in find_files(args.docs_dir):
print(f"Adding links for imports in {file}") # noqa: T201
file_imports = replace_imports(file)
if file_imports:
@ -179,6 +178,8 @@ def replace_imports(file):
# Use re.sub to replace each Python code block
data = code_block_re.sub(replacer, data)
if all_imports:
print(f"Adding {len(all_imports)} links for imports in {file}") # noqa: T201
with open(file, "w") as f:
f.write(data)
return all_imports