update import for is_hosted

deepset-ai · Aug 7, 2024 · 7d71017 · 7d71017
1 parent 2d2e9dd
commit 7d71017
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 7 deletions.
diff --git a/...rations/nvidia/src/haystack_integrations/components/embedders/nvidia/document_embedder.py b/...rations/nvidia/src/haystack_integrations/components/embedders/nvidia/document_embedder.py
@@ -3,8 +3,7 @@
 
 from haystack import Document, component, default_from_dict, default_to_dict
 from haystack.utils import Secret, deserialize_secrets_inplace
-from haystack_integrations.util.nvidia import NimBackend
-from haystack_integrations.util.nvidia.util import is_hosted
+from haystack_integrations.util.nvidia import NimBackend, is_hosted
 from tqdm import tqdm
 
 from .truncate import EmbeddingTruncateMode

diff --git a/integrations/nvidia/src/haystack_integrations/components/embedders/nvidia/text_embedder.py b/integrations/nvidia/src/haystack_integrations/components/embedders/nvidia/text_embedder.py
@@ -3,8 +3,7 @@
 
 from haystack import component, default_from_dict, default_to_dict
 from haystack.utils import Secret, deserialize_secrets_inplace
-from haystack_integrations.util.nvidia import NimBackend
-from haystack_integrations.util.nvidia.util import is_hosted
+from haystack_integrations.util.nvidia import NimBackend, is_hosted
 
 from .truncate import EmbeddingTruncateMode
 

diff --git a/integrations/nvidia/src/haystack_integrations/components/generators/nvidia/generator.py b/integrations/nvidia/src/haystack_integrations/components/generators/nvidia/generator.py
@@ -6,8 +6,7 @@
 
 from haystack import component, default_from_dict, default_to_dict
 from haystack.utils.auth import Secret, deserialize_secrets_inplace
-from haystack_integrations.util.nvidia import NimBackend
-from haystack_integrations.util.nvidia.util import is_hosted
+from haystack_integrations.util.nvidia import NimBackend, is_hosted
 
 _DEFAULT_API_URL = "https://integrate.api.nvidia.com/v1"
 

diff --git a/integrations/nvidia/src/haystack_integrations/util/nvidia/__init__.py b/integrations/nvidia/src/haystack_integrations/util/nvidia/__init__.py
@@ -1,3 +1,4 @@
 from .nim_backend import Model, NimBackend
+from .util import is_hosted
 
-__all__ = ["NimBackend", "Model"]
+__all__ = ["NimBackend", "Model", "is_hosted"]