cwyodmodules 0.3.36__py3-none-any.whl → 0.3.37__py3-none-any.whl

cwyodmodules/batch/utilities/integrated_vectorization/azure_search_index.py

@@ -26,6 +26,7 @@ from ..helpers.llm_helper import LLMHelper
 
 from logging_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
 
@@ -48,7 +49,7 @@ class AzureSearchIndex:
         )
 
     @property
-    @logger.trace_function(log_args=log_args, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
     def search_dimensions(self) -> int:
         if AzureSearchIndex._search_dimension is None:
             AzureSearchIndex._search_dimension = len(
@@ -56,7 +57,7 @@ class AzureSearchIndex:
             )
         return AzureSearchIndex._search_dimension
 
-    @logger.trace_function(log_args=log_args, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
     def create_or_update_index(self):
         # Create a search index
         fields = [
@@ -119,7 +120,7 @@ class AzureSearchIndex:
         logger.info(f"{result.name} index created successfully.")
         return result
 
-    @logger.trace_function(log_args=log_args, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
     def get_vector_search_config(self):
         if self.env_helper.is_auth_type_keys():
             azure_open_ai_parameters = AzureOpenAIParameters(
@@ -172,7 +173,7 @@ class AzureSearchIndex:
             ],
         )
 
-    @logger.trace_function(log_args=log_args, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
     def get_semantic_search_config(self):
         semantic_config = SemanticConfiguration(
             name=self.env_helper.AZURE_SEARCH_SEMANTIC_SEARCH_CONFIG,

cwyodmodules/batch/utilities/integrated_vectorization/azure_search_indexer.py

@@ -8,6 +8,7 @@ from azure.core.credentials import AzureKeyCredential
 from ...utilities.helpers.env_helper import EnvHelper
 from logging_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
 
@@ -26,7 +27,7 @@ class AzureSearchIndexer:
             ),
         )
 
-    @logger.trace_function(log_args=log_args, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
     def create_or_update_indexer(self, indexer_name: str, skillset_name: str):
         indexer = SearchIndexer(
             name=indexer_name,
@@ -64,7 +65,7 @@ class AzureSearchIndexer:
         )
         return indexer_result
 
-    @logger.trace_function(log_args=log_args, log_result=log_result)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
     def run_indexer(self, indexer_name: str):
         self.indexer_client.reset_indexer(indexer_name)
         self.indexer_client.run_indexer(indexer_name)
@@ -72,7 +73,7 @@ class AzureSearchIndexer:
             f" {indexer_name} is created and running. If queries return no results, please wait a bit and try again."
         )
 
-    @logger.trace_function(log_args=log_args, log_result=log_result)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
     def indexer_exists(self, indexer_name: str):
         return indexer_name in [
             name for name in self.indexer_client.get_indexer_names()

cwyodmodules/batch/utilities/integrated_vectorization/azure_search_skillset.py

@@ -20,6 +20,7 @@ from azure.core.credentials import AzureKeyCredential
 from ...utilities.helpers.env_helper import EnvHelper
 from logging_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
 
@@ -43,7 +44,7 @@ class AzureSearchSkillset:
         )
         self.integrated_vectorization_config = integrated_vectorization_config
 
-    @logger.trace_function(log_args=log_args, log_result=log_result)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
     def create_skillset(self):
         skillset_name = f"{self.env_helper.AZURE_SEARCH_INDEX}-skillset"
 

cwyodmodules/batch/utilities/orchestrator/lang_chain_agent.py

@@ -15,6 +15,7 @@ from ..common.answer import Answer
 from ...utilities.helpers.env_helper import EnvHelper
 from logging_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
 
@@ -51,7 +52,7 @@ class LangChainAgent(OrchestratorBase):
             ),
         ]
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def run_tool(self, user_message: str) -> str:
         """
         Executes the question answering tool with the provided user message.
@@ -67,7 +68,7 @@ class LangChainAgent(OrchestratorBase):
         )
         return answer.to_json()
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def run_text_processing_tool(self, user_message: str) -> str:
         """
         Executes the text processing tool with the provided user message.
@@ -83,7 +84,7 @@ class LangChainAgent(OrchestratorBase):
         )
         return answer.to_json()
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     async def orchestrate(
         self, user_message: str, chat_history: List[dict], **kwargs: dict
     ) -> list[dict]:

cwyodmodules/batch/utilities/orchestrator/open_ai_functions.py

@@ -12,6 +12,7 @@ from ..common.answer import Answer
 from ...utilities.helpers.env_helper import EnvHelper
 from logging_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
 
@@ -68,7 +69,7 @@ class OpenAIFunctionsOrchestrator(OrchestratorBase):
             },
         ]
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     async def orchestrate(
         self, user_message: str, chat_history: List[dict], **kwargs: dict
     ) -> list[dict]:

cwyodmodules/batch/utilities/orchestrator/orchestrator_base.py

@@ -9,6 +9,7 @@ from ..tools.content_safety_checker import ContentSafetyChecker
 from ...utilities.helpers.env_helper import EnvHelper
 from logging_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
 
@@ -34,7 +35,7 @@ class OrchestratorBase(ABC):
         self.content_safety_checker = ContentSafetyChecker()
         self.output_parser = OutputParserTool()
 
-    @logger.trace_function(log_args=log_args, log_result=log_result)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
     def log_tokens(self, prompt_tokens: int, completion_tokens: int) -> None:
         """
         Logs the number of tokens used in the prompt and completion phases of a conversation.
@@ -48,7 +49,7 @@ class OrchestratorBase(ABC):
         self.tokens["total"] += prompt_tokens + completion_tokens
 
     @abstractmethod
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     async def orchestrate(
         self,
         user_message: str,
@@ -94,7 +95,7 @@ class OrchestratorBase(ABC):
 
         return None
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def call_content_safety_output(
         self, user_message: str, answer: str
     ) -> Optional[list[dict]]:
@@ -121,7 +122,7 @@ class OrchestratorBase(ABC):
 
         return None
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     async def handle_message(
         self,
         user_message: str,

cwyodmodules/batch/utilities/orchestrator/prompt_flow.py

@@ -10,6 +10,7 @@ from ..helpers.env_helper import EnvHelper
 
 from logging_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
 
@@ -37,7 +38,7 @@ class PromptFlowOrchestrator(OrchestratorBase):
 
         logger.info("PromptFlowOrchestrator initialized.")
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     async def orchestrate(
         self, user_message: str, chat_history: List[dict], **kwargs: dict
     ) -> list[dict]:
@@ -111,7 +112,7 @@ class PromptFlowOrchestrator(OrchestratorBase):
         logger.info("Orchestration completed successfully.")
         return messages
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def transform_chat_history(self, chat_history: List[dict]) -> List[dict]:
         """
         Transform the chat history into a format suitable for the Prompt Flow service.
@@ -142,7 +143,7 @@ class PromptFlowOrchestrator(OrchestratorBase):
         logger.info("Chat history transformation completed.")
         return transformed_chat_history
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def transform_data_into_file(
         self, user_message: str, chat_history: List[dict]
     ) -> str:
@@ -164,7 +165,7 @@ class PromptFlowOrchestrator(OrchestratorBase):
         logger.info("Temporary file created")
         return file.name
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def transform_citations_into_source_documents(
         self, citations: dict
     ) -> List[SourceDocument]:

cwyodmodules/batch/utilities/orchestrator/semantic_kernel_orchestrator.py

@@ -22,6 +22,7 @@ from .orchestrator_base import OrchestratorBase
 
 from logging_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
 
@@ -42,7 +43,7 @@ class SemanticKernelOrchestrator(OrchestratorBase):
             plugin=PostAnsweringPlugin(), plugin_name="PostAnswering"
         )
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     async def orchestrate(
         self, user_message: str, chat_history: list[dict], user_info, **kwargs: dict
     ) -> list[dict]:

cwyodmodules/batch/utilities/parser/output_parser_tool.py

@@ -7,6 +7,7 @@ from ..common.source_document import SourceDocument
 from ...utilities.helpers.env_helper import EnvHelper
 from logging_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
 
@@ -23,7 +24,7 @@ class OutputParserTool(ParserBase):
         """
         self.name = "OutputParser"
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def _clean_up_answer(self, answer: str) -> str:
         """
         Cleans up the answer by replacing double spaces with single spaces.
@@ -36,7 +37,7 @@ class OutputParserTool(ParserBase):
         """
         return answer.replace("  ", " ")
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def _get_source_docs_from_answer(self, answer: str) -> List[int]:
         """
         Extracts all document references from the answer and returns them as a list of integers.
@@ -50,7 +51,7 @@ class OutputParserTool(ParserBase):
         results = re.findall(r"\[doc(\d+)\]", answer)
         return [int(i) for i in results]
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def _make_doc_references_sequential(self, answer: str) -> str:
         """
         Makes document references in the answer sequential.
@@ -72,7 +73,7 @@ class OutputParserTool(ParserBase):
             offset += len(f"[doc{i + 1}]") - (end - start)
         return updated_answer
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def parse(
         self,
         question: str,

cwyodmodules/batch/utilities/plugins/outlook_calendar_plugin.py

@@ -6,6 +6,7 @@ from ...utilities.helpers.env_helper import EnvHelper
 
 from logging_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
 
@@ -16,7 +17,7 @@ class OutlookCalendarPlugin:
         self.user_info = user_info
         self.env_helper = EnvHelper()
 
-    @logger.trace_function(log_args=log_args, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
     def _get_access_token(self) -> str:
         logger.info("Retrieving access token from user info")
         access_token = self.user_info.get("access_token", None)
@@ -27,7 +28,7 @@ class OutlookCalendarPlugin:
         return access_token
 
     @kernel_function(name="get_calendar_events", description="Get upcoming Outlook calendar events, appointments, metings, etc.")
-    @logger.trace_function(log_args=log_args, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
     def get_calendar_events(self, days: int = 1) -> str:
         language = self.env_helper.AZURE_MAIN_CHAT_LANGUAGE
         logger.info("Method get_calendar_events of OutlookCalendarPlugin started")
@@ -75,7 +76,7 @@ class OutlookCalendarPlugin:
         return answer
 
     @kernel_function(name="schedule_appointment", description="Schedule a new Outlook calendar appointment, meeting, etc.")
-    @logger.trace_function(log_args=log_args, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
     def schedule_appointment(self, subject: str, start_time: str, end_time: str) -> str:
         logger.info("Method schedule_appointment of OutlookCalendarPlugin started")
         language = self.env_helper.AZURE_MAIN_CHAT_LANGUAGE

cwyodmodules/batch/utilities/search/azure_search_handler.py

@@ -11,6 +11,7 @@ import tiktoken
 from ...utilities.helpers.env_helper import EnvHelper
 from logging_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
 
@@ -23,17 +24,17 @@ class AzureSearchHandler(SearchHandlerBase):
         self.llm_helper = LLMHelper()
         self.azure_computer_vision_client = AzureComputerVisionClient(env_helper)
 
-    @logger.trace_function(log_args=log_args, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
     def create_search_client(self):
         return AzureSearchHelper().get_search_client()
 
-    @logger.trace_function(log_args=log_args, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
     def perform_search(self, filename):
         return self.search_client.search(
             "*", select="title, content, metadata", filter=f"title eq '{filename}'"
         )
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def process_results(self, results):
         logger.info("Processing search results")
         if results is None:
@@ -47,13 +48,13 @@ class AzureSearchHandler(SearchHandlerBase):
         logger.info("Processed results")
         return data
 
-    @logger.trace_function(log_args=log_args, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
     def get_files(self):
         return self.search_client.search(
             "*", select="id, title", include_total_count=True
         )
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def output_results(self, results):
         files = {}
         for result in results:
@@ -66,7 +67,7 @@ class AzureSearchHandler(SearchHandlerBase):
 
         return files
 
-    @logger.trace_function(log_args=log_args, log_result=log_result)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
     def delete_files(self, files):
         ids_to_delete = []
         files_to_delete = []
@@ -78,7 +79,7 @@ class AzureSearchHandler(SearchHandlerBase):
 
         return ", ".join(files_to_delete)
 
-    @logger.trace_function(log_args=log_args, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
     def search_by_blob_url(self, blob_url):
         return self.search_client.search(
             "*",
@@ -87,7 +88,7 @@ class AzureSearchHandler(SearchHandlerBase):
             filter=f"source eq '{blob_url}_SAS_TOKEN_PLACEHOLDER_'",
         )
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def query_search(self, question) -> List[SourceDocument]:
         logger.info(f"Performing query search for question: {question}")
         encoding = tiktoken.get_encoding(self._ENCODER_NAME)
@@ -116,7 +117,7 @@ class AzureSearchHandler(SearchHandlerBase):
         logger.info("Converting search results to SourceDocument list")
         return self._convert_to_source_documents(results)
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def _semantic_search(
         self,
         question: str,
@@ -151,7 +152,7 @@ class AzureSearchHandler(SearchHandlerBase):
             top=self.env_helper.AZURE_SEARCH_TOP_K,
         )
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def _hybrid_search(
         self,
         question: str,
@@ -184,7 +185,7 @@ class AzureSearchHandler(SearchHandlerBase):
             top=self.env_helper.AZURE_SEARCH_TOP_K,
         )
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def _convert_to_source_documents(self, search_results) -> List[SourceDocument]:
         source_documents = []
         for source in search_results:

cwyodmodules/batch/utilities/search/azure_search_handler_light_rag.py

@@ -7,6 +7,7 @@ import json
 from ...utilities.helpers.env_helper import EnvHelper
 from logging_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
 
@@ -16,17 +17,17 @@ class AzureSearchHandlerLightRag(SearchHandlerBase):
         super().__init__(env_helper)
         self.light_rag_helper = LightRAGHelper(env_helper)
 
-    @logger.trace_function(log_args=log_args, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
     def create_search_client(self):
         return self.light_rag_helper.get_search_client()
 
-    @logger.trace_function(log_args=log_args, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
     def perform_search(self, filename):
         return self.light_rag_helper.search(
             "*", select="title, content, metadata", filter=f"title eq '{filename}'"
         )
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def process_results(self, results):
         logger.info("Processing search results")
         if results is None:
@@ -39,11 +40,11 @@ class AzureSearchHandlerLightRag(SearchHandlerBase):
         logger.info("Processed results")
         return data
 
-    @logger.trace_function(log_args=log_args, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
     def get_files(self):
         return self.light_rag_helper.get_files()
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def output_results(self, results):
         files = {}
         for result in results:
@@ -56,7 +57,7 @@ class AzureSearchHandlerLightRag(SearchHandlerBase):
 
         return files
 
-    @logger.trace_function(log_args=log_args, log_result=log_result)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
     def delete_files(self, files):
         ids_to_delete = []
         files_to_delete = []
@@ -68,18 +69,18 @@ class AzureSearchHandlerLightRag(SearchHandlerBase):
 
         return ", ".join(files_to_delete)
 
-    @logger.trace_function(log_args=log_args, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
     def search_by_blob_url(self, blob_url):
         return self.light_rag_helper.search_by_blob_url(blob_url)
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def query_search(self, question) -> List[SourceDocument]:
         logger.info(f"Performing query search for question: {question}")
         results = self.light_rag_helper.query_search(question)
         logger.info("Converting search results to SourceDocument list")
         return self._convert_to_source_documents(results)
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def _convert_to_source_documents(self, search_results) -> List[SourceDocument]:
         source_documents = []
         for source in search_results:

cwyodmodules/batch/utilities/search/integrated_vectorization_search_handler.py

@@ -10,13 +10,14 @@ import re
 from ...utilities.helpers.env_helper import EnvHelper
 from logging_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 IntegratedVectorizationSearchHandler(SearchHandlerBase):
     def __init__(self):
         self.azure_identity_helper = AzureIdentityHelper()
-    @logger.trace_function(log_args=log_args, log_result=False)    
+    @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.")
         if self._check_index_exists():
@@ -31,7 +32,7 @@ class IntegratedVectorizationSearchHandler(SearchHandlerBase):
                 ),
             )
 
-    @logger.trace_function(log_args=log_args, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
     def perform_search(self, filename):
         logger.info(f"Performing search for file: {filename}.")
         if self._check_index_exists():
@@ -41,7 +42,7 @@ class IntegratedVectorizationSearchHandler(SearchHandlerBase):
                 filter=f"title eq '{filename}'",
             )
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def process_results(self, results):
         logger.info("Processing search results.")
         if results is None:
@@ -54,7 +55,7 @@ class IntegratedVectorizationSearchHandler(SearchHandlerBase):
         logger.info(f"Processed {len(data)} results.")
         return data
 
-    @logger.trace_function(log_args=log_args, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
     def get_files(self):
         logger.info("Fetching files from search index.")
         if self._check_index_exists():
@@ -62,7 +63,7 @@ class IntegratedVectorizationSearchHandler(SearchHandlerBase):
                 "*", select="id, chunk_id, title", include_total_count=True
             )
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def output_results(self, results):
         logger.info("Organizing search results into output format.")
         files = {}
@@ -75,7 +76,7 @@ class IntegratedVectorizationSearchHandler(SearchHandlerBase):
                 files[filename] = [id]
         return files
 
-    @logger.trace_function(log_args=log_args, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=False)
     def search_by_blob_url(self, blob_url: str):
         logger.info(f"Searching by blob URL: {blob_url}.")
         if self._check_index_exists():
@@ -87,7 +88,7 @@ class IntegratedVectorizationSearchHandler(SearchHandlerBase):
                 filter=f"title eq '{title}'",
             )
 
-    @logger.trace_function(log_args=log_args, log_result=log_result)
+    @logger.trace_function(log_execution=log_execution, log_args=log_args, log_result=log_result)
     def delete_files(self, files):
         logger.info("Deleting files.")
         ids_to_delete = []
@@ -102,7 +103,7 @@ class IntegratedVectorizationSearchHandler(SearchHandlerBase):
         logger.info(f"Deleted files: {', '.join(files_to_delete)}.")
         return ", ".join(files_to_delete)
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def query_search(self, question) -> List[SourceDocument]:
         logger.info(f"Querying search for question: {question}.")
         if self._check_index_exists():
@@ -116,7 +117,7 @@ class IntegratedVectorizationSearchHandler(SearchHandlerBase):
             logger.info("Search completed. Converting results to SourceDocuments.")
             return self._convert_to_source_documents(search_results)
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def _hybrid_search(self, question: str):
         logger.info(f"Performing hybrid search for question: {question}.")
         vector_query = VectorizableTextQuery(
@@ -131,7 +132,7 @@ class IntegratedVectorizationSearchHandler(SearchHandlerBase):
             top=self.env_helper.AZURE_SEARCH_TOP_K,
         )
 
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def _semantic_search(self, question: str):
         logger.info(f"Performing semantic search for question: {question}.")
         vector_query = VectorizableTextQuery(
@@ -150,7 +151,7 @@ class IntegratedVectorizationSearchHandler(SearchHandlerBase):
             query_answer="extractive",
             top=self.env_helper.AZURE_SEARCH_TOP_K,
         )
-    @logger.trace_function(log_args=False, log_result=False)
+    @logger.trace_function(log_execution=log_execution, log_args=False, log_result=False)
     def _convert_to_source_documents(self, search_results) -> List[SourceDocument]:
         logger.info("Converting search results to SourceDocument objects.")
         source_documents = []