cwyodmodules 0.3.45__py3-none-any.whl → 0.3.47__py3-none-any.whl

cwyodmodules/batch/utilities/helpers/config/config_helper.py

@@ -1,362 +1,353 @@
-import os
-import json
-import functools
-from string import Template
-
-from ..azure_blob_storage_client import AzureBlobStorageClient
-from ...document_chunking.chunking_strategy import ChunkingStrategy, ChunkingSettings
-from ...document_loading import LoadingSettings, LoadingStrategy
-from .embedding_config import EmbeddingConfig
-from ...orchestrator.orchestration_strategy import OrchestrationStrategy
-from ...orchestrator import OrchestrationSettings
-from ..env_helper import EnvHelper
-from .assistant_strategy import AssistantStrategy
-from .conversation_flow import ConversationFlow
-from .database_type import DatabaseType
-from .agent_mode import AgentMode
-
-CONFIG_CONTAINER_NAME = "config"
-CONFIG_FILE_NAME = "active.json"
-ADVANCED_IMAGE_PROCESSING_FILE_TYPES = ["jpeg", "jpg", "png", "tiff", "bmp"]
-
-from mgmt_config import logger
-env_helper: EnvHelper = EnvHelper()
-log_execution = env_helper.LOG_EXECUTION
-log_args = env_helper.LOG_ARGS
-log_result = env_helper.LOG_RESULT
-
-
-class Config:
-    def __init__(self, config: dict):
-        self.prompts = Prompts(config["prompts"])
-        self.messages = Messages(config["messages"])
-        self.example = Example(config["example"])
-        self.logging = Logging(config["logging"])
-        self.document_processors = [
-            EmbeddingConfig(
-                document_type=c["document_type"],
-                chunking=ChunkingSettings(c["chunking"]),
-                loading=LoadingSettings(c["loading"]),
-                use_advanced_image_processing=c.get(
-                    "use_advanced_image_processing", False
-                ),
-            )
-            for c in config["document_processors"]
-        ]
-        self.env_helper = EnvHelper()
-        self.default_orchestration_settings = {
-            "strategy": self.env_helper.ORCHESTRATION_STRATEGY
-        }
-        self.orchestrator = OrchestrationSettings(
-            config.get("orchestrator", self.default_orchestration_settings)
-        )
-        self.integrated_vectorization_config = (
-            IntegratedVectorizationConfig(config["integrated_vectorization_config"])
-            if self.env_helper.AZURE_SEARCH_USE_INTEGRATED_VECTORIZATION
-            else None
-        )
-        self.enable_chat_history = config["enable_chat_history"]
-        self.database_type = config.get("database_type", self.env_helper.DATABASE_TYPE)
-        self.conversational_flow = config.get(
-            "conversational_flow", self.env_helper.CONVERSATION_FLOW
-        )
-        self.agent_mode = config.get("agent_mode", AgentMode.NORMAL.value)
-
-    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
-    def get_available_document_types(self) -> list[str]:
-        document_types = {
-            "txt",
-            "pdf",
-            "url",
-            "html",
-            "htm",
-            "md",
-            "jpeg",
-            "jpg",
-            "png",
-            "docx",
-        }
-        if self.env_helper.USE_ADVANCED_IMAGE_PROCESSING:
-            document_types.update(ADVANCED_IMAGE_PROCESSING_FILE_TYPES)
-
-        return sorted(document_types)
-
-    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
-    def get_advanced_image_processing_image_types(self):
-        return ADVANCED_IMAGE_PROCESSING_FILE_TYPES
-
-    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
-    def get_available_chunking_strategies(self):
-        return [c.value for c in ChunkingStrategy]
-
-    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
-    def get_available_loading_strategies(self):
-        return [c.value for c in LoadingStrategy]
-
-    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
-    def get_available_orchestration_strategies(self):
-        return [c.value for c in OrchestrationStrategy]
-
-    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
-    def get_available_ai_assistant_types(self):
-        return [c.value for c in AssistantStrategy]
-
-    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
-    def get_available_conversational_flows(self):
-        return [c.value for c in ConversationFlow]
-
-    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
-    def get_available_agent_modes(self):
-        return [c.value for c in AgentMode]
-
-
-# TODO: Change to AnsweringChain or something, Prompts is not a good name
-class Prompts:
-    def __init__(self, prompts: dict):
-        self.condense_question_prompt = prompts["condense_question_prompt"]
-        self.answering_system_prompt = prompts["answering_system_prompt"]
-        self.answering_user_prompt = prompts["answering_user_prompt"]
-        self.post_answering_prompt = prompts["post_answering_prompt"]
-        self.use_on_your_data_format = prompts["use_on_your_data_format"]
-        self.enable_post_answering_prompt = prompts["enable_post_answering_prompt"]
-        self.enable_content_safety = prompts["enable_content_safety"]
-        self.ai_assistant_type = prompts["ai_assistant_type"]
-        self.conversational_flow = prompts["conversational_flow"]
-
-
-class Example:
-    def __init__(self, example: dict):
-        self.documents = example["documents"]
-        self.user_question = example["user_question"]
-        self.answer = example["answer"]
-
-
-class Messages:
-    def __init__(self, messages: dict):
-        self.post_answering_filter = messages["post_answering_filter"]
-
-
-class Logging:
-    def __init__(self, logging: dict):
-        self.log_user_interactions = (
-            str(logging["log_user_interactions"]).lower() == "true"
-        )
-        self.log_tokens = str(logging["log_tokens"]).lower() == "true"
-
-
-class IntegratedVectorizationConfig:
-    def __init__(self, integrated_vectorization_config: dict):
-        self.max_page_length = integrated_vectorization_config["max_page_length"]
-        self.page_overlap_length = integrated_vectorization_config[
-            "page_overlap_length"
-        ]
-
-
-class ConfigHelper:
-    _default_config = None
-
-    @staticmethod
-    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
-    def _set_new_config_properties(config: dict, default_config: dict):
-        """
-        Function used to set newer properties that will not be present in older configs.
-        The function mutates the config object.
-        """
-        if config["prompts"].get("answering_system_prompt") is None:
-            config["prompts"]["answering_system_prompt"] = default_config["prompts"][
-                "answering_system_prompt"
-            ]
-
-        prompt_modified = (
-            config["prompts"].get("answering_prompt")
-            != default_config["prompts"]["answering_prompt"]
-        )
-
-        if config["prompts"].get("answering_user_prompt") is None:
-            if prompt_modified:
-                config["prompts"]["answering_user_prompt"] = config["prompts"].get(
-                    "answering_prompt"
-                )
-            else:
-                config["prompts"]["answering_user_prompt"] = default_config["prompts"][
-                    "answering_user_prompt"
-                ]
-
-        if config["prompts"].get("use_on_your_data_format") is None:
-            config["prompts"]["use_on_your_data_format"] = not prompt_modified
-
-        if config.get("example") is None:
-            config["example"] = default_config["example"]
-
-        if config["prompts"].get("ai_assistant_type") is None:
-            config["prompts"]["ai_assistant_type"] = default_config["prompts"][
-                "ai_assistant_type"
-            ]
-
-        if config.get("integrated_vectorization_config") is None:
-            config["integrated_vectorization_config"] = default_config[
-                "integrated_vectorization_config"
-            ]
-
-        if config["prompts"].get("conversational_flow") is None:
-            config["prompts"]["conversational_flow"] = default_config["prompts"][
-                "conversational_flow"
-            ]
-        if config.get("enable_chat_history") is None:
-            config["enable_chat_history"] = default_config["enable_chat_history"]
-            
-        if config.get("agent_mode") is None:
-            config["agent_mode"] = default_config["agent_mode"]
-
-    @staticmethod
-    @functools.cache
-    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
-    def get_active_config_or_default():
-        logger.info("Method get_active_config_or_default started")
-        env_helper = EnvHelper()
-        config = ConfigHelper.get_default_config()
-        if env_helper.LOAD_CONFIG_FROM_BLOB_STORAGE:
-            logger.info("Loading configuration from Blob Storage")
-            blob_client = AzureBlobStorageClient(container_name=CONFIG_CONTAINER_NAME)
-
-            if blob_client.file_exists(CONFIG_FILE_NAME):
-                logger.info("Configuration file found in Blob Storage")
-                default_config = config
-                config_file = blob_client.download_file(CONFIG_FILE_NAME)
-                config = json.loads(config_file)
-
-                ConfigHelper._set_new_config_properties(config, default_config)
-            else:
-                logger.info(
-                    "Configuration file not found in Blob Storage, using default configuration"
-                )
-
-        logger.info("Method get_active_config_or_default ended")
-        return Config(config)
-
-    @staticmethod
-    @functools.cache
-    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
-    def get_default_assistant_prompt():
-        config = ConfigHelper.get_default_config()
-        return config["prompts"]["answering_user_prompt"]
-
-    @staticmethod
-    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=log_result)
-    def save_config_as_active(config):
-        ConfigHelper.validate_config(config)
-        blob_client = AzureBlobStorageClient(container_name=CONFIG_CONTAINER_NAME)
-        blob_client = blob_client.upload_file(
-            json.dumps(config, indent=2),
-            CONFIG_FILE_NAME,
-            content_type="application/json",
-        )
-        ConfigHelper.get_active_config_or_default.cache_clear()
-
-    @staticmethod
-    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=log_result)
-    def validate_config(config: dict):
-        for document_processor in config.get("document_processors"):
-            document_type = document_processor.get("document_type")
-            unsupported_advanced_image_processing_file_type = (
-                document_type not in ADVANCED_IMAGE_PROCESSING_FILE_TYPES
-            )
-            if (
-                document_processor.get("use_advanced_image_processing")
-                and unsupported_advanced_image_processing_file_type
-            ):
-                raise Exception(
-                    f"Advanced image processing has not been enabled for document type {document_type}, as only {ADVANCED_IMAGE_PROCESSING_FILE_TYPES} file types are supported."
-                )
-
-    @staticmethod
-    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
-    def get_default_config():
-        if ConfigHelper._default_config is None:
-            env_helper = EnvHelper()
-
-            config_file_path = os.path.join(os.path.dirname(__file__), "default.json")
-            logger.info("Loading default config from %s", config_file_path)
-            with open(config_file_path, encoding="utf-8") as f:
-                ConfigHelper._default_config = json.loads(
-                    Template(f.read()).substitute(
-                        ORCHESTRATION_STRATEGY=env_helper.ORCHESTRATION_STRATEGY,
-                        LOG_USER_INTERACTIONS=(
-                            False
-                            if env_helper.DATABASE_TYPE == DatabaseType.POSTGRESQL.value
-                            else True
-                        ),
-                        LOG_TOKENS=(
-                            False
-                            if env_helper.DATABASE_TYPE == DatabaseType.POSTGRESQL.value
-                            else True
-                        ),
-                        CONVERSATION_FLOW=env_helper.CONVERSATION_FLOW,
-                        DATABASE_TYPE=env_helper.DATABASE_TYPE,
-                    )
-                )
-                if env_helper.USE_ADVANCED_IMAGE_PROCESSING:
-                    ConfigHelper._append_advanced_image_processors()
-
-        return ConfigHelper._default_config
-
-    @staticmethod
-    @functools.cache
-    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
-    def get_default_contract_assistant():
-        contract_file_path = os.path.join(
-            os.path.dirname(__file__), "default_contract_assistant_prompt.txt"
-        )
-        contract_assistant = ""
-        with open(contract_file_path, encoding="utf-8") as f:
-            contract_assistant = f.readlines()
-
-        return "".join([str(elem) for elem in contract_assistant])
-
-    @staticmethod
-    @functools.cache
-    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
-    def get_default_employee_assistant():
-        employee_file_path = os.path.join(
-            os.path.dirname(__file__), "default_employee_assistant_prompt.txt"
-        )
-        employee_assistant = ""
-        with open(employee_file_path, encoding="utf-8") as f:
-            employee_assistant = f.readlines()
-
-        return "".join([str(elem) for elem in employee_assistant])
-
-    @staticmethod
-    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
-    def clear_config():
-        ConfigHelper._default_config = None
-        ConfigHelper.get_active_config_or_default.cache_clear()
-
-    @staticmethod
-    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
-    def _append_advanced_image_processors():
-        image_file_types = ["jpeg", "jpg", "png", "tiff", "bmp"]
-        ConfigHelper._remove_processors_for_file_types(image_file_types)
-        ConfigHelper._default_config["document_processors"].extend(
-            [
-                {"document_type": file_type, "use_advanced_image_processing": True}
-                for file_type in image_file_types
-            ]
-        )
-
-    @staticmethod
-    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
-    def _remove_processors_for_file_types(file_types: list[str]):
-        document_processors = ConfigHelper._default_config["document_processors"]
-        document_processors = [
-            document_processor
-            for document_processor in document_processors
-            if document_processor["document_type"] not in file_types
-        ]
-        ConfigHelper._default_config["document_processors"] = document_processors
-
-    @staticmethod
-    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
-    def delete_config():
-        blob_client = AzureBlobStorageClient(container_name=CONFIG_CONTAINER_NAME)
-        blob_client.delete_file(CONFIG_FILE_NAME)
+import os
+import json
+import functools
+from string import Template
+
+from ..azure_blob_storage_client import AzureBlobStorageClient
+from ...document_chunking.chunking_strategy import ChunkingStrategy, ChunkingSettings
+from ...document_loading import LoadingSettings, LoadingStrategy
+from .embedding_config import EmbeddingConfig
+from ...orchestrator.orchestration_strategy import OrchestrationStrategy
+from ...orchestrator import OrchestrationSettings
+from ..env_helper import EnvHelper
+from .assistant_strategy import AssistantStrategy
+from .conversation_flow import ConversationFlow
+from .database_type import DatabaseType
+
+CONFIG_CONTAINER_NAME = "config"
+CONFIG_FILE_NAME = "active.json"
+ADVANCED_IMAGE_PROCESSING_FILE_TYPES = ["jpeg", "jpg", "png", "tiff", "bmp"]
+
+from mgmt_config import logger
+env_helper: EnvHelper = EnvHelper()
+log_execution = env_helper.LOG_EXECUTION
+log_args = env_helper.LOG_ARGS
+log_result = env_helper.LOG_RESULT
+
+
+class Config:
+    def __init__(self, config: dict):
+        self.prompts = Prompts(config["prompts"])
+        self.messages = Messages(config["messages"])
+        self.example = Example(config["example"])
+        self.logging = Logging(config["logging"])
+        self.document_processors = [
+            EmbeddingConfig(
+                document_type=c["document_type"],
+                chunking=ChunkingSettings(c["chunking"]),
+                loading=LoadingSettings(c["loading"]),
+                use_advanced_image_processing=c.get(
+                    "use_advanced_image_processing", False
+                ),
+            )
+            for c in config["document_processors"]
+        ]
+        self.env_helper = EnvHelper()
+        self.default_orchestration_settings = {
+            "strategy": self.env_helper.ORCHESTRATION_STRATEGY
+        }
+        self.orchestrator = OrchestrationSettings(
+            config.get("orchestrator", self.default_orchestration_settings)
+        )
+        self.integrated_vectorization_config = (
+            IntegratedVectorizationConfig(config["integrated_vectorization_config"])
+            if self.env_helper.AZURE_SEARCH_USE_INTEGRATED_VECTORIZATION
+            else None
+        )
+        self.enable_chat_history = config["enable_chat_history"]
+        self.database_type = config.get("database_type", self.env_helper.DATABASE_TYPE)
+        self.conversational_flow = config.get(
+            "conversational_flow", self.env_helper.CONVERSATION_FLOW
+        )
+
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
+    def get_available_document_types(self) -> list[str]:
+        document_types = {
+            "txt",
+            "pdf",
+            "url",
+            "html",
+            "htm",
+            "md",
+            "jpeg",
+            "jpg",
+            "png",
+            "docx",
+        }
+        if self.env_helper.USE_ADVANCED_IMAGE_PROCESSING:
+            document_types.update(ADVANCED_IMAGE_PROCESSING_FILE_TYPES)
+
+        return sorted(document_types)
+
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
+    def get_advanced_image_processing_image_types(self):
+        return ADVANCED_IMAGE_PROCESSING_FILE_TYPES
+
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
+    def get_available_chunking_strategies(self):
+        return [c.value for c in ChunkingStrategy]
+
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
+    def get_available_loading_strategies(self):
+        return [c.value for c in LoadingStrategy]
+
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
+    def get_available_orchestration_strategies(self):
+        return [c.value for c in OrchestrationStrategy]
+
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
+    def get_available_ai_assistant_types(self):
+        return [c.value for c in AssistantStrategy]
+
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
+    def get_available_conversational_flows(self):
+        return [c.value for c in ConversationFlow]
+
+
+# TODO: Change to AnsweringChain or something, Prompts is not a good name
+class Prompts:
+    def __init__(self, prompts: dict):
+        self.condense_question_prompt = prompts["condense_question_prompt"]
+        self.answering_system_prompt = prompts["answering_system_prompt"]
+        self.answering_user_prompt = prompts["answering_user_prompt"]
+        self.post_answering_prompt = prompts["post_answering_prompt"]
+        self.use_on_your_data_format = prompts["use_on_your_data_format"]
+        self.enable_post_answering_prompt = prompts["enable_post_answering_prompt"]
+        self.enable_content_safety = prompts["enable_content_safety"]
+        self.ai_assistant_type = prompts["ai_assistant_type"]
+        self.conversational_flow = prompts["conversational_flow"]
+
+
+class Example:
+    def __init__(self, example: dict):
+        self.documents = example["documents"]
+        self.user_question = example["user_question"]
+        self.answer = example["answer"]
+
+
+class Messages:
+    def __init__(self, messages: dict):
+        self.post_answering_filter = messages["post_answering_filter"]
+
+
+class Logging:
+    def __init__(self, logging: dict):
+        self.log_user_interactions = (
+            str(logging["log_user_interactions"]).lower() == "true"
+        )
+        self.log_tokens = str(logging["log_tokens"]).lower() == "true"
+
+
+class IntegratedVectorizationConfig:
+    def __init__(self, integrated_vectorization_config: dict):
+        self.max_page_length = integrated_vectorization_config["max_page_length"]
+        self.page_overlap_length = integrated_vectorization_config[
+            "page_overlap_length"
+        ]
+
+
+class ConfigHelper:
+    _default_config = None
+
+    @staticmethod
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
+    def _set_new_config_properties(config: dict, default_config: dict):
+        """
+        Function used to set newer properties that will not be present in older configs.
+        The function mutates the config object.
+        """
+        if config["prompts"].get("answering_system_prompt") is None:
+            config["prompts"]["answering_system_prompt"] = default_config["prompts"][
+                "answering_system_prompt"
+            ]
+
+        prompt_modified = (
+            config["prompts"].get("answering_prompt")
+            != default_config["prompts"]["answering_prompt"]
+        )
+
+        if config["prompts"].get("answering_user_prompt") is None:
+            if prompt_modified:
+                config["prompts"]["answering_user_prompt"] = config["prompts"].get(
+                    "answering_prompt"
+                )
+            else:
+                config["prompts"]["answering_user_prompt"] = default_config["prompts"][
+                    "answering_user_prompt"
+                ]
+
+        if config["prompts"].get("use_on_your_data_format") is None:
+            config["prompts"]["use_on_your_data_format"] = not prompt_modified
+
+        if config.get("example") is None:
+            config["example"] = default_config["example"]
+
+        if config["prompts"].get("ai_assistant_type") is None:
+            config["prompts"]["ai_assistant_type"] = default_config["prompts"][
+                "ai_assistant_type"
+            ]
+
+        if config.get("integrated_vectorization_config") is None:
+            config["integrated_vectorization_config"] = default_config[
+                "integrated_vectorization_config"
+            ]
+
+        if config["prompts"].get("conversational_flow") is None:
+            config["prompts"]["conversational_flow"] = default_config["prompts"][
+                "conversational_flow"
+            ]
+        if config.get("enable_chat_history") is None:
+            config["enable_chat_history"] = default_config["enable_chat_history"]
+
+    @staticmethod
+    @functools.cache
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
+    def get_active_config_or_default():
+        logger.info("Method get_active_config_or_default started")
+        env_helper = EnvHelper()
+        config = ConfigHelper.get_default_config()
+        if env_helper.LOAD_CONFIG_FROM_BLOB_STORAGE:
+            logger.info("Loading configuration from Blob Storage")
+            blob_client = AzureBlobStorageClient(container_name=CONFIG_CONTAINER_NAME)
+
+            if blob_client.file_exists(CONFIG_FILE_NAME):
+                logger.info("Configuration file found in Blob Storage")
+                default_config = config
+                config_file = blob_client.download_file(CONFIG_FILE_NAME)
+                config = json.loads(config_file)
+
+                ConfigHelper._set_new_config_properties(config, default_config)
+            else:
+                logger.info(
+                    "Configuration file not found in Blob Storage, using default configuration"
+                )
+
+        logger.info("Method get_active_config_or_default ended")
+        return Config(config)
+
+    @staticmethod
+    @functools.cache
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
+    def get_default_assistant_prompt():
+        config = ConfigHelper.get_default_config()
+        return config["prompts"]["answering_user_prompt"]
+
+    @staticmethod
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=log_result)
+    def save_config_as_active(config):
+        ConfigHelper.validate_config(config)
+        blob_client = AzureBlobStorageClient(container_name=CONFIG_CONTAINER_NAME)
+        blob_client = blob_client.upload_file(
+            json.dumps(config, indent=2),
+            CONFIG_FILE_NAME,
+            content_type="application/json",
+        )
+        ConfigHelper.get_active_config_or_default.cache_clear()
+
+    @staticmethod
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=log_result)
+    def validate_config(config: dict):
+        for document_processor in config.get("document_processors"):
+            document_type = document_processor.get("document_type")
+            unsupported_advanced_image_processing_file_type = (
+                document_type not in ADVANCED_IMAGE_PROCESSING_FILE_TYPES
+            )
+            if (
+                document_processor.get("use_advanced_image_processing")
+                and unsupported_advanced_image_processing_file_type
+            ):
+                raise Exception(
+                    f"Advanced image processing has not been enabled for document type {document_type}, as only {ADVANCED_IMAGE_PROCESSING_FILE_TYPES} file types are supported."
+                )
+
+    @staticmethod
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
+    def get_default_config():
+        if ConfigHelper._default_config is None:
+            env_helper = EnvHelper()
+
+            config_file_path = os.path.join(os.path.dirname(__file__), "default.json")
+            logger.info("Loading default config from %s", config_file_path)
+            with open(config_file_path, encoding="utf-8") as f:
+                ConfigHelper._default_config = json.loads(
+                    Template(f.read()).substitute(
+                        ORCHESTRATION_STRATEGY=env_helper.ORCHESTRATION_STRATEGY,
+                        LOG_USER_INTERACTIONS=(
+                            False
+                            if env_helper.DATABASE_TYPE == DatabaseType.POSTGRESQL.value
+                            else True
+                        ),
+                        LOG_TOKENS=(
+                            False
+                            if env_helper.DATABASE_TYPE == DatabaseType.POSTGRESQL.value
+                            else True
+                        ),
+                        CONVERSATION_FLOW=env_helper.CONVERSATION_FLOW,
+                        DATABASE_TYPE=env_helper.DATABASE_TYPE,
+                    )
+                )
+                if env_helper.USE_ADVANCED_IMAGE_PROCESSING:
+                    ConfigHelper._append_advanced_image_processors()
+
+        return ConfigHelper._default_config
+
+    @staticmethod
+    @functools.cache
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
+    def get_default_contract_assistant():
+        contract_file_path = os.path.join(
+            os.path.dirname(__file__), "default_contract_assistant_prompt.txt"
+        )
+        contract_assistant = ""
+        with open(contract_file_path, encoding="utf-8") as f:
+            contract_assistant = f.readlines()
+
+        return "".join([str(elem) for elem in contract_assistant])
+
+    @staticmethod
+    @functools.cache
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
+    def get_default_employee_assistant():
+        employee_file_path = os.path.join(
+            os.path.dirname(__file__), "default_employee_assistant_prompt.txt"
+        )
+        employee_assistant = ""
+        with open(employee_file_path, encoding="utf-8") as f:
+            employee_assistant = f.readlines()
+
+        return "".join([str(elem) for elem in employee_assistant])
+
+    @staticmethod
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
+    def clear_config():
+        ConfigHelper._default_config = None
+        ConfigHelper.get_active_config_or_default.cache_clear()
+
+    @staticmethod
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
+    def _append_advanced_image_processors():
+        image_file_types = ["jpeg", "jpg", "png", "tiff", "bmp"]
+        ConfigHelper._remove_processors_for_file_types(image_file_types)
+        ConfigHelper._default_config["document_processors"].extend(
+            [
+                {"document_type": file_type, "use_advanced_image_processing": True}
+                for file_type in image_file_types
+            ]
+        )
+
+    @staticmethod
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
+    def _remove_processors_for_file_types(file_types: list[str]):
+        document_processors = ConfigHelper._default_config["document_processors"]
+        document_processors = [
+            document_processor
+            for document_processor in document_processors
+            if document_processor["document_type"] not in file_types
+        ]
+        ConfigHelper._default_config["document_processors"] = document_processors
+
+    @staticmethod
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
+    def delete_config():
+        blob_client = AzureBlobStorageClient(container_name=CONFIG_CONTAINER_NAME)
+        blob_client.delete_file(CONFIG_FILE_NAME)

cwyodmodules/batch/utilities/helpers/config/default.json

@@ -143,6 +143,5 @@
     "strategy": "${ORCHESTRATION_STRATEGY}"
   },
   "enable_chat_history": true,
-  "database_type": "${DATABASE_TYPE}",
-  "agent_mode": "off"
+  "database_type": "${DATABASE_TYPE}"
 }

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, create_keyvault_client
 
 
 
@@ -33,10 +32,10 @@ class EnvHelper:
         # Wrapper for Azure Key Vault
         os.environ["APPLICATIONINSIGHTS_ENABLED"] = "true"
 
-        self.secretHelper = SecretHelper(keyvault_uri=os.environ["key_vault_uri"])
+        self.secretHelper = keyvault
 
-        self.secretHelperHead = SecretHelper(
-            keyvault_uri=os.environ["head_key_vault_uri"]
+        self.secretHelperHead = create_keyvault_client(
+            vault_url=os.environ["head_key_vault_uri"]
         )
 
         # self.secretHelper = SecretHelper(

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.45.dist-info → cwyodmodules-0.3.47.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cwyodmodules
-Version: 0.3.45
+Version: 0.3.47
 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.5
 Dynamic: license-file
 
 # paddypy

{cwyodmodules-0.3.45.dist-info → cwyodmodules-0.3.47.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,17 +37,16 @@ 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=avZCU8eZOjrhGLu8LcjQr_kjoLzkL_75UemCGQTINxY,15827
 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=rHP9YgkwG9UodAlRALesQ54M_dDsl801s_gP710vqHk,15685
+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
 cwyodmodules/batch/utilities/helpers/config/database_type.py,sha256=Zmmlh1NAKDdd-2ei478boncRKcx8v3lDkPf4kO2j4ss,132
-cwyodmodules/batch/utilities/helpers/config/default.json,sha256=kwZg04XoLlKnva22o_YIuC31jWeSTpuewJTG9oyJP9o,15430
+cwyodmodules/batch/utilities/helpers/config/default.json,sha256=865avujlLpO2to1dlh7yXfs0F57noJNrld2zwVYirwM,15406
 cwyodmodules/batch/utilities/helpers/config/default_contract_assistant_prompt.txt,sha256=X39WGcxzQPIvqG7NpAMPsgmSwSyMEoK1DVWiuEHEHRg,3210
 cwyodmodules/batch/utilities/helpers/config/default_employee_assistant_prompt.txt,sha256=toQFo0wXYrEK7zAItAS9rbtyhT6DJZKBhiL6C9VPUQk,3942
 cwyodmodules/batch/utilities/helpers/config/embedding_config.py,sha256=9pCJxpsouln9dngjVHaKGFYP14PrwmSts_UFDytSiVk,950
@@ -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.45.dist-info/licenses/LICENSE,sha256=UqBDTipijsSW2ZSOXyTZnMsXmLoEHTgNEM0tL4g-Sso,1150
-cwyodmodules-0.3.45.dist-info/METADATA,sha256=dhCo5n3jzElUG7B7kF2FimtLv8dXN5kYYF15zVZaaNU,2002
-cwyodmodules-0.3.45.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-cwyodmodules-0.3.45.dist-info/top_level.txt,sha256=99RENLbkdRX-qpJvsxZ5AfmTL5s6shSaKOWYpz1vwzg,13
-cwyodmodules-0.3.45.dist-info/RECORD,,
+cwyodmodules-0.3.47.dist-info/licenses/LICENSE,sha256=UqBDTipijsSW2ZSOXyTZnMsXmLoEHTgNEM0tL4g-Sso,1150
+cwyodmodules-0.3.47.dist-info/METADATA,sha256=YSDKlXiAglRuJPkC06tOr1GIB9zXKgRfGgMTKl-cFWE,2002
+cwyodmodules-0.3.47.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+cwyodmodules-0.3.47.dist-info/top_level.txt,sha256=99RENLbkdRX-qpJvsxZ5AfmTL5s6shSaKOWYpz1vwzg,13
+cwyodmodules-0.3.47.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