From 72cbf3ee0b190eafc01e20039de047fbd76edb23 Mon Sep 17 00:00:00 2001
From: Stefano Fiorucci <44616784+anakin87@users.noreply.github.com>
Date: Wed, 8 Nov 2023 17:33:07 +0100
Subject: [PATCH] fix: replace `haystack.lazy_imports` with
 `haystack.preview.lazy_imports` (#6255)

* lazy import transformers in tgi

* fix pylint

* fix wrong import
---
 haystack/preview/components/generators/chat/hugging_face_tgi.py | 2 +-
 haystack/preview/components/generators/hf_utils.py              | 2 +-
 haystack/preview/components/generators/hugging_face_tgi.py      | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/haystack/preview/components/generators/chat/hugging_face_tgi.py b/haystack/preview/components/generators/chat/hugging_face_tgi.py
index a5681e380..4d4062ef2 100644
--- a/haystack/preview/components/generators/chat/hugging_face_tgi.py
+++ b/haystack/preview/components/generators/chat/hugging_face_tgi.py
@@ -7,7 +7,7 @@ from haystack.preview import component, default_to_dict, default_from_dict
 from haystack.preview.components.generators.utils import serialize_callback_handler, deserialize_callback_handler
 from haystack.preview.dataclasses import ChatMessage, StreamingChunk
 from haystack.preview.components.generators.hf_utils import check_valid_model, check_generation_params
-from haystack.lazy_imports import LazyImport
+from haystack.preview.lazy_imports import LazyImport
 
 with LazyImport(message="Run 'pip install transformers'") as transformers_import:
     from huggingface_hub import InferenceClient
diff --git a/haystack/preview/components/generators/hf_utils.py b/haystack/preview/components/generators/hf_utils.py
index 832a99628..9eca92ae5 100644
--- a/haystack/preview/components/generators/hf_utils.py
+++ b/haystack/preview/components/generators/hf_utils.py
@@ -1,7 +1,7 @@
 import inspect
 from typing import Any, Dict, List, Optional
 
-from haystack.lazy_imports import LazyImport
+from haystack.preview.lazy_imports import LazyImport
 
 with LazyImport(message="Run 'pip install transformers'") as transformers_import:
     from huggingface_hub import InferenceClient, HfApi
diff --git a/haystack/preview/components/generators/hugging_face_tgi.py b/haystack/preview/components/generators/hugging_face_tgi.py
index 5d3526f03..71dc64acd 100644
--- a/haystack/preview/components/generators/hugging_face_tgi.py
+++ b/haystack/preview/components/generators/hugging_face_tgi.py
@@ -6,8 +6,8 @@ from urllib.parse import urlparse
 from haystack.preview import component, default_to_dict, default_from_dict
 from haystack.preview.components.generators.utils import serialize_callback_handler, deserialize_callback_handler
 from haystack.preview.dataclasses import StreamingChunk
-from haystack.lazy_imports import LazyImport
 from haystack.preview.components.generators.hf_utils import check_generation_params, check_valid_model
+from haystack.preview.lazy_imports import LazyImport
 
 with LazyImport(message="Run 'pip install transformers'") as transformers_import:
     from huggingface_hub import InferenceClient