Spaces:
Runtime error
Runtime error
Update direct from local
Browse files- 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 = "
|
39 |
|
40 |
docs = load_pdf_data("data/Explorer.pdf")
|
41 |
|
42 |
vector_db = Qdrant.from_documents(
|
43 |
-
|
44 |
-
embedding = embedding_model,
|
45 |
location=qdrant_url,
|
46 |
collection_name=collection_name,
|
47 |
api_key=api_key,
|
48 |
-
timeout=
|
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 |
|