Update app.py
Browse files
app.py
CHANGED
@@ -71,17 +71,17 @@ def initialize_embeddings():
|
|
71 |
return None
|
72 |
|
73 |
@st.cache_resource
|
74 |
-
def store_in_qdrant(
|
75 |
"""Store document chunks in a hosted Qdrant instance."""
|
76 |
try:
|
77 |
vector_store = Qdrant.from_documents(
|
78 |
-
documents=
|
79 |
embedding=embeddings,
|
80 |
url=os.getenv("QDRANT_URL"),
|
81 |
api_key=os.getenv("QDRANT_API_KEY"),
|
82 |
collection_name="wikipedia_chunks"
|
83 |
)
|
84 |
-
logger.info(f"Stored {len(
|
85 |
return vector_store
|
86 |
except Exception as e:
|
87 |
logger.error(f"Error storing in Qdrant: {e}")
|
|
|
71 |
return None
|
72 |
|
73 |
@st.cache_resource
|
74 |
+
def store_in_qdrant(_chunks, embeddings):
|
75 |
"""Store document chunks in a hosted Qdrant instance."""
|
76 |
try:
|
77 |
vector_store = Qdrant.from_documents(
|
78 |
+
documents=_chunks,
|
79 |
embedding=embeddings,
|
80 |
url=os.getenv("QDRANT_URL"),
|
81 |
api_key=os.getenv("QDRANT_API_KEY"),
|
82 |
collection_name="wikipedia_chunks"
|
83 |
)
|
84 |
+
logger.info(f"Stored {len(_chunks)} chunks in Qdrant at {os.getenv('QDRANT_URL')}")
|
85 |
return vector_store
|
86 |
except Exception as e:
|
87 |
logger.error(f"Error storing in Qdrant: {e}")
|