Skip to main content

A Cosmos DB storage library for Microsoft Agents

Project description

Microsoft Agents Storage - Cosmos DB

PyPI version

Azure Cosmos DB storage integration for Microsoft 365 Agents SDK. This library provides enterprise-grade persistent storage for conversation state, user data, and custom agent information using Azure Cosmos DB's globally distributed, multi-model database service.

This library implements the storage interface for the Microsoft 365 Agents SDK using Azure Cosmos DB as the backend. It provides automatic partitioning, global distribution, and low-latency access to your agent data. Perfect for production deployments requiring high availability, scalability, and multi-region support.

What is this?

This library is part of the Microsoft 365 Agents SDK for Python - a comprehensive framework for building enterprise-grade conversational AI agents. The SDK enables developers to create intelligent agents that work across multiple platforms including Microsoft Teams, M365 Copilot, Copilot Studio, and web chat, with support for third-party integrations like Slack, Facebook Messenger, and Twilio.

Release Notes

Version Date Release Notes
0.5.0 2025-10-22 0.5.0 Release Notes

Packages Overview

We offer the following PyPI packages to create conversational experiences based on Agents:

Package Name PyPI Version Description
microsoft-agents-activity PyPI Types and validators implementing the Activity protocol spec.
microsoft-agents-hosting-core PyPI Core library for Microsoft Agents hosting.
microsoft-agents-hosting-aiohttp PyPI Configures aiohttp to run the Agent.
microsoft-agents-hosting-teams PyPI Provides classes to host an Agent for Teams.
microsoft-agents-storage-blob PyPI Extension to use Azure Blob as storage.
microsoft-agents-storage-cosmos PyPI Extension to use CosmosDB as storage.
microsoft-agents-authentication-msal PyPI MSAL-based authentication for Microsoft Agents.

Additionally we provide a Copilot Studio Client, to interact with Agents created in CopilotStudio:

Package Name PyPI Version Description
microsoft-agents-copilotstudio-client PyPI Direct to Engine client to interact with Agents created in CopilotStudio

Why Cosmos DB?

  • 🌍 Global distribution with multi-region writes
  • ⚡ Single-digit millisecond latency
  • 📈 Automatic and instant scalability
  • 🔄 Multiple consistency models
  • 💪 99.999% availability SLA

Installation

pip install microsoft-agents-storage-cosmos

Environment Setup

Local Development with Cosmos DB Emulator

Install and run the Azure Cosmos DB Emulator for local testing:

Download: Azure Cosmos DB Emulator

Best Practices

  1. Use Managed Identity in Production - Avoid storing auth keys in code or environment variables
  2. Initialize Once - Call storage.initialize() during app startup, not per request
  3. Batch Operations - Read/write multiple items together when possible
  4. Monitor RU Consumption - Use Azure Monitor to track Request Units usage
  5. Set Appropriate Throughput - Start with 400 RU/s, scale up based on metrics
  6. Use Session Consistency - Default consistency level for most scenarios
  7. Implement Retry Logic - Handle transient failures with exponential backoff
  8. Partition Wisely - Current implementation uses /id partitioning (automatic)
  9. Enable Diagnostics - Configure Azure diagnostic logs for troubleshooting
  10. Test with Emulator - Use local emulator for development and testing

Key Classes Reference

  • CosmosDBStorage - Main storage implementation using Azure Cosmos DB
  • CosmosDBStorageConfig - Configuration settings for connection and behavior
  • StoreItem - Base class for data models (inherit to create custom types)

Quick Links

Sample Applications

Name Description README
Quickstart Simplest agent Quickstart
Auto Sign In Simple OAuth agent using Graph and GitHub auto-signin
OBO Authorization OBO flow to access a Copilot Studio Agent obo-authorization
Semantic Kernel Integration A weather agent built with Semantic Kernel semantic-kernel-multiturn
Streaming Agent Streams OpenAI responses azure-ai-streaming
Copilot Studio Client Console app to consume a Copilot Studio Agent copilotstudio-client
Cards Agent Agent that uses rich cards to enhance conversation design cards

Project details


Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

microsoft_agents_storage_cosmos-0.7.0.dev1.tar.gz (11.5 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

File details

Details for the file microsoft_agents_storage_cosmos-0.7.0.dev1.tar.gz.

File metadata

File hashes

Hashes for microsoft_agents_storage_cosmos-0.7.0.dev1.tar.gz
Algorithm Hash digest
SHA256 0e4b28d710170c5dc1b7a148652c4d5d2a680134eda815a3229a29199e294ab4
MD5 4f436467305fc89c4d62505fcc869dc3
BLAKE2b-256 ca35f2c550e652e994e209a776f228b4cc4a68b833dbb001ca358789eb34bf75

See more details on using hashes here.

File details

Details for the file microsoft_agents_storage_cosmos-0.7.0.dev1-py3-none-any.whl.

File metadata

File hashes

Hashes for microsoft_agents_storage_cosmos-0.7.0.dev1-py3-none-any.whl
Algorithm Hash digest
SHA256 23c15244a78f82804403ae089fd11062d6282c9d570bf86c43c975d0095ac7b8
MD5 76ae4ce4d7c0c98696bfa47f84509532
BLAKE2b-256 e3e2d076c6fcf701e5fa9a92fa79d140db16d8ace712ad454ea0855431a91d1c

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page