Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ from langchain.schema import Document
|
|
12 |
from chunk_python_code import chunk_python_code_with_metadata
|
13 |
from vectorstore import get_chroma_vectorstore
|
14 |
from download_repo_to_huggingface import download_gitlab_repo_to_hfspace
|
15 |
-
from process_repo_zipfile import
|
16 |
from ragchain import RAGChain
|
17 |
from llm import get_groq_llm
|
18 |
|
@@ -155,9 +155,9 @@ def initialize():
|
|
155 |
version=config2["gitlab"]["project"]["version"]
|
156 |
)
|
157 |
|
158 |
-
code_texts, code_references =
|
159 |
#doc_texts, doc_references = extract_files_and_filepath_from_dir(DATA_DIR, ['docs/source/'], [])
|
160 |
-
doc_texts, doc_references =
|
161 |
|
162 |
print("LEEEEEEEEEEEENGTH of code_texts: ", len(code_texts))
|
163 |
print("LEEEEEEEEEEEENGTH of doc_files: ", len(doc_texts))
|
|
|
12 |
from chunk_python_code import chunk_python_code_with_metadata
|
13 |
from vectorstore import get_chroma_vectorstore
|
14 |
from download_repo_to_huggingface import download_gitlab_repo_to_hfspace
|
15 |
+
from process_repo_zipfile import extract_repo_files
|
16 |
from ragchain import RAGChain
|
17 |
from llm import get_groq_llm
|
18 |
|
|
|
155 |
version=config2["gitlab"]["project"]["version"]
|
156 |
)
|
157 |
|
158 |
+
code_texts, code_references = extract_repo_files(DATA_DIR, ['kadi_apy'], [])
|
159 |
#doc_texts, doc_references = extract_files_and_filepath_from_dir(DATA_DIR, ['docs/source/'], [])
|
160 |
+
doc_texts, doc_references = extract_repo_files(DATA_DIR, [], [])
|
161 |
|
162 |
print("LEEEEEEEEEEEENGTH of code_texts: ", len(code_texts))
|
163 |
print("LEEEEEEEEEEEENGTH of doc_files: ", len(doc_texts))
|