bupa1018 commited on
Commit
624df56
·
1 Parent(s): aef0888

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 extract_files_and_filepath_from_dir
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 = extract_files_and_filepath_from_dir(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_files_and_filepath_from_dir(DATA_DIR, [], [])
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))