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

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

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

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

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

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

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