Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,8 @@ with open('config.json') as config_file:
|
|
24 |
config = json.load(config_file)
|
25 |
|
26 |
|
27 |
-
|
|
|
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,
|
373 |
-
codestore = setup_vectorstore(code_chunks, EMBEDDING_MODEL_NAME,
|
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 |
|