diff --git a/integrations/nvidia/src/haystack_integrations/components/embedders/nvidia/document_embedder.py b/integrations/nvidia/src/haystack_integrations/components/embedders/nvidia/document_embedder.py index 2c349423b..53988cdaa 100644 --- a/integrations/nvidia/src/haystack_integrations/components/embedders/nvidia/document_embedder.py +++ b/integrations/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 index c594375d8..83ce29fff 100644 --- 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 index f5c7e231f..1739d6142 100644 --- 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 index 32da64a6c..6977f1aa1 100644 --- 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"]