microsoft-agents-storage-cosmos 0.6.1__py3-none-any.whl → 0.7.0__py3-none-any.whl

microsoft_agents/storage/cosmos/cosmos_db_storage.py

@@ -61,7 +61,7 @@ class CosmosDBStorage(AsyncStorageBase):
                     )
                 )
             return CosmosClient(
-                account_url=self._config.url, credential=self._config.credential
+                url=self._config.url, credential=self._config.credential
             )
 
         connection_policy = self._config.cosmos_client_options.get(
@@ -139,32 +139,30 @@ class CosmosDBStorage(AsyncStorageBase):
             "kind": documents.PartitionKind.Hash,
         }
         try:
+            kwargs = {}
+            if self._config.container_throughput:
+                kwargs["offer_throughput"] = self._config.container_throughput
             self._container = await self._database.create_container(
-                self._config.container_id,
-                partition_key,
-                offer_throughput=self._config.container_throughput,
+                self._config.container_id, partition_key, **kwargs
             )
-        except cosmos_exceptions.CosmosHttpResponseError as err:
-            if err.status_code == http_constants.StatusCodes.CONFLICT:
-                self._container = self._database.get_container_client(
-                    self._config.container_id
-                )
-                properties = await self._container.read()
-                # if "partitionKey" not in properties:
-                #     self._compatability_mode_partition_key = True
-                # else:
-                # containers created had no partition key, so the default was "/_partitionKey"
-                paths = properties["partitionKey"]["paths"]
-                if "/_partitionKey" in paths:
-                    self._compatability_mode_partition_key = True
-                elif "/id" not in paths:
-                    raise Exception(
-                        storage_errors.InvalidConfiguration.format(
-                            f"Custom Partition Key Paths are not supported. {self._config.container_id} has a custom Partition Key Path of {paths[0]}."
-                        )
+        except Exception as err:
+            self._container = self._database.get_container_client(
+                self._config.container_id
+            )
+            properties = await self._container.read()
+            # if "partitionKey" not in properties:
+            #     self._compatability_mode_partition_key = True
+            # else:
+            # containers created had no partition key, so the default was "/_partitionKey"
+            paths = properties["partitionKey"]["paths"]
+            if "/_partitionKey" in paths:
+                self._compatability_mode_partition_key = True
+            elif "/id" not in paths:
+                raise Exception(
+                    storage_errors.InvalidConfiguration.format(
+                        f"Custom Partition Key Paths are not supported. {self._config.container_id} has a custom Partition Key Path of {paths[0]}."
                     )
-            else:
-                raise err
+                )
 
     async def initialize(self) -> None:
         if not self._container:
@@ -182,3 +180,6 @@ class CosmosDBStorage(AsyncStorageBase):
 
     def _get_partition_key(self, key: str):
         return NonePartitionKeyValue if self._compatability_mode_partition_key else key
+
+    async def _close(self) -> None:
+        await self._client.close()

microsoft_agents/storage/cosmos/cosmos_db_storage_config.py

@@ -1,7 +1,7 @@
 import json
 from typing import Union
 
-from azure.core.credentials import TokenCredential
+from azure.core.credentials_async import AsyncTokenCredential
 from microsoft_agents.storage.cosmos.errors import storage_errors
 
 from .key_ops import sanitize_key
@@ -17,11 +17,11 @@ class CosmosDBStorageConfig:
         database_id: str = "",
         container_id: str = "",
         cosmos_client_options: dict = None,
-        container_throughput: int = 0,
+        container_throughput: int | None = None,
         key_suffix: str = "",
         compatibility_mode: bool = False,
         url: str = "",
-        credential: Union[TokenCredential, None] = None,
+        credential: Union[AsyncTokenCredential, None] = None,
         **kwargs,
     ):
         """Create the Config object.
@@ -55,14 +55,14 @@ class CosmosDBStorageConfig:
             "cosmos_client_options", {}
         )
         self.container_throughput: int = container_throughput or kwargs.get(
-            "container_throughput", 400
+            "container_throughput"
         )
         self.key_suffix: str = key_suffix or kwargs.get("key_suffix", "")
         self.compatibility_mode: bool = compatibility_mode or kwargs.get(
             "compatibility_mode", False
         )
         self.url = url or kwargs.get("url", "")
-        self.credential: Union[TokenCredential, None] = credential
+        self.credential: Union[AsyncTokenCredential, None] = credential
 
     @staticmethod
     def validate_cosmos_db_config(config: "CosmosDBStorageConfig") -> None:
@@ -71,10 +71,6 @@ class CosmosDBStorageConfig:
         This is used prior to the creation of the CosmosDBStorage object."""
         if not config:
             raise ValueError(str(storage_errors.CosmosDbConfigRequired))
-        if not config.cosmos_db_endpoint:
-            raise ValueError(str(storage_errors.CosmosDbEndpointRequired))
-        if not config.auth_key:
-            raise ValueError(str(storage_errors.CosmosDbAuthKeyRequired))
         if not config.database_id:
             raise ValueError(str(storage_errors.CosmosDbDatabaseIdRequired))
         if not config.container_id:

{microsoft_agents_storage_cosmos-0.6.1.dist-info → microsoft_agents_storage_cosmos-0.7.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: microsoft-agents-storage-cosmos
-Version: 0.6.1
+Version: 0.7.0
 Summary: A Cosmos DB storage library for Microsoft Agents
 Author: Microsoft Corporation
 License-Expression: MIT
@@ -15,9 +15,10 @@ Classifier: Operating System :: OS Independent
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: microsoft-agents-hosting-core==0.6.1
+Requires-Dist: microsoft-agents-hosting-core==0.7.0
 Requires-Dist: azure-core
 Requires-Dist: azure-cosmos
+Requires-Dist: azure-identity
 Dynamic: license-file
 Dynamic: requires-dist
 
@@ -39,11 +40,29 @@ This library is part of the **Microsoft 365 Agents SDK for Python** - a comprehe
     <th style="width:20%">Date</th>
     <th style="width:60%">Release Notes</th>
   </tr>
+  <tr>
+    <td>0.6.1</td>
+    <td>2025-12-01</td>
+    <td>
+      <a href="https://github.com/microsoft/Agents-for-python/blob/main/changelog.md#microsoft-365-agents-sdk-for-python---release-notes-v061">
+        0.6.1 Release Notes
+      </a>
+    </td>
+  </tr>
+  <tr>
+    <td>0.6.0</td>
+    <td>2025-11-18</td>
+    <td>
+      <a href="https://github.com/microsoft/Agents-for-python/blob/main/changelog.md#microsoft-365-agents-sdk-for-python---release-notes-v060">
+        0.6.0 Release Notes
+      </a>
+    </td>
+  </tr>
   <tr>
     <td>0.5.0</td>
     <td>2025-10-22</td>
     <td>
-      <a href="https://github.com/microsoft/Agents-for-python/blob/main/changelog.md">
+      <a href="https://github.com/microsoft/Agents-for-python/blob/main/changelog.md#microsoft-365-agents-sdk-for-python---release-notes-v050">
         0.5.0 Release Notes
       </a>
     </td>

{microsoft_agents_storage_cosmos-0.6.1.dist-info → microsoft_agents_storage_cosmos-0.7.0.dist-info}/RECORD

@@ -1,11 +1,11 @@
 microsoft_agents/storage/cosmos/__init__.py,sha256=2JVNhbohUB-Pm8LmgagHQH3vlEgljpUzbfPhdOS-rCQ,174
-microsoft_agents/storage/cosmos/cosmos_db_storage.py,sha256=TLvvVKmRqKWrXsy9ddOjXPbPjPvLFXW6oB3KE9cuIC4,6753
-microsoft_agents/storage/cosmos/cosmos_db_storage_config.py,sha256=ji--EihsTE9FiGyUpeaw0or1iMN8-Z6O-gUd0fMcfx0,4210
+microsoft_agents/storage/cosmos/cosmos_db_storage.py,sha256=7AivID9eWk2XB_Ar7J3__T7fETqNiGqqUfdL6xWjo0w,6680
+microsoft_agents/storage/cosmos/cosmos_db_storage_config.py,sha256=Qyt01_zXWLL6Ho8WBmWep9C0nqed8LrRC0YjUBC5PR8,4013
 microsoft_agents/storage/cosmos/key_ops.py,sha256=2wnSdSpKzwoCvw4giNn68pBDOa1Gc1jyvEhk2tc_gPQ,1747
 microsoft_agents/storage/cosmos/errors/__init__.py,sha256=3J6VOVCyB2NxCrodoLeVxLAtHMmsrZKKbz4LfUHfk-0,409
 microsoft_agents/storage/cosmos/errors/error_resources.py,sha256=7mXt0Dj1LOGt2fg9h2iV5ZGPu480DMeFEGSN2uue4A0,2386
-microsoft_agents_storage_cosmos-0.6.1.dist-info/licenses/LICENSE,sha256=ws_MuBL-SCEBqPBFl9_FqZkaaydIJmxHrJG2parhU4M,1141
-microsoft_agents_storage_cosmos-0.6.1.dist-info/METADATA,sha256=VVaKPqkNQBbEdDhz8YIvRuiGGLkGQeThuZu8zmnZbQA,7774
-microsoft_agents_storage_cosmos-0.6.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-microsoft_agents_storage_cosmos-0.6.1.dist-info/top_level.txt,sha256=lWKcT4v6fTA_NgsuHdNvuMjSrkiBMXohn64ApY7Xi8A,17
-microsoft_agents_storage_cosmos-0.6.1.dist-info/RECORD,,
+microsoft_agents_storage_cosmos-0.7.0.dist-info/licenses/LICENSE,sha256=ws_MuBL-SCEBqPBFl9_FqZkaaydIJmxHrJG2parhU4M,1141
+microsoft_agents_storage_cosmos-0.7.0.dist-info/METADATA,sha256=xll37oMmS4tEbhFS7FqCBEpC3K_WTn1ph3zNZ-EQQL0,8381
+microsoft_agents_storage_cosmos-0.7.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+microsoft_agents_storage_cosmos-0.7.0.dist-info/top_level.txt,sha256=lWKcT4v6fTA_NgsuHdNvuMjSrkiBMXohn64ApY7Xi8A,17
+microsoft_agents_storage_cosmos-0.7.0.dist-info/RECORD,,