Make azure-core optional

langchain/utilities/powerbi.py

@@ -4,13 +4,11 @@ from __future__ import annotations
 
 import logging
 import os
-from typing import Any, Dict, Iterable, List, Optional, Union
+from typing import TYPE_CHECKING, Any, Dict, Iterable, List, Optional, Union
 
 import aiohttp
 import requests
 from aiohttp import ServerTimeoutError
-from azure.core.credentials import TokenCredential
-from azure.core.exceptions import ClientAuthenticationError
 from pydantic import BaseModel, Field, root_validator
 from requests.exceptions import Timeout
 
@@ -18,6 +16,10 @@ from langchain.tools.powerbi.prompt import SCHEMA_ERROR_RESPONSE, UNAUTHORIZED_R
 
 _LOGGER = logging.getLogger(__name__)
 
+if TYPE_CHECKING:
+    from azure.core.credentials import TokenCredential
+    from azure.core.exceptions import ClientAuthenticationError
+
 BASE_URL = os.getenv("POWERBI_BASE_URL", "https://api.powerbi.com/v1.0/myorg/datasets/")
 
 

pyproject.toml

@@ -17,7 +17,7 @@ SQLAlchemy = ">1.3,<3"
 requests = "^2"
 PyYAML = ">=5.4.1"
 numpy = "^1"
-azure-core = {version = "^1.26.4"}
+azure-core = {version = "^1.26.4", optional = true}
 tqdm = {version = ">=4.48.0", optional = true}
 openapi-schema-pydantic = "^1.2"
 faiss-cpu = {version = "^1", optional = true}