credential as param

Author: Kristian Nylund

text_2_sql/autogen/pyproject.toml

@@ -19,7 +19,6 @@ dependencies = [
     "nltk>=3.8.1",
     "cachetools>=5.5.1",
     "azure-cosmos>=4.9.0",
-    "azure-identity>=1.19.0",
 ]
 
 [dependency-groups]

text_2_sql/autogen/src/autogen_text_2_sql/state_store.py

@@ -1,7 +1,7 @@
 from abc import ABC, abstractmethod
 from cachetools import TTLCache
 from azure.cosmos import CosmosClient, exceptions
-from azure.identity import DefaultAzureCredential
+
 
 class StateStore(ABC):
     @abstractmethod
@@ -25,10 +25,10 @@ def save_state(self, thread_id: str, state: dict) -> None:
 
 
 class CosmosStateStore(StateStore):
-    def __init__(self, endpoint, database, container, partition_key = None):
+    def __init__(self, endpoint, database, container, credential, partition_key = None):
         client = CosmosClient(
             url=endpoint,
-            credential=DefaultAzureCredential(),
+            credential=credential
         )
         database_client = client.get_database_client(database)
         self._db = database_client.get_container_client(container)