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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: galileo-core
-Version: 3.87.1
+Version: 3.87.2
 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 → galileo_core-3.87.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "galileo-core"
-version = "3.87.1"
+version = "3.87.2"
 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.87.2/src/galileo_core/__init__.py

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

{galileo_core-3.87.1 → galileo_core-3.87.2}/src/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-3.87.1 → galileo_core-3.87.2}/src/galileo_core/schemas/logging/tree.py

@@ -1,9 +1,9 @@
 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.span import Span
 from galileo_core.schemas.logging.step import BaseStep, StepType
 from galileo_core.schemas.logging.trace import Trace
 
@@ -15,13 +15,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 +51,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/src/galileo_core/__init__.py

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