Update __init__.py in ads.llm

qiuosier · qiuosier · commit b7027c92e5fc · 2023-12-06T11:04:20.000-05:00
diff --git a/ads/llm/__init__.py b/ads/llm/__init__.py
@@ -16,3 +16,11 @@
             f"{ex.msg}\nPlease install/update langchain with `pip install langchain -U`"
         ) from ex
     raise ex
+
+
+__all__ = [
+    "GenerativeAI",
+    "GenerativeAIEmbeddings",
+    "ModelDeploymentTGI",
+    "ModelDeploymentVLLM",
+]
