revoke deletion of _functions.py

sdk/ai/azure-ai-generative/azure/ai/generative/index/__init__.py

@@ -6,7 +6,17 @@
 __path__ = __import__("pkgutil").extend_path(__path__, __name__)
 
 from ._build_mlindex import build_index
+from ._functions import (
+    get_langchain_embeddings_from_index,
+    get_langchain_retriever_from_index,
+    get_langchain_vectorstore_from_index,
+    get_native_index_client_from_index
+)
 
 __all__ = [
     "build_index",
+    "get_langchain_embeddings_from_index",
+    "get_langchain_retriever_from_index",
+    "get_langchain_vectorstore_from_index",
+    "get_native_index_client_from_index"
 ]

sdk/ai/azure-ai-generative/azure/ai/generative/index/_functions.py

@@ -0,0 +1,39 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+
+import os
+from pathlib import Path
+from typing import Optional, Union
+from azure.core.credentials import TokenCredential
+from azure.ai.resources.entities import Index
+
+from ._mlindex import MLIndex as DataplaneMLIndex
+
+def get_langchain_embeddings_from_index(
+    path: Optional[Union[str, Path]],
+    credential: Optional[TokenCredential] = None,
+):
+    return DataplaneMLIndex(path).get_langchain_embeddings(credential=credential)
+
+
+def get_langchain_vectorstore_from_index(
+    path: Optional[Union[str, Path]],
+    credential: Optional[TokenCredential] = None,
+):
+    return DataplaneMLIndex(path).as_langchain_vectorstore(credential=credential)
+
+
+def get_langchain_retriever_from_index(
+    path: Optional[Union[str, Path]],
+    credential: Optional[TokenCredential] = None,
+    **kwargs,
+):
+    return DataplaneMLIndex(path).as_langchain_retriever(credential=credential, **kwargs)
+
+
+def get_native_index_client_from_index(
+    path: Optional[Union[str, Path]],
+    credential: Optional[TokenCredential] = None,
+):
+    return DataplaneMLIndex(path).as_native_index_client(credential=credential)