Update weaviate_datastore.py

shrix · May 15, 2023 · af10132 · af10132
1 parent ddba10f
commit af10132
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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"],