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.6.0.dev15.tar.gz (11.6 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.6.0.dev15.tar.gz.

File metadata

File hashes

Hashes for microsoft_agents_storage_cosmos-0.6.0.dev15.tar.gz
Algorithm Hash digest
SHA256 ef3a34ee30501a97bcec73892697b7688ceee842b8a0250303ca0c9b05ca61d3
MD5 849414cc0837cb1a84ee1d29286883b6
BLAKE2b-256 3fb979937e0f3e1befca3b40188b1c2b112ef201ce8ba4f870e27b294cc7f5cb

See more details on using hashes here.

File details

Details for the file microsoft_agents_storage_cosmos-0.6.0.dev15-py3-none-any.whl.

File metadata

File hashes

Hashes for microsoft_agents_storage_cosmos-0.6.0.dev15-py3-none-any.whl
Algorithm Hash digest
SHA256 463259937fe62ee342a87e07b7a2eb6824b42d7ffaeea91163b3ac587ef36450
MD5 f443e43c3fa3a0098a12482e9dabf2f6
BLAKE2b-256 321f001db05f7cbe65abd48201e9423150a336289d55006aa0bf4777dcea0bac

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