Spaces:
Sleeping
Sleeping
Commit
·
341806e
1
Parent(s):
4bcb630
feat: add metadata
Browse files- custom_pgvector.py +5 -1
custom_pgvector.py
CHANGED
@@ -353,7 +353,11 @@ class CustomPGVector(VectorStore):
|
|
353 |
(
|
354 |
Document(
|
355 |
page_content=result.Article.abstract,
|
356 |
-
|
|
|
|
|
|
|
|
|
357 |
),
|
358 |
result.distance if self.embedding_function is not None else None,
|
359 |
)
|
|
|
353 |
(
|
354 |
Document(
|
355 |
page_content=result.Article.abstract,
|
356 |
+
metadata={
|
357 |
+
"id": result.Article.id,
|
358 |
+
"title": result.Article.title,
|
359 |
+
"doi": result.Article.doi
|
360 |
+
},
|
361 |
),
|
362 |
result.distance if self.embedding_function is not None else None,
|
363 |
)
|