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

opik/rest_api/types/provider_api_key_public.py

@@ -13,6 +13,11 @@ class ProviderApiKeyPublic(UniversalBaseModel):
     provider: ProviderApiKeyPublicProvider
     api_key: typing.Optional[str] = None
     name: typing.Optional[str] = None
+    provider_name: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Provider name - required for custom LLM providers to uniquely identify them (e.g., 'ollama', 'vllm'). Must not be blank for custom providers. Should not be set for standard providers (OpenAI, Anthropic, etc.). This requirement is conditional and validation is enforced programmatically.
+    """
+
     headers: typing.Optional[typing.Dict[str, str]] = None
     configuration: typing.Optional[typing.Dict[str, str]] = None
     base_url: typing.Optional[str] = None
@@ -20,6 +25,10 @@ class ProviderApiKeyPublic(UniversalBaseModel):
     created_by: typing.Optional[str] = None
     last_updated_at: typing.Optional[dt.datetime] = None
     last_updated_by: typing.Optional[str] = None
+    read_only: typing.Optional[bool] = pydantic.Field(default=None)
+    """
+    If true, this provider is system-managed and cannot be edited or deleted
+    """
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

opik/rest_api/types/provider_api_key_public_provider.py

@@ -3,5 +3,5 @@
 import typing
 
 ProviderApiKeyPublicProvider = typing.Union[
-    typing.Literal["openai", "anthropic", "gemini", "openrouter", "vertex-ai", "custom-llm"], typing.Any
+    typing.Literal["openai", "anthropic", "gemini", "openrouter", "vertex-ai", "custom-llm", "opik-free"], typing.Any
 ]

opik/rest_api/types/score_name.py

@@ -8,6 +8,7 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 class ScoreName(UniversalBaseModel):
     name: typing.Optional[str] = None
+    type: typing.Optional[str] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

opik/rest_api/types/service_toggles_config.py

@@ -13,8 +13,26 @@ class ServiceTogglesConfig(UniversalBaseModel):
     trace_thread_python_evaluator_enabled: typing_extensions.Annotated[
         bool, FieldMetadata(alias="traceThreadPythonEvaluatorEnabled")
     ]
+    span_llm_as_judge_enabled: typing_extensions.Annotated[bool, FieldMetadata(alias="spanLlmAsJudgeEnabled")]
+    span_user_defined_metric_python_enabled: typing_extensions.Annotated[
+        bool, FieldMetadata(alias="spanUserDefinedMetricPythonEnabled")
+    ]
     guardrails_enabled: typing_extensions.Annotated[bool, FieldMetadata(alias="guardrailsEnabled")]
     opik_ai_enabled: typing_extensions.Annotated[bool, FieldMetadata(alias="opikAIEnabled")]
+    alerts_enabled: typing_extensions.Annotated[bool, FieldMetadata(alias="alertsEnabled")]
+    welcome_wizard_enabled: typing_extensions.Annotated[bool, FieldMetadata(alias="welcomeWizardEnabled")]
+    csv_upload_enabled: typing_extensions.Annotated[bool, FieldMetadata(alias="csvUploadEnabled")]
+    export_enabled: typing_extensions.Annotated[bool, FieldMetadata(alias="exportEnabled")]
+    optimization_studio_enabled: typing_extensions.Annotated[bool, FieldMetadata(alias="optimizationStudioEnabled")]
+    dashboards_enabled: typing_extensions.Annotated[bool, FieldMetadata(alias="dashboardsEnabled")]
+    dataset_versioning_enabled: typing_extensions.Annotated[bool, FieldMetadata(alias="datasetVersioningEnabled")]
+    openai_provider_enabled: typing_extensions.Annotated[bool, FieldMetadata(alias="openaiProviderEnabled")]
+    anthropic_provider_enabled: typing_extensions.Annotated[bool, FieldMetadata(alias="anthropicProviderEnabled")]
+    gemini_provider_enabled: typing_extensions.Annotated[bool, FieldMetadata(alias="geminiProviderEnabled")]
+    openrouter_provider_enabled: typing_extensions.Annotated[bool, FieldMetadata(alias="openrouterProviderEnabled")]
+    vertexai_provider_enabled: typing_extensions.Annotated[bool, FieldMetadata(alias="vertexaiProviderEnabled")]
+    customllm_provider_enabled: typing_extensions.Annotated[bool, FieldMetadata(alias="customllmProviderEnabled")]
+    collaborators_tab_enabled: typing_extensions.Annotated[bool, FieldMetadata(alias="collaboratorsTabEnabled")]
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

opik/rest_api/types/span.py

@@ -9,7 +9,6 @@ from .comment import Comment
 from .error_info import ErrorInfo
 from .feedback_score import FeedbackScore
 from .json_list_string import JsonListString
-from .json_node import JsonNode
 from .span_type import SpanType
 
 
@@ -29,7 +28,7 @@ class Span(UniversalBaseModel):
     end_time: typing.Optional[dt.datetime] = None
     input: typing.Optional[JsonListString] = None
     output: typing.Optional[JsonListString] = None
-    metadata: typing.Optional[JsonNode] = None
+    metadata: typing.Optional[JsonListString] = None
     model: typing.Optional[str] = None
     provider: typing.Optional[str] = None
     tags: typing.Optional[typing.List[str]] = None

opik/rest_api/types/span_enrichment_options.py

@@ -0,0 +1,31 @@
+# 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 SpanEnrichmentOptions(UniversalBaseModel):
+    """
+    Options for enriching span data
+    """
+
+    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/span_experiment_item_bulk_write_view.py

@@ -7,7 +7,6 @@ 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
-from .json_node_experiment_item_bulk_write_view import JsonNodeExperimentItemBulkWriteView
 from .span_experiment_item_bulk_write_view_type import SpanExperimentItemBulkWriteViewType
 
 
@@ -20,7 +19,7 @@ class SpanExperimentItemBulkWriteView(UniversalBaseModel):
     end_time: typing.Optional[dt.datetime] = None
     input: typing.Optional[JsonListStringExperimentItemBulkWriteView] = None
     output: typing.Optional[JsonListStringExperimentItemBulkWriteView] = None
-    metadata: typing.Optional[JsonNodeExperimentItemBulkWriteView] = None
+    metadata: typing.Optional[JsonListStringExperimentItemBulkWriteView] = None
     model: typing.Optional[str] = None
     provider: typing.Optional[str] = None
     tags: typing.Optional[typing.List[str]] = None

opik/rest_api/types/span_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 .span_filter_operator import SpanFilterOperator
+
+
+class SpanFilter(UniversalBaseModel):
+    field: typing.Optional[str] = None
+    operator: typing.Optional[SpanFilterOperator] = 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/span_filter_operator.py

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

opik/rest_api/types/span_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 .span_filter_write_operator import SpanFilterWriteOperator
+
+
+class SpanFilterWrite(UniversalBaseModel):
+    field: typing.Optional[str] = None
+    operator: typing.Optional[SpanFilterWriteOperator] = 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/span_filter_write_operator.py

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

opik/rest_api/types/span_llm_as_judge_code.py

@@ -0,0 +1,27 @@
+# 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
+from .llm_as_judge_message import LlmAsJudgeMessage
+from .llm_as_judge_model_parameters import LlmAsJudgeModelParameters
+from .llm_as_judge_output_schema import LlmAsJudgeOutputSchema
+
+
+class SpanLlmAsJudgeCode(UniversalBaseModel):
+    model: LlmAsJudgeModelParameters
+    messages: typing.List[LlmAsJudgeMessage]
+    variables: typing.Dict[str, str]
+    schema_: typing_extensions.Annotated[typing.List[LlmAsJudgeOutputSchema], FieldMetadata(alias="schema")]
+
+    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/span_llm_as_judge_code_public.py

@@ -0,0 +1,27 @@
+# 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
+from .llm_as_judge_message_public import LlmAsJudgeMessagePublic
+from .llm_as_judge_model_parameters_public import LlmAsJudgeModelParametersPublic
+from .llm_as_judge_output_schema_public import LlmAsJudgeOutputSchemaPublic
+
+
+class SpanLlmAsJudgeCodePublic(UniversalBaseModel):
+    model: LlmAsJudgeModelParametersPublic
+    messages: typing.List[LlmAsJudgeMessagePublic]
+    variables: typing.Dict[str, str]
+    schema_: typing_extensions.Annotated[typing.List[LlmAsJudgeOutputSchemaPublic], FieldMetadata(alias="schema")]
+
+    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/span_llm_as_judge_code_write.py

@@ -0,0 +1,27 @@
+# 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
+from .llm_as_judge_message_write import LlmAsJudgeMessageWrite
+from .llm_as_judge_model_parameters_write import LlmAsJudgeModelParametersWrite
+from .llm_as_judge_output_schema_write import LlmAsJudgeOutputSchemaWrite
+
+
+class SpanLlmAsJudgeCodeWrite(UniversalBaseModel):
+    model: LlmAsJudgeModelParametersWrite
+    messages: typing.List[LlmAsJudgeMessageWrite]
+    variables: typing.Dict[str, str]
+    schema_: typing_extensions.Annotated[typing.List[LlmAsJudgeOutputSchemaWrite], FieldMetadata(alias="schema")]
+
+    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/span_public.py

@@ -9,7 +9,6 @@ from .comment_public import CommentPublic
 from .error_info_public import ErrorInfoPublic
 from .feedback_score_public import FeedbackScorePublic
 from .json_list_string_public import JsonListStringPublic
-from .json_node_public import JsonNodePublic
 from .span_public_type import SpanPublicType
 
 
@@ -29,7 +28,7 @@ class SpanPublic(UniversalBaseModel):
     end_time: typing.Optional[dt.datetime] = None
     input: typing.Optional[JsonListStringPublic] = None
     output: typing.Optional[JsonListStringPublic] = None
-    metadata: typing.Optional[JsonNodePublic] = None
+    metadata: typing.Optional[JsonListStringPublic] = None
     model: typing.Optional[str] = None
     provider: typing.Optional[str] = None
     tags: typing.Optional[typing.List[str]] = None

opik/rest_api/types/span_update.py

@@ -0,0 +1,46 @@
+# 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 import ErrorInfo
+from .json_list_string import JsonListString
+from .span_update_type import SpanUpdateType
+
+
+class SpanUpdate(UniversalBaseModel):
+    project_name: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    If null and project_id not specified, Default Project is assumed
+    """
+
+    project_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    If null and project_name not specified, Default Project is assumed
+    """
+
+    trace_id: str
+    parent_span_id: typing.Optional[str] = None
+    name: typing.Optional[str] = None
+    type: typing.Optional[SpanUpdateType] = None
+    end_time: typing.Optional[dt.datetime] = None
+    input: typing.Optional[JsonListString] = None
+    output: typing.Optional[JsonListString] = None
+    metadata: typing.Optional[JsonListString] = None
+    model: typing.Optional[str] = None
+    provider: typing.Optional[str] = None
+    tags: typing.Optional[typing.List[str]] = None
+    usage: typing.Optional[typing.Dict[str, int]] = None
+    total_estimated_cost: typing.Optional[float] = None
+    error_info: typing.Optional[ErrorInfo] = 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/span_user_defined_metric_python_code.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 SpanUserDefinedMetricPythonCode(UniversalBaseModel):
+    metric: str
+    arguments: typing.Dict[str, 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/span_user_defined_metric_python_code_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
+
+
+class SpanUserDefinedMetricPythonCodePublic(UniversalBaseModel):
+    metric: str
+    arguments: typing.Dict[str, 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/span_user_defined_metric_python_code_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 SpanUserDefinedMetricPythonCodeWrite(UniversalBaseModel):
+    metric: str
+    arguments: typing.Dict[str, 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/span_write.py

@@ -7,7 +7,6 @@ import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .error_info_write import ErrorInfoWrite
 from .json_list_string_write import JsonListStringWrite
-from .json_node_write import JsonNodeWrite
 from .span_write_type import SpanWriteType
 
 
@@ -26,7 +25,7 @@ class SpanWrite(UniversalBaseModel):
     end_time: typing.Optional[dt.datetime] = None
     input: typing.Optional[JsonListStringWrite] = None
     output: typing.Optional[JsonListStringWrite] = None
-    metadata: typing.Optional[JsonNodeWrite] = None
+    metadata: typing.Optional[JsonListStringWrite] = None
     model: typing.Optional[str] = None
     provider: typing.Optional[str] = None
     tags: typing.Optional[typing.List[str]] = None

opik/rest_api/types/studio_evaluation.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_metric import StudioMetric
+
+
+class StudioEvaluation(UniversalBaseModel):
+    metrics: typing.List[StudioMetric]
+
+    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_evaluation_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_metric_public import StudioMetricPublic
+
+
+class StudioEvaluationPublic(UniversalBaseModel):
+    metrics: typing.List[StudioMetricPublic]
+
+    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_evaluation_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_metric_write import StudioMetricWrite
+
+
+class StudioEvaluationWrite(UniversalBaseModel):
+    metrics: typing.List[StudioMetricWrite]
+
+    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_llm_model.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 StudioLlmModel(UniversalBaseModel):
+    model: 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_llm_model_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 StudioLlmModelPublic(UniversalBaseModel):
+    model: 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_llm_model_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 StudioLlmModelWrite(UniversalBaseModel):
+    model: 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_message.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 StudioMessage(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_message_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
+
+
+class StudioMessagePublic(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_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