Minor import enhancement

chandrasekharan-zipstack · chandrasekharan-zipstack · commit 4feac345604d · 2024-06-13T21:03:21.000+05:30
diff --git a/src/unstract/llmwhisperer/__init__.py b/src/unstract/llmwhisperer/__init__.py
@@ -1,5 +1,7 @@
 __version__ = "0.2.0"
 
+from .client import LLMWhispererClient  # noqa: F401
+
 
 def get_sdk_version():
     """Returns the SDK version."""
diff --git a/tests/client_test.py b/tests/client_test.py
@@ -5,7 +5,7 @@
 
 import pytest
 
-from unstract.llmwhisperer.client import LLMWhispererClient
+from unstract.llmwhisperer import LLMWhispererClient
 
 logger = logging.getLogger(__name__)
 
