sunholo 0.127.3__py3-none-any.whl → 0.129.0__py3-none-any.whl

sunholo/discovery_engine/chunker_handler.py

@@ -136,6 +136,7 @@ def check_write_memories(config:ConfigManager):
     
     return write_mem
 
+
 def discovery_engine_chunker_check(message_data, 
                                    metadata, 
                                    vector_name:str=None, 
@@ -172,8 +173,6 @@ def discovery_engine_chunker_check(message_data,
         except Exception as err:
             log.error(f"Error processing discovery engine: {str(err)} {traceback.format_exc()}")
             disc_meta = None
-            
-        return disc_meta
 
     # If discovery engine is the only entry, return
     if total_discovery_memories == total_memories:

sunholo/traced_logging.py

@@ -0,0 +1,70 @@
+from .custom_logging import setup_logging, GoogleCloudLogging
+
+class TracedGoogleCloudLogging(GoogleCloudLogging):
+    """Extends GoogleCloudLogging with trace ID functionality."""
+    
+    def __init__(self, project_id=None, log_level=None, logger_name=None, trace_id=None):
+        super().__init__(project_id, log_level, logger_name)
+        self.trace_id = trace_id
+        
+    def update_trace_id(self, trace_id):
+        """Update the trace ID to be included in all logs."""
+        self.trace_id = trace_id
+        
+    def _get_trace_prefix(self):
+        """Get the trace prefix if a trace ID is set."""
+        return f"aitana-{self.trace_id} " if self.trace_id else ""
+        
+    def structured_log(self, log_text=None, log_struct=None, logger_name=None, severity="INFO"):
+        """Override to add trace ID to logs."""
+        if log_text and self.trace_id:
+            log_text = f"{self._get_trace_prefix()}{log_text}"
+            
+        if log_struct and self.trace_id:
+            if isinstance(log_struct, dict):
+                log_struct = {**log_struct, "trace_id": self.trace_id}
+                
+        # Call the parent method with the modified parameters
+        super().structured_log(log_text, log_struct, logger_name, severity)
+        
+def setup_traced_logging(logger_name=None, log_level=None, project_id=None, trace_id=None):
+    """Sets up traced logging that includes a trace ID in all logs."""
+    # First get the base logger from the original setup_logging
+    base_logger = setup_logging(logger_name, log_level, project_id)
+    
+    # If it's a GoogleCloudLogging instance, wrap it with our traced version
+    if isinstance(base_logger, GoogleCloudLogging):
+        traced_logger = TracedGoogleCloudLogging(
+            project_id=base_logger.project_id,
+            log_level=base_logger.log_level,
+            logger_name=base_logger.logger_name,
+            trace_id=trace_id
+        )
+        traced_logger.client = base_logger.client  # Reuse the client
+        return traced_logger
+    
+    # For standard Python logging, we can add a filter
+    import logging
+    class TraceFilter(logging.Filter):
+        def __init__(self, trace_id=None):
+            super().__init__()
+            self.trace_id = trace_id
+            
+        def update_trace_id(self, trace_id):
+            self.trace_id = trace_id
+            
+        def filter(self, record):
+            if self.trace_id:
+                prefix = f"aitana-{self.trace_id} "
+                if not record.msg.startswith(prefix):
+                    record.msg = f"{prefix}{record.msg}"
+            return True
+    
+    # Add the trace filter to the logger
+    trace_filter = TraceFilter(trace_id)
+    base_logger.addFilter(trace_filter)
+    
+    # Add the update method to the standard logger
+    base_logger.update_trace_id = trace_filter.update_trace_id
+    
+    return base_logger

{sunholo-0.127.3.dist-info → sunholo-0.129.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sunholo
-Version: 0.127.3
+Version: 0.129.0
 Summary: Large Language Model DevOps - a package to help deploy LLMs to the Cloud.
 Author-email: Holosun ApS <multivac@sunholo.com>
 License: Apache License, Version 2.0

{sunholo-0.127.3.dist-info → sunholo-0.129.0.dist-info}/RECORD

@@ -1,6 +1,7 @@
 sunholo/__init__.py,sha256=InRbX4V0-qdNHo9zYH3GEye7ASLR6LX8-SMvPV4Jsaw,1212
 sunholo/custom_logging.py,sha256=YfIN1oP3dOEkkYkyRBU8BGS3uJFGwUDsFCl8mIVbwvE,12225
 sunholo/langchain_types.py,sha256=uZ4zvgej_f7pLqjtu4YP7qMC_eZD5ym_5x4pyvA1Ih4,1834
+sunholo/traced_logging.py,sha256=TaAK9gFQ63h700dXfCK-nLcyMvh00N-oZ-xlz4VAZ_4,2891
 sunholo/agents/__init__.py,sha256=X2I3pPkGeKWjc3d0QgSpkTyqD8J8JtrEWqwrumf1MMc,391
 sunholo/agents/chat_history.py,sha256=Gph_CdlP2otYnNdR1q1Umyyyvcad2F6K3LxU5yBQ9l0,5387
 sunholo/agents/dispatch_to_qa.py,sha256=NHihwAoCJ5_Lk11e_jZnucVUGQyZHCB-YpkfMHBCpQk,8882
@@ -72,7 +73,7 @@ sunholo/database/sql/sb/delete_source_row.sql,sha256=r6fEuUKdbiLHCDGKSbKINDCpJjs
 sunholo/database/sql/sb/return_sources.sql,sha256=89KAnxfK8n_qGK9jy1OQT8f9n4uYUtYL5cCxbC2mj_c,255
 sunholo/database/sql/sb/setup.sql,sha256=CvoFvZQev2uWjmFa3aj3m3iuPFzAAJZ0S7Qi3L3-zZI,89
 sunholo/discovery_engine/__init__.py,sha256=hLgqRDJ22Aov9o2QjAEfsVgnL3kMdM-g5p8RJ9OyKdQ,130
-sunholo/discovery_engine/chunker_handler.py,sha256=44qlTpdtz2GKzrhoQrxVMk-RPVFp7vQDPJoe9KmCcsw,7517
+sunholo/discovery_engine/chunker_handler.py,sha256=wkvXl4rFtYfN6AZUKdW9_QD49Whf77BukDbO82UwlAg,7480
 sunholo/discovery_engine/cli.py,sha256=KGVle5rkLL49oF9TQhrGI--8017IvvLOEoYur545Qb0,12790
 sunholo/discovery_engine/create_new.py,sha256=WUi4_xh_dFaGX3xA9jkNKZhaR6LCELjMPeRb0hyj4FU,1226
 sunholo/discovery_engine/discovery_engine_client.py,sha256=0KhKRFKCvqvtkUOIrCXNk5353t9duuEtUQDhQnN2B24,37335
@@ -168,9 +169,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.127.3.dist-info/licenses/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
-sunholo-0.127.3.dist-info/METADATA,sha256=Cw-b89amd_DtSRdUVOV-kPMUnVKfMzHj1m9oAnvzam4,10084
-sunholo-0.127.3.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-sunholo-0.127.3.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
-sunholo-0.127.3.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
-sunholo-0.127.3.dist-info/RECORD,,
+sunholo-0.129.0.dist-info/licenses/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.129.0.dist-info/METADATA,sha256=57vcuIcATCDaw7YigDz2bXfPqGp3PBSToaTLJLIiJRo,10084
+sunholo-0.129.0.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+sunholo-0.129.0.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.129.0.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.129.0.dist-info/RECORD,,