leo-bourrel commited on
Commit
341806e
·
1 Parent(s): 4bcb630

feat: add metadata

Browse files
Files changed (1) hide show
  1. 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
- # metadata={"title": result.Article.title},
 
 
 
 
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
  )