Updates

Author: sjrl

integrations/nvidia/src/haystack_integrations/components/embedders/nvidia/document_embedder.py

@@ -247,14 +247,12 @@ def run(self, documents: list[Document]) -> dict[str, Union[list[Document], dict
             A dictionary with the following keys and values:
             - `documents` - List of processed Documents with embeddings.
             - `meta` - Metadata on usage statistics, etc.
-        :raises RuntimeError:
-            If the component was not initialized.
         :raises TypeError:
             If the input is not a string.
         """
         if not self._initialized:
-            msg = "The embedding model has not been loaded. Please call warm_up() before running."
-            raise RuntimeError(msg)
+            self.warm_up()
+
         elif not isinstance(documents, list) or (documents and not isinstance(documents[0], Document)):
             msg = (
                 "NvidiaDocumentEmbedder expects a list of Documents as input."

integrations/nvidia/src/haystack_integrations/components/embedders/nvidia/text_embedder.py

@@ -197,14 +197,14 @@ def run(self, text: str) -> dict[str, Union[list[float], dict[str, Any]]]:
             A dictionary with the following keys and values:
             - `embedding` - Embedding of the text.
             - `meta` - Metadata on usage statistics, etc.
-        :raises RuntimeError:
-            If the component was not initialized.
         :raises TypeError:
             If the input is not a string.
+        :raises ValueError:
+            If the input string is empty.
         """
         if not self._initialized:
-            msg = "The embedding model has not been loaded. Please call warm_up() before running."
-            raise RuntimeError(msg)
+            self.warm_up()
+
         elif not isinstance(text, str):
             msg = (
                 "NvidiaTextEmbedder expects a string as an input."