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

cwyodmodules/batch/utilities/helpers/azure_search_helper.py

@@ -1,4 +1,3 @@
-import logging
 from azure.identity import ChainedTokenCredential, DefaultAzureCredential
 from typing import Union
 from langchain_community.vectorstores import AzureSearch
@@ -27,11 +26,12 @@ from azure.search.documents.indexes.models import (
 )
 
 from ..helpers.azure_computer_vision_client import AzureComputerVisionClient
-from .llm_helper import LLMHelper
-from .env_helper import EnvHelper
-
-logger = logging.getLogger("__main__")
-
+from ..helpers.llm_helper import LLMHelper
+from ..helpers.env_helper import EnvHelper
+from logging_config import logger
+env_helper: EnvHelper = EnvHelper()
+log_args = env_helper.LOG_ARGS
+log_result = env_helper.LOG_RESULT
 
 class AzureSearchHelper:
     _search_dimension: int | None = None
@@ -47,6 +47,7 @@ class AzureSearchHelper:
         self.search_index_client = self._create_search_index_client(search_credential)
         self.azure_computer_vision_client = AzureComputerVisionClient(self.env_helper)
 
+    @logger.trace_function(log_args=log_args, log_result=False)
     def _search_credential(self):
         if self.env_helper.is_auth_type_keys():
             return AzureKeyCredential(self.env_helper.AZURE_SEARCH_KEY)
@@ -54,6 +55,7 @@ class AzureSearchHelper:
             credential = self.azure_identity_helper.get_credential()
             return credential
 
+    @logger.trace_function(log_args=False, log_result=False)
     def _create_search_client(
         self, search_credential: Union[AzureKeyCredential, ChainedTokenCredential, DefaultAzureCredential]
     ) -> SearchClient:
@@ -63,6 +65,7 @@ class AzureSearchHelper:
             credential=search_credential,
         )
 
+    @logger.trace_function(log_args=False, log_result=False)
     def _create_search_index_client(
         self, search_credential: Union[AzureKeyCredential, ChainedTokenCredential, DefaultAzureCredential]
     ):
@@ -70,11 +73,13 @@ class AzureSearchHelper:
             endpoint=self.env_helper.AZURE_SEARCH_SERVICE, credential=search_credential
         )
 
+    @logger.trace_function(log_args=log_args, log_result=False)
     def get_search_client(self) -> SearchClient:
         self.create_index()
         return self.search_client
 
     @property
+    @logger.trace_function(log_args=log_args, log_result=False)
     def search_dimensions(self) -> int:
         if AzureSearchHelper._search_dimension is None:
             AzureSearchHelper._search_dimension = len(
@@ -83,6 +88,7 @@ class AzureSearchHelper:
         return AzureSearchHelper._search_dimension
 
     @property
+    @logger.trace_function(log_args=log_args, log_result=False)
     def image_search_dimensions(self) -> int:
         if AzureSearchHelper._image_search_dimension is None:
             AzureSearchHelper._image_search_dimension = len(
@@ -90,6 +96,7 @@ class AzureSearchHelper:
             )
         return AzureSearchHelper._image_search_dimension
 
+    @logger.trace_function(log_args=log_args, log_result=False)
     def create_index(self):
         fields = [
             SimpleField(
@@ -204,11 +211,13 @@ class AzureSearchHelper:
             )
             self.search_index_client.create_index(index)
 
+    @logger.trace_function(log_args=log_args, log_result=log_result)
     def _index_not_exists(self, index_name: str) -> bool:
         return index_name not in [
             name for name in self.search_index_client.list_index_names()
         ]
 
+    @logger.trace_function(log_args=log_args, log_result=False)
     def get_conversation_logger(self):
         fields = [
             SimpleField(

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

@@ -1,6 +1,5 @@
 import os
 import json
-import logging
 import functools
 from string import Template
 
@@ -18,7 +17,11 @@ from .database_type import DatabaseType
 CONFIG_CONTAINER_NAME = "config"
 CONFIG_FILE_NAME = "active.json"
 ADVANCED_IMAGE_PROCESSING_FILE_TYPES = ["jpeg", "jpg", "png", "tiff", "bmp"]
-logger = logging.getLogger("__main__")
+
+from logging_config import logger
+env_helper: EnvHelper = EnvHelper()
+log_args = env_helper.LOG_ARGS
+log_result = env_helper.LOG_RESULT
 
 
 class Config:
@@ -56,6 +59,7 @@ class Config:
             "conversational_flow", self.env_helper.CONVERSATION_FLOW
         )
 
+    @logger.trace_function(log_args=log_args, log_result=log_result)
     def get_available_document_types(self) -> list[str]:
         document_types = {
             "txt",
@@ -74,21 +78,27 @@ class Config:
 
         return sorted(document_types)
 
+    @logger.trace_function(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_args=log_args, log_result=log_result)
     def get_available_chunking_strategies(self):
         return [c.value for c in ChunkingStrategy]
 
+    @logger.trace_function(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_args=log_args, log_result=log_result)
     def get_available_orchestration_strategies(self):
         return [c.value for c in OrchestrationStrategy]
 
+    @logger.trace_function(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_args=log_args, log_result=log_result)
     def get_available_conversational_flows(self):
         return [c.value for c in ConversationFlow]
 
@@ -139,6 +149,7 @@ class ConfigHelper:
     _default_config = None
 
     @staticmethod
+    @logger.trace_function(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.
@@ -189,6 +200,7 @@ class ConfigHelper:
 
     @staticmethod
     @functools.cache
+    @logger.trace_function(log_args=False, log_result=False)
     def get_active_config_or_default():
         logger.info("Method get_active_config_or_default started")
         env_helper = EnvHelper()
@@ -214,11 +226,13 @@ class ConfigHelper:
 
     @staticmethod
     @functools.cache
+    @logger.trace_function(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_args=False, log_result=log_result)
     def save_config_as_active(config):
         ConfigHelper.validate_config(config)
         blob_client = AzureBlobStorageClient(container_name=CONFIG_CONTAINER_NAME)
@@ -230,6 +244,7 @@ class ConfigHelper:
         ConfigHelper.get_active_config_or_default.cache_clear()
 
     @staticmethod
+    @logger.trace_function(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")
@@ -245,6 +260,7 @@ class ConfigHelper:
                 )
 
     @staticmethod
+    @logger.trace_function(log_args=log_args, log_result=False)
     def get_default_config():
         if ConfigHelper._default_config is None:
             env_helper = EnvHelper()
@@ -276,6 +292,7 @@ class ConfigHelper:
 
     @staticmethod
     @functools.cache
+    @logger.trace_function(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"
@@ -288,6 +305,7 @@ class ConfigHelper:
 
     @staticmethod
     @functools.cache
+    @logger.trace_function(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"
@@ -299,11 +317,13 @@ class ConfigHelper:
         return "".join([str(elem) for elem in employee_assistant])
 
     @staticmethod
+    @logger.trace_function(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_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)
@@ -315,6 +335,7 @@ class ConfigHelper:
         )
 
     @staticmethod
+    @logger.trace_function(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 = [
@@ -325,6 +346,7 @@ class ConfigHelper:
         ConfigHelper._default_config["document_processors"] = document_processors
 
     @staticmethod
+    @logger.trace_function(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/env_helper.py

@@ -9,14 +9,7 @@ from ..helpers.config.database_type import DatabaseType
 from ..helpers.secret_helper import SecretHelper
 from ..helpers.azure_identity_helper import AzureIdentityHelper
 
-from logging import getLogger
-from opentelemetry import trace, baggage
-from opentelemetry.propagate import extract
-
-# logger = getLogger("__main__" + ".base_package")
-logger = getLogger("__main__")
-# tracer = trace.get_tracer("__main__" + ".base_package")
-tracer = trace.get_tracer("__main__")
+from logging_config import logger
 
 
 
@@ -31,7 +24,8 @@ class EnvHelper:
                 instance.__load_config()
                 cls._instance = instance
             return cls._instance
-
+        
+    @logger.trace_function(log_args=False, log_result=False)
     def __load_config(self, **kwargs) -> None:
         # load_dotenv()
 
@@ -54,6 +48,12 @@ class EnvHelper:
         # )
 
         self.LOGLEVEL = self.secretHelper.get_secret("logging-level")
+        self.LOG_ARGS = self.get_env_var_bool(
+                "LOG_ARGS", "True"
+            )
+        self.LOG_RESULT = self.get_env_var_bool(
+                "LOG_RESULT", "True"
+            )
 
         # Azure
         self.AZURE_SUBSCRIPTION_ID = self.secretHelper.get_secret("subscription-id")

cwyodmodules/batch/utilities/helpers/lightrag_helper.py

@@ -1,9 +1,12 @@
-import logging
 import psycopg2
 from psycopg2.extras import execute_values, RealDictCursor
 from ..helpers.azure_identity_helper import AzureIdentityHelper
 
-logger = logging.getLogger("__main__")
+from ...utilities.helpers.env_helper import EnvHelper
+from logging_config import logger
+env_helper: EnvHelper = EnvHelper()
+log_args = env_helper.LOG_ARGS
+log_result = env_helper.LOG_RESULT
 
 
 class LightRAGHelper:
@@ -12,6 +15,7 @@ class LightRAGHelper:
         self.azure_identity_helper = AzureIdentityHelper()
         self.conn = None
 
+    @logger.trace_function(log_args=log_args, log_result=False)
     def _create_connection(self):
         """
         Establishes a connection to PostgreSQL using AAD authentication.
@@ -41,6 +45,7 @@ class LightRAGHelper:
             logger.error(f"Error establishing a connection to PostgreSQL: {e}")
             raise
 
+    @logger.trace_function(log_args=log_args, log_result=False)
     def get_connection(self):
         """
         Provides a reusable database connection.
@@ -49,6 +54,7 @@ class LightRAGHelper:
             self.conn = self._create_connection()
         return self.conn
 
+    @logger.trace_function(log_args=False, log_result=False)
     def store_vector_and_text(self, vector, text, metadata):
         """
         Stores a vector and associated text in the PostgreSQL database.
@@ -70,6 +76,7 @@ class LightRAGHelper:
         finally:
             conn.close()
 
+    @logger.trace_function(log_args=False, log_result=False)
     def retrieve_vectors(self, query_vector, top_k):
         """
         Retrieves the top K vectors similar to the provided query vector.

cwyodmodules/batch/utilities/helpers/llm_helper.py

@@ -1,4 +1,3 @@
-import logging
 from openai import AzureOpenAI
 from typing import List, Union, cast
 from langchain_openai import AzureChatOpenAI, AzureOpenAIEmbeddings
@@ -11,7 +10,10 @@ from azure.ai.ml import MLClient
 from ..helpers.azure_identity_helper import AzureIdentityHelper
 from .env_helper import EnvHelper
 
-logger = logging.getLogger("__main__")
+from logging_config import logger
+env_helper: EnvHelper = EnvHelper()
+log_args = env_helper.LOG_ARGS
+log_result = env_helper.LOG_RESULT
 
 
 class LLMHelper:
@@ -46,6 +48,7 @@ class LLMHelper:
 
         logger.info("Initializing LLMHelper completed")
 
+    @logger.trace_function(log_args=log_args, log_result=False)
     def get_llm(self):
         if self.auth_type_keys:
             return AzureChatOpenAI(
@@ -67,6 +70,7 @@ class LLMHelper:
             )
 
     # TODO: This needs to have a custom callback to stream back to the UI
+    @logger.trace_function(log_args=log_args, log_result=False)
     def get_streaming_llm(self):
         if self.auth_type_keys:
             return AzureChatOpenAI(
@@ -92,6 +96,7 @@ class LLMHelper:
                 azure_ad_token_provider=self.token_provider,
             )
 
+    @logger.trace_function(log_args=log_args, log_result=False)
     def get_embedding_model(self):
         if self.auth_type_keys:
             return AzureOpenAIEmbeddings(
@@ -108,6 +113,7 @@ class LLMHelper:
                 azure_ad_token_provider=self.token_provider,
             )
 
+    @logger.trace_function(log_args=False, log_result=False)
     def generate_embeddings(self, input: Union[str, list[int]]) -> List[float]:
         return (
             self.openai_client.embeddings.create(
@@ -117,6 +123,7 @@ class LLMHelper:
             .embedding
         )
 
+    @logger.trace_function(log_args=False, log_result=False)
     def get_chat_completion_with_functions(
         self, messages: list[dict], functions: list[dict], function_call: str = "auto"
     ):
@@ -127,6 +134,7 @@ class LLMHelper:
             function_call=function_call,
         )
 
+    @logger.trace_function(log_args=False, log_result=False)
     def get_chat_completion(
         self, messages: list[dict], model: str | None = None, **kwargs
     ):
@@ -137,6 +145,7 @@ class LLMHelper:
             **kwargs,
         )
 
+    @logger.trace_function(log_args=False, log_result=False)
     def get_sk_chat_completion_service(self, service_id: str):
         if self.auth_type_keys:
             return AzureChatCompletion(
@@ -166,6 +175,7 @@ class LLMHelper:
                 ad_token_provider=self.token_provider,
             )
 
+    @logger.trace_function(log_args=False, log_result=False)
     def get_sk_service_settings(self, service: AzureChatCompletion):
         return cast(
             AzureChatPromptExecutionSettings,
@@ -176,6 +186,7 @@ class LLMHelper:
             ),
         )
 
+    @logger.trace_function(log_args=False, log_result=False)
     def get_ml_client(self):
         if not hasattr(self, "_ml_client"):
             credential = self.azure_identity_helper.get_credential()

cwyodmodules/batch/utilities/helpers/secret_helper.py

@@ -1,8 +1,11 @@
-import logging
 from ..helpers.azure_identity_helper import AzureIdentityHelper
 from azure.keyvault.secrets import SecretClient
 
-logger = logging.getLogger("__main__")
+from ...utilities.helpers.env_helper import EnvHelper
+from logging_config import logger
+env_helper: EnvHelper = EnvHelper()
+log_args = env_helper.LOG_ARGS
+log_result = env_helper.LOG_RESULT
 
 
 class SecretHelper:
@@ -31,6 +34,7 @@ class SecretHelper:
                 connection_verify=True,
             )
 
+    @logger.trace_function(log_args=log_args, 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.
@@ -45,11 +49,10 @@ class SecretHelper:
             None
 
         """
-        logger.warning(f"Trying to get secret: {secret_name}")
         secret_value = self.secret_client.get_secret(name=secret_name).value
-        logger.warning(f"Fetched secret value: {secret_value}")
         return secret_value
 
+    @logger.trace_function(log_args=log_args, log_result=log_result)
     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.
@@ -64,12 +67,10 @@ class SecretHelper:
         Raises:
             None
         """
-        logger.warning(f"Trying to set secret: {secret_name}")
         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)
-                logger.warning(f"Secret {secret_name} has been updated with new value")
             else:
                 logger.warning(
                     f"Secret {secret_name} already has the same value, skipping update"
@@ -78,9 +79,8 @@ class SecretHelper:
             self.secret_client.set_secret(name=secret_name, value=secret_value)
             logger.warning(f"Secret {secret_name} has been created")
 
-    def get_secret_from_json(self, secret_name: str) -> str:
 
-        logger.warning(f"Trying to get secret from json: {secret_name}")
+    @logger.trace_function(log_args=log_args, log_result=False)
+    def get_secret_from_json(self, secret_name: str) -> str:
         secret_value = self.secret_client.get_secret(secret_name).value
-        logger.warning(f"Fetched secret value: {secret_value}")
         return secret_value

cwyodmodules/batch/utilities/integrated_vectorization/azure_search_index.py

@@ -1,4 +1,3 @@
-import logging
 from azure.search.documents.indexes import SearchIndexClient
 from azure.search.documents.indexes.models import (
     SearchField,
@@ -25,7 +24,10 @@ from ..helpers.azure_identity_helper import AzureIdentityHelper
 from azure.core.credentials import AzureKeyCredential
 from ..helpers.llm_helper import LLMHelper
 
-logger = logging.getLogger("__main__")
+from logging_config import logger
+env_helper: EnvHelper = EnvHelper()
+log_args = env_helper.LOG_ARGS
+log_result = env_helper.LOG_RESULT
 
 
 class AzureSearchIndex:
@@ -46,6 +48,7 @@ class AzureSearchIndex:
         )
 
     @property
+    @logger.trace_function(log_args=log_args, log_result=False)
     def search_dimensions(self) -> int:
         if AzureSearchIndex._search_dimension is None:
             AzureSearchIndex._search_dimension = len(
@@ -53,6 +56,7 @@ class AzureSearchIndex:
             )
         return AzureSearchIndex._search_dimension
 
+    @logger.trace_function(log_args=log_args, log_result=False)
     def create_or_update_index(self):
         # Create a search index
         fields = [
@@ -115,6 +119,7 @@ class AzureSearchIndex:
         logger.info(f"{result.name} index created successfully.")
         return result
 
+    @logger.trace_function(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(
@@ -167,6 +172,7 @@ class AzureSearchIndex:
             ],
         )
 
+    @logger.trace_function(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

@@ -1,11 +1,15 @@
-import logging
 from azure.search.documents.indexes.models import SearchIndexer, FieldMapping
 from azure.search.documents.indexes import SearchIndexerClient
 from ..helpers.env_helper import EnvHelper
 from ..helpers.azure_identity_helper import AzureIdentityHelper
 from azure.core.credentials import AzureKeyCredential
 
-logger = logging.getLogger("__main__")
+
+from ...utilities.helpers.env_helper import EnvHelper
+from logging_config import logger
+env_helper: EnvHelper = EnvHelper()
+log_args = env_helper.LOG_ARGS
+log_result = env_helper.LOG_RESULT
 
 
 class AzureSearchIndexer:
@@ -22,6 +26,7 @@ class AzureSearchIndexer:
             ),
         )
 
+    @logger.trace_function(log_args=log_args, log_result=False)
     def create_or_update_indexer(self, indexer_name: str, skillset_name: str):
         indexer = SearchIndexer(
             name=indexer_name,
@@ -59,6 +64,7 @@ class AzureSearchIndexer:
         )
         return indexer_result
 
+    @logger.trace_function(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)
@@ -66,6 +72,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)
     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

@@ -1,4 +1,3 @@
-import logging
 from azure.search.documents.indexes.models import (
     SplitSkill,
     InputFieldMappingEntry,
@@ -18,7 +17,11 @@ from ..helpers.env_helper import EnvHelper
 from ..helpers.azure_identity_helper import AzureIdentityHelper
 from azure.core.credentials import AzureKeyCredential
 
-logger = logging.getLogger("__main__")
+from ...utilities.helpers.env_helper import EnvHelper
+from logging_config import logger
+env_helper: EnvHelper = EnvHelper()
+log_args = env_helper.LOG_ARGS
+log_result = env_helper.LOG_RESULT
 
 
 class AzureSearchSkillset:
@@ -40,6 +43,7 @@ class AzureSearchSkillset:
         )
         self.integrated_vectorization_config = integrated_vectorization_config
 
+    @logger.trace_function(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

@@ -1,4 +1,3 @@
-import logging
 from typing import List
 from langchain.agents import Tool
 from langchain.memory import ConversationBufferMemory
@@ -13,7 +12,11 @@ from ..tools.question_answer_tool import QuestionAnswerTool
 from ..tools.text_processing_tool import TextProcessingTool
 from ..common.answer import Answer
 
-logger = logging.getLogger("__main__")
+from ...utilities.helpers.env_helper import EnvHelper
+from logging_config import logger
+env_helper: EnvHelper = EnvHelper()
+log_args = env_helper.LOG_ARGS
+log_result = env_helper.LOG_RESULT
 
 
 class LangChainAgent(OrchestratorBase):
@@ -48,6 +51,7 @@ class LangChainAgent(OrchestratorBase):
             ),
         ]
 
+    @logger.trace_function(log_args=False, log_result=False)
     def run_tool(self, user_message: str) -> str:
         """
         Executes the question answering tool with the provided user message.
@@ -63,6 +67,7 @@ class LangChainAgent(OrchestratorBase):
         )
         return answer.to_json()
 
+    @logger.trace_function(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.
@@ -78,6 +83,7 @@ class LangChainAgent(OrchestratorBase):
         )
         return answer.to_json()
 
+    @logger.trace_function(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

@@ -1,4 +1,3 @@
-import logging
 from typing import List
 import json
 
@@ -10,7 +9,11 @@ from ..tools.question_answer_tool import QuestionAnswerTool
 from ..tools.text_processing_tool import TextProcessingTool
 from ..common.answer import Answer
 
-logger = logging.getLogger("__main__")
+from ...utilities.helpers.env_helper import EnvHelper
+from logging_config import logger
+env_helper: EnvHelper = EnvHelper()
+log_args = env_helper.LOG_ARGS
+log_result = env_helper.LOG_RESULT
 
 
 class OpenAIFunctionsOrchestrator(OrchestratorBase):
@@ -65,6 +68,7 @@ class OpenAIFunctionsOrchestrator(OrchestratorBase):
             },
         ]
 
+    @logger.trace_function(log_args=False, log_result=False)
     async def orchestrate(
         self, user_message: str, chat_history: List[dict], **kwargs: dict
     ) -> list[dict]: