galileo-core 3.80.0__py3-none-any.whl → 3.81.0__py3-none-any.whl

galileo_core/__init__.py

@@ -1 +1 @@
-__version__ = "3.80.0"
+__version__ = "3.81.0"

galileo_core/schemas/shared/records.py

@@ -1,7 +1,8 @@
 from datetime import datetime, timezone
-from typing import Dict, List, Optional, Sequence, Type, Union
+from typing import Dict, List, Optional, Sequence, Type, TypeAlias, Union
 
 from pydantic import UUID4, BaseModel, Field, TypeAdapter, field_validator
+from pydantic_partial import PartialModelMixin
 from typing_extensions import Annotated, Any, get_args
 
 from galileo_core.schemas.logging.session import BaseSession
@@ -20,7 +21,7 @@ class RecordIdsWithMetrics(BaseModel):
     metrics: Dict[str, Any]
 
 
-class BaseRecord(BaseStep):
+class BaseRecord(PartialModelMixin, BaseStep):
     id: UUID4 = Field(title="ID", description="Galileo ID of the session, trace or span")
     session_id: UUID4 = Field(title="Session ID", description="Galileo ID of the session")
     trace_id: Optional[UUID4] = Field(
@@ -234,3 +235,43 @@ class SessionRecordWithChildren(SessionRecord, RecordWithChildTraces):
 
 
 SpanRecordWithChildrenAdapter: TypeAdapter[SpanRecordWithChildren] = TypeAdapter(SpanRecordWithChildren)
+
+
+PartialAgentSpanRecord: TypeAlias = AgentSpanRecord.model_as_partial()  # type: ignore[valid-type]
+PartialWorkflowSpanRecord: TypeAlias = WorkflowSpanRecord.model_as_partial()  # type: ignore[valid-type]
+PartialLlmSpanRecord: TypeAlias = LlmSpanRecord.model_as_partial()  # type: ignore[valid-type]
+PartialToolSpanRecord: TypeAlias = ToolSpanRecord.model_as_partial()  # type: ignore[valid-type]
+PartialRetrieverSpanRecord: TypeAlias = RetrieverSpanRecord.model_as_partial()  # type: ignore[valid-type]
+PartialTraceRecord: TypeAlias = TraceRecord.model_as_partial()  # type: ignore[valid-type]
+PartialSessionRecord: TypeAlias = SessionRecord.model_as_partial()  # type: ignore[valid-type]
+
+
+PartialSpanRecord: TypeAlias = Annotated[
+    Union[
+        PartialAgentSpanRecord,
+        PartialWorkflowSpanRecord,
+        PartialLlmSpanRecord,
+        PartialToolSpanRecord,
+        PartialRetrieverSpanRecord,
+    ],
+    Field(discriminator="type"),
+]
+
+
+PartialSpanRecordAdapter: TypeAdapter[PartialSpanRecord] = TypeAdapter(PartialSpanRecord)  # type: ignore[valid-type]
+
+
+PartialSpanRecordTypes: List[Type[BaseRecord]] = list(get_args(get_args(PartialSpanRecord)[0]))
+
+
+PartialRecordType: TypeAlias = Annotated[
+    Union[
+        PartialTraceRecord,
+        PartialSpanRecord,
+        PartialSessionRecord,
+    ],
+    Field(discriminator="type"),
+]
+
+
+PartialRecordTypeAdapter: TypeAdapter[PartialRecordType] = TypeAdapter(PartialRecordType)  # type: ignore[valid-type]

{galileo_core-3.80.0.dist-info → galileo_core-3.81.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: galileo-core
-Version: 3.80.0
+Version: 3.81.0
 Summary: Shared schemas and configuration for Galileo's Python packages.
 License: Apache-2.0
 Keywords: llm,quality,language_models,galileo
@@ -31,6 +31,7 @@ Classifier: Typing :: Typed
 Provides-Extra: testing
 Requires-Dist: httpx (>=0.27.0,<0.29.0)
 Requires-Dist: pydantic (>=2.6.0,<3.0.0)
+Requires-Dist: pydantic-partial (>=0.10.1,<0.11.0)
 Requires-Dist: pydantic-settings (>=2.2.1,<3.0.0)
 Requires-Dist: pyjwt (>=2.8.0,<3.0.0)
 Requires-Dist: pytest (>=8.2.1,<9.0.0) ; extra == "testing"

{galileo_core-3.80.0.dist-info → galileo_core-3.81.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-galileo_core/__init__.py,sha256=9kHWlwK_NdaRDkSZqYPi2gl6X15DwsW3dOeK5xPOh5U,23
+galileo_core/__init__.py,sha256=HNRblLuZoS2yJW9FwHR4inHitiqokCTC0JFja9KuSNo,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
@@ -99,7 +99,7 @@ galileo_core/schemas/shared/message.py,sha256=-LQoadmHzeMYWAtmQEBwy0gDHw2VYfaSYf
 galileo_core/schemas/shared/message_role.py,sha256=fljQ9y6Rqgp2tAEsgRA25ys59waaR1F7JYZoG6sj-NA,186
 galileo_core/schemas/shared/messages.py,sha256=ENLwaLVSRbBL6HyOAXCbre-MoecjXG088hluVzastTk,423
 galileo_core/schemas/shared/metric.py,sha256=_Hw64RsLK7FBpx03minpQZwWb3oSfqiTU4KYEIBbpiI,249
-galileo_core/schemas/shared/records.py,sha256=0NctLyPrsXPomHrNVrC5zTZU0lRZ_AtFSLRx__Uok-8,8359
+galileo_core/schemas/shared/records.py,sha256=W8BHCqTQq-oA3zIk3_KyqqiTsLs1E02r097AaHM6zeE,9908
 galileo_core/schemas/shared/scorers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 galileo_core/schemas/shared/scorers/base_configs.py,sha256=ejZpnCmyj_n_6FPhDsRG2Y16IwCMH0mc27fUvJNUT2E,2631
 galileo_core/schemas/shared/scorers/chain_aggregation.py,sha256=OpuCmR6wJ8GGFJEyJHKT5yvPJqdmx9YI51i-gdwQDDo,145
@@ -122,8 +122,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.80.0.dist-info/LICENSE,sha256=4D_6tarPnIYkAqF-0LzrodE9wFtSulO3V4RZf__MriE,10946
-galileo_core-3.80.0.dist-info/METADATA,sha256=oDcQb9GteeJlWkPjLrQktjhUAd4VhH8dPRKBp0SupVk,2680
-galileo_core-3.80.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-galileo_core-3.80.0.dist-info/entry_points.txt,sha256=epOvdlN6_7ua8stNHURhJhXDIIrheccwYv0Z-QFmHJU,61
-galileo_core-3.80.0.dist-info/RECORD,,
+galileo_core-3.81.0.dist-info/LICENSE,sha256=4D_6tarPnIYkAqF-0LzrodE9wFtSulO3V4RZf__MriE,10946
+galileo_core-3.81.0.dist-info/METADATA,sha256=a01eAPr43IVC3tTjoEPrLk-kzAVkzHd_VQGQZYFm_2w,2731
+galileo_core-3.81.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+galileo_core-3.81.0.dist-info/entry_points.txt,sha256=epOvdlN6_7ua8stNHURhJhXDIIrheccwYv0Z-QFmHJU,61
+galileo_core-3.81.0.dist-info/RECORD,,