LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2025-11-29 16:43:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update ingest.py

Browse Source
This commit is contained in:
Robbie Walmsley
2023-10-09 11:04:55 +01:00
committed by GitHub
parent 57fb29b600
commit 2d0b6bcfcc
1 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
scripts/ingest.py
View File

@@ -108,14 +108,19 @@ def convert(dir: Optional[str] = typer.Option("inputs",
Creates documentation linked to original functions from specified location.
By default /inputs folder is used, .py is parsed.
"""
if formats == 'py':
functions_dict, classes_dict = extract_py(dir)
elif formats == 'js':
functions_dict, classes_dict = extract_js(dir)
elif formats == 'java':
functions_dict, classes_dict = extract_java(dir)
# Using a dictionary to map between the formats and their respective extraction functions
# makes the code more scalable. When adding more formats in the future,
# you only need to update the extraction_functions dictionary.
extraction_functions = {
'py': extract_py,
'js': extract_js,
'java': extract_java
}
if formats in extraction_functions:
functions_dict, classes_dict = extraction_functions[formats](dir)
else:
raise Exception("Sorry, language not supported yet")
raise Exception("Sorry, language not supported yet")
transform_to_docs(functions_dict, classes_dict, formats, dir)