Azure Databricks连接Azure Cognitive Search超时问题求助

问题现象 #

执行create_index('test22_index_db', config)时触发超时错误：

ServiceRequestError: (<urllib3.connection.HTTPSConnection object at 0x7f081820e1a0>, 'Connection to xxxxx-aisearch-02.search.windows.net timed out. (connect timeout=300)')

已确认信息：

• 配置中传入了正确的搜索URL、密钥及其他参数
• Cognitive Search使用专用终结点部署
• Databricks与Document Intelligence、OpenAI、存储账户、密钥保管库等服务连接正常
• 个人订阅环境可复现相同错误

排查步骤 #

1. 验证专用终结点网络连通性 #

登录Databricks集群所在VNet内的虚拟机/容器，用以下命令测试Cognitive Search专用终结点的443端口连通性：

nc -zv xxxxx-aisearch-02.search.windows.net 443

若连通失败，检查：

• Cognitive Search的专用终结点是否已关联到Databricks集群所在VNet
• VNet的网络安全组(NSG)是否允许出站443端口到该专用终结点的IP段
• 路由表是否存在阻止流量的自定义规则

2. 检查Cognitive Search的网络配置 #

在Azure门户进入Cognitive Search资源→网络→专用终结点连接，确认：

• 专用终结点状态为「已批准」
• 未误将防火墙规则设置为仅允许特定公网IP访问（专用终结点场景需确保允许专用终结点流量）

3. 确认SDK版本与参数合法性 #

• 检查azure-search-documents SDK版本，旧版本可能存在专用终结点兼容性问题：

  pip show azure-search-documents
  

  建议升级至最新稳定版
• 验证config中的参数：
  • endpoint必须是完整的专用终结点URL（格式：https://xxxxx-aisearch-02.search.windows.net）
  • 使用的密钥为管理员密钥（查询密钥无创建索引权限）

4. 调整超时与重试策略 #

临时调高超时时间并添加重试策略，排查是否为偶发性网络波动：

from azure.core.credentials import AzureKeyCredential
from azure.search.documents import SearchIndexClient
from azure.core.pipeline.policies import RetryPolicy

# 配置重试策略与更长超时
retry_policy = RetryPolicy(total_retries=3, retry_backoff_factor=2)
index_client = SearchIndexClient(
    endpoint=config["endpoint"],
    credential=AzureKeyCredential(config["key"]),
    retry_policy=retry_policy,
    connection_timeout=600  # 调整为600秒
)
index_client.create_index(你的索引定义)

5. 检查Databricks集群网络配置 #

• 确认集群已启用VNet注入，且所在子网与Cognitive Search专用终结点子网处于同一VNet或已通过VNet对等互联连通
• 排查集群出站网络规则，是否存在限制Azure Cognitive Search服务流量的策略

内容的提问来源于stack exchange，提问作者mac