cwyodmodules 0.3.42__py3-none-any.whl → 0.3.43__py3-none-any.whl

cwyodmodules/batch/utilities/chat_history/database_factory.py

@@ -3,14 +3,12 @@ from ..helpers.env_helper import EnvHelper
 from .cosmosdb import CosmosConversationClient
 from .postgresdbservice import PostgresConversationClient
 from ..helpers.config.database_type import DatabaseType
-from ..helpers.azure_identity_helper import AzureIdentityHelper
-
+from mgmt_config import identity
 
 class DatabaseFactory:
     @staticmethod
     def get_conversation_client():
         env_helper: EnvHelper = EnvHelper()
-        azure_identity_helper = AzureIdentityHelper()
 
         if env_helper.DATABASE_TYPE == DatabaseType.COSMOSDB.value:
             DatabaseFactory._validate_env_vars(
@@ -26,7 +24,7 @@ class DatabaseFactory:
                 f"https://{env_helper.AZURE_COSMOSDB_ACCOUNT}.documents.azure.com:443/"
             )
             credential = (
-                azure_identity_helper.get_credential()
+                identity.get_credential()
                 if not env_helper.AZURE_COSMOSDB_ACCOUNT_KEY
                 else env_helper.AZURE_COSMOSDB_ACCOUNT_KEY
             )

cwyodmodules/batch/utilities/chat_history/postgresdbservice.py

@@ -1,10 +1,8 @@
 import asyncpg
 from datetime import datetime, timezone
-from ..helpers.azure_identity_helper import AzureIdentityHelper
 from .database_client_base import DatabaseClientBase
-
 from ...utilities.helpers.env_helper import EnvHelper
-from mgmt_config import logger
+from mgmt_config import logger, identity
 env_helper: EnvHelper = EnvHelper()
 log_execution = env_helper.LOG_EXECUTION
 log_args = env_helper.LOG_ARGS
@@ -16,7 +14,6 @@ class PostgresConversationClient(DatabaseClientBase):
     def __init__(
         self, user: str, host: str, database: str, enable_message_feedback: bool = False
     ):
-        self.azure_identity_helper = AzureIdentityHelper()
         self.user = user
         self.host = host
         self.database = database
@@ -26,7 +23,7 @@ class PostgresConversationClient(DatabaseClientBase):
     @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
     async def connect(self):
         try:
-            access_information = self.azure_identity_helper.get_token(scopes="https://ossrdbms-aad.database.windows.net/.default")
+            access_information = identity.get_token(scopes="https://ossrdbms-aad.database.windows.net/.default")
             token = access_information.token
             self.conn = await asyncpg.connect(
                 user=self.user,

cwyodmodules/batch/utilities/helpers/azure_blob_storage_client.py

@@ -12,7 +12,7 @@ from azure.core.credentials import AzureNamedKeyCredential
 from azure.storage.queue import QueueClient, BinaryBase64EncodePolicy
 import chardet
 from .env_helper import EnvHelper
-from ..helpers.azure_identity_helper import AzureIdentityHelper
+from mgmt_config import identity
 
 
 def connection_string(account_name: str, account_key: str):
@@ -22,8 +22,7 @@ def connection_string(account_name: str, account_key: str):
 def create_queue_client():
     env_helper: EnvHelper = EnvHelper()
     if env_helper.AZURE_AUTH_TYPE == "rbac":
-        azure_identity_helper = AzureIdentityHelper()
-        credential = azure_identity_helper.get_credential()
+        credential = identity.get_credential()
         return QueueClient(
             account_url=f"https://{env_helper.AZURE_BLOB_ACCOUNT_NAME}.queue.core.windows.net/",
             queue_name=env_helper.DOCUMENT_PROCESSING_QUEUE_NAME,
@@ -49,7 +48,6 @@ class AzureBlobStorageClient:
         container_name: Optional[str] = None,
     ):
         env_helper: EnvHelper = EnvHelper()
-        self.azure_identity_helper = AzureIdentityHelper()
 
         self.auth_type = env_helper.AZURE_AUTH_TYPE
         self.account_name = account_name or env_helper.AZURE_BLOB_ACCOUNT_NAME
@@ -58,7 +56,7 @@ class AzureBlobStorageClient:
 
         if self.auth_type == "rbac":
             self.account_key = None
-            credential = self.azure_identity_helper.get_credential()
+            credential = identity.get_credential()
             self.blob_service_client = BlobServiceClient(
                 account_url=self.endpoint, credential=credential
             )

cwyodmodules/batch/utilities/helpers/azure_computer_vision_client.py

@@ -1,5 +1,5 @@
 from urllib.parse import urljoin
-from ..helpers.azure_identity_helper import AzureIdentityHelper
+from mgmt_config import identity
 
 import requests
 from requests import Response
@@ -20,7 +20,6 @@ class AzureComputerVisionClient:
     __RESPONSE_VECTOR_KEY = "vector"
 
     def __init__(self, env_helper: EnvHelper) -> None:
-        self.azure_identity_helper = AzureIdentityHelper()
         self.host = env_helper.AZURE_COMPUTER_VISION_ENDPOINT
         self.timeout = env_helper.AZURE_COMPUTER_VISION_TIMEOUT
         self.key = env_helper.AZURE_COMPUTER_VISION_KEY
@@ -61,10 +60,10 @@ class AzureComputerVisionClient:
             if self.use_keys:
                 headers["Ocp-Apim-Subscription-Key"] = self.key
             else:
-                access_information = self.azure_identity_helper.get_token_provider(
+                access_information = identity.get_token_provider(
                     scopes=self.__TOKEN_SCOPE
                 )
-                headers["Authorization"] = "Bearer " + access_information()
+                headers["Authorization"] = "Bearer " + access_information.get_token()
 
             return requests.post(
                 url=urljoin(self.host, path),

cwyodmodules/batch/utilities/helpers/azure_form_recognizer_helper.py

@@ -1,6 +1,6 @@
 from azure.core.credentials import AzureKeyCredential
 from azure.ai.formrecognizer import DocumentAnalysisClient
-from ..helpers.azure_identity_helper import AzureIdentityHelper
+from mgmt_config import identity
 import html
 import traceback
 from .env_helper import EnvHelper
@@ -20,8 +20,7 @@ class AzureFormRecognizerClient:
             env_helper.AZURE_FORM_RECOGNIZER_ENDPOINT
         )
         if env_helper.AZURE_AUTH_TYPE == "rbac":
-            self.azure_identity_helper = AzureIdentityHelper()
-            credential = self.azure_identity_helper.get_credential()
+            credential = identity.get_credential()
             self.document_analysis_client = DocumentAnalysisClient(
                 endpoint=self.AZURE_FORM_RECOGNIZER_ENDPOINT,
                 credential=credential,

cwyodmodules/batch/utilities/helpers/azure_postgres_helper.py

@@ -1,6 +1,6 @@
 import psycopg2
 from psycopg2.extras import execute_values, RealDictCursor
-from ..helpers.azure_identity_helper import AzureIdentityHelper
+from mgmt_config import identity
 from .llm_helper import LLMHelper
 from .env_helper import EnvHelper
 
@@ -15,7 +15,6 @@ class AzurePostgresHelper:
     def __init__(self):
         self.llm_helper = LLMHelper()
         self.env_helper = EnvHelper()
-        self.azure_identity_helper = AzureIdentityHelper()
         self.conn = None
 
     @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
@@ -29,7 +28,7 @@ class AzurePostgresHelper:
             dbname = self.env_helper.POSTGRESQL_DATABASE
 
             # Acquire the access token
-            access_information = self.azure_identity_helper.get_token(
+            access_information = identity.get_token(
                 scopes="https://ossrdbms-aad.database.windows.net/.default"
             )
             token = access_information.token

cwyodmodules/batch/utilities/helpers/azure_postgres_helper_light_rag.py

@@ -3,9 +3,8 @@ from psycopg2.extras import execute_values, RealDictCursor
 from .llm_helper import LLMHelper
 from .env_helper import EnvHelper
 from .lightrag_helper import LightRAGHelper
-from ..helpers.azure_identity_helper import AzureIdentityHelper
 
-from mgmt_config import logger
+from mgmt_config import logger, identity
 env_helper: EnvHelper = EnvHelper()
 log_execution = env_helper.LOG_EXECUTION
 log_args = env_helper.LOG_ARGS
@@ -17,7 +16,6 @@ class AzurePostgresHelper:
         self.llm_helper = LLMHelper()
         self.env_helper = EnvHelper()
         self.lightrag_helper = LightRAGHelper()
-        self.azure_identity_helper = AzureIdentityHelper()
         self.conn = None
 
     @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
@@ -31,7 +29,7 @@ class AzurePostgresHelper:
             dbname = self.env_helper.POSTGRESQL_DATABASE
 
             # Acquire the access token
-            access_information = self.azure_identity_helper.get_token(
+            access_information = identity.get_token(
                 scopes="https://ossrdbms-aad.database.windows.net/.default"
             )
             token = access_information.token

cwyodmodules/batch/utilities/helpers/azure_search_helper.py

@@ -2,7 +2,6 @@ from azure.identity import ChainedTokenCredential, DefaultAzureCredential
 from typing import Union
 from langchain_community.vectorstores import AzureSearch
 from azure.core.credentials import AzureKeyCredential
-from ..helpers.azure_identity_helper import AzureIdentityHelper
 from azure.search.documents import SearchClient
 from azure.search.documents.indexes import SearchIndexClient
 from azure.search.documents.indexes.models import (
@@ -28,7 +27,7 @@ from azure.search.documents.indexes.models import (
 from ..helpers.azure_computer_vision_client import AzureComputerVisionClient
 from ..helpers.llm_helper import LLMHelper
 from ..helpers.env_helper import EnvHelper
-from mgmt_config import logger
+from mgmt_config import logger, identity
 env_helper: EnvHelper = EnvHelper()
 log_execution = env_helper.LOG_EXECUTION
 log_args = env_helper.LOG_ARGS
@@ -41,7 +40,6 @@ class AzureSearchHelper:
     def __init__(self):
         self.llm_helper = LLMHelper()
         self.env_helper = EnvHelper()
-        self.azure_identity_helper = AzureIdentityHelper()
 
         search_credential = self._search_credential()
         self.search_client = self._create_search_client(search_credential)
@@ -53,7 +51,7 @@ class AzureSearchHelper:
         if self.env_helper.is_auth_type_keys():
             return AzureKeyCredential(self.env_helper.AZURE_SEARCH_KEY)
         else:
-            credential = self.azure_identity_helper.get_credential()
+            credential = identity.get_credential()
             return credential
 
     @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)

cwyodmodules/batch/utilities/helpers/env_helper.py

@@ -1,15 +1,14 @@
 import json
 import os
 
-import threading
+import threading                        
 # from dotenv import load_dotenv
 from ..orchestrator.orchestration_strategy import OrchestrationStrategy
 from ..helpers.config.conversation_flow import ConversationFlow
 from ..helpers.config.database_type import DatabaseType
 from ..helpers.secret_helper import SecretHelper
-from ..helpers.azure_identity_helper import AzureIdentityHelper
 
-from mgmt_config import logger
+from mgmt_config import logger, identity
 
 
 
@@ -76,8 +75,7 @@ class EnvHelper:
             f"psql-main-{self.PROJECT_CODE}-{self.AZURE_RESOURCE_ENVIRONMENT}"
         )
         self.AZURE_AUTH_TYPE = "rbac"
-        azure_identity_helper = AzureIdentityHelper()# Extract context from the incoming request
-        access_information = azure_identity_helper.get_token_provider(scopes="https://cognitiveservices.azure.com/.default")
+        access_information = identity.get_token_provider(scopes="https://cognitiveservices.azure.com/.default")
         self.AZURE_TOKEN_PROVIDER = access_information
         self.AZURE_BLOB_ACCOUNT_NAME = (
             f"stqueue{self.PROJECT_CODE}{self.AZURE_RESOURCE_ENVIRONMENT}"

cwyodmodules/batch/utilities/helpers/lightrag_helper.py

@@ -1,9 +1,8 @@
 import psycopg2
 from psycopg2.extras import execute_values, RealDictCursor
-from ..helpers.azure_identity_helper import AzureIdentityHelper
 
 from ...utilities.helpers.env_helper import EnvHelper
-from mgmt_config import logger
+from mgmt_config import logger, identity
 env_helper: EnvHelper = EnvHelper()
 log_execution = env_helper.LOG_EXECUTION
 log_args = env_helper.LOG_ARGS
@@ -13,7 +12,6 @@ log_result = env_helper.LOG_RESULT
 class LightRAGHelper:
     def __init__(self, env_helper):
         self.env_helper = env_helper
-        self.azure_identity_helper = AzureIdentityHelper()
         self.conn = None
 
     @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
@@ -27,7 +25,7 @@ class LightRAGHelper:
             dbname = self.env_helper.POSTGRESQL_DATABASE
 
             # Acquire the access token
-            access_information = self.azure_identity_helper.get_token(
+            access_information = identity.get_token(
                 scopes="https://ossrdbms-aad.database.windows.net/.default"
             )
             token = access_information.token

cwyodmodules/batch/utilities/helpers/llm_helper.py

@@ -7,10 +7,9 @@ from semantic_kernel.connectors.ai.open_ai.prompt_execution_settings.azure_chat_
     AzureChatPromptExecutionSettings,
 )
 from azure.ai.ml import MLClient
-from ..helpers.azure_identity_helper import AzureIdentityHelper
 from .env_helper import EnvHelper
 
-from mgmt_config import logger
+from mgmt_config import logger, identity
 env_helper: EnvHelper = EnvHelper()
 log_execution = env_helper.LOG_EXECUTION
 log_args = env_helper.LOG_ARGS
@@ -21,7 +20,6 @@ class LLMHelper:
     def __init__(self):
         logger.info("Initializing LLMHelper")
         self.env_helper: EnvHelper = EnvHelper()
-        self.azure_identity_helper = AzureIdentityHelper()
 
         self.auth_type_keys = self.env_helper.is_auth_type_keys()
         self.token_provider = self.env_helper.AZURE_TOKEN_PROVIDER
@@ -190,7 +188,7 @@ class LLMHelper:
     @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def get_ml_client(self):
         if not hasattr(self, "_ml_client"):
-            credential = self.azure_identity_helper.get_credential()
+            credential = identity.get_credential()
             self._ml_client = MLClient(
                 credential,
                 self.env_helper.AZURE_SUBSCRIPTION_ID,

cwyodmodules/batch/utilities/helpers/secret_helper.py

@@ -1,7 +1,6 @@
-from ..helpers.azure_identity_helper import AzureIdentityHelper
 from azure.keyvault.secrets import SecretClient
 
-from mgmt_config import logger
+from mgmt_config import logger, identity
 
 
 class SecretHelper:
@@ -19,11 +18,10 @@ class SecretHelper:
         Returns:
             None
         """
-        self.azure_identity_helper = AzureIdentityHelper()
         self.USE_KEY_VAULT = True
         self.secret_client = None
         if self.USE_KEY_VAULT:
-            credential = self.azure_identity_helper.get_credential()
+            credential = identity.get_credential()
             self.secret_client = SecretClient(
                 vault_url=keyvault_uri,
                 credential=credential,

cwyodmodules/batch/utilities/integrated_vectorization/azure_search_datasource.py

@@ -7,21 +7,20 @@ from azure.search.documents.indexes._generated.models import (
 )
 from azure.search.documents.indexes import SearchIndexerClient
 from ..helpers.env_helper import EnvHelper
-from ..helpers.azure_identity_helper import AzureIdentityHelper
 from azure.core.credentials import AzureKeyCredential
+from mgmt_config import identity
 
 
 class AzureSearchDatasource:
     def __init__(self, env_helper: EnvHelper):
         self.env_helper = env_helper
-        self.azure_identity_helper = AzureIdentityHelper()
         
         self.indexer_client = SearchIndexerClient(
             self.env_helper.AZURE_SEARCH_SERVICE,
             (
                 AzureKeyCredential(self.env_helper.AZURE_SEARCH_KEY)
                 if self.env_helper.is_auth_type_keys()
-                else self.azure_identity_helper.get_credential()
+                else identity.get_credential()
             ),
         )
 

cwyodmodules/batch/utilities/integrated_vectorization/azure_search_index.py

@@ -20,11 +20,9 @@ from azure.search.documents.indexes.models import (
     SearchIndex,
 )
 from ..helpers.env_helper import EnvHelper
-from ..helpers.azure_identity_helper import AzureIdentityHelper
 from azure.core.credentials import AzureKeyCredential
 from ..helpers.llm_helper import LLMHelper
-
-from mgmt_config import logger
+from mgmt_config import logger, identity
 env_helper: EnvHelper = EnvHelper()
 log_execution = env_helper.LOG_EXECUTION
 log_args = env_helper.LOG_ARGS
@@ -37,14 +35,13 @@ class AzureSearchIndex:
     def __init__(self, env_helper: EnvHelper, llm_helper: LLMHelper):
         self.env_helper = env_helper
         self.llm_helper = llm_helper
-        self.azure_identity_helper = AzureIdentityHelper()
 
         self.index_client = SearchIndexClient(
             self.env_helper.AZURE_SEARCH_SERVICE,
             (
                 AzureKeyCredential(self.env_helper.AZURE_SEARCH_KEY)
                 if self.env_helper.is_auth_type_keys()
-                else self.azure_identity_helper.get_credential()
+                else identity.get_credential()
             ),
         )
 

cwyodmodules/batch/utilities/integrated_vectorization/azure_search_indexer.py

@@ -1,12 +1,11 @@
 from azure.search.documents.indexes.models import SearchIndexer, FieldMapping
 from azure.search.documents.indexes import SearchIndexerClient
 from ..helpers.env_helper import EnvHelper
-from ..helpers.azure_identity_helper import AzureIdentityHelper
 from azure.core.credentials import AzureKeyCredential
 
 
 from ...utilities.helpers.env_helper import EnvHelper
-from mgmt_config import logger
+from mgmt_config import logger, identity
 env_helper: EnvHelper = EnvHelper()
 log_execution = env_helper.LOG_EXECUTION
 log_args = env_helper.LOG_ARGS
@@ -16,14 +15,13 @@ log_result = env_helper.LOG_RESULT
 class AzureSearchIndexer:
     def __init__(self, env_helper: EnvHelper):
         self.env_helper = env_helper
-        self.azure_identity_helper = AzureIdentityHelper()
 
         self.indexer_client = SearchIndexerClient(
             self.env_helper.AZURE_SEARCH_SERVICE,
             (
                 AzureKeyCredential(self.env_helper.AZURE_SEARCH_KEY)
                 if self.env_helper.is_auth_type_keys()
-                else self.azure_identity_helper.get_credential()
+                else identity.get_credential()
             ),
         )
 

cwyodmodules/batch/utilities/integrated_vectorization/azure_search_skillset.py

@@ -14,11 +14,10 @@ from azure.search.documents.indexes.models import (
 from azure.search.documents.indexes import SearchIndexerClient
 from ..helpers.config.config_helper import IntegratedVectorizationConfig
 from ..helpers.env_helper import EnvHelper
-from ..helpers.azure_identity_helper import AzureIdentityHelper
 from azure.core.credentials import AzureKeyCredential
 
 from ...utilities.helpers.env_helper import EnvHelper
-from mgmt_config import logger
+from mgmt_config import logger, identity
 env_helper: EnvHelper = EnvHelper()
 log_execution = env_helper.LOG_EXECUTION
 log_args = env_helper.LOG_ARGS
@@ -32,14 +31,13 @@ class AzureSearchSkillset:
         integrated_vectorization_config: IntegratedVectorizationConfig,
     ):
         self.env_helper = env_helper
-        self.azure_identity_helper = AzureIdentityHelper()
 
         self.indexer_client = SearchIndexerClient(
             self.env_helper.AZURE_SEARCH_SERVICE,
             (
                 AzureKeyCredential(self.env_helper.AZURE_SEARCH_KEY)
                 if self.env_helper.is_auth_type_keys()
-                else self.azure_identity_helper.get_credential()
+                else identity.get_credential()
             ),
         )
         self.integrated_vectorization_config = integrated_vectorization_config

cwyodmodules/batch/utilities/search/integrated_vectorization_search_handler.py

@@ -4,11 +4,10 @@ from azure.search.documents import SearchClient
 from azure.search.documents.indexes import SearchIndexClient
 from azure.search.documents.models import VectorizableTextQuery
 from azure.core.credentials import AzureKeyCredential
-from ..helpers.azure_identity_helper import AzureIdentityHelper
 from ..common.source_document import SourceDocument
 import re
 from ...utilities.helpers.env_helper import EnvHelper
-from mgmt_config import logger
+from mgmt_config import logger, identity
 env_helper: EnvHelper = EnvHelper()
 log_execution = env_helper.LOG_EXECUTION
 log_args = env_helper.LOG_ARGS
@@ -16,7 +15,8 @@ log_result = env_helper.LOG_RESULT
 
 class IntegratedVectorizationSearchHandler(SearchHandlerBase):
     def __init__(self):
-        self.azure_identity_helper = AzureIdentityHelper()
+        self.env_helper = EnvHelper()
+
     @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)    
     def create_search_client(self):
         logger.info("Creating Azure Search Client.")
@@ -28,7 +28,7 @@ class IntegratedVectorizationSearchHandler(SearchHandlerBase):
                 credential=(
                     AzureKeyCredential(self.env_helper.AZURE_SEARCH_KEY)
                     if self.env_helper.is_auth_type_keys()
-                    else self.azure_identity_helper.get_credential()
+                    else identity.get_credential()
                 ),
             )
 
@@ -186,7 +186,7 @@ class IntegratedVectorizationSearchHandler(SearchHandlerBase):
             credential=(
                 AzureKeyCredential(self.env_helper.AZURE_SEARCH_KEY)
                 if self.env_helper.is_auth_type_keys()
-                else self.azure_identity_helper.get_credential()
+                else identity.get_credential()
             ),
         )
 

cwyodmodules/batch/utilities/tools/content_safety_checker.py

@@ -1,13 +1,12 @@
 from azure.ai.contentsafety import ContentSafetyClient
 from azure.core.credentials import AzureKeyCredential
-from ..helpers.azure_identity_helper import AzureIdentityHelper
 from azure.core.exceptions import HttpResponseError
 from azure.ai.contentsafety.models import AnalyzeTextOptions
 from ..helpers.env_helper import EnvHelper
 from .answer_processing_base import AnswerProcessingBase
 from ..common.answer import Answer
 
-from mgmt_config import logger
+from mgmt_config import logger, identity
 env_helper: EnvHelper = EnvHelper()
 log_execution = env_helper.LOG_EXECUTION
 log_args = env_helper.LOG_ARGS
@@ -16,10 +15,8 @@ log_result = env_helper.LOG_RESULT
 
 class ContentSafetyChecker(AnswerProcessingBase):
     def __init__(self):
-        self.azure_identity_helper = AzureIdentityHelper()
-        
         if env_helper.AZURE_AUTH_TYPE == "rbac":
-            credential = self.azure_identity_helper.get_credential()
+            credential = identity.get_credential()
             logger.info(
                 "Initializing ContentSafetyClient with RBAC authentication."
             )

{cwyodmodules-0.3.42.dist-info → cwyodmodules-0.3.43.dist-info}/METADATA

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

{cwyodmodules-0.3.42.dist-info → cwyodmodules-0.3.43.dist-info}/RECORD

@@ -6,8 +6,8 @@ cwyodmodules/batch/utilities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NM
 cwyodmodules/batch/utilities/chat_history/auth_utils.py,sha256=4DipngRgRwkbwt-R1j0_I_KCazLuV5RSFtXn1zJ-5vE,2109
 cwyodmodules/batch/utilities/chat_history/cosmosdb.py,sha256=St5wWqVm4IOsDmlJDNlJHv9PVrPmdN04-9MYmc5udMY,8777
 cwyodmodules/batch/utilities/chat_history/database_client_base.py,sha256=6y7h2iL0Uxn4c11N99Ao-8nplQGVGQAOnev3GlHIDXA,2328
-cwyodmodules/batch/utilities/chat_history/database_factory.py,sha256=kdleC4P8u8rYPsx2HcPQwyOgFf_mph3Nh-Mmzz1V7oM,2401
-cwyodmodules/batch/utilities/chat_history/postgresdbservice.py,sha256=tkE7dpv95GNXwHah_Oj33LYmAuDerf5YhEAot1BjGt4,13027
+cwyodmodules/batch/utilities/chat_history/database_factory.py,sha256=3fUmxTmxLZ01DUwBfxZ8raZjsEppVgAnTpchMGcH2oU,2302
+cwyodmodules/batch/utilities/chat_history/postgresdbservice.py,sha256=71TtWBNzGq4T-4TubzFGEBSCTRRm28wB04q0AZTyQ9o,12892
 cwyodmodules/batch/utilities/chat_history/sample_user.py,sha256=GNXZ_yTjud8Zj0vgHnoU96RJMiJt0YRjEVO3pt7203A,3037
 cwyodmodules/batch/utilities/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cwyodmodules/batch/utilities/common/answer.py,sha256=C1g_Xt85TfXK2qV1F6J5_o3MyIUdFpUmvgkIpuEcp4I,3093
@@ -28,20 +28,19 @@ cwyodmodules/batch/utilities/document_loading/strategies.py,sha256=ZBKYPJD8UJmPB
 cwyodmodules/batch/utilities/document_loading/web.py,sha256=LRTNYs_7CN8nfMOaCoW7Py_obrLpj3vI4kneNVEHGXE,1186
 cwyodmodules/batch/utilities/document_loading/word_document.py,sha256=-F1asMaupQk4swEeCoAD8tyYENE4Qq-05-VmPUjRdeA,1569
 cwyodmodules/batch/utilities/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cwyodmodules/batch/utilities/helpers/azure_blob_storage_client.py,sha256=FN2XnEThmtbWnwSi1sEfgekuPH7aJBOAU5n2DBmQ9ww,10315
-cwyodmodules/batch/utilities/helpers/azure_computer_vision_client.py,sha256=tCSJM0SsC_ris4NZ-WCBvP5n2mRZBhVHjnpYjUVHUVQ,4333
-cwyodmodules/batch/utilities/helpers/azure_form_recognizer_helper.py,sha256=agZEbw3vJAOiX7fpzCsId527VMsLOez3cVELYlpUJy0,7002
-cwyodmodules/batch/utilities/helpers/azure_identity_helper.py,sha256=aiYZH4F9aZBd5DDRuVvIn6jiIDgC0i-R-1-NqHqosnI,1806
-cwyodmodules/batch/utilities/helpers/azure_postgres_helper.py,sha256=J0JZEuH0OE0sGtC4G2E5X4df4Ua1Jg8LKspL-sVxxxQ,11894
-cwyodmodules/batch/utilities/helpers/azure_postgres_helper_light_rag.py,sha256=-6CPczt8vbkedZliSXmmoLat8EiKm4oHIFldbQe_Q5k,11966
-cwyodmodules/batch/utilities/helpers/azure_search_helper.py,sha256=_HDHbrpWbbM8AJfcd2deTtseZK-hM4ar9QCrXvVSfYM,11445
+cwyodmodules/batch/utilities/helpers/azure_blob_storage_client.py,sha256=WgJFr1o4ZLqTTogS5oFwqSTHs6ofrycjrnSvffhIjoA,10140
+cwyodmodules/batch/utilities/helpers/azure_computer_vision_client.py,sha256=zxpUx_Z5T2VmtgMuAdwLtIwQXF9QI-v11ShTnaSfjRA,4235
+cwyodmodules/batch/utilities/helpers/azure_form_recognizer_helper.py,sha256=wEbSJjZuV_u-4yWYNCiG5RwifjxNO8hdfXDuDoV-KiY,6890
+cwyodmodules/batch/utilities/helpers/azure_postgres_helper.py,sha256=efDNCnhrY-IJBKEwrJN_xbg4I72_r2HY3g6vyMYw5AM,11786
+cwyodmodules/batch/utilities/helpers/azure_postgres_helper_light_rag.py,sha256=M92Ir9vzXRMpCXcENE3Jux13C6mrMc6o9julcM6b3uY,11835
+cwyodmodules/batch/utilities/helpers/azure_search_helper.py,sha256=vIIMEck1wPg9oRlWweE2gSZ1nUYc_tmEe4QeFlsrwKk,11314
 cwyodmodules/batch/utilities/helpers/document_chunking_helper.py,sha256=2MZOjW-fHXgYijP3m9O-nizOlk96Yg0axyxT0K6fTnM,725
 cwyodmodules/batch/utilities/helpers/document_loading_helper.py,sha256=2HBEl3vW-_PKbX5pPntTC_R5eToTk2Qb-q3M4Mt6hCU,603
-cwyodmodules/batch/utilities/helpers/env_helper.py,sha256=sgpaLlw3SlW99j9mvei-XE8wb1lX1f_vhEXeKf7hZvs,16021
-cwyodmodules/batch/utilities/helpers/lightrag_helper.py,sha256=18J8G-N7Rt1Yk2Hf2VlM_5LzKTcFTf72U_RCRXvgtMI,3976
-cwyodmodules/batch/utilities/helpers/llm_helper.py,sha256=hgxfNhAZGmWLakKTUu6d8FVrmUjFKJyAGUfpls9elL8,8413
+cwyodmodules/batch/utilities/helpers/env_helper.py,sha256=S0EG0XzILrJ_iPCAwKN-kzx-yPl5K__MtBZee924DKE,15881
+cwyodmodules/batch/utilities/helpers/lightrag_helper.py,sha256=7lb9JMm5IohsO73LWo5bWmlzWCGYNsx_fYl-aFdwATQ,3845
+cwyodmodules/batch/utilities/helpers/llm_helper.py,sha256=lHLYrUidtaemmKrVbWoo7oIvwluUoPUk16U5lV-YIX8,8282
 cwyodmodules/batch/utilities/helpers/orchestrator_helper.py,sha256=mCcZyMFG0otnw9gzWd-PYocHmDdFDVg-RT9oDPiDZPk,897
-cwyodmodules/batch/utilities/helpers/secret_helper.py,sha256=eRMDFo7NIjzJpgkKfCSKUwNsrq6gfjoujyRrgu27ci0,2887
+cwyodmodules/batch/utilities/helpers/secret_helper.py,sha256=RYDPR-1c7uChfmAJfzvN_GiI4u6czyWZ67XQah0hla4,2756
 cwyodmodules/batch/utilities/helpers/config/assistant_strategy.py,sha256=uT8h646zEURU9x8oDOH7pWoZKb0Mw6dA2nJtA2M-ufg,171
 cwyodmodules/batch/utilities/helpers/config/config_helper.py,sha256=26na6YrLqRLhdSZxTSlOnJOkIcPbTcbFVuPEQTPT3WY,14908
 cwyodmodules/batch/utilities/helpers/config/conversation_flow.py,sha256=4nP8a-I-sME5-2unzWWBNpTzWdfpfc5_EAYU6Pn6LAQ,94
@@ -55,10 +54,10 @@ cwyodmodules/batch/utilities/helpers/embedders/embedder_factory.py,sha256=cJ9ZTX
 cwyodmodules/batch/utilities/helpers/embedders/integrated_vectorization_embedder.py,sha256=qguAc5gRxbZdSXe9qjkn1eqRgVc8LrsiZAE3ggeJaSg,3338
 cwyodmodules/batch/utilities/helpers/embedders/postgres_embedder.py,sha256=_Wc4gkMDsXvGLkz7NpslkTAumvyZt6CHKlktU3bzbcU,4865
 cwyodmodules/batch/utilities/helpers/embedders/push_embedder.py,sha256=jxOp-AW1vTQMQHR6f-UoD0gFxD2ZJ6E0SLt1KvzdykQ,9143
-cwyodmodules/batch/utilities/integrated_vectorization/azure_search_datasource.py,sha256=rDwPgr-UCSYscc7hPOUJMwP09a9rX1MXAGf94TubdQo,2231
-cwyodmodules/batch/utilities/integrated_vectorization/azure_search_index.py,sha256=VwyfKBUDcWt0uBW_8mtq0XlFzaBZNUdKUjF8WUvt9uU,6733
-cwyodmodules/batch/utilities/integrated_vectorization/azure_search_indexer.py,sha256=xhrF783aC1lWSVPHiX8zF_zX6npT4NSE3_UWvGSjcYQ,3361
-cwyodmodules/batch/utilities/integrated_vectorization/azure_search_skillset.py,sha256=cfIZbiJAQ15LB0_j_xno8EQt6cY_HfhL07zrU-wBAYI,5902
+cwyodmodules/batch/utilities/integrated_vectorization/azure_search_datasource.py,sha256=_412aNCETv21Pb2kAE_L2My_4svxxpIXcY8L7zNmoG0,2123
+cwyodmodules/batch/utilities/integrated_vectorization/azure_search_index.py,sha256=19qKloRV6_CYJbuGnAwV60sH9lkzyGFGA3SvvKciuDM,6601
+cwyodmodules/batch/utilities/integrated_vectorization/azure_search_indexer.py,sha256=qWBsFGIJkrUEcgV8cPv93_FZatVLeLyKD5R-gnSUfj0,3230
+cwyodmodules/batch/utilities/integrated_vectorization/azure_search_skillset.py,sha256=5X0cSf-BqiXLbWQ6dVbOOkGQrmsIZp1FD5seJq9YfCI,5771
 cwyodmodules/batch/utilities/loggers/conversation_logger.py,sha256=0aXsL475-6WTqg18nHFJMFRBo34oIXWrZ_eVZwULcdk,3014
 cwyodmodules/batch/utilities/orchestrator/__init__.py,sha256=4nCkoUWTROUHJMolgMwPgFIUsJrFUuu0zlHXMUTchRc,479
 cwyodmodules/batch/utilities/orchestrator/lang_chain_agent.py,sha256=X-rnXhd20XUcu2YUYMzKi0acJsBhCADMypauOZkbJas,6892
@@ -76,7 +75,7 @@ cwyodmodules/batch/utilities/plugins/outlook_calendar_plugin.py,sha256=q4RUHa9ij
 cwyodmodules/batch/utilities/plugins/post_answering_plugin.py,sha256=U1zzf_ztxzl4y-9Qah_n7ylHDZfnDSp2ork5ctdkA5I,1117
 cwyodmodules/batch/utilities/search/azure_search_handler.py,sha256=SiPy-C2H6pncwTCyUNF5cjL3TZZ4wZmFCMy2xsBjrwk,8046
 cwyodmodules/batch/utilities/search/azure_search_handler_light_rag.py,sha256=OHyp-QWU-7HTeKxOLNqR7G5r8CIlhDzzuZX-7xVLMQQ,3917
-cwyodmodules/batch/utilities/search/integrated_vectorization_search_handler.py,sha256=nX5JSQd3e-DOcjJsueuDqeqfNgR7ZhkzWQqIInQ8ayU,8600
+cwyodmodules/batch/utilities/search/integrated_vectorization_search_handler.py,sha256=uphMlUfZiju96w2SC0KyL5n5rS_4OUvrXL03o1aU7to,8490
 cwyodmodules/batch/utilities/search/lightrag_search_handler.py,sha256=e3AAPyeKEAazJcjz3INy5EJZIUs80MPgKDMFjhgCdaA,3312
 cwyodmodules/batch/utilities/search/postgres_search_handler.py,sha256=pSkbDwimPuIVyyYKiZIg6jmItLYJ59Xugyiuy9kLEV8,5571
 cwyodmodules/batch/utilities/search/postgres_search_handler_light_rag.py,sha256=4N5VRK0c0jvRlVL_JozPmBm_tBrloSHRDXexqo1jdis,5981
@@ -85,7 +84,7 @@ cwyodmodules/batch/utilities/search/search_handler_base.py,sha256=UyS9dFoY-Sp4b-
 cwyodmodules/batch/utilities/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cwyodmodules/batch/utilities/tools/answer_processing_base.py,sha256=N3Dz7HfN-zl0yPl9cSvnChcEufFMqwRsDAJ5Lgck-ho,293
 cwyodmodules/batch/utilities/tools/answering_tool_base.py,sha256=aN2ND5Ud_1ZlIPfhLRrOe_m4MUf_SaXvO6q7GcbGiU8,348
-cwyodmodules/batch/utilities/tools/content_safety_checker.py,sha256=ViikBC3wwySqcFUB6TcVQSjutCUu5KRgg4InZaHegms,4307
+cwyodmodules/batch/utilities/tools/content_safety_checker.py,sha256=AOCzRkpacBAWVhHtuhK42ON5q8hqHEArmY_lE_G_DJQ,4167
 cwyodmodules/batch/utilities/tools/post_prompt_tool.py,sha256=tGxw8qMNS7jp8P3KqpqHplicmX064sL2J85VwnCWBbA,2495
 cwyodmodules/batch/utilities/tools/question_answer_tool.py,sha256=8GWurj50bUND_4x0i1_UPfwl1HMLUIpvq5xBv-q5UDA,12213
 cwyodmodules/batch/utilities/tools/text_processing_tool.py,sha256=zCF83SQaC8k6z6-Gevxj6AdDKzhU3P_6FG0hAnrG5Zk,1824
@@ -109,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.42.dist-info/licenses/LICENSE,sha256=UqBDTipijsSW2ZSOXyTZnMsXmLoEHTgNEM0tL4g-Sso,1150
-cwyodmodules-0.3.42.dist-info/METADATA,sha256=48pZpR1Wj91W8zN8om8njkNcQZEJfwk9BdyrpPpHHZk,2002
-cwyodmodules-0.3.42.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-cwyodmodules-0.3.42.dist-info/top_level.txt,sha256=99RENLbkdRX-qpJvsxZ5AfmTL5s6shSaKOWYpz1vwzg,13
-cwyodmodules-0.3.42.dist-info/RECORD,,
+cwyodmodules-0.3.43.dist-info/licenses/LICENSE,sha256=UqBDTipijsSW2ZSOXyTZnMsXmLoEHTgNEM0tL4g-Sso,1150
+cwyodmodules-0.3.43.dist-info/METADATA,sha256=CJHnTq4Q2E-Dcn8GiIYWPqI4jyvxQ1S44KIVClTitQY,2002
+cwyodmodules-0.3.43.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+cwyodmodules-0.3.43.dist-info/top_level.txt,sha256=99RENLbkdRX-qpJvsxZ5AfmTL5s6shSaKOWYpz1vwzg,13
+cwyodmodules-0.3.43.dist-info/RECORD,,

cwyodmodules/batch/utilities/helpers/azure_identity_helper.py

@@ -1,50 +0,0 @@
-from azure.identity import (
-    TokenCachePersistenceOptions,
-    get_bearer_token_provider,
-    DefaultAzureCredential
-)
-
-from logging import getLogger
-from opentelemetry import trace, baggage
-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.
-    It prioritizes Managed Identity, then Environment variables.
-    Token caching is configured for in-memory persistence.
-    """
-    def __init__(self):
-        
-        token_cache_options = TokenCachePersistenceOptions(allow_unencrypted_storage=True)        
-
-        self._credential = DefaultAzureCredential(
-            token_cache_persistence_options=token_cache_options
-        )
-
-    def get_credential(self):
-        """
-        Returns the configured ChainedTokenCredential.
-        """
-        with tracer.start_as_current_span("AzureIdentityHelper.get_credential"):
-            logger.info("Retrieving ChainedTokenCredential.")
-        return self._credential
-    
-    def get_token(self, scopes):
-        """
-        Returns the configured ChainedTokenCredential.
-        """
-        with tracer.start_as_current_span("AzureIdentityHelper.get_token_provider"):
-            logger.info("Retrieving ChainedTokenCredential provider.")
-        return self._credential.get_token(scopes)
-    
-    def get_token_provider(self, scopes):
-        """
-        Returns the configured ChainedTokenCredential.
-        """
-        with tracer.start_as_current_span("AzureIdentityHelper.get_token_provider"):
-            logger.info("Retrieving ChainedTokenCredential provider.")
-        return get_bearer_token_provider(self._credential, scopes)