Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -360,26 +360,26 @@ def rag_workflow(query):
|
|
360 |
def initialize():
|
361 |
global docstore, codestore, chunks, llm
|
362 |
|
363 |
-
code_partial_paths = ['kadi_apy/lib/']
|
364 |
-
code_file_path = []
|
365 |
-
doc_partial_paths = ['docs/source/setup/']
|
366 |
doc_file_paths = ['docs/source/usage/lib.rst']
|
367 |
|
368 |
|
369 |
-
code_files, code_file_references = process_directory(REPOSITORY_DIRECTORY, code_partial_paths, code_file_path)
|
370 |
|
371 |
doc_files, doc_file_references = process_directory(REPOSITORY_DIRECTORY, doc_partial_paths, doc_file_paths)
|
372 |
|
373 |
-
code_chunks = split_pythoncode_into_chunks(code_files, code_file_references, 1500, 0)
|
374 |
doc_chunks = split_into_chunks(doc_files, doc_file_references, CHUNK_SIZE, CHUNK_OVERLAP)
|
375 |
|
376 |
-
print(f"Total number of code_chunks: {len(code_chunks)}")
|
377 |
print(f"Total number of doc_chunks: {len(doc_chunks)}")
|
378 |
|
379 |
docstore = setup_vectorstore(doc_chunks, EMBEDDING_MODEL_NAME, PERSIST_DOC_DIRECTORY)
|
380 |
-
codestore = setup_vectorstore(code_chunks, EMBEDDING_MODEL_NAME, PERSIST_CODE_DIRECTORY)
|
381 |
|
382 |
-
llm = setup_llm(LLM_MODEL_NAME, LLM_TEMPERATURE, GROQ_API_KEY)
|
383 |
|
384 |
|
385 |
initialize()
|
|
|
360 |
def initialize():
|
361 |
global docstore, codestore, chunks, llm
|
362 |
|
363 |
+
#code_partial_paths = ['kadi_apy/lib/']
|
364 |
+
#code_file_path = []
|
365 |
+
#doc_partial_paths = ['docs/source/setup/']
|
366 |
doc_file_paths = ['docs/source/usage/lib.rst']
|
367 |
|
368 |
|
369 |
+
#code_files, code_file_references = process_directory(REPOSITORY_DIRECTORY, code_partial_paths, code_file_path)
|
370 |
|
371 |
doc_files, doc_file_references = process_directory(REPOSITORY_DIRECTORY, doc_partial_paths, doc_file_paths)
|
372 |
|
373 |
+
#code_chunks = split_pythoncode_into_chunks(code_files, code_file_references, 1500, 0)
|
374 |
doc_chunks = split_into_chunks(doc_files, doc_file_references, CHUNK_SIZE, CHUNK_OVERLAP)
|
375 |
|
376 |
+
#print(f"Total number of code_chunks: {len(code_chunks)}")
|
377 |
print(f"Total number of doc_chunks: {len(doc_chunks)}")
|
378 |
|
379 |
docstore = setup_vectorstore(doc_chunks, EMBEDDING_MODEL_NAME, PERSIST_DOC_DIRECTORY)
|
380 |
+
#codestore = setup_vectorstore(code_chunks, EMBEDDING_MODEL_NAME, PERSIST_CODE_DIRECTORY)
|
381 |
|
382 |
+
#llm = setup_llm(LLM_MODEL_NAME, LLM_TEMPERATURE, GROQ_API_KEY)
|
383 |
|
384 |
|
385 |
initialize()
|