galileo-core 3.87.2__tar.gz → 3.88.0__tar.gz

{galileo_core-3.87.2 → galileo_core-3.88.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: galileo-core
-Version: 3.87.2
+Version: 3.88.0
 Summary: Shared schemas and configuration for Galileo's Python packages.
 License: Apache-2.0
 Keywords: llm,quality,language_models,galileo

{galileo_core-3.87.2 → galileo_core-3.88.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "galileo-core"
-version = "3.87.2"
+version = "3.88.0"
 description = "Shared schemas and configuration for Galileo's Python packages."
 authors = [{ name = "Galileo Technologies Inc.", email = "team@rungalileo.io" }]
 readme = "README.md"

galileo_core-3.88.0/src/galileo_core/__init__.py

@@ -0,0 +1 @@
+__version__ = "3.88.0"

{galileo_core-3.87.2 → galileo_core-3.88.0}/src/galileo_core/schemas/logging/tree.py

@@ -3,9 +3,42 @@ from typing import Dict, List, Sequence, cast
 from uuid import UUID
 
 from galileo_core.schemas.logging.session import BaseSession, Session
-from galileo_core.schemas.logging.span import Span
+from galileo_core.schemas.logging.span import Span, SpanAdapter
 from galileo_core.schemas.logging.step import BaseStep, StepType
 from galileo_core.schemas.logging.trace import Trace
+from galileo_core.schemas.shared.records import RecordType
+
+
+def records_to_base_steps(records: Sequence[RecordType]) -> List[BaseStep]:
+    """
+    Convert RecordType objects to BaseStep objects by validating from dicts.
+
+    This avoids re-validation issues that occur when passing model instances
+    directly to create_tree_for_trace/create_tree_for_session. By converting
+    to dicts first, Pydantic validates fresh data instead of re-validating
+    existing model instances.
+
+    Args:
+        records: Sequence of RecordType objects (TraceRecord, SpanRecord, SessionRecord, etc.)
+
+    Returns:
+        List of BaseStep objects suitable for passing to create_tree_for_trace/create_tree_for_session
+    """
+    result: List[BaseStep] = []
+    for record in records:
+        record_dict = record.model_dump()
+        record_type = record_dict.get("type")
+
+        if record_type == StepType.trace:
+            # Validate as Trace directly so create_tree_for_trace can use it
+            result.append(Trace.model_validate(record_dict))
+        elif record_type == StepType.session:
+            # Validate as BaseSession - create_tree_for_session will convert to Session
+            result.append(BaseSession.model_validate(record_dict))
+        else:
+            # For spans, validate through SpanAdapter to get correct span type
+            result.append(SpanAdapter.validate_python(record_dict))
+    return result
 
 
 def create_tree_for_trace(records: Sequence[BaseStep]) -> Sequence[Trace]:

galileo_core-3.87.2/src/galileo_core/__init__.py

@@ -1 +0,0 @@
-__version__ = "3.87.2"