Skip to content

Comments

Adding Cosmos DB support for storage#68

Merged
axelsrz merged 14 commits intomainfrom
users/robrandao/cosmos-db-storage
Jul 24, 2025
Merged

Adding Cosmos DB support for storage#68
axelsrz merged 14 commits intomainfrom
users/robrandao/cosmos-db-storage

Conversation

@rodrigobr-msft
Copy link
Contributor

This pull request introduces a new CosmosDB storage library (microsoft-agents-cosmos) and includes updates to existing storage libraries and testing. The most significant changes involve the addition of the new CosmosDB library and modifications to test structures.

New CosmosDB Storage Library:

  • Added a new library microsoft-agents-cosmos, including core files:
  • Exposed CosmosDBStorage and CosmosDBStorageConfig in the module’s __init__.py.
  • Added pyproject.toml for packaging and dependencies, including azure-cosmos and microsoft.agents.storage.

Updates to CI/CD Pipelines:

  • Updated .azdo/ci-pr.yaml and .github/workflows/python-package.yml to include installation of the new microsoft_agents_cosmos package during CI builds. [1] [2]

Codebase Simplifications:

  • Refactored imports in blob_storage.py to group related imports together.
  • Removed the unused BlobStorageMock class and related references in test_blob_storage.py. [1] [2]

Enhancements to Tests:

  • Added a new test test_external_change_is_visible to verify that external changes to blob storage are correctly reflected.
  • Removed redundant teardown comments in test_blob_storage.py for improved clarity.

@rodrigobr-msft rodrigobr-msft requested a review from axelsrz July 21, 2025 20:21
@hexi-bumblebee
Copy link

Hi Axel, once this is merged, how long do we expect the libraries changes to be published to test pipy? I saw the last published release was May 19th 2025. I'm interested in picking up this cosmoDB change, as well as the streaming response change.

@axelsrz axelsrz merged commit 1f2b1ed into main Jul 24, 2025
5 checks passed
@axelsrz axelsrz deleted the users/robrandao/cosmos-db-storage branch July 24, 2025 16:35
kylerohnmsft pushed a commit that referenced this pull request Jul 24, 2025
* Adding Cosmos DB storage

* Removed unnecessary file

* Added quick storage testing functionality for debugging and improved documentation

* Simplified storage test setup and added unit test

* Fixed legacy container support and added relevant tests

* Revised AsyncStorageBase export/imports

* Fixed final unit tests and refined documentation

* Added microsoft-agent-cosmos to CI files

* Minor fixes to storage tests

* Moved files around

* Fixed imports and formatted with black

* Made key operation tests more data driven

* Fixed minor issues in unit tests
@rodrigobr-msft rodrigobr-msft linked an issue Jul 29, 2025 that may be closed by this pull request
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Cosmos Storage integration package

3 participants