anhkhoiphan commited on
Commit
a1523c0
verified
1 Parent(s): 4efabab

Update direct from local

Browse files
Files changed (1) hide show
  1. chatbot/retrieval.py +3 -4
chatbot/retrieval.py CHANGED
@@ -35,17 +35,16 @@ def load_pdf_data(pdf_path):
35
  def get_vector_db():
36
  qdrant_url = QDRANT_HOST
37
  api_key = QDRANT_API_KEY
38
- collection_name = "expl_embeddings_01"
39
 
40
  docs = load_pdf_data("data/Explorer.pdf")
41
 
42
  vector_db = Qdrant.from_documents(
43
- documents = docs,
44
- embedding = embedding_model,
45
  location=qdrant_url,
46
  collection_name=collection_name,
47
  api_key=api_key,
48
- timeout=300
49
  )
50
  return vector_db
51
 
 
35
  def get_vector_db():
36
  qdrant_url = QDRANT_HOST
37
  api_key = QDRANT_API_KEY
38
+ collection_name = "expl_embeddings"
39
 
40
  docs = load_pdf_data("data/Explorer.pdf")
41
 
42
  vector_db = Qdrant.from_documents(
43
+ docs, embedding_model,
 
44
  location=qdrant_url,
45
  collection_name=collection_name,
46
  api_key=api_key,
47
+ timeout=500
48
  )
49
  return vector_db
50