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

galileo_core/__init__.py

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

galileo_core/schemas/logging/llm.py

@@ -65,6 +65,7 @@ class EventType(str, Enum):
     mcp_call = "mcp_call"
     mcp_list_tools = "mcp_list_tools"
     mcp_approval_request = "mcp_approval_request"
+    web_search_call = "web_search_call"
 
 
 class EventStatus(str, Enum):
@@ -105,7 +106,7 @@ class ReasoningEvent(BaseEvent):
 
     type: Literal[EventType.reasoning] = EventType.reasoning
     content: Optional[str] = Field(default=None, description="The reasoning/thinking content")
-    summary: Optional[str] = Field(default=None, description="Summary of the reasoning")
+    summary: Optional[Union[str, List[Dict[str, Any]]]] = Field(default=None, description="Summary of the reasoning")
 
 
 class InternalToolCall(BaseEvent):
@@ -121,6 +122,21 @@ class InternalToolCall(BaseEvent):
     output: Optional[Dict[str, Any]] = Field(default=None, description="Output/results from the tool call")
 
 
+class WebSearchAction(BaseModel):
+    """Action payload for a web search call event."""
+
+    type: Literal["search"] = Field(description="Type of web search action")
+    query: Optional[str] = Field(default=None, description="Search query string")
+    sources: Optional[Any] = Field(default=None, description="Optional provider-specific sources")
+
+
+class WebSearchCallEvent(BaseEvent):
+    """An OpenAI-style web search call event."""
+
+    type: Literal[EventType.web_search_call] = EventType.web_search_call
+    action: WebSearchAction = Field(description="Web search action payload")
+
+
 class ImageGenerationEvent(BaseEvent):
     """An image generation event from the model."""
 
@@ -171,6 +187,7 @@ Event = Annotated[
         MessageEvent,
         ReasoningEvent,
         InternalToolCall,
+        WebSearchCallEvent,
         ImageGenerationEvent,
         MCPCallEvent,
         MCPListToolsEvent,

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/schemas/shared/scorers/scorer_name.py

@@ -86,6 +86,8 @@ class ScorerName(str, Enum):
 
     rouge = "rouge"
 
+    reasoning_coherence = "reasoning_coherence"
+
     sql_efficiency = "sql_efficiency"
 
     sql_adherence = "sql_adherence"

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: galileo-core
-Version: 3.80.0
+Version: 3.82.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.82.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-galileo_core/__init__.py,sha256=9kHWlwK_NdaRDkSZqYPi2gl6X15DwsW3dOeK5xPOh5U,23
+galileo_core/__init__.py,sha256=Za0uTMfFaftHcrL2K7d8-r1a_Sek_wUhBGK_J52fbow,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
@@ -71,7 +71,7 @@ galileo_core/schemas/core/user_project.py,sha256=A9pxfmL8AFbwIhLHgBf22zadmEzHAOV
 galileo_core/schemas/core/user_role.py,sha256=HsQ-5fgGt4bWwIbhAE4PqzJWhHvINAZW2-WD5YTivng,329
 galileo_core/schemas/logging/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 galileo_core/schemas/logging/agent.py,sha256=zexqP7o_wjicj4H1ZhrF9wkaYBxcj-skLv_LXj8H7gU,252
-galileo_core/schemas/logging/llm.py,sha256=CqTGbq9Q26w8A0Wgk0PpioYV_cx270K3JIUMPXbM334,6836
+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=k4UdVNV4fbUaCQ4ZsKir125jbAnN3ZD1vDNFA7fVB1I,12480
 galileo_core/schemas/logging/step.py,sha256=rRwRSeFbUFbnWg6mOfclo3s6VTtPayLE2-7qBMwqKFQ,8702
@@ -99,14 +99,14 @@ 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
 galileo_core/schemas/shared/scorers/filters.py,sha256=CEiHddUQzwFq0m_2pnEMsIc5hZvO7m-ErE73BnJCPyc,782
 galileo_core/schemas/shared/scorers/mixins.py,sha256=PvyTqTJtrey5Buo6lJiDngMcUJxCKxcymK3OzXQJJC8,906
 galileo_core/schemas/shared/scorers/scorer_configuration.py,sha256=hSJ-rObA_Wqff__74wA1srmKKZplCBqSRmPuHB6EJhA,1120
-galileo_core/schemas/shared/scorers/scorer_name.py,sha256=7u2bwOIO1MUTf6DbhEy-NNCZZJPtPC7fUIPOy1YqUk0,3537
+galileo_core/schemas/shared/scorers/scorer_name.py,sha256=9MbDQzObVcmJl53AmmD-xrBM7Qxmm2YhNBr4RNhn2YE,3586
 galileo_core/schemas/shared/scorers/scorer_type.py,sha256=04ZPqHkvRR6b9ARitQJhGfn-9nKP4fNH_n1noeEt08c,1401
 galileo_core/schemas/shared/scorers/scorers.py,sha256=3OA8-QYkCslo7IOHrHQ5kVKRfBbaFvt5gG7WVTzi9pc,4962
 galileo_core/schemas/shared/traces_logger.py,sha256=HXydSyfHmizxcqYFzwAKYdamH1lfgRrxXThh1yirjYU,23498
@@ -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.82.0.dist-info/LICENSE,sha256=4D_6tarPnIYkAqF-0LzrodE9wFtSulO3V4RZf__MriE,10946
+galileo_core-3.82.0.dist-info/METADATA,sha256=wBo3uAicViV37ZZy7RtvYX4rpgie7DQJVlevlvsKICY,2731
+galileo_core-3.82.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+galileo_core-3.82.0.dist-info/entry_points.txt,sha256=epOvdlN6_7ua8stNHURhJhXDIIrheccwYv0Z-QFmHJU,61
+galileo_core-3.82.0.dist-info/RECORD,,