diff --git a/docs/source/conf.py b/docs/source/conf.py index eec21fbb..f9e85c42 100644 --- a/docs/source/conf.py +++ b/docs/source/conf.py @@ -77,6 +77,7 @@ None, ), "pydantic": ("https://docs.pydantic.dev/latest/", None), + "unitycatalog": ("https://docs.unitycatalog.io/", None), } extlinks = { diff --git a/integrations/langchain/src/databricks_langchain/__init__.py b/integrations/langchain/src/databricks_langchain/__init__.py index e281fff6..40135d97 100644 --- a/integrations/langchain/src/databricks_langchain/__init__.py +++ b/integrations/langchain/src/databricks_langchain/__init__.py @@ -1,3 +1,18 @@ +""" +**Re-exported Unity Catalog Utilities** + +This module re-exports selected utilities from the Unity Catalog open source package. + +Available aliases: + +- :class:`databricks_langchain.UCFunctionToolkit` +- :class:`databricks_langchain.UnityCatalogTool` +- :class:`databricks_langchain.DatabricksFunctionClient` +- :func:`databricks_langchain.set_uc_function_client` + +Refer to the Unity Catalog `documentation `_ for more information. +""" + from unitycatalog.ai.core.base import set_uc_function_client from unitycatalog.ai.core.databricks import DatabricksFunctionClient from unitycatalog.ai.langchain.toolkit import UCFunctionToolkit, UnityCatalogTool diff --git a/integrations/llamaindex/pyproject.toml b/integrations/llamaindex/pyproject.toml index c88ad7d9..3d781c6a 100644 --- a/integrations/llamaindex/pyproject.toml +++ b/integrations/llamaindex/pyproject.toml @@ -12,6 +12,7 @@ dependencies = [ "databricks-vectorsearch>=0.40", "databricks-ai-bridge>=0.1.0", "llama-index>=0.11.0", + "unitycatalog-llamaindex[databricks]>=0.2.0", ] [project.optional-dependencies] diff --git a/integrations/llamaindex/src/databricks_llamaindex/__init__.py b/integrations/llamaindex/src/databricks_llamaindex/__init__.py index ceccb60d..fc224b84 100644 --- a/integrations/llamaindex/src/databricks_llamaindex/__init__.py +++ b/integrations/llamaindex/src/databricks_llamaindex/__init__.py @@ -1,6 +1,28 @@ +""" +**Re-exported Unity Catalog Utilities** + +This module re-exports selected utilities from the Unity Catalog open source package. + +Available aliases: + +- :class:`databricks_llamaindex.UCFunctionToolkit` +- :class:`databricks_llamaindex.UnityCatalogTool` +- :class:`databricks_llamaindex.DatabricksFunctionClient` +- :func:`databricks_llamaindex.set_uc_function_client` + +Refer to the Unity Catalog `documentation `_ for more information. +""" + +from unitycatalog.ai.core.base import set_uc_function_client +from unitycatalog.ai.core.databricks import DatabricksFunctionClient +from unitycatalog.ai.llama_index.toolkit import UCFunctionToolkit + from databricks_llamaindex.vector_search_retriever_tool import VectorSearchRetrieverTool # Expose all integrations to users under databricks-langchain __all__ = [ "VectorSearchRetrieverTool", + "set_uc_function_client", + "DatabricksFunctionClient", + "UCFunctionToolkit", ] diff --git a/integrations/openai/src/databricks_openai/__init__.py b/integrations/openai/src/databricks_openai/__init__.py index c0cafa0a..1224284e 100644 --- a/integrations/openai/src/databricks_openai/__init__.py +++ b/integrations/openai/src/databricks_openai/__init__.py @@ -1,3 +1,17 @@ +""" +**Re-exported Unity Catalog Utilities** + +This module re-exports selected utilities from the Unity Catalog open source package. + +Available aliases: + +- :class:`databricks_openai.UCFunctionToolkit` +- :class:`databricks_openai.DatabricksFunctionClient` +- :func:`databricks_openai.set_uc_function_client` + +Refer to the Unity Catalog `documentation `_ for more information. +""" + from unitycatalog.ai.core.base import set_uc_function_client from unitycatalog.ai.core.databricks import DatabricksFunctionClient from unitycatalog.ai.openai.toolkit import UCFunctionToolkit