Azure Cosmos DB SDK Acceptance Criteria

SDK: azure-cosmosRepository: https://github.com/Azure/azure-sdk-for-pythonCommit: 5c5d6eb014da472e71937aceddbf2f19fdb9aa40Purpose: Skill testing acceptance criteria for validating generated code correctness

---

1. Correct Import Patterns

1.1 Client Imports

✅ CORRECT: Sync Client

from azure.cosmos import CosmosClient, PartitionKey
from azure.identity import DefaultAzureCredential

✅ CORRECT: Async Client

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

✅ CORRECT: Exceptions (Optional)

from azure.cosmos import exceptions

1.2 Anti-Patterns (ERRORS)

❌ INCORRECT: Wrong module for identity

# WRONG - DefaultAzureCredential is in azure.identity
from azure.cosmos import DefaultAzureCredential

❌ INCORRECT: Wrong module for PartitionKey

# WRONG - PartitionKey is not in azure.cosmos.aio
from azure.cosmos.aio import PartitionKey

---

2. Authentication Patterns

2.1 ✅ CORRECT: DefaultAzureCredential (AAD)

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

credential = DefaultAzureCredential()
client = CosmosClient(
    url=os.environ["ACCOUNT_URI"],
    credential=credential,
)

2.2 ✅ CORRECT: Connection String

import os
from azure.cosmos import CosmosClient

client = CosmosClient.from_connection_string(
    conn_str=os.environ["AZURE_COSMOS_CONNECTIONSTRING"],
)

2.3 Anti-Patterns (ERRORS)

❌ INCORRECT: Passing connection string as URL

# WRONG - conn string is not the account URL
client = CosmosClient(
    url=os.environ["AZURE_COSMOS_CONNECTIONSTRING"],
    credential=os.environ["ACCOUNT_KEY"],
)

❌ INCORRECT: Hardcoded credentials

# WRONG - do not hardcode secrets
client = CosmosClient(
    url="https://my-account.documents.azure.com:443/",