bupa1018 commited on
Commit
f9bbc36
·
1 Parent(s): 0c7d2d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
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()