juliaturc commited on
Commit
9ce7fb5
·
1 Parent(s): 21f5cea

Update vector_store.py

Browse files
Files changed (1) hide show
  1. sage/vector_store.py +2 -1
sage/vector_store.py CHANGED
@@ -89,7 +89,8 @@ class PineconeVectorStore(VectorStore):
89
  def patched_query(*args, **kwargs):
90
  result = original_query(*args, **kwargs)
91
  for res in result["matches"]:
92
- res["metadata"]["context"] = res["metadata"][TEXT_FIELD]
 
93
  return result
94
 
95
  index.query = patched_query
 
89
  def patched_query(*args, **kwargs):
90
  result = original_query(*args, **kwargs)
91
  for res in result["matches"]:
92
+ if TEXT_FIELD in res["metadata"]:
93
+ res["metadata"]["context"] = res["metadata"][TEXT_FIELD]
94
  return result
95
 
96
  index.query = patched_query