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

microsoft_agents/storage/cosmos/cosmos_db_storage.py

@@ -61,7 +61,7 @@ class CosmosDBStorage(AsyncStorageBase):
                     )
                 )
             return CosmosClient(
-                url=self._config.url, credential=self._config.credential
+                account_url=self._config.url, credential=self._config.credential
             )
 
         connection_policy = self._config.cosmos_client_options.get(
@@ -139,30 +139,32 @@ 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, **kwargs
+                self._config.container_id,
+                partition_key,
+                offer_throughput=self._config.container_throughput,
             )
-        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]}."
-                    )
+        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]}."
+                        )
+                    )
+            else:
+                raise err
 
     async def initialize(self) -> None:
         if not self._container:
@@ -180,6 +182,3 @@ 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_async import AsyncTokenCredential
+from azure.core.credentials import TokenCredential
 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 | None = None,
+        container_throughput: int = 0,
         key_suffix: str = "",
         compatibility_mode: bool = False,
         url: str = "",
-        credential: Union[AsyncTokenCredential, None] = None,
+        credential: Union[TokenCredential, 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"
+            "container_throughput", 400
         )
         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[AsyncTokenCredential, None] = credential
+        self.credential: Union[TokenCredential, None] = credential
 
     @staticmethod
     def validate_cosmos_db_config(config: "CosmosDBStorageConfig") -> None:
@@ -71,6 +71,10 @@ 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.7.0.dist-info → microsoft_agents_storage_cosmos-0.7.0.dev1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: microsoft-agents-storage-cosmos
-Version: 0.7.0
+Version: 0.7.0.dev1
 Summary: A Cosmos DB storage library for Microsoft Agents
 Author: Microsoft Corporation
 License-Expression: MIT
@@ -15,10 +15,9 @@ Classifier: Operating System :: OS Independent
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: microsoft-agents-hosting-core==0.7.0
+Requires-Dist: microsoft-agents-hosting-core==0.7.0.dev1
 Requires-Dist: azure-core
 Requires-Dist: azure-cosmos
-Requires-Dist: azure-identity
 Dynamic: license-file
 Dynamic: requires-dist
 
@@ -40,29 +39,11 @@ 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#microsoft-365-agents-sdk-for-python---release-notes-v050">
+      <a href="https://github.com/microsoft/Agents-for-python/blob/main/changelog.md">
         0.5.0 Release Notes
       </a>
     </td>

microsoft_agents_storage_cosmos-0.7.0.dev1.dist-info/RECORD

@@ -0,0 +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/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.7.0.dev1.dist-info/licenses/LICENSE,sha256=ws_MuBL-SCEBqPBFl9_FqZkaaydIJmxHrJG2parhU4M,1141
+microsoft_agents_storage_cosmos-0.7.0.dev1.dist-info/METADATA,sha256=0TxCCAgfY-Ndmc2NunPYVjwcinKSDn4U27ilPkhBaL8,7784
+microsoft_agents_storage_cosmos-0.7.0.dev1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+microsoft_agents_storage_cosmos-0.7.0.dev1.dist-info/top_level.txt,sha256=lWKcT4v6fTA_NgsuHdNvuMjSrkiBMXohn64ApY7Xi8A,17
+microsoft_agents_storage_cosmos-0.7.0.dev1.dist-info/RECORD,,

microsoft_agents_storage_cosmos-0.7.0.dist-info/RECORD

@@ -1,11 +0,0 @@
-microsoft_agents/storage/cosmos/__init__.py,sha256=2JVNhbohUB-Pm8LmgagHQH3vlEgljpUzbfPhdOS-rCQ,174
-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.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,,