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 |
|