From af10132cb6a299fb8c1e927d6fbada7b7415def7 Mon Sep 17 00:00:00 2001
From: isafulf <51974293+isafulf@users.noreply.github.com>
Date: Mon, 15 May 2023 13:46:47 -0700
Subject: [PATCH] Update weaviate_datastore.py

---
 datastore/providers/weaviate_datastore.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/datastore/providers/weaviate_datastore.py b/datastore/providers/weaviate_datastore.py
index 9202835e5..4baae392a 100644
--- a/datastore/providers/weaviate_datastore.py
+++ b/datastore/providers/weaviate_datastore.py
@@ -245,11 +245,11 @@ async def _single_query(query: QueryWithEmbedding) -> QueryResult:
                 result = DocumentChunkWithScore(
                     id=resp["chunk_id"],
                     text=resp["text"],
-                    embedding=resp["_additional"]["vector"],
+                    # embedding=resp["_additional"]["vector"],
                     score=resp["_additional"]["score"],
                     metadata=DocumentChunkMetadata(
                         document_id=resp["document_id"] if resp["document_id"] else "",
-                        source=Source(resp["source"]),
+                        source=Source(resp["source"]) if resp["source"] else None,
                         source_id=resp["source_id"],
                         url=resp["url"],
                         created_at=resp["created_at"],