bupa1018 commited on
Commit
7d60fcf
·
1 Parent(s): 724f085

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -24,7 +24,8 @@ with open('config.json') as config_file:
24
  config = json.load(config_file)
25
 
26
 
27
- PERSIST_DIRECTORY = config["persist_directory"]
 
28
  CHUNK_SIZE = config["chunk_size"]
29
  CHUNK_OVERLAP = config["chunk_overlap"]
30
  EMBEDDING_MODEL_NAME = config["embedding_model"]
@@ -369,8 +370,8 @@ def initialize():
369
  print(f"Total number of code_chunks: {len(code_chunks)}")
370
  print(f"Total number of doc_chunks: {len(doc_chunks)}")
371
 
372
- docstore = setup_vectorstore(doc_chunks, EMBEDDING_MODEL_NAME, PERSIST_DIRECTORY)
373
- codestore = setup_vectorstore(code_chunks, EMBEDDING_MODEL_NAME, PERSIST_DIRECTORY)
374
 
375
  llm = setup_llm(LLM_MODEL_NAME, LLM_TEMPERATURE, GROQ_API_KEY)
376
 
 
24
  config = json.load(config_file)
25
 
26
 
27
+ PERSIST_DOC_DIRECTORY = config["chroma_doc_data_"],
28
+ PERSIST_CODE_DIRECTORY =config["chroma_code_data"]
29
  CHUNK_SIZE = config["chunk_size"]
30
  CHUNK_OVERLAP = config["chunk_overlap"]
31
  EMBEDDING_MODEL_NAME = config["embedding_model"]
 
370
  print(f"Total number of code_chunks: {len(code_chunks)}")
371
  print(f"Total number of doc_chunks: {len(doc_chunks)}")
372
 
373
+ docstore = setup_vectorstore(doc_chunks, EMBEDDING_MODEL_NAME, PERSIST_DOC_DIRECTORY)
374
+ codestore = setup_vectorstore(code_chunks, EMBEDDING_MODEL_NAME, PERSIST_CODE_DIRECTORY)
375
 
376
  llm = setup_llm(LLM_MODEL_NAME, LLM_TEMPERATURE, GROQ_API_KEY)
377