opik 1.6.4__py3-none-any.whl → 1.9.71__py3-none-any.whl

opik/rest_api/types/studio_message_write.py

@@ -0,0 +1,20 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class StudioMessageWrite(UniversalBaseModel):
+    role: str
+    content: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/studio_metric.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .json_node import JsonNode
+
+
+class StudioMetric(UniversalBaseModel):
+    type: str
+    parameters: typing.Optional[JsonNode] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/studio_metric_public.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .json_node_public import JsonNodePublic
+
+
+class StudioMetricPublic(UniversalBaseModel):
+    type: str
+    parameters: typing.Optional[JsonNodePublic] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/studio_metric_write.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .json_node_write import JsonNodeWrite
+
+
+class StudioMetricWrite(UniversalBaseModel):
+    type: str
+    parameters: typing.Optional[JsonNodeWrite] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/studio_optimizer.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .json_node import JsonNode
+
+
+class StudioOptimizer(UniversalBaseModel):
+    type: str
+    parameters: typing.Optional[JsonNode] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/studio_optimizer_public.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .json_node_public import JsonNodePublic
+
+
+class StudioOptimizerPublic(UniversalBaseModel):
+    type: str
+    parameters: typing.Optional[JsonNodePublic] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/studio_optimizer_write.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .json_node_write import JsonNodeWrite
+
+
+class StudioOptimizerWrite(UniversalBaseModel):
+    type: str
+    parameters: typing.Optional[JsonNodeWrite] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/studio_prompt.py

@@ -0,0 +1,20 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .studio_message import StudioMessage
+
+
+class StudioPrompt(UniversalBaseModel):
+    messages: typing.List[StudioMessage]
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/studio_prompt_public.py

@@ -0,0 +1,20 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .studio_message_public import StudioMessagePublic
+
+
+class StudioPromptPublic(UniversalBaseModel):
+    messages: typing.List[StudioMessagePublic]
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/studio_prompt_write.py

@@ -0,0 +1,20 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .studio_message_write import StudioMessageWrite
+
+
+class StudioPromptWrite(UniversalBaseModel):
+    messages: typing.List[StudioMessageWrite]
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/tool.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .function import Function
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .function import Function
 
 
 class Tool(UniversalBaseModel):
@@ -12,9 +12,7 @@ class Tool(UniversalBaseModel):
     function: typing.Optional[Function] = None
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:
 
         class Config:

opik/rest_api/types/tool_call.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .function_call import FunctionCall
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .function_call import FunctionCall
 
 
 class ToolCall(UniversalBaseModel):
@@ -14,9 +14,7 @@ class ToolCall(UniversalBaseModel):
     function: typing.Optional[FunctionCall] = None
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:
 
         class Config:

opik/rest_api/types/trace.py

@@ -1,14 +1,16 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+import datetime as dt
 import typing
+
 import pydantic
-import datetime as dt
-from .json_node import JsonNode
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .comment import Comment
 from .error_info import ErrorInfo
 from .feedback_score import FeedbackScore
-from .comment import Comment
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from .guardrails_validation import GuardrailsValidation
+from .json_list_string import JsonListString
+from .trace_visibility_mode import TraceVisibilityMode
 
 
 class Trace(UniversalBaseModel):
@@ -19,12 +21,12 @@ class Trace(UniversalBaseModel):
     """
 
     project_id: typing.Optional[str] = None
-    name: str
+    name: typing.Optional[str] = None
     start_time: dt.datetime
     end_time: typing.Optional[dt.datetime] = None
-    input: typing.Optional[JsonNode] = None
-    output: typing.Optional[JsonNode] = None
-    metadata: typing.Optional[JsonNode] = None
+    input: typing.Optional[JsonListString] = None
+    output: typing.Optional[JsonListString] = None
+    metadata: typing.Optional[JsonListString] = None
     tags: typing.Optional[typing.List[str]] = None
     error_info: typing.Optional[ErrorInfo] = None
     usage: typing.Optional[typing.Dict[str, int]] = None
@@ -33,19 +35,31 @@ class Trace(UniversalBaseModel):
     created_by: typing.Optional[str] = None
     last_updated_by: typing.Optional[str] = None
     feedback_scores: typing.Optional[typing.List[FeedbackScore]] = None
+    span_feedback_scores: typing.Optional[typing.List[FeedbackScore]] = pydantic.Field(default=None)
+    """
+    Aggregated feedback scores from all spans in this trace, averaged by score name
+    """
+
     comments: typing.Optional[typing.List[Comment]] = None
+    guardrails_validations: typing.Optional[typing.List[GuardrailsValidation]] = None
     total_estimated_cost: typing.Optional[float] = None
+    span_count: typing.Optional[int] = None
     duration: typing.Optional[float] = pydantic.Field(default=None)
     """
     Duration in milliseconds as a decimal number to support sub-millisecond precision
     """
 
     thread_id: typing.Optional[str] = None
+    visibility_mode: typing.Optional[TraceVisibilityMode] = None
+    llm_span_count: typing.Optional[int] = None
+    has_tool_spans: typing.Optional[bool] = None
+    providers: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    """
+    List of unique provider names from all spans in this trace, sorted alphabetically
+    """
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:
 
         class Config:

opik/rest_api/types/trace_batch.py

@@ -1,19 +1,17 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .trace import Trace
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .trace import Trace
 
 
 class TraceBatch(UniversalBaseModel):
     traces: typing.List[Trace]
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:
 
         class Config:

opik/rest_api/types/trace_count_response.py

@@ -1,19 +1,17 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .workspace_trace_count import WorkspaceTraceCount
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .workspace_trace_count import WorkspaceTraceCount
 
 
 class TraceCountResponse(UniversalBaseModel):
     workspaces_traces_count: typing.Optional[typing.List[WorkspaceTraceCount]] = None
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:
 
         class Config:

opik/rest_api/types/trace_enrichment_options.py

@@ -0,0 +1,32 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+import typing_extensions
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from ..core.serialization import FieldMetadata
+
+
+class TraceEnrichmentOptions(UniversalBaseModel):
+    """
+    Options for enriching trace data
+    """
+
+    include_spans: typing_extensions.Annotated[typing.Optional[bool], FieldMetadata(alias="includeSpans")] = None
+    include_tags: typing_extensions.Annotated[typing.Optional[bool], FieldMetadata(alias="includeTags")] = None
+    include_feedback_scores: typing_extensions.Annotated[
+        typing.Optional[bool], FieldMetadata(alias="includeFeedbackScores")
+    ] = None
+    include_comments: typing_extensions.Annotated[typing.Optional[bool], FieldMetadata(alias="includeComments")] = None
+    include_usage: typing_extensions.Annotated[typing.Optional[bool], FieldMetadata(alias="includeUsage")] = None
+    include_metadata: typing_extensions.Annotated[typing.Optional[bool], FieldMetadata(alias="includeMetadata")] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/trace_experiment_item_bulk_write_view.py

@@ -0,0 +1,41 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .error_info_experiment_item_bulk_write_view import ErrorInfoExperimentItemBulkWriteView
+from .json_list_string_experiment_item_bulk_write_view import JsonListStringExperimentItemBulkWriteView
+
+
+class TraceExperimentItemBulkWriteView(UniversalBaseModel):
+    """
+    Please provide either none, only one of evaluate_task_result or trace, but never both
+    """
+
+    id: typing.Optional[str] = None
+    project_name: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    If null, the default project is used
+    """
+
+    name: typing.Optional[str] = None
+    start_time: dt.datetime
+    end_time: typing.Optional[dt.datetime] = None
+    input: typing.Optional[JsonListStringExperimentItemBulkWriteView] = None
+    output: typing.Optional[JsonListStringExperimentItemBulkWriteView] = None
+    metadata: typing.Optional[JsonListStringExperimentItemBulkWriteView] = None
+    tags: typing.Optional[typing.List[str]] = None
+    error_info: typing.Optional[ErrorInfoExperimentItemBulkWriteView] = None
+    last_updated_at: typing.Optional[dt.datetime] = None
+    thread_id: typing.Optional[str] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/trace_filter.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .trace_filter_operator import TraceFilterOperator
+
+
+class TraceFilter(UniversalBaseModel):
+    field: typing.Optional[str] = None
+    operator: typing.Optional[TraceFilterOperator] = None
+    key: typing.Optional[str] = None
+    value: typing.Optional[str] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/trace_filter_operator.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+TraceFilterOperator = typing.Union[
+    typing.Literal[
+        "contains",
+        "not_contains",
+        "starts_with",
+        "ends_with",
+        "=",
+        "!=",
+        ">",
+        ">=",
+        "<",
+        "<=",
+        "is_empty",
+        "is_not_empty",
+    ],
+    typing.Any,
+]

opik/rest_api/types/trace_filter_public.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .trace_filter_public_operator import TraceFilterPublicOperator
+
+
+class TraceFilterPublic(UniversalBaseModel):
+    field: typing.Optional[str] = None
+    operator: typing.Optional[TraceFilterPublicOperator] = None
+    key: typing.Optional[str] = None
+    value: typing.Optional[str] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/trace_filter_public_operator.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+TraceFilterPublicOperator = typing.Union[
+    typing.Literal[
+        "contains",
+        "not_contains",
+        "starts_with",
+        "ends_with",
+        "=",
+        "!=",
+        ">",
+        ">=",
+        "<",
+        "<=",
+        "is_empty",
+        "is_not_empty",
+    ],
+    typing.Any,
+]

opik/rest_api/types/trace_filter_write.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .trace_filter_write_operator import TraceFilterWriteOperator
+
+
+class TraceFilterWrite(UniversalBaseModel):
+    field: typing.Optional[str] = None
+    operator: typing.Optional[TraceFilterWriteOperator] = None
+    key: typing.Optional[str] = None
+    value: typing.Optional[str] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/trace_filter_write_operator.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+TraceFilterWriteOperator = typing.Union[
+    typing.Literal[
+        "contains",
+        "not_contains",
+        "starts_with",
+        "ends_with",
+        "=",
+        "!=",
+        ">",
+        ">=",
+        "<",
+        "<=",
+        "is_empty",
+        "is_not_empty",
+    ],
+    typing.Any,
+]

opik/rest_api/types/trace_page_public.py

@@ -1,12 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .trace_public import TracePublic
+
+import pydantic
 import typing_extensions
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from ..core.serialization import FieldMetadata
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
+from .trace_public import TracePublic
 
 
 class TracePagePublic(UniversalBaseModel):
@@ -14,14 +14,12 @@ class TracePagePublic(UniversalBaseModel):
     size: typing.Optional[int] = None
     total: typing.Optional[int] = None
     content: typing.Optional[typing.List[TracePublic]] = None
-    sortable_by: typing_extensions.Annotated[
-        typing.Optional[typing.List[str]], FieldMetadata(alias="sortableBy")
-    ] = None
+    sortable_by: typing_extensions.Annotated[typing.Optional[typing.List[str]], FieldMetadata(alias="sortableBy")] = (
+        None
+    )
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:
 
         class Config: