updated pipeline to avoid corner case with empty item

wikidatachat/api.py

@@ -145,7 +145,13 @@ async def api(
     logger.debug(f'{answer=}')
 
     # Return the processed answer and sources.
-    return {
-        "answer": answer.data.content,
-        "sources": sources
-    }
+    if hasattr(answer.data, 'content'):
+        return {
+            "answer": answer.data.content,
+            "sources": sources
+        }
+    else:
+        return {
+            "answer": '',
+            "sources": sources
+        }

wikidatachat/vector_store_interface.py

@@ -4,6 +4,7 @@
 from tqdm import tqdm
 
 from haystack import Document  # , Pipeline
+from haystack.utils import ComponentDevice
 from haystack.components.embedders import SentenceTransformersDocumentEmbedder
 from haystack.document_stores.in_memory import InMemoryDocumentStore
 from haystack.components.retrievers.in_memory import InMemoryEmbeddingRetriever
@@ -189,6 +190,7 @@ def make_embedder(embedding_model: str = EMBEDDING_MODEL, device: str = 'cpu'):
     """
     # Use GPU if available, otherwise fallback to CPU.
     device = "cuda" if torch.cuda.is_available() else device
+    device = ComponentDevice.from_str(device)
 
     logger.info(  # Log the name of the embedding model being used.
         'GPU is available. Using GPU.'