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

cwyodmodules/batch/utilities/plugins/outlook_calendar_plugin.py

@@ -1,16 +1,13 @@
 import requests
 import datetime
 from semantic_kernel.functions import kernel_function
-from logging import getLogger
-from opentelemetry import trace, baggage
-from opentelemetry.propagate import extract
-from typing import Annotated
 from ..tools.text_processing_tool import TextProcessingTool
-from ..helpers.env_helper import EnvHelper
-# logger = getLogger("__main__" + ".base_package")
-logger = getLogger("__main__")
-# tracer = trace.get_tracer("__main__" + ".base_package")
-tracer = trace.get_tracer("__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 OutlookCalendarPlugin:
     def __init__(self, question: str, chat_history: list[dict], user_info: dict):
@@ -19,6 +16,7 @@ class OutlookCalendarPlugin:
         self.user_info = user_info
         self.env_helper = EnvHelper()
 
+    @logger.trace_function(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)
@@ -29,93 +27,93 @@ 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)
     def get_calendar_events(self, days: int = 1) -> str:
-        with tracer.start_as_current_span("SemanticKernelOrchestrator_orchestrate.get_calendar_events"):
-            language = self.env_helper.AZURE_MAIN_CHAT_LANGUAGE
-            logger.info("Method get_calendar_events of OutlookCalendarPlugin started")
-            try:
-                logger.info("Retrieving access token for calendar events")
-                token = self._get_access_token()
-            except Exception as e:
-                answer = TextProcessingTool().answer_question(
-                    question=self.question,
-                    chat_history=self.chat_history,
-                    text=f"Authentication error: {str(e)}",
-                    operation="Explain the user in his language {language} that you failed to get calendar appointment due to an error.",
-                )
-                return answer
-            headers = {"Authorization": f"Bearer {token}"}
-            now = datetime.datetime.utcnow().isoformat() + "Z"
-            end = (datetime.datetime.utcnow() + datetime.timedelta(days=days)).isoformat() + "Z"
-            url = f"https://graph.microsoft.com/v1.0/me/calendarview?startDateTime={now}&endDateTime={end}"
-            resp = requests.get(url, headers=headers)
-            logger.info("Calendar get results: %s", resp.text[0:120])
-            if resp.status_code != 200:
-                answer = TextProcessingTool().answer_question(
-                    question=self.question,
-                    chat_history=self.chat_history,
-                    text=f"Failed to fetch events: {resp.text[0:120]}",
-                    operation=f"Explain the user in his language {language} that you failed to fetch calendar events due to an error.",
-                )
-                return answer
-            events = resp.json().get("value", [])
-            if not events:
-                answer = TextProcessingTool().answer_question(
-                    question=self.question,
-                    chat_history=self.chat_history,
-                    text="No events found.",
-                    operation=f"Explain the user in his language {language} that no events were found in the calendar.",
-                )
-                return answer
-            events_text = "\n".join([f"{e.get('subject', 'No subject')} at {e.get('start', {}).get('dateTime', 'Unknown time')}" for e in events])
+        language = self.env_helper.AZURE_MAIN_CHAT_LANGUAGE
+        logger.info("Method get_calendar_events of OutlookCalendarPlugin started")
+        try:
+            logger.info("Retrieving access token for calendar events")
+            token = self._get_access_token()
+        except Exception as e:
+            answer = TextProcessingTool().answer_question(
+                question=self.question,
+                chat_history=self.chat_history,
+                text=f"Authentication error: {str(e)}",
+                operation="Explain the user in his language {language} that you failed to get calendar appointment due to an error.",
+            )
+            return answer
+        headers = {"Authorization": f"Bearer {token}"}
+        now = datetime.datetime.utcnow().isoformat() + "Z"
+        end = (datetime.datetime.utcnow() + datetime.timedelta(days=days)).isoformat() + "Z"
+        url = f"https://graph.microsoft.com/v1.0/me/calendarview?startDateTime={now}&endDateTime={end}"
+        resp = requests.get(url, headers=headers)
+        logger.info("Calendar get results: %s", resp.text[0:120])
+        if resp.status_code != 200:
             answer = TextProcessingTool().answer_question(
-                    question=self.question,
-                    chat_history=self.chat_history,
-                    text=events_text,
-                    operation=f"Summarize the calendar schedule in the user's {language}.",
-                )
+                question=self.question,
+                chat_history=self.chat_history,
+                text=f"Failed to fetch events: {resp.text[0:120]}",
+                operation=f"Explain the user in his language {language} that you failed to fetch calendar events due to an error.",
+            )
             return answer
+        events = resp.json().get("value", [])
+        if not events:
+            answer = TextProcessingTool().answer_question(
+                question=self.question,
+                chat_history=self.chat_history,
+                text="No events found.",
+                operation=f"Explain the user in his language {language} that no events were found in the calendar.",
+            )
+            return answer
+        events_text = "\n".join([f"{e.get('subject', 'No subject')} at {e.get('start', {}).get('dateTime', 'Unknown time')}" for e in events])
+        answer = TextProcessingTool().answer_question(
+                question=self.question,
+                chat_history=self.chat_history,
+                text=events_text,
+                operation=f"Summarize the calendar schedule in the user's {language}.",
+            )
+        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)
     def schedule_appointment(self, subject: str, start_time: str, end_time: str) -> str:
-        with tracer.start_as_current_span("SemanticKernelOrchestrator_orchestrate.schedule_appointment"):
-            logger.info("Method schedule_appointment of OutlookCalendarPlugin started")
-            language = self.env_helper.AZURE_MAIN_CHAT_LANGUAGE
-            try:
-                token = self._get_access_token()
-            except Exception as e:
-                answer = TextProcessingTool().answer_question(
-                    question=self.question,
-                    chat_history=self.chat_history,
-                    text=f"Failed to schedule appointment: {str(e)}",
-                    operation=f"Explain the user in his language {language} that you failed to schedule a calendar appointment due to an error.",
-                )
-                return answer
-            headers = {
-                "Authorization": f"Bearer {token}",
-                "Content-Type": "application/json"
-            }
-            url = "https://graph.microsoft.com/v1.0/me/events"
-            event = {
-                "subject": subject,
-                "start": {"dateTime": start_time, "timeZone": "UTC"},
-                "end": {"dateTime": end_time, "timeZone": "UTC"},
-            }
-            resp = requests.post(url, headers=headers, json=event)
-            logger.info("Calendar set results: %s", resp.text[0:120])
-            if resp.status_code == 201:
-                answer = TextProcessingTool().answer_question(
-                    question=self.question,
-                    chat_history=self.chat_history,
-                    text="Appointment scheduled successfully.",
-                    operation=f"Explain to the user in his language {language} that the appointment was scheduled successfully. And summarize the appointment details. And all appointments shortly before and after the scheduled appointment.",
-                )
-                return answer
-            else:
-                answer = TextProcessingTool().answer_question(
-                    question=self.question,
-                    chat_history=self.chat_history,
-                    text=f"Failed to schedule appointment: {resp.text[0:120]}",
-                    operation=f"Explain to the user in his language {language} that the appointment scheduling failed.",
-                )
-                return answer
+        logger.info("Method schedule_appointment of OutlookCalendarPlugin started")
+        language = self.env_helper.AZURE_MAIN_CHAT_LANGUAGE
+        try:
+            token = self._get_access_token()
+        except Exception as e:
+            answer = TextProcessingTool().answer_question(
+                question=self.question,
+                chat_history=self.chat_history,
+                text=f"Failed to schedule appointment: {str(e)}",
+                operation=f"Explain the user in his language {language} that you failed to schedule a calendar appointment due to an error.",
+            )
+            return answer
+        headers = {
+            "Authorization": f"Bearer {token}",
+            "Content-Type": "application/json"
+        }
+        url = "https://graph.microsoft.com/v1.0/me/events"
+        event = {
+            "subject": subject,
+            "start": {"dateTime": start_time, "timeZone": "UTC"},
+            "end": {"dateTime": end_time, "timeZone": "UTC"},
+        }
+        resp = requests.post(url, headers=headers, json=event)
+        logger.info("Calendar set results: %s", resp.text[0:120])
+        if resp.status_code == 201:
+            answer = TextProcessingTool().answer_question(
+                question=self.question,
+                chat_history=self.chat_history,
+                text="Appointment scheduled successfully.",
+                operation=f"Explain to the user in his language {language} that the appointment was scheduled successfully. And summarize the appointment details. And all appointments shortly before and after the scheduled appointment.",
+            )
+            return answer
+        else:
+            answer = TextProcessingTool().answer_question(
+                question=self.question,
+                chat_history=self.chat_history,
+                text=f"Failed to schedule appointment: {resp.text[0:120]}",
+                operation=f"Explain to the user in his language {language} that the appointment scheduling failed.",
+            )
+            return answer

cwyodmodules/batch/utilities/search/azure_search_handler.py

@@ -1,6 +1,4 @@
-import logging
 from typing import List
-
 from .search_handler_base import SearchHandlerBase
 from ..helpers.llm_helper import LLMHelper
 from ..helpers.azure_computer_vision_client import AzureComputerVisionClient
@@ -10,7 +8,11 @@ import json
 from azure.search.documents.models import VectorizedQuery
 import tiktoken
 
-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 AzureSearchHandler(SearchHandlerBase):
@@ -21,14 +23,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)
     def create_search_client(self):
         return AzureSearchHelper().get_search_client()
 
+    @logger.trace_function(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)
     def process_results(self, results):
         logger.info("Processing search results")
         if results is None:
@@ -42,11 +47,13 @@ class AzureSearchHandler(SearchHandlerBase):
         logger.info("Processed results")
         return data
 
+    @logger.trace_function(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)
     def output_results(self, results):
         files = {}
         for result in results:
@@ -59,6 +66,7 @@ class AzureSearchHandler(SearchHandlerBase):
 
         return files
 
+    @logger.trace_function(log_args=log_args, log_result=log_result)
     def delete_files(self, files):
         ids_to_delete = []
         files_to_delete = []
@@ -70,6 +78,7 @@ class AzureSearchHandler(SearchHandlerBase):
 
         return ", ".join(files_to_delete)
 
+    @logger.trace_function(log_args=log_args, log_result=False)
     def search_by_blob_url(self, blob_url):
         return self.search_client.search(
             "*",
@@ -78,6 +87,7 @@ class AzureSearchHandler(SearchHandlerBase):
             filter=f"source eq '{blob_url}_SAS_TOKEN_PLACEHOLDER_'",
         )
 
+    @logger.trace_function(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)
@@ -106,6 +116,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)
     def _semantic_search(
         self,
         question: str,
@@ -140,6 +151,7 @@ class AzureSearchHandler(SearchHandlerBase):
             top=self.env_helper.AZURE_SEARCH_TOP_K,
         )
 
+    @logger.trace_function(log_args=False, log_result=False)
     def _hybrid_search(
         self,
         question: str,
@@ -172,6 +184,7 @@ class AzureSearchHandler(SearchHandlerBase):
             top=self.env_helper.AZURE_SEARCH_TOP_K,
         )
 
+    @logger.trace_function(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

@@ -1,11 +1,14 @@
-import logging
 from typing import List
 from ..helpers.lightrag_helper import LightRAGHelper
 from .search_handler_base import SearchHandlerBase
 from ..common.source_document import SourceDocument
 import json
 
-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 AzureSearchHandlerLightRag(SearchHandlerBase):
@@ -13,14 +16,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)
     def create_search_client(self):
         return self.light_rag_helper.get_search_client()
 
+    @logger.trace_function(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)
     def process_results(self, results):
         logger.info("Processing search results")
         if results is None:
@@ -33,9 +39,11 @@ class AzureSearchHandlerLightRag(SearchHandlerBase):
         logger.info("Processed results")
         return data
 
+    @logger.trace_function(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)
     def output_results(self, results):
         files = {}
         for result in results:
@@ -48,6 +56,7 @@ class AzureSearchHandlerLightRag(SearchHandlerBase):
 
         return files
 
+    @logger.trace_function(log_args=log_args, log_result=log_result)
     def delete_files(self, files):
         ids_to_delete = []
         files_to_delete = []
@@ -59,15 +68,18 @@ class AzureSearchHandlerLightRag(SearchHandlerBase):
 
         return ", ".join(files_to_delete)
 
+    @logger.trace_function(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)
     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)
     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

@@ -1,4 +1,3 @@
-import logging
 from typing import List
 from .search_handler_base import SearchHandlerBase
 from azure.search.documents import SearchClient
@@ -8,16 +7,20 @@ from azure.core.credentials import AzureKeyCredential
 from ..helpers.azure_identity_helper import AzureIdentityHelper
 from ..common.source_document import SourceDocument
 import re
-
+from ...utilities.helpers.env_helper import EnvHelper
+from logging_config import logger
+env_helper: EnvHelper = EnvHelper()
+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)    
     def create_search_client(self):
-        logging.info("Creating Azure Search Client.")
+        logger.info("Creating Azure Search Client.")
         if self._check_index_exists():
-            logging.info("Search index exists. Returning Search Client.")
+            logger.info("Search index exists. Returning Search Client.")
             return SearchClient(
                 endpoint=self.env_helper.AZURE_SEARCH_SERVICE,
                 index_name=self.env_helper.AZURE_SEARCH_INDEX,
@@ -28,8 +31,9 @@ class IntegratedVectorizationSearchHandler(SearchHandlerBase):
                 ),
             )
 
+    @logger.trace_function(log_args=log_args, log_result=False)
     def perform_search(self, filename):
-        logging.info(f"Performing search for file: {filename}.")
+        logger.info(f"Performing search for file: {filename}.")
         if self._check_index_exists():
             return self.search_client.search(
                 search_text="*",
@@ -37,27 +41,30 @@ class IntegratedVectorizationSearchHandler(SearchHandlerBase):
                 filter=f"title eq '{filename}'",
             )
 
+    @logger.trace_function(log_args=False, log_result=False)
     def process_results(self, results):
-        logging.info("Processing search results.")
+        logger.info("Processing search results.")
         if results is None:
-            logging.warning("No results found to process.")
+            logger.warning("No results found to process.")
             return []
         data = [
             [re.findall(r"\d+", result["chunk_id"])[-1], result["content"]]
             for result in results
         ]
-        logging.info(f"Processed {len(data)} results.")
+        logger.info(f"Processed {len(data)} results.")
         return data
 
+    @logger.trace_function(log_args=log_args, log_result=False)
     def get_files(self):
-        logging.info("Fetching files from search index.")
+        logger.info("Fetching files from search index.")
         if self._check_index_exists():
             return self.search_client.search(
                 "*", select="id, chunk_id, title", include_total_count=True
             )
 
+    @logger.trace_function(log_args=False, log_result=False)
     def output_results(self, results):
-        logging.info("Organizing search results into output format.")
+        logger.info("Organizing search results into output format.")
         files = {}
         for result in results:
             id = result["chunk_id"]
@@ -68,8 +75,9 @@ class IntegratedVectorizationSearchHandler(SearchHandlerBase):
                 files[filename] = [id]
         return files
 
+    @logger.trace_function(log_args=log_args, log_result=False)
     def search_by_blob_url(self, blob_url: str):
-        logging.info(f"Searching by blob URL: {blob_url}.")
+        logger.info(f"Searching by blob URL: {blob_url}.")
         if self._check_index_exists():
             title = blob_url.split(f"{self.env_helper.AZURE_BLOB_CONTAINER_NAME}/")[1]
             return self.search_client.search(
@@ -79,8 +87,9 @@ class IntegratedVectorizationSearchHandler(SearchHandlerBase):
                 filter=f"title eq '{title}'",
             )
 
+    @logger.trace_function(log_args=log_args, log_result=log_result)
     def delete_files(self, files):
-        logging.info("Deleting files.")
+        logger.info("Deleting files.")
         ids_to_delete = []
         files_to_delete = []
 
@@ -90,24 +99,26 @@ class IntegratedVectorizationSearchHandler(SearchHandlerBase):
 
         self.search_client.delete_documents(ids_to_delete)
 
-        logging.info(f"Deleted files: {', '.join(files_to_delete)}.")
+        logger.info(f"Deleted files: {', '.join(files_to_delete)}.")
         return ", ".join(files_to_delete)
 
+    @logger.trace_function(log_args=False, log_result=False)
     def query_search(self, question) -> List[SourceDocument]:
-        logging.info(f"Querying search for question: {question}.")
+        logger.info(f"Querying search for question: {question}.")
         if self._check_index_exists():
-            logging.info("Search index exists. Proceeding with search.")
+            logger.info("Search index exists. Proceeding with search.")
             if self.env_helper.AZURE_SEARCH_USE_SEMANTIC_SEARCH:
-                logging.info("Using semantic search.")
+                logger.info("Using semantic search.")
                 search_results = self._semantic_search(question)
             else:
-                logging.info("Using hybrid search.")
+                logger.info("Using hybrid search.")
                 search_results = self._hybrid_search(question)
-            logging.info("Search completed. Converting results to SourceDocuments.")
+            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)
     def _hybrid_search(self, question: str):
-        logging.info(f"Performing hybrid search for question: {question}.")
+        logger.info(f"Performing hybrid search for question: {question}.")
         vector_query = VectorizableTextQuery(
             text=question,
             k_nearest_neighbors=self.env_helper.AZURE_SEARCH_TOP_K,
@@ -120,8 +131,9 @@ class IntegratedVectorizationSearchHandler(SearchHandlerBase):
             top=self.env_helper.AZURE_SEARCH_TOP_K,
         )
 
+    @logger.trace_function(log_args=False, log_result=False)
     def _semantic_search(self, question: str):
-        logging.info(f"Performing semantic search for question: {question}.")
+        logger.info(f"Performing semantic search for question: {question}.")
         vector_query = VectorizableTextQuery(
             text=question,
             k_nearest_neighbors=self.env_helper.AZURE_SEARCH_TOP_K,
@@ -138,9 +150,9 @@ class IntegratedVectorizationSearchHandler(SearchHandlerBase):
             query_answer="extractive",
             top=self.env_helper.AZURE_SEARCH_TOP_K,
         )
-
+    @logger.trace_function(log_args=False, log_result=False)
     def _convert_to_source_documents(self, search_results) -> List[SourceDocument]:
-        logging.info("Converting search results to SourceDocument objects.")
+        logger.info("Converting search results to SourceDocument objects.")
         source_documents = []
         for source in search_results:
             source_documents.append(
@@ -152,7 +164,7 @@ class IntegratedVectorizationSearchHandler(SearchHandlerBase):
                     chunk_id=source.get("chunk_id"),
                 )
             )
-        logging.info("Converted SourceDocument objects.")
+        logger.info("Converted SourceDocument objects.")
         return source_documents
 
     def _extract_source_url(self, original_source: str) -> str:

cwyodmodules/batch/utilities/search/lightrag_search_handler.py

@@ -1,10 +1,14 @@
-import logging
 from typing import List
 from ..helpers.lightrag_helper import LightRAGHelper
 from .search_handler_base import SearchHandlerBase
 from ..common.source_document import SourceDocument
 
-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 LightRAGSearchHandler(SearchHandlerBase):
@@ -12,6 +16,7 @@ class LightRAGSearchHandler(SearchHandlerBase):
         super().__init__(env_helper)
         self.lightrag_helper = LightRAGHelper()
 
+    @logger.trace_function(log_args=log_args, log_result=False)
     def query_search(self, question) -> List[SourceDocument]:
         logger.info(f"Performing query search for question: {question}")
         search_results = self.lightrag_helper.search(question)
@@ -19,6 +24,7 @@ class LightRAGSearchHandler(SearchHandlerBase):
         logger.info(f"Found {len(source_documents)} source documents.")
         return source_documents
 
+    @logger.trace_function(log_args=log_args, log_result=False)
     def _convert_to_source_documents(self, search_results) -> List[SourceDocument]:
         source_documents = []
         for source in search_results:
@@ -35,26 +41,32 @@ class LightRAGSearchHandler(SearchHandlerBase):
             )
         return source_documents
 
+    @logger.trace_function(log_args=False, log_result=False)
     def create_vector_store(self, documents_to_upload):
         logger.info(f"Creating vector store with {len(documents_to_upload)} documents.")
         return self.lightrag_helper.create_vector_store(documents_to_upload)
 
+    @logger.trace_function(log_args=log_args, log_result=False)
     def perform_search(self, filename):
         logger.info(f"Performing search for filename: {filename}")
         return self.lightrag_helper.perform_search(filename)
 
+    @logger.trace_function(log_args=log_args, log_result=False)
     def get_files(self):
         logger.info("Fetching files from LightRAG.")
         return self.lightrag_helper.get_files()
 
+    @logger.trace_function(log_args=log_args, log_result=log_result)
     def delete_files(self, files):
         logger.info(f"Deleting files: {files}")
         return self.lightrag_helper.delete_files(files)
 
+    @logger.trace_function(log_args=log_args, log_result=False)
     def search_by_blob_url(self, blob_url):
         logger.info(f"Searching by blob URL: {blob_url}")
         return self.lightrag_helper.search_by_blob_url(blob_url)
 
+    @logger.trace_function(log_args=log_args, log_result=False) 
     def get_unique_files(self):
         logger.info("Fetching unique files from LightRAG.")
         return self.lightrag_helper.get_unique_files()