opik 1.9.5__py3-none-any.whl → 1.9.39__py3-none-any.whl

opik/rest_api/types/optimization_studio_config_write.py

@@ -0,0 +1,27 @@
+# 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_evaluation_write import StudioEvaluationWrite
+from .studio_llm_model_write import StudioLlmModelWrite
+from .studio_optimizer_write import StudioOptimizerWrite
+from .studio_prompt_write import StudioPromptWrite
+
+
+class OptimizationStudioConfigWrite(UniversalBaseModel):
+    dataset_name: str
+    prompt: StudioPromptWrite
+    llm_model: StudioLlmModelWrite
+    evaluation: StudioEvaluationWrite
+    optimizer: StudioOptimizerWrite
+
+    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/optimization_studio_log.py

@@ -0,0 +1,22 @@
+# 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
+
+
+class OptimizationStudioLog(UniversalBaseModel):
+    url: typing.Optional[str] = None
+    last_modified: typing.Optional[dt.datetime] = None
+    expires_at: typing.Optional[dt.datetime] = 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/optimization_write.py

@@ -6,6 +6,7 @@ import typing
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .json_list_string_write import JsonListStringWrite
+from .optimization_studio_config_write import OptimizationStudioConfigWrite
 from .optimization_write_status import OptimizationWriteStatus
 
 
@@ -16,6 +17,7 @@ class OptimizationWrite(UniversalBaseModel):
     objective_name: str
     status: OptimizationWriteStatus
     metadata: typing.Optional[JsonListStringWrite] = None
+    studio_config: typing.Optional[OptimizationStudioConfigWrite] = None
     last_updated_at: typing.Optional[dt.datetime] = None
 
     if IS_PYDANTIC_V2:

opik/rest_api/types/optimization_write_status.py

@@ -2,4 +2,6 @@
 
 import typing
 
-OptimizationWriteStatus = typing.Union[typing.Literal["running", "completed", "cancelled"], typing.Any]
+OptimizationWriteStatus = typing.Union[
+    typing.Literal["running", "completed", "cancelled", "initialized", "error"], typing.Any
+]

opik/rest_api/types/prompt.py

@@ -6,6 +6,7 @@ import typing
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .json_node import JsonNode
+from .prompt_template_structure import PromptTemplateStructure
 from .prompt_type import PromptType
 from .prompt_version import PromptVersion
 
@@ -18,6 +19,11 @@ class Prompt(UniversalBaseModel):
     metadata: typing.Optional[JsonNode] = None
     change_description: typing.Optional[str] = None
     type: typing.Optional[PromptType] = None
+    template_structure: typing.Optional[PromptTemplateStructure] = pydantic.Field(default=None)
+    """
+    Template structure type: 'text' or 'chat'. Immutable after creation.
+    """
+
     tags: typing.Optional[typing.List[str]] = None
     created_at: typing.Optional[dt.datetime] = None
     created_by: typing.Optional[str] = None

opik/rest_api/types/prompt_detail.py

@@ -5,6 +5,7 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .prompt_detail_template_structure import PromptDetailTemplateStructure
 from .prompt_version_detail import PromptVersionDetail
 
 
@@ -12,6 +13,11 @@ class PromptDetail(UniversalBaseModel):
     id: typing.Optional[str] = None
     name: str
     description: typing.Optional[str] = None
+    template_structure: typing.Optional[PromptDetailTemplateStructure] = pydantic.Field(default=None)
+    """
+    Template structure type: 'text' or 'chat'. Immutable after creation.
+    """
+
     tags: typing.Optional[typing.List[str]] = None
     created_at: typing.Optional[dt.datetime] = None
     created_by: typing.Optional[str] = None

opik/rest_api/types/prompt_detail_template_structure.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+PromptDetailTemplateStructure = typing.Union[typing.Literal["text", "chat"], typing.Any]

opik/rest_api/types/prompt_public.py

@@ -5,12 +5,18 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .prompt_public_template_structure import PromptPublicTemplateStructure
 
 
 class PromptPublic(UniversalBaseModel):
     id: typing.Optional[str] = None
     name: str
     description: typing.Optional[str] = None
+    template_structure: typing.Optional[PromptPublicTemplateStructure] = pydantic.Field(default=None)
+    """
+    Template structure type: 'text' or 'chat'. Immutable after creation.
+    """
+
     tags: typing.Optional[typing.List[str]] = None
     created_at: typing.Optional[dt.datetime] = None
     created_by: typing.Optional[str] = None

opik/rest_api/types/prompt_public_template_structure.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+PromptPublicTemplateStructure = typing.Union[typing.Literal["text", "chat"], typing.Any]

opik/rest_api/types/prompt_template_structure.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+PromptTemplateStructure = typing.Union[typing.Literal["text", "chat"], typing.Any]

opik/rest_api/types/prompt_version.py

@@ -6,6 +6,7 @@ import typing
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .json_node import JsonNode
+from .prompt_version_template_structure import PromptVersionTemplateStructure
 from .prompt_version_type import PromptVersionType
 
 
@@ -26,6 +27,7 @@ class PromptVersion(UniversalBaseModel):
     type: typing.Optional[PromptVersionType] = None
     change_description: typing.Optional[str] = None
     variables: typing.Optional[typing.List[str]] = None
+    template_structure: typing.Optional[PromptVersionTemplateStructure] = None
     created_at: typing.Optional[dt.datetime] = None
     created_by: typing.Optional[str] = None
 

opik/rest_api/types/prompt_version_detail.py

@@ -6,6 +6,7 @@ import typing
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .json_node_detail import JsonNodeDetail
+from .prompt_version_detail_template_structure import PromptVersionDetailTemplateStructure
 from .prompt_version_detail_type import PromptVersionDetailType
 
 
@@ -26,6 +27,7 @@ class PromptVersionDetail(UniversalBaseModel):
     type: typing.Optional[PromptVersionDetailType] = None
     change_description: typing.Optional[str] = None
     variables: typing.Optional[typing.List[str]] = None
+    template_structure: typing.Optional[PromptVersionDetailTemplateStructure] = None
     created_at: typing.Optional[dt.datetime] = None
     created_by: typing.Optional[str] = None
 

opik/rest_api/types/prompt_version_detail_template_structure.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+PromptVersionDetailTemplateStructure = typing.Union[typing.Literal["text", "chat"], typing.Any]

opik/rest_api/types/prompt_version_public.py

@@ -6,6 +6,7 @@ import typing
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .json_node_public import JsonNodePublic
+from .prompt_version_public_template_structure import PromptVersionPublicTemplateStructure
 from .prompt_version_public_type import PromptVersionPublicType
 
 
@@ -25,6 +26,7 @@ class PromptVersionPublic(UniversalBaseModel):
     metadata: typing.Optional[JsonNodePublic] = None
     type: typing.Optional[PromptVersionPublicType] = None
     change_description: typing.Optional[str] = None
+    template_structure: typing.Optional[PromptVersionPublicTemplateStructure] = None
     created_at: typing.Optional[dt.datetime] = None
     created_by: typing.Optional[str] = None
 

opik/rest_api/types/prompt_version_public_template_structure.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+PromptVersionPublicTemplateStructure = typing.Union[typing.Literal["text", "chat"], typing.Any]

opik/rest_api/types/prompt_version_template_structure.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+PromptVersionTemplateStructure = typing.Union[typing.Literal["text", "chat"], 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,10 +13,16 @@ 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")]
     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")]
 
     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_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_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_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/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