sunholo 0.129.1__tar.gz → 0.130.2__tar.gz

{sunholo-0.129.1/src/sunholo.egg-info → sunholo-0.130.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sunholo
-Version: 0.129.1
+Version: 0.130.2
 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.129.1 → sunholo-0.130.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "sunholo"
-version = "0.129.1"
+version = "0.130.2"
 description = "Large Language Model DevOps - a package to help deploy LLMs to the Cloud."
 readme = "README.md"
 requires-python = ">=3.10"

{sunholo-0.129.1 → sunholo-0.130.2}/src/sunholo/agents/flask/vac_routes.py

@@ -238,12 +238,10 @@ if __name__ == "__main__":
 
         log.info(f'Streaming data with: {all_input}')
         if span:
-            generation = span.generation(
+            span.update(
                 name="start_streaming_chat",
                 metadata=vac_config.configs_by_kind,
-                input=all_input,
-                completion_start_time=str(int(datetime.datetime.now().timestamp())),
-                model=vac_config.vacConfig("model") or vac_config.vacConfig("llm")
+                input=all_input
             )
 
         def generate_response_content():
@@ -272,7 +270,6 @@ if __name__ == "__main__":
                                         if trace:
                                             chunk["trace_id"] = trace.id
                                             chunk["trace_url"] = trace.get_trace_url()
-                                            generation.end(output=json.dumps(chunk))
                                             span.end(output=json.dumps(chunk))
                                             trace.update(output=json.dumps(chunk))
                                         archive_qa(chunk, vector_name)
@@ -314,7 +311,6 @@ if __name__ == "__main__":
                                 chunk["trace_url"] = trace.get_trace_url()
                             archive_qa(chunk, vector_name)
                             if trace:
-                                generation.end(output=json.dumps(chunk))
                                 span.end(output=json.dumps(chunk))
                                 trace.update(output=json.dumps(chunk))
                             yield json.dumps(chunk)
@@ -330,7 +326,6 @@ if __name__ == "__main__":
 
         log.debug(f"streaming response: {response}")
         if trace:
-            generation.end(output=response)
             span.end(output=response)
             trace.update(output=response)
             self.langfuse_eval_response(trace_id=trace.id, eval_percent=all_input.get('eval_percent'))

{sunholo-0.129.1 → sunholo-0.130.2}/src/sunholo/custom_logging.py

@@ -142,14 +142,25 @@ class GoogleCloudLogging:
         """
         from .utils.version import sunholo_version
         
+        # Add version to log_text if it exists
         if log_text is not None:
             log_text = f"[{sunholo_version()}] {log_text}"
-            # Apply trace ID to log text
             log_text = self._append_trace_id(log_text)
         
-        if log_struct is not None:
-            # Add trace ID to structured log
-            log_struct = self._add_trace_to_struct(log_struct)
+        # Always create or update log_struct with trace_id if available
+        if not log_struct:
+            log_struct = {}
+        
+        # Make sure log_struct is a dictionary
+        if not isinstance(log_struct, dict):
+            log_struct = {"original_non_dict_value": str(log_struct)}
+        
+        # Add trace ID to log_struct
+        if hasattr(self, 'trace_id') and self.trace_id:
+            log_struct["trace_id"] = self.trace_id
+        
+        # Add version to log_struct
+        log_struct["version"] = sunholo_version()
         
         if not logger_name and not self.logger_name:
             raise ValueError("Must provide a logger name e.g. 'run.googleapis.com%2Fstderr'")
@@ -159,30 +170,28 @@ class GoogleCloudLogging:
             import logging as log
             log.basicConfig(level=log.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
             if log_text:
-                log.info(f"[{severity}][{logger_name or self.logger_name}][{self.version}] - {log_text}")
-            elif log_struct:
+                log.info(f"[{severity}][{logger_name or self.logger_name}][{self.version}] - {log_text} - {log_struct}")
+            else:
                 log.info(f"[{severity}][{logger_name or self.logger_name}][{self.version}] - {str(log_struct)}")
             return
         
         logger = self.client.logger(logger_name or self.logger_name)
-        
         caller_info = self._get_caller_info()
         
+        # Always log struct, and include message if provided
         if log_text:
-            try:
-                turn_to_text = str(log_text)
-                logger.log_text(turn_to_text, severity=severity, source_location=caller_info)
-            except Exception as err:
-                print(f"Could not log this: {log_text=} - {str(err)}")
+            log_struct["message"] = log_text
         
-        if log_struct:
-            if not isinstance(log_struct, dict):
-                print(f"Warning: log_struct must be a dictionary, got {type(log_struct)}")
-            else:
-                try:
-                    logger.log_struct(log_struct, severity=severity, source_location=caller_info)
-                except Exception as err:
-                    print(f"Could not log struct: {log_struct=} - {str(err)}")
+        try:
+            logger.log_struct(log_struct, severity=severity, source_location=caller_info)
+        except Exception as err:
+            print(f"Failed to log struct: {err}")
+            # Fallback to text logging
+            fallback_message = log_text if log_text else str(log_struct)
+            try:
+                logger.log_text(fallback_message, severity=severity, source_location=caller_info)
+            except Exception as text_err:
+                print(f"Even fallback text logging failed: {text_err}")
 
     def debug(self, log_text=None, log_struct=None):
 

{sunholo-0.129.1 → sunholo-0.130.2}/src/sunholo/database/alloydb_client.py

@@ -1167,7 +1167,8 @@ class AlloyDBClient:
             'total_rows': len(rows),
             'inserted_rows': 0,
             'failed_rows': 0,
-            'errors': []
+            'errors': [],
+            'return_ids': []
         }
         
         for i, row in enumerate(rows):
@@ -1197,6 +1198,7 @@ class AlloyDBClient:
                 
                 # Insert the row
                 result = await self._insert_single_row(table_name, filtered_row, primary_key_column=primary_key_column)
+                results['return_ids'].append(result)
                 results['inserted_rows'] += 1
                 
             except Exception as e:

{sunholo-0.129.1 → sunholo-0.130.2/src/sunholo.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sunholo
-Version: 0.129.1
+Version: 0.130.2
 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