Goodnight7 commited on
Commit
6573c7f
·
verified ·
1 Parent(s): ccfc75f

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -16,12 +16,12 @@ from qdrant_client import models
16
  load_dotenv()
17
  #Retriever
18
  def retriever(n_docs=5):
19
- vector_database_path = "knowledge-base"
20
 
21
  embeddings_model = NomicEmbeddings(model="nomic-embed-text-v1.5", inference_mode="local")
22
 
23
 
24
- vectorstore = Chroma(collection_name="knowledge-base",
25
  persist_directory=vector_database_path,
26
  embedding_function=embeddings_model)
27
 
 
16
  load_dotenv()
17
  #Retriever
18
  def retriever(n_docs=5):
19
+ vector_database_path = "chromadb"
20
 
21
  embeddings_model = NomicEmbeddings(model="nomic-embed-text-v1.5", inference_mode="local")
22
 
23
 
24
+ vectorstore = Chroma(collection_name="chromadb",
25
  persist_directory=vector_database_path,
26
  embedding_function=embeddings_model)
27