From 6cb29451640ebd06132b0125ca1e67f9440c14e2 Mon Sep 17 00:00:00 2001 From: Abdul Basit Anees Date: Thu, 6 Mar 2025 22:43:51 +0000 Subject: [PATCH 1/2] update cosmosdb vector store auth --- graphrag/vector_stores/cosmosdb.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/graphrag/vector_stores/cosmosdb.py b/graphrag/vector_stores/cosmosdb.py index c674ac4b52..1004453b35 100644 --- a/graphrag/vector_stores/cosmosdb.py +++ b/graphrag/vector_stores/cosmosdb.py @@ -39,9 +39,15 @@ def connect(self, **kwargs: Any) -> Any: if not url: msg = "Either connection_string or url must be provided." raise ValueError(msg) - self._cosmos_client = CosmosClient( - url=url, credential=DefaultAzureCredential() - ) + access_key = kwargs.get("api_key") + if access_key: + self._cosmos_client = CosmosClient( + url=url, credential=access_key + ) + else: + self._cosmos_client = CosmosClient( + url=url, credential=DefaultAzureCredential() + ) database_name = kwargs.get("database_name") if database_name is None: From e470ddcfbb9585ed29979ed6959c359b09829b91 Mon Sep 17 00:00:00 2001 From: Abdul Basit Anees Date: Thu, 6 Mar 2025 22:47:45 +0000 Subject: [PATCH 2/2] cosmosdb auth update-semversioner --- .semversioner/next-release/patch-20250306224613336336.json | 4 ++++ 1 file changed, 4 insertions(+) create mode 100644 .semversioner/next-release/patch-20250306224613336336.json diff --git a/.semversioner/next-release/patch-20250306224613336336.json b/.semversioner/next-release/patch-20250306224613336336.json new file mode 100644 index 0000000000..087d1d4aa3 --- /dev/null +++ b/.semversioner/next-release/patch-20250306224613336336.json @@ -0,0 +1,4 @@ +{ + "type": "patch", + "description": "add support for cosmosdb to use url+AccountKey based authentication" +}