galileo-core 3.80.0__tar.gz → 3.81.0__tar.gz

{galileo_core-3.80.0 → galileo_core-3.81.0}/PKG-INFO

@@ -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 → galileo_core-3.81.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "galileo-core"
-version = "3.80.0"
+version = "3.81.0"
 description = "Shared schemas and configuration for Galileo's Python packages."
 authors = [{ name = "Galileo Technologies Inc.", email = "team@rungalileo.io" }]
 readme = "README.md"
@@ -39,6 +39,7 @@ homepage = "https://www.galileo.ai/"
 [tool.poetry.dependencies]
 python = "^3.10.0"
 pydantic = "^2.6.0"
+pydantic-partial = "^0.10.1"
 pydantic-settings = "^2.2.1"
 pyjwt = "^2.8.0"
 httpx = ">=0.27.0,<0.29.0"

galileo_core-3.81.0/src/galileo_core/__init__.py

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

{galileo_core-3.80.0 → galileo_core-3.81.0}/src/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/src/galileo_core/__init__.py

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