sunholo 0.134.7__py3-none-any.whl → 0.135.0__py3-none-any.whl

sunholo/custom_logging.py

@@ -22,6 +22,8 @@ from .utils.version import sunholo_version
 import logging
 import inspect
 import os
+import json
+from functools import wraps
 
 class GoogleCloudLogging:
     
@@ -270,6 +272,91 @@ class GoogleCloudLogging:
         """
         self.structured_log(log_text=log_text, log_struct=log_struct, severity="CRITICAL")
 
+class StandardLoggerWrapper:
+    """
+    A wrapper for standard Python logger that mimics the interface of GoogleCloudLogging.
+    """
+    def __init__(self, logger):
+        self.logger = logger
+        self.trace_id = None
+        self.version = None
+
+    def _format_message(self, log_text=None, log_struct=None, severity=None):
+        """Format message to include structured data as JSON"""
+        parts = []
+        
+        # Add severity if provided
+        if severity:
+            parts.append(f"[{severity}]")
+            
+        # Add trace ID if available
+        if self.trace_id:
+            parts.append(f"[trace_id:{self.trace_id}]")
+            
+        # Add version if available
+        if self.version:
+            parts.append(f"[version:{self.version}]")
+            
+        # Add the log text if provided
+        if log_text:
+            parts.append(log_text)
+            
+        # Add structured data as JSON if provided
+        if log_struct:
+            if self.trace_id and isinstance(log_struct, dict):
+                log_struct["trace_id"] = self.trace_id
+            if self.version and isinstance(log_struct, dict):
+                log_struct["version"] = self.version
+            parts.append(f"STRUCT: {json.dumps(log_struct, default=str)}")
+            
+        return " ".join(parts)
+
+    def update_trace_id(self, trace_id):
+        """Set trace ID to be included in all logs."""
+        self.trace_id = trace_id
+        
+    def set_version(self, version):
+        """Set version to be included in all logs."""
+        self.version = version
+        
+    def structured_log(self, log_text=None, log_struct=None, logger_name=None, severity="INFO"):
+        """
+        Emulates Google Cloud's structured_log method using standard logging.
+        """
+        message = self._format_message(log_text, log_struct, severity)
+        
+        if severity == "DEBUG":
+            self.logger.debug(message)
+        elif severity == "INFO":
+            self.logger.info(message)
+        elif severity == "WARNING":
+            self.logger.warning(message)
+        elif severity == "ERROR":
+            self.logger.error(message)
+        elif severity == "CRITICAL":
+            self.logger.critical(message)
+        else:
+            self.logger.info(message)  # Default to info
+            
+    def debug(self, log_text=None, log_struct=None):
+        self.structured_log(log_text=log_text, log_struct=log_struct, severity="DEBUG")
+        
+    def info(self, log_text=None, log_struct=None):
+        self.structured_log(log_text=log_text, log_struct=log_struct, severity="INFO")
+        
+    def warning(self, log_text=None, log_struct=None):
+        self.structured_log(log_text=log_text, log_struct=log_struct, severity="WARNING")
+        
+    def error(self, log_text=None, log_struct=None):
+        self.structured_log(log_text=log_text, log_struct=log_struct, severity="ERROR")
+        
+    def exception(self, log_text=None, log_struct=None):
+        self.structured_log(log_text=log_text, log_struct=log_struct, severity="CRITICAL")
+        
+    # Forward any other standard logging methods to the underlying logger
+    def __getattr__(self, name):
+        return getattr(self.logger, name)
+    
 def is_logging_setup(logger=None):
     logger = logger or logging.getLogger()  # If no logger is specified, use the root logger
     return bool(logger.handlers)
@@ -352,7 +439,14 @@ def setup_logging(logger_name=None, log_level=logging.INFO, project_id=None):
 
     logger = logging.getLogger(logger_name)
     if is_logging_setup(logger):
-        return logger
+        # Check if it's already our wrapper
+        if hasattr(logger, 'structured_log'):
+            return logger
+        else:
+            # Wrap the existing logger
+            wrapper = StandardLoggerWrapper(logger)
+            wrapper.set_version(sunholo_version())
+            return wrapper
     
     if logger_name is None:
         logger_name = "sunholo"
@@ -370,10 +464,13 @@ def setup_logging(logger_name=None, log_level=logging.INFO, project_id=None):
         return gc_logger.setup_logging()
     else:
         if not logging.getLogger().hasHandlers():
-            logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+            logging.basicConfig(level=log_level, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
         log = logging.getLogger(logger_name)
-
-        return log
+        
+        # Wrap the standard logger to support log_struct parameter
+        wrapper = StandardLoggerWrapper(log)
+        wrapper.set_version(sunholo_version())
+        return wrapper
     
 # for debugging
 def safe_log_struct(log, severity, message, struct):

sunholo/discovery_engine/discovery_engine_client.py

@@ -925,18 +925,19 @@ class DiscoveryEngineClient:
         if parse_chunks_to_string:
             if content_search_spec_type=="chunks":
                 if parse_chunks_to_string:
-                    big_string = self.async_process_chunks(search_response)
+                    big_string = await self.async_process_chunks(search_response)
                     log.info(f"Discovery engine chunks string sample: {big_string[:100]}")
 
                     return big_string
                 
             elif content_search_spec_type=="documents":
-                big_string = self.async_process_documents(search_response)
+                big_string = await self.async_process_documents(search_response)
                 log.info(f"Discovery engine documents string sample: {big_string[:100]}")
 
                 return big_string
         
         log.info("Discovery engine response object")
+        
         return search_response
 
     def search_by_objectId_and_or_date(self, query, objectId=None, date=None, **kwargs):

{sunholo-0.134.7.dist-info → sunholo-0.135.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sunholo
-Version: 0.134.7
+Version: 0.135.0
 Summary: AI DevOps - a package to help deploy GenAI to the Cloud.
 Author-email: Holosun ApS <multivac@sunholo.com>
 License: Apache License, Version 2.0

{sunholo-0.134.7.dist-info → sunholo-0.135.0.dist-info}/RECORD

@@ -1,5 +1,5 @@
 sunholo/__init__.py,sha256=InRbX4V0-qdNHo9zYH3GEye7ASLR6LX8-SMvPV4Jsaw,1212
-sunholo/custom_logging.py,sha256=PrUXreBF0nZJMfa8A8cbFhTWoaOWWB_i6qwZfnjvRlo,18296
+sunholo/custom_logging.py,sha256=JXZTnXp_DixP3jwYfKw4LYRDS9IuTq7ctCgfZbI2rxA,22023
 sunholo/langchain_types.py,sha256=uZ4zvgej_f7pLqjtu4YP7qMC_eZD5ym_5x4pyvA1Ih4,1834
 sunholo/agents/__init__.py,sha256=X2I3pPkGeKWjc3d0QgSpkTyqD8J8JtrEWqwrumf1MMc,391
 sunholo/agents/chat_history.py,sha256=Gph_CdlP2otYnNdR1q1Umyyyvcad2F6K3LxU5yBQ9l0,5387
@@ -75,7 +75,7 @@ sunholo/discovery_engine/__init__.py,sha256=hLgqRDJ22Aov9o2QjAEfsVgnL3kMdM-g5p8R
 sunholo/discovery_engine/chunker_handler.py,sha256=wkvXl4rFtYfN6AZUKdW9_QD49Whf77BukDbO82UwlAg,7480
 sunholo/discovery_engine/cli.py,sha256=tsKqNSDCEsDTz5-wuNwjttb3Xt35D97-KyyEiaqolMQ,35628
 sunholo/discovery_engine/create_new.py,sha256=WUi4_xh_dFaGX3xA9jkNKZhaR6LCELjMPeRb0hyj4FU,1226
-sunholo/discovery_engine/discovery_engine_client.py,sha256=SqzFNirl5FMldF2T8E9NMRXgn3AwRwG_WkAdQQzq5AQ,58684
+sunholo/discovery_engine/discovery_engine_client.py,sha256=Sf7Sr6FYKA_jn19Ba2ENShrB1jnZ4HgPScuytDIuK9c,58705
 sunholo/discovery_engine/get_ai_search_chunks.py,sha256=I6Dt1CznqEvE7XIZ2PkLqopmjpO96iVEWJJqL5cJjOU,5554
 sunholo/embedder/__init__.py,sha256=sI4N_CqgEVcrMDxXgxKp1FsfsB4FpjoXgPGkl4N_u4I,44
 sunholo/embedder/embed_chunk.py,sha256=did2pKkWM2o0KkRcb0H9l2x_WjCq6OyuHDxGbITFKPM,6530
@@ -168,9 +168,9 @@ sunholo/vertex/init.py,sha256=1OQwcPBKZYBTDPdyU7IM4X4OmiXLdsNV30C-fee2scQ,2875
 sunholo/vertex/memory_tools.py,sha256=tBZxqVZ4InTmdBvLlOYwoSEWu4-kGquc-gxDwZCC4FA,7667
 sunholo/vertex/safety.py,sha256=S9PgQT1O_BQAkcqauWncRJaydiP8Q_Jzmu9gxYfy1VA,2482
 sunholo/vertex/type_dict_to_json.py,sha256=uTzL4o9tJRao4u-gJOFcACgWGkBOtqACmb6ihvCErL8,4694
-sunholo-0.134.7.dist-info/licenses/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
-sunholo-0.134.7.dist-info/METADATA,sha256=cogo1VdlgznP_MC7vo-EwbzYuXsk1jg1eClmG1ExBqU,10067
-sunholo-0.134.7.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-sunholo-0.134.7.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
-sunholo-0.134.7.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
-sunholo-0.134.7.dist-info/RECORD,,
+sunholo-0.135.0.dist-info/licenses/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.135.0.dist-info/METADATA,sha256=jLaY76jTW-W8S-9V7_9THZZ3-FroKO6HHoFKvloXXPI,10067
+sunholo-0.135.0.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+sunholo-0.135.0.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.135.0.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.135.0.dist-info/RECORD,,

{sunholo-0.134.7.dist-info → sunholo-0.135.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (79.0.0)
+Generator: setuptools (79.0.1)
 Root-Is-Purelib: true
 Tag: py3-none-any