galileo-core 3.87.1__py3-none-any.whl → 3.88.0__py3-none-any.whl

galileo_core/__init__.py

@@ -1 +1 @@
-__version__ = "3.87.1"
+__version__ = "3.88.0"

galileo_core/schemas/logging/step.py

@@ -182,7 +182,7 @@ class BaseStep(BaseModel):
         description="Galileo ID of the trace containing the span (or the same value as id for a trace)",
     )
 
-    model_config = ConfigDict(validate_assignment=True, from_attributes=True)
+    model_config = ConfigDict(validate_assignment=True)
 
     def __init__(self, **data: Any):
         for k, v in list(data.items()):

galileo_core/schemas/logging/tree.py

@@ -1,11 +1,44 @@
 from operator import attrgetter
-from typing import Dict, List, Sequence
+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, 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]:
@@ -15,13 +48,10 @@ def create_tree_for_trace(records: Sequence[BaseStep]) -> Sequence[Trace]:
     for record in records:
         if record.id is None:
             raise ValueError("Node ID cannot be None")
-        # Handle both enum and string values for type
-        record_type = record.type if isinstance(record.type, StepType) else StepType(record.type)
-        if record_type == StepType.trace:
-            trace_id_to_trace[record.id] = Trace.model_validate(record, from_attributes=True)
-        elif record_type != StepType.session:
-            # Use SpanAdapter with discriminator to pick the correct span class
-            span_id_to_span[record.id] = SpanAdapter.validate_python(record, from_attributes=True)
+        if record.type is StepType.trace:
+            trace_id_to_trace[record.id] = Trace.model_validate(record)
+        elif record.type is not StepType.session:
+            span_id_to_span[record.id] = cast(Span, record)
             if (
                 hasattr(record, "parent_id") and record.parent_id is not None
             ):  # Only add value to the parent-child relationship if there is a parent
@@ -54,10 +84,8 @@ def create_tree_for_session(records: Sequence[BaseStep]) -> List[Session]:
     for record in records:
         if record.id is None:
             raise ValueError("Node ID cannot be None")
-        # Handle both enum and string values for type
-        record_type = record.type if isinstance(record.type, StepType) else StepType(record.type)
-        if record_type == StepType.session:
-            session = BaseSession.model_validate(record, from_attributes=True)
+        if record.type is StepType.session:
+            session = BaseSession.model_validate(record)
             session_id_to_session[record.id] = Session(**session.model_dump(), traces=[])
         else:
             if (

{galileo_core-3.87.1.dist-info → galileo_core-3.88.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: galileo-core
-Version: 3.87.1
+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.1.dist-info → galileo_core-3.88.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-galileo_core/__init__.py,sha256=JFk7X0ZpQQNC5vJgOTAgZQ7hqeaSnhrPcjt1w7A0GVg,23
+galileo_core/__init__.py,sha256=F8ra0owkga0tPRG5jB7lK5XkSV5lY8Fd_Z-wmAIHr4E,23
 galileo_core/constants/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 galileo_core/constants/config.py,sha256=MYvixj3X3CoiC38QVMdidApURKFOAaPaYoHhFESOJlU,329
 galileo_core/constants/dataset_format.py,sha256=H6ZigDd2QMpRcRhLC38e0hRoMh0kKbvdhbmS5yK1gZU,116
@@ -74,9 +74,9 @@ galileo_core/schemas/logging/agent.py,sha256=zexqP7o_wjicj4H1ZhrF9wkaYBxcj-skLv_
 galileo_core/schemas/logging/llm.py,sha256=BN2oya2sqAWdOxthN2bCWCfDXEsjKBTCXjBYWh4Q758,7521
 galileo_core/schemas/logging/session.py,sha256=R7J5B33ffdQrNpzj7HEBLjQ3rKqNRuHOYttO_-PK378,751
 galileo_core/schemas/logging/span.py,sha256=smaGm3kvghRi3seaFdMKcgdvRF0ZLerptBIZAh-I6i4,12722
-galileo_core/schemas/logging/step.py,sha256=Ar0S61lxGAABvZGHt6AeXlPErydvEMy0pXACXWW5uNA,8724
+galileo_core/schemas/logging/step.py,sha256=rRwRSeFbUFbnWg6mOfclo3s6VTtPayLE2-7qBMwqKFQ,8702
 galileo_core/schemas/logging/trace.py,sha256=irLlf6XyPL0k326A3u_dfN7sb6TzEdfR8veCXWB3c3o,925
-galileo_core/schemas/logging/tree.py,sha256=zOiblz_WZIYV0AK_92HZchAyCjabtSr9zs9iEekOUOQ,3658
+galileo_core/schemas/logging/tree.py,sha256=QZT2b77ASckcH3DOAF7_hMEDpoh4Rq1A0XvpQcF58qE,4622
 galileo_core/schemas/logging/union.py,sha256=iSQYPqlJkVBxkmxXaGNe6XPIL6nzP-6Ckb5fiO7zrCQ,396
 galileo_core/schemas/protect/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 galileo_core/schemas/protect/action.py,sha256=PmCdiMOdC72m3C--avb4o_IY7DI8Coj61EV80krv7zA,1720
@@ -123,8 +123,8 @@ galileo_core/utils/dataset.py,sha256=wjm3NdNRuxj5mK7GqQUdA-Map_EB0SpFfM1QKapd6-A
 galileo_core/utils/json.py,sha256=ug5P-iozgmqXRhLZS3Dm_JPViTICboKukg1mqqBq2i4,327
 galileo_core/utils/name.py,sha256=kaDeZJ1K3-SSazYrUeSbUsQrnZlnOBYU42t-QvX-coE,204
 galileo_core/utils/scorer_validation.py,sha256=bjNmDZhNsmmb8qHBfndXVAL60W5nCvoDdKrAoAegxJg,11259
-galileo_core-3.87.1.dist-info/LICENSE,sha256=4D_6tarPnIYkAqF-0LzrodE9wFtSulO3V4RZf__MriE,10946
-galileo_core-3.87.1.dist-info/METADATA,sha256=C2cs_-9OqLEgie5Cks7m_rHuOTYDRe0TI_3vebMxvCU,2731
-galileo_core-3.87.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-galileo_core-3.87.1.dist-info/entry_points.txt,sha256=epOvdlN6_7ua8stNHURhJhXDIIrheccwYv0Z-QFmHJU,61
-galileo_core-3.87.1.dist-info/RECORD,,
+galileo_core-3.88.0.dist-info/LICENSE,sha256=4D_6tarPnIYkAqF-0LzrodE9wFtSulO3V4RZf__MriE,10946
+galileo_core-3.88.0.dist-info/METADATA,sha256=ferBML3uAZ_CaVUheREXRi5FvmqXWgzYi4hh_ULbdpQ,2731
+galileo_core-3.88.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+galileo_core-3.88.0.dist-info/entry_points.txt,sha256=epOvdlN6_7ua8stNHURhJhXDIIrheccwYv0Z-QFmHJU,61
+galileo_core-3.88.0.dist-info/RECORD,,