search

yoheinakajima / babyagi

MIT License
19.64k stars 2.57k forks source link

fix: enriched result vector not passed to storage when using weaviate #342

Open saad039 opened 1 year ago

saad039 commented 1 year ago

Enriched result vector is required to be added inside result_storage when using Weaviate (i.e WEAVIATE_USE_EMBEDDED=True).

Therefore, it modifies the existing code to 

vector = enriched_result["data"] if WEAVIATE_USE_EMBEDDED else None
...
results_storage.add(task, result, result_id, vector=vector)