cwyodmodules 0.3.31__py3-none-any.whl → 0.3.32__py3-none-any.whl

cwyodmodules/batch/utilities/helpers/azure_identity_helper.py

@@ -1,8 +1,4 @@
-import os
 from azure.identity import (
-    ChainedTokenCredential,
-    ManagedIdentityCredential,
-    EnvironmentCredential,
     TokenCachePersistenceOptions,
     get_bearer_token_provider,
     DefaultAzureCredential
@@ -14,6 +10,7 @@ from opentelemetry.propagate import extract
 
 logger = getLogger("__main__")
 tracer = trace.get_tracer("__main__")
+
 class AzureIdentityHelper:
     """
     A helper class to provide a chained Azure token credential.
@@ -21,40 +18,12 @@ class AzureIdentityHelper:
     Token caching is configured for in-memory persistence.
     """
     def __init__(self):
-        # Configure in-memory token cache persistence
-        # For in-memory, unencrypted storage is typically allowed for simplicity during development.
-        # In production, especially with shared environments, consider the security implications.
-        client_secret_available =  os.getenv("AZURE_CLIENT_SECRET") is not None
         
-        token_cache_options = TokenCachePersistenceOptions(allow_unencrypted_storage=True)
+        token_cache_options = TokenCachePersistenceOptions(allow_unencrypted_storage=True)        
 
-        # Create individual credential instances
-        managed_identity_credential = ManagedIdentityCredential(
-            token_cache_persistence_options=token_cache_options
-        )
-        environment_credential = EnvironmentCredential(
+        self._credential = DefaultAzureCredential(
             token_cache_persistence_options=token_cache_options
         )
-        
-
-        # Create a chain of credentials
-        # The chain will try credentials in the order they are provided.
-        if client_secret_available:
-            logger.info("Using Environment Credential first with token cache persistence.")
-            self._credential = ChainedTokenCredential(
-                environment_credential,
-                managed_identity_credential
-            )
-        else:
-            logger.info("Using Managed Identity Credential first with token cache persistence.")
-            # self._credential = ChainedTokenCredential(
-
-            #     managed_identity_credential,
-            #     environment_credential
-            # )
-            self._credential = DefaultAzureCredential(
-                token_cache_persistence_options=token_cache_options
-            )
 
     def get_credential(self):
         """

{cwyodmodules-0.3.31.dist-info → cwyodmodules-0.3.32.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cwyodmodules
-Version: 0.3.31
+Version: 0.3.32
 Summary: Add your description here
 Author-email: Patrik <patrikhartl@gmail.com>
 Classifier: Operating System :: OS Independent

{cwyodmodules-0.3.31.dist-info → cwyodmodules-0.3.32.dist-info}/RECORD

@@ -30,7 +30,7 @@ cwyodmodules/batch/utilities/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeu
 cwyodmodules/batch/utilities/helpers/azure_blob_storage_client.py,sha256=FN2XnEThmtbWnwSi1sEfgekuPH7aJBOAU5n2DBmQ9ww,10315
 cwyodmodules/batch/utilities/helpers/azure_computer_vision_client.py,sha256=5qhxriTEaKkRc8K4E9oKQjCOzV_JSkad37z1FskCpOg,3661
 cwyodmodules/batch/utilities/helpers/azure_form_recognizer_helper.py,sha256=kN9pjanKj_IsGM0XQA8QwICuLaB0yBdtRMINmhSD_mU,6699
-cwyodmodules/batch/utilities/helpers/azure_identity_helper.py,sha256=lq10tDgH4kMKnmj5cAUj97-P_RIdav-Mbac93WZVE8U,3264
+cwyodmodules/batch/utilities/helpers/azure_identity_helper.py,sha256=aiYZH4F9aZBd5DDRuVvIn6jiIDgC0i-R-1-NqHqosnI,1806
 cwyodmodules/batch/utilities/helpers/azure_postgres_helper.py,sha256=RV2jcqFuTofbz1gk8CDYdECPgl_ypH3K84Yeu3kjU7o,10823
 cwyodmodules/batch/utilities/helpers/azure_postgres_helper_light_rag.py,sha256=DLgJ7KNK71qSUlRHTWgRgp6-4ufnC8frL_pOWE-oKgM,10915
 cwyodmodules/batch/utilities/helpers/azure_search_helper.py,sha256=GniJ8JhRTBdzI1kN0yHtyR8ocZBav5C635uFfD1eHrc,10473
@@ -108,8 +108,8 @@ cwyodmodules/graphrag/query/generate.py,sha256=xBnZs2U9xFWtPk4AfAZgYKbHdcxNcIO6Q
 cwyodmodules/graphrag/query/graph_search.py,sha256=95h3ecSWx4864XgKABtG0fh3Nk8HkqJVzoCrO8daJ-Y,7724
 cwyodmodules/graphrag/query/types.py,sha256=1Iq1dp4I4a56_cuFjOZ0NTgd0A2_MpVFznp_czgt6cI,617
 cwyodmodules/graphrag/query/vector_search.py,sha256=9Gwu9LPjtoAYUU8WKqCvbCHAIg3dpk71reoYd1scLnQ,1807
-cwyodmodules-0.3.31.dist-info/licenses/LICENSE,sha256=UqBDTipijsSW2ZSOXyTZnMsXmLoEHTgNEM0tL4g-Sso,1150
-cwyodmodules-0.3.31.dist-info/METADATA,sha256=8klc8o4pwRr-E20ts80y9Q-2dbUzQGm4KdyXHh8RJ_k,1969
-cwyodmodules-0.3.31.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-cwyodmodules-0.3.31.dist-info/top_level.txt,sha256=99RENLbkdRX-qpJvsxZ5AfmTL5s6shSaKOWYpz1vwzg,13
-cwyodmodules-0.3.31.dist-info/RECORD,,
+cwyodmodules-0.3.32.dist-info/licenses/LICENSE,sha256=UqBDTipijsSW2ZSOXyTZnMsXmLoEHTgNEM0tL4g-Sso,1150
+cwyodmodules-0.3.32.dist-info/METADATA,sha256=3q9b9xT2oTfuaNwVlF9KFffml5vpTTBDoDF4MWDLVvE,1969
+cwyodmodules-0.3.32.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+cwyodmodules-0.3.32.dist-info/top_level.txt,sha256=99RENLbkdRX-qpJvsxZ5AfmTL5s6shSaKOWYpz1vwzg,13
+cwyodmodules-0.3.32.dist-info/RECORD,,