如何修复Azure Cosmos DB的KeyError问题？附代码及报错信息

错误原因 #

你代码里的os.environ["https://davidjohns.documents.azure.com/"]用法完全错误。os.environ[key]是用来读取系统中名为key的环境变量，而你直接把Cosmos DB的endpoint URL当成了环境变量的名称，系统里根本不存在这个名称的环境变量，所以触发了KeyError。

解决步骤 #

1. 设置对应名称的环境变量

   • 临时设置（仅当前终端会话有效）：
     • macOS/Linux终端：export AZURE_COSMOS_ENDPOINT="https://davidjohns.documents.azure.com/"
     • Windows命令提示符：set AZURE_COSMOS_ENDPOINT=https://davidjohns.documents.azure.com/
   • 持久化设置（推荐）：可以把该环境变量添加到你的shell配置文件（如_/.bashrc、/.zshrc），或者用.env文件配合python-dotenv库加载（需先执行pip install python-dotenv安装）。

2. 修改代码读取正确的环境变量
   将读取endpoint的代码行，改为读取你设置的环境变量名称：

import os
from azure.cosmos import CosmosClient
from azure.identity import DefaultAzureCredential

# 读取名为AZURE_COSMOS_ENDPOINT的环境变量
endpoint = os.environ["AZURE_COSMOS_ENDPOINT"]
DATABASE_NAME = "cosmicworks"
CONTAINER_NAME = "products"
credential = DefaultAzureCredential()
client = CosmosClient(url=endpoint, credential=credential)

额外提示 #

如果使用DefaultAzureCredential，请确保你的本地开发环境已配置Azure AD身份验证（比如通过az login登录Azure CLI），或者你的服务主体/托管身份拥有访问该Cosmos DB账户的权限。

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