cwyodmodules 0.3.46__py3-none-any.whl → 0.3.48__py3-none-any.whl

cwyodmodules/batch/utilities/helpers/env_helper.py

@@ -6,9 +6,8 @@ import threading
 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 mgmt_config import logger, identity
+from mgmt_config import logger, identity, keyvault, head_keyvault
 
 
 
@@ -30,23 +29,23 @@ class EnvHelper:
 
         logger.info("Initializing EnvHelper")
         
+        if not keyvault:
+            raise ValueError("keyvault is not configured. Please set 'key_vault_uri' environment variable.")
+        
+        if not head_keyvault:
+            raise ValueError("head_keyvault is not configured. Please set 'head_key_vault_uri' environment variable.")
+
         # Wrapper for Azure Key Vault
         os.environ["APPLICATIONINSIGHTS_ENABLED"] = "true"
 
-        self.secretHelper = SecretHelper(keyvault_uri=os.environ["key_vault_uri"])
-
-        self.secretHelperHead = SecretHelper(
-            keyvault_uri=os.environ["head_key_vault_uri"]
-        )
-
-        # self.secretHelper = SecretHelper(
+        # keyvault = SecretHelper(
         #     keyvault_uri="https://www.kv-main-cwyod-res1.vault.azure.net/"
         # )
-        # self.secretHelperHead = SecretHelper(
+        # head_keyvault = SecretHelper(
         #     keyvault_uri="https://www.kv-main-cwyod-hd-res1.vault.azure.net/"
         # )
 
-        self.LOGLEVEL = self.secretHelper.get_secret("logging-level")
+        self.LOGLEVEL = keyvault.get_secret("logging-level")
         self.LOG_EXECUTION = self.get_env_var_bool(
                 "LOG_EXECUTION", "True"
             )
@@ -58,18 +57,18 @@ class EnvHelper:
             )
 
         # Azure
-        self.AZURE_SUBSCRIPTION_ID = self.secretHelper.get_secret("subscription-id")
-        self.AZURE_RESOURCE_GROUP = self.secretHelper.get_secret("resource-group-name")
-        self.AZURE_HEAD_RESOURCE_GROUP = self.secretHelperHead.get_secret(
+        self.AZURE_SUBSCRIPTION_ID = keyvault.get_secret("subscription-id")
+        self.AZURE_RESOURCE_GROUP = keyvault.get_secret("resource-group-name")
+        self.AZURE_HEAD_RESOURCE_GROUP = head_keyvault.get_secret(
             "resource-group-name"
         )
-        self.AZURE_RESOURCE_ENVIRONMENT = self.secretHelper.get_secret(
+        self.AZURE_RESOURCE_ENVIRONMENT = keyvault.get_secret(
             "resource-group-environment"
         )
         self.AZURE_RESOURCE_PRIVATE = (
-            self.secretHelper.get_secret("run-private-endpoint").lower() == "true"
+            keyvault.get_secret("run-private-endpoint").lower() == "true"
         )
-        self.PROJECT_CODE = self.secretHelper.get_secret("project-code")
+        self.PROJECT_CODE = keyvault.get_secret("project-code")
         self.APP_NAME = os.getenv("REFLECTION_NAME", "Default")
         self.POSTGRESQL_NAME = (
             f"psql-main-{self.PROJECT_CODE}-{self.AZURE_RESOURCE_ENVIRONMENT}"
@@ -84,7 +83,7 @@ class EnvHelper:
             f"https://{self.AZURE_BLOB_ACCOUNT_NAME}.blob.core.windows.net/"
         )
 
-        self.AZURE_FUNCTION_APP_ENDPOINT = self.secretHelper.get_secret(
+        self.AZURE_FUNCTION_APP_ENDPOINT = keyvault.get_secret(
             f"func-backend-{self.PROJECT_CODE}-{self.AZURE_RESOURCE_ENVIRONMENT}-endpoint"
         )
         self.AZURE_BLOB_CONTAINER_NAME = "documents"
@@ -116,7 +115,7 @@ class EnvHelper:
                 self.AZURE_COSMOSDB_CONVERSATIONS_CONTAINER = os.getenv(
                     "AZURE_COSMOSDB_CONVERSATIONS_CONTAINER_NAME", ""
                 )
-            self.AZURE_COSMOSDB_ACCOUNT_KEY = self.secretHelper.get_secret(
+            self.AZURE_COSMOSDB_ACCOUNT_KEY = keyvault.get_secret(
                 "AZURE_COSMOSDB_ACCOUNT_KEY"
             )
             self.AZURE_COSMOSDB_ENABLE_FEEDBACK = (
@@ -143,10 +142,10 @@ class EnvHelper:
                 self.POSTGRESQL_HOST = azure_postgresql_info.get("host", "")
             else:
                 self.POSTGRESQL_USER = "cwyod_project_uai"
-                self.POSTGRESQL_DATABASE = self.secretHelper.get_secret(
+                self.POSTGRESQL_DATABASE = keyvault.get_secret(
                     f"{self.POSTGRESQL_NAME}-default-database-name"
                 )
-                self.POSTGRESQL_HOST = self.secretHelper.get_secret(
+                self.POSTGRESQL_HOST = keyvault.get_secret(
                     f"{self.POSTGRESQL_NAME}-server-name"
                 )
             # Ensure integrated vectorization is disabled for PostgreSQL
@@ -178,11 +177,11 @@ class EnvHelper:
             True if self.AZURE_OPENAI_STREAM.lower() == "true" else False
         )
 
-        # self.AZURE_COMPUTER_VISION_NAME = self.secretHelperHead.get_secret(
+        # self.AZURE_COMPUTER_VISION_NAME = head_keyvault.get_secret(
         #     "cognitive-kind-ComputerVision"
         # )
         self.AZURE_COMPUTER_VISION_NAME = ""
-        # self.AZURE_COMPUTER_VISION_ENDPOINT = self.secretHelperHead.get_secret(
+        # self.AZURE_COMPUTER_VISION_ENDPOINT = head_keyvault.get_secret(
         #     f"{self.AZURE_COMPUTER_VISION_NAME}-endpoint"
         # )
         self.AZURE_COMPUTER_VISION_ENDPOINT = ""
@@ -198,16 +197,16 @@ class EnvHelper:
             self.AZURE_OPENAI_API_KEY = ""
             self.AZURE_COMPUTER_VISION_KEY = None
         else:
-            self.AZURE_SEARCH_KEY = self.secretHelper.get_secret("AZURE_SEARCH_KEY")
-            self.AZURE_OPENAI_API_KEY = self.secretHelper.get_secret(
+            self.AZURE_SEARCH_KEY = keyvault.get_secret("AZURE_SEARCH_KEY")
+            self.AZURE_OPENAI_API_KEY = keyvault.get_secret(
                 "AZURE_OPENAI_API_KEY"
             )
-            self.AZURE_COMPUTER_VISION_KEY = self.secretHelper.get_secret(
+            self.AZURE_COMPUTER_VISION_KEY = keyvault.get_secret(
                 "AZURE_COMPUTER_VISION_KEY"
             )
 
         # Set env for Azure OpenAI
-        self.AZURE_AI_SERVICES_NAME = self.secretHelperHead.get_secret(
+        self.AZURE_AI_SERVICES_NAME = head_keyvault.get_secret(
             "cognitive-kind-AIServices"
         )
         self.AZURE_OPENAI_ENDPOINT = (
@@ -227,10 +226,10 @@ class EnvHelper:
 
         # Azure Form Recognizer
 
-        self.AZURE_FORM_RECOGNIZER_NAME = self.secretHelper.get_secret(
+        self.AZURE_FORM_RECOGNIZER_NAME = keyvault.get_secret(
             "cognitive-kind-FormRecognizer"
         )
-        self.AZURE_FORM_RECOGNIZER_ENDPOINT = self.secretHelper.get_secret(
+        self.AZURE_FORM_RECOGNIZER_ENDPOINT = keyvault.get_secret(
             f"{self.AZURE_FORM_RECOGNIZER_NAME}-endpoint"
         )
 
@@ -239,18 +238,18 @@ class EnvHelper:
         self.APPLICATIONINSIGHTS_ENABLED = "True"
 
         # Azure AI Content Safety
-        self.AZURE_CONTENT_SAFETY_NAME = self.secretHelperHead.get_secret(
+        self.AZURE_CONTENT_SAFETY_NAME = head_keyvault.get_secret(
             "cognitive-kind-ContentSafety"
         )
-        self.AZURE_CONTENT_SAFETY_ENDPOINT = self.secretHelperHead.get_secret(
+        self.AZURE_CONTENT_SAFETY_ENDPOINT = head_keyvault.get_secret(
             f"{self.AZURE_CONTENT_SAFETY_NAME}-endpoint"
         )
 
         # Speech Service
-        self.AZURE_SPEECH_SERVICE_NAME = self.secretHelperHead.get_secret(
+        self.AZURE_SPEECH_SERVICE_NAME = head_keyvault.get_secret(
             "cognitive-kind-SpeechServices"
         )
-        self.AZURE_SPEECH_ENDPOINT = self.secretHelperHead.get_secret(
+        self.AZURE_SPEECH_ENDPOINT = head_keyvault.get_secret(
             f"{self.AZURE_SPEECH_SERVICE_NAME}-endpoint"
         )
 
@@ -263,7 +262,7 @@ class EnvHelper:
         self.AZURE_SPEECH_REGION_ENDPOINT = (
             f"https://{self.AZURE_SPEECH_SERVICE_REGION}.api.cognitive.microsoft.com/"
         )
-        self.AZURE_SPEECH_KEY = self.secretHelperHead.get_secret(f"{self.AZURE_SPEECH_SERVICE_NAME}-key") # changed from None to allow private endpoint communication
+        self.AZURE_SPEECH_KEY = head_keyvault.get_secret(f"{self.AZURE_SPEECH_SERVICE_NAME}-key") # changed from None to allow private endpoint communication
 
         self.LOAD_CONFIG_FROM_BLOB_STORAGE = self.get_env_var_bool(
             "LOAD_CONFIG_FROM_BLOB_STORAGE"

cwyodmodules/mgmt_config.py

@@ -1,132 +1,181 @@
-"""
-Azure Project Management Configuration Template
-
-This template provides standardized configuration for Azure logging and identity
-management across projects. It creates singleton instances of AzureLogger and
-AzureIdentity that can be imported and used throughout your application.
-
-Usage:
-    from mgmt_config import logger, identity
-    
-    logger.info("Application started")
-    credential = identity.get_credential()
-"""
-
-import os
-from typing import Optional, Dict, Any
-from azpaddypy.mgmt.logging import create_app_logger, create_function_logger
-from azpaddypy.mgmt.identity import create_azure_identity
-
-# =============================================================================
-# SERVICE CONFIGURATION
-# =============================================================================
-
-# Service identity - customize these for your project
-SERVICE_NAME = os.getenv("SERVICE_NAME", "cwyodmodules-pacakge")
-SERVICE_VERSION = os.getenv("SERVICE_VERSION", "1.0.0")
-
-# =============================================================================
-# LOGGING CONFIGURATION
-# =============================================================================
-
-# Enable console output (useful for local development)
-LOGGER_ENABLE_CONSOLE = os.getenv("LOGGER_ENABLE_CONSOLE", "true").lower() == "true"
-
-# Application Insights connection string (optional, will use environment variable if not set)
-LOGGER_CONNECTION_STRING = os.getenv("APPLICATIONINSIGHTS_CONNECTION_STRING")
-
-# Configure which Azure SDK components to instrument
-LOGGER_INSTRUMENTATION_OPTIONS = {
-    "azure_sdk": {"enabled": True},
-    "django": {"enabled": False},
-    "fastapi": {"enabled": False},
-    "flask": {"enabled": True},
-    "psycopg2": {"enabled": True},
-    "requests": {"enabled": True},
-    "urllib": {"enabled": True},
-    "urllib3": {"enabled": True},
-}
-
-# =============================================================================
-# IDENTITY CONFIGURATION
-# =============================================================================
-
-# Token caching settings
-IDENTITY_ENABLE_TOKEN_CACHE = os.getenv("IDENTITY_ENABLE_TOKEN_CACHE", "true").lower() == "true"
-IDENTITY_ALLOW_UNENCRYPTED_STORAGE = os.getenv("IDENTITY_ALLOW_UNENCRYPTED_STORAGE", "true").lower() == "true"
-
-# Custom credential options (None means use defaults)
-IDENTITY_CUSTOM_CREDENTIAL_OPTIONS: Optional[Dict[str, Any]] = None
-
-# Connection string for identity logging (uses same as logger by default)
-IDENTITY_CONNECTION_STRING = LOGGER_CONNECTION_STRING
-
-# =============================================================================
-# INITIALIZE SERVICES
-# =============================================================================
-# Create logger instance
-if "functionapp" in os.getenv("REFLECTION_KIND", "app"):
-    logger = create_function_logger(
-        function_app_name=os.getenv("REFLECTION_NAME", "app"),
-        function_name="backend",
-        service_version=SERVICE_VERSION,
-        connection_string=LOGGER_CONNECTION_STRING,
-        instrumentation_options=LOGGER_INSTRUMENTATION_OPTIONS,
-    )
-    logger.info("Function logger initialized")
-else:
-    logger = create_app_logger(
-        service_name=SERVICE_NAME,
-        service_version=SERVICE_VERSION,
-        connection_string=LOGGER_CONNECTION_STRING,
-        enable_console_logging=LOGGER_ENABLE_CONSOLE,
-        instrumentation_options=LOGGER_INSTRUMENTATION_OPTIONS,
-    )
-    logger.info("App logger initialized")
-
-# Create identity instance with shared logger
-identity = create_azure_identity(
-    service_name=SERVICE_NAME,
-    service_version=SERVICE_VERSION,
-    enable_token_cache=IDENTITY_ENABLE_TOKEN_CACHE,
-    allow_unencrypted_storage=IDENTITY_ALLOW_UNENCRYPTED_STORAGE,
-    custom_credential_options=IDENTITY_CUSTOM_CREDENTIAL_OPTIONS,
-    connection_string=IDENTITY_CONNECTION_STRING,
-    logger=logger,
-)
-
-# =============================================================================
-# VALIDATION & STARTUP
-# =============================================================================
-
-# Validate critical configuration
-if SERVICE_NAME == __name__:
-    logger.warning(
-        "SERVICE_NAME is not configured. Please set SERVICE_NAME environment variable or update this template.",
-        extra={"configuration_issue": "service_name_not_set"}
-    )
-
-if not LOGGER_CONNECTION_STRING:
-    logger.info(
-        "No Application Insights connection string configured. Telemetry will be disabled.",
-        extra={"telemetry_status": "disabled"}
-    )
-
-# Log successful initialization
-logger.info(
-    f"Management configuration initialized for service '{SERVICE_NAME}' v{SERVICE_VERSION}",
-    extra={
-        "service_name": SERVICE_NAME,
-        "service_version": SERVICE_VERSION,
-        "console_logging": LOGGER_ENABLE_CONSOLE,
-        "token_cache_enabled": IDENTITY_ENABLE_TOKEN_CACHE,
-        "telemetry_enabled": bool(LOGGER_CONNECTION_STRING),
-    }
-)
-
-# =============================================================================
-# EXPORTS
-# =============================================================================
-
-# Export both logger and identity for use in applications
-__all__ = ["logger", "identity"]
+"""
+Azure Project Management Configuration Template
+
+This template provides standardized configuration for Azure logging, identity,
+and KeyVault management across projects. It creates singleton instances of
+AzureLogger, AzureIdentity, and AzureKeyVault that can be imported and used
+throughout your application.
+
+Usage:
+    from mgmt_config import logger, identity, keyvault
+    
+    logger.info("Application started")
+    credential = identity.get_credential()
+    secret = keyvault.get_secret("my-secret") if keyvault else None
+"""
+
+import os
+from typing import Optional, Dict, Any
+from azpaddypy.mgmt.logging import create_app_logger
+from azpaddypy.mgmt.identity import create_azure_identity
+from azpaddypy.resources.keyvault import create_azure_keyvault, AzureKeyVault
+
+# =============================================================================
+# SERVICE CONFIGURATION
+# =============================================================================
+
+# Service identity - customize these for your project
+SERVICE_NAME = os.getenv("SERVICE_NAME", "cwyodmodules-pacakge")
+SERVICE_VERSION = os.getenv("SERVICE_VERSION", "1.0.0")
+
+# =============================================================================
+# LOGGING CONFIGURATION
+# =============================================================================
+
+# Enable console output (useful for local development)
+LOGGER_ENABLE_CONSOLE = os.getenv("LOGGER_ENABLE_CONSOLE", "true").lower() == "true"
+
+# Application Insights connection string (optional, will use environment variable if not set)
+LOGGER_CONNECTION_STRING = os.getenv("APPLICATIONINSIGHTS_CONNECTION_STRING")
+
+# Configure which Azure SDK components to instrument
+LOGGER_INSTRUMENTATION_OPTIONS = {
+    "azure_sdk": {"enabled": True},
+    "django": {"enabled": False},
+    "fastapi": {"enabled": False},
+    "flask": {"enabled": True},
+    "psycopg2": {"enabled": True},
+    "requests": {"enabled": True},
+    "urllib": {"enabled": True},
+    "urllib3": {"enabled": True},
+}
+
+# =============================================================================
+# IDENTITY CONFIGURATION
+# =============================================================================
+
+# Token caching settings
+IDENTITY_ENABLE_TOKEN_CACHE = os.getenv("IDENTITY_ENABLE_TOKEN_CACHE", "true").lower() == "true"
+IDENTITY_ALLOW_UNENCRYPTED_STORAGE = os.getenv("IDENTITY_ALLOW_UNENCRYPTED_STORAGE", "true").lower() == "true"
+
+# Custom credential options (None means use defaults)
+IDENTITY_CUSTOM_CREDENTIAL_OPTIONS: Optional[Dict[str, Any]] = None
+
+# Connection string for identity logging (uses same as logger by default)
+IDENTITY_CONNECTION_STRING = LOGGER_CONNECTION_STRING
+
+# =============================================================================
+# KEYVAULT CONFIGURATION
+# =============================================================================
+
+# KeyVault URL (optional, will be None if not configured)
+KEYVAULT_URL = os.getenv("key_vault_uri")
+
+# KeyVault client enablement settings
+KEYVAULT_ENABLE_SECRETS = os.getenv("KEYVAULT_ENABLE_SECRETS", "true").lower() == "true"
+KEYVAULT_ENABLE_KEYS = os.getenv("KEYVAULT_ENABLE_KEYS", "false").lower() == "true"
+KEYVAULT_ENABLE_CERTIFICATES = os.getenv("KEYVAULT_ENABLE_CERTIFICATES", "false").lower() == "true"
+
+# Connection string for KeyVault logging (uses same as logger by default)
+KEYVAULT_CONNECTION_STRING = LOGGER_CONNECTION_STRING
+
+# =============================================================================
+# INITIALIZE SERVICES
+# =============================================================================
+
+# Create logger instance
+logger = create_app_logger(
+    service_name=SERVICE_NAME,
+    service_version=SERVICE_VERSION,
+    connection_string=LOGGER_CONNECTION_STRING,
+    enable_console_logging=LOGGER_ENABLE_CONSOLE,
+    instrumentation_options=LOGGER_INSTRUMENTATION_OPTIONS,
+)
+
+# Create identity instance with shared logger
+identity = create_azure_identity(
+    service_name=SERVICE_NAME,
+    service_version=SERVICE_VERSION,
+    enable_token_cache=IDENTITY_ENABLE_TOKEN_CACHE,
+    allow_unencrypted_storage=IDENTITY_ALLOW_UNENCRYPTED_STORAGE,
+    custom_credential_options=IDENTITY_CUSTOM_CREDENTIAL_OPTIONS,
+    connection_string=IDENTITY_CONNECTION_STRING,
+    logger=logger,
+)
+
+# Create KeyVault instance with shared logger and identity (only if URL is configured)
+keyvault = None
+if KEYVAULT_URL:
+    keyvault = create_azure_keyvault(
+        vault_url=KEYVAULT_URL,
+        azure_identity=identity,
+        service_name=SERVICE_NAME,
+        service_version=SERVICE_VERSION,
+        logger=logger,
+        connection_string=KEYVAULT_CONNECTION_STRING,
+        enable_secrets=KEYVAULT_ENABLE_SECRETS,
+        enable_keys=KEYVAULT_ENABLE_KEYS,
+        enable_certificates=KEYVAULT_ENABLE_CERTIFICATES,
+    )
+
+def create_keyvault_client(vault_url: str) -> Optional[AzureKeyVault]:
+    """Creates a secondary KeyVault client for a different vault."""
+    if not vault_url:
+        return None
+    return create_azure_keyvault(
+        vault_url=vault_url,
+        azure_identity=identity,
+        service_name=SERVICE_NAME,
+        service_version=SERVICE_VERSION,
+        logger=logger,
+        connection_string=KEYVAULT_CONNECTION_STRING,
+        enable_secrets=KEYVAULT_ENABLE_SECRETS,
+        enable_keys=KEYVAULT_ENABLE_KEYS,
+        enable_certificates=KEYVAULT_ENABLE_CERTIFICATES,
+    )
+
+# =============================================================================
+# VALIDATION & STARTUP
+# =============================================================================
+
+# Validate critical configuration
+if SERVICE_NAME == "your-service-name":
+    logger.warning(
+        "SERVICE_NAME is not configured. Please set SERVICE_NAME environment variable or update this template.",
+        extra={"configuration_issue": "service_name_not_set"}
+    )
+
+if not LOGGER_CONNECTION_STRING:
+    logger.info(
+        "No Application Insights connection string configured. Telemetry will be disabled.",
+        extra={"telemetry_status": "disabled"}
+    )
+
+if not KEYVAULT_URL:
+    logger.info(
+        "No KeyVault URL configured. KeyVault operations will be disabled.",
+        extra={"keyvault_status": "disabled"}
+    )
+
+# Log successful initialization
+logger.info(
+    f"Management configuration initialized for service '{SERVICE_NAME}' v{SERVICE_VERSION}",
+    extra={
+        "service_name": SERVICE_NAME,
+        "service_version": SERVICE_VERSION,
+        "console_logging": LOGGER_ENABLE_CONSOLE,
+        "token_cache_enabled": IDENTITY_ENABLE_TOKEN_CACHE,
+        "telemetry_enabled": bool(LOGGER_CONNECTION_STRING),
+        "keyvault_enabled": bool(KEYVAULT_URL),
+        "keyvault_secrets_enabled": KEYVAULT_ENABLE_SECRETS,
+        "keyvault_keys_enabled": KEYVAULT_ENABLE_KEYS,
+        "keyvault_certificates_enabled": KEYVAULT_ENABLE_CERTIFICATES,
+    }
+)
+
+# =============================================================================
+# EXPORTS
+# =============================================================================
+
+# Export logger, identity, and keyvault for use in applications
+__all__ = ["logger", "identity", "keyvault", "create_keyvault_client"]

{cwyodmodules-0.3.46.dist-info → cwyodmodules-0.3.48.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cwyodmodules
-Version: 0.3.46
+Version: 0.3.48
 Summary: Add your description here
 Author-email: Patrik <patrikhartl@gmail.com>
 Classifier: Operating System :: OS Independent
@@ -40,7 +40,7 @@ Requires-Dist: azure-search-documents==11.6.0b4
 Requires-Dist: semantic-kernel==1.3.0
 Requires-Dist: pydantic==2.7.4
 Requires-Dist: pandas>=2.2.3
-Requires-Dist: azpaddypy>=0.3.4
+Requires-Dist: azpaddypy>=0.3.7
 Dynamic: license-file
 
 # paddypy

{cwyodmodules-0.3.46.dist-info → cwyodmodules-0.3.48.dist-info}/RECORD

@@ -1,5 +1,5 @@
 cwyodmodules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cwyodmodules/mgmt_config.py,sha256=xRp9absSmQifcKKCn6fyqcMSJ9hDvMTqPPAViezJHZw,5308
+cwyodmodules/mgmt_config.py,sha256=YFVPC_Q3enDDf3tw9mQ4dWeGfvtCnxsEkIFu0616C4w,7163
 cwyodmodules/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cwyodmodules/api/chat_history.py,sha256=bVXFmhTHIfEiHv_nBrfizO-cQRHhKgrdcZ07OD1b0Tw,20683
 cwyodmodules/batch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -37,11 +37,10 @@ cwyodmodules/batch/utilities/helpers/azure_postgres_helper_light_rag.py,sha256=M
 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=S0EG0XzILrJ_iPCAwKN-kzx-yPl5K__MtBZee924DKE,15881
+cwyodmodules/batch/utilities/helpers/env_helper.py,sha256=0wJTaoMAR8rBo1-q2TQx2xqkYktbgk9EsUmZSsvwIJ0,15727
 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=KacIc2hHpWtgButUnR3sFC9StLS36HjYvNpTmf2f4gc,2834
 cwyodmodules/batch/utilities/helpers/config/agent_mode.py,sha256=8XMbu8dwMXva_xxeZNDlwOjDaZwIcwc-xJK1-QsaJ3w,82
 cwyodmodules/batch/utilities/helpers/config/assistant_strategy.py,sha256=uT8h646zEURU9x8oDOH7pWoZKb0Mw6dA2nJtA2M-ufg,171
 cwyodmodules/batch/utilities/helpers/config/config_helper.py,sha256=26na6YrLqRLhdSZxTSlOnJOkIcPbTcbFVuPEQTPT3WY,14908
@@ -110,8 +109,8 @@ cwyodmodules/graphrag/query/generate.py,sha256=BZiB6iw7PkIovw-CyYFogMHnDxK0Qu_4u
 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.46.dist-info/licenses/LICENSE,sha256=UqBDTipijsSW2ZSOXyTZnMsXmLoEHTgNEM0tL4g-Sso,1150
-cwyodmodules-0.3.46.dist-info/METADATA,sha256=RAKWb18OzQioSxPjwq-xR7ZX9NBrRL4jELP1p5gUmMM,2002
-cwyodmodules-0.3.46.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-cwyodmodules-0.3.46.dist-info/top_level.txt,sha256=99RENLbkdRX-qpJvsxZ5AfmTL5s6shSaKOWYpz1vwzg,13
-cwyodmodules-0.3.46.dist-info/RECORD,,
+cwyodmodules-0.3.48.dist-info/licenses/LICENSE,sha256=UqBDTipijsSW2ZSOXyTZnMsXmLoEHTgNEM0tL4g-Sso,1150
+cwyodmodules-0.3.48.dist-info/METADATA,sha256=9YMDuaHhW4n_XMuMVucVnaefjDVdVch9u7s6bl4O5Bg,2002
+cwyodmodules-0.3.48.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+cwyodmodules-0.3.48.dist-info/top_level.txt,sha256=99RENLbkdRX-qpJvsxZ5AfmTL5s6shSaKOWYpz1vwzg,13
+cwyodmodules-0.3.48.dist-info/RECORD,,

cwyodmodules/batch/utilities/helpers/secret_helper.py

@@ -1,79 +0,0 @@
-from azure.keyvault.secrets import SecretClient
-from mgmt_config import logger, identity
-
-
-class SecretHelper:
-    def __init__(self, keyvault_uri) -> None:
-        """
-        Initializes an instance of the SecretHelper class.
-
-        The constructor sets the USE_KEY_VAULT attribute based on the value of the USE_KEY_VAULT environment variable.
-        If USE_KEY_VAULT is set to "true" (case-insensitive), it initializes a SecretClient object using the
-        AZURE_KEY_VAULT_ENDPOINT environment variable and the DefaultAzureCredential.
-
-        Args:
-            None
-
-        Returns:
-            None
-        """
-        self.USE_KEY_VAULT = True
-        self.secret_client = None
-        if self.USE_KEY_VAULT:
-            credential = identity.get_credential()
-            self.secret_client = SecretClient(
-                vault_url=keyvault_uri,
-                credential=credential,
-                connection_verify=True,
-            )
-
-    @logger.trace_function(log_execution=True, log_args=False, log_result=False)
-    def get_secret(self, secret_name: str) -> str:
-        """
-        Retrieves the value of a secret from the environment variables or Azure Key Vault.
-
-        Args:
-            secret_name (str): The name of the secret or "".
-
-        Returns:
-            str: The value of the secret.
-
-        Raises:
-            None
-
-        """
-        secret_value = self.secret_client.get_secret(name=secret_name).value
-        return secret_value
-
-    @logger.trace_function(log_execution=True, log_args=False, log_result=False)
-    def set_secret(self, secret_name: str, secret_value: str) -> None:
-        """
-        Sets the value of a secret in Azure Key Vault only if it doesn't exist or has a different value.
-
-        Args:
-            secret_name (str): The name of the secret.
-            secret_value (str): The value to be stored.
-
-        Returns:
-            None
-
-        Raises:
-            None
-        """
-        try:
-            current_secret = self.secret_client.get_secret(name=secret_name)
-            if current_secret.value != secret_value:
-                self.secret_client.set_secret(name=secret_name, value=secret_value)
-            else:
-                logger.warning(
-                    f"Secret {secret_name} already has the same value, skipping update"
-                )
-        except Exception:
-            self.secret_client.set_secret(name=secret_name, value=secret_value)
-            logger.warning(f"Secret {secret_name} has been created")
-
-
-    @logger.trace_function(log_execution=True, log_args=False, log_result=False)
-    def get_secret_from_json(self, secret_name: str) -> str:
-        secret_value = self.secret_client.get_secret(secret_name).value
-        return secret_value