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

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_page_public.py

@@ -1,10 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .span_public import SpanPublic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+import typing_extensions
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from ..core.serialization import FieldMetadata
+from .span_public import SpanPublic
 
 
 class SpanPagePublic(UniversalBaseModel):
@@ -12,11 +14,12 @@ class SpanPagePublic(UniversalBaseModel):
     size: typing.Optional[int] = None
     total: typing.Optional[int] = None
     content: typing.Optional[typing.List[SpanPublic]] = 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:

opik/rest_api/types/span_public.py

@@ -1,29 +1,34 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from .span_public_type import SpanPublicType
 import datetime as dt
-from .json_node_public import JsonNodePublic
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .comment_public import CommentPublic
 from .error_info_public import ErrorInfoPublic
 from .feedback_score_public import FeedbackScorePublic
-from .comment_public import CommentPublic
-import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from .json_list_string_public import JsonListStringPublic
+from .span_public_type import SpanPublicType
 
 
 class SpanPublic(UniversalBaseModel):
     id: typing.Optional[str] = None
+    project_name: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    If null, the default project is used
+    """
+
     project_id: typing.Optional[str] = None
-    trace_id: str
+    trace_id: typing.Optional[str] = None
     parent_span_id: typing.Optional[str] = None
-    name: str
-    type: SpanPublicType
+    name: typing.Optional[str] = None
+    type: typing.Optional[SpanPublicType] = None
     start_time: dt.datetime
     end_time: typing.Optional[dt.datetime] = None
-    input: typing.Optional[JsonNodePublic] = None
-    output: typing.Optional[JsonNodePublic] = None
-    metadata: typing.Optional[JsonNodePublic] = None
+    input: typing.Optional[JsonListStringPublic] = None
+    output: typing.Optional[JsonListStringPublic] = None
+    metadata: typing.Optional[JsonListStringPublic] = None
     model: typing.Optional[str] = None
     provider: typing.Optional[str] = None
     tags: typing.Optional[typing.List[str]] = None
@@ -43,9 +48,7 @@ class SpanPublic(UniversalBaseModel):
     """
 
     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/span_public_type.py

@@ -2,4 +2,4 @@
 
 import typing
 
-SpanPublicType = typing.Union[typing.Literal["general", "tool", "llm"], typing.Any]
+SpanPublicType = typing.Union[typing.Literal["general", "tool", "llm", "guardrail"], typing.Any]

opik/rest_api/types/span_type.py

@@ -2,4 +2,4 @@
 
 import typing
 
-SpanType = typing.Union[typing.Literal["general", "tool", "llm"], typing.Any]
+SpanType = typing.Union[typing.Literal["general", "tool", "llm", "guardrail"], typing.Any]

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_update_type.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+SpanUpdateType = typing.Union[typing.Literal["general", "tool", "llm", "guardrail"], typing.Any]

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

@@ -1,13 +1,13 @@
 # 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
-from .span_write_type import SpanWriteType
-import datetime as dt
-from .json_node_write import JsonNodeWrite
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .error_info_write import ErrorInfoWrite
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from .json_list_string_write import JsonListStringWrite
+from .span_write_type import SpanWriteType
 
 
 class SpanWrite(UniversalBaseModel):
@@ -17,27 +17,26 @@ class SpanWrite(UniversalBaseModel):
     If null, the default project is used
     """
 
-    trace_id: str
+    trace_id: typing.Optional[str] = None
     parent_span_id: typing.Optional[str] = None
-    name: str
-    type: SpanWriteType
+    name: typing.Optional[str] = None
+    type: typing.Optional[SpanWriteType] = None
     start_time: dt.datetime
     end_time: typing.Optional[dt.datetime] = None
-    input: typing.Optional[JsonNodeWrite] = None
-    output: typing.Optional[JsonNodeWrite] = None
-    metadata: typing.Optional[JsonNodeWrite] = None
+    input: typing.Optional[JsonListStringWrite] = None
+    output: typing.Optional[JsonListStringWrite] = None
+    metadata: typing.Optional[JsonListStringWrite] = 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
     error_info: typing.Optional[ErrorInfoWrite] = None
+    last_updated_at: typing.Optional[dt.datetime] = None
     total_estimated_cost: typing.Optional[float] = None
     total_estimated_cost_version: typing.Optional[str] = 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/span_write_type.py

@@ -2,4 +2,4 @@
 
 import typing
 
-SpanWriteType = typing.Union[typing.Literal["general", "tool", "llm"], typing.Any]
+SpanWriteType = typing.Union[typing.Literal["general", "tool", "llm", "guardrail"], typing.Any]

opik/rest_api/types/spans_count_response.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 .workspace_spans_count import WorkspaceSpansCount
+
+
+class SpansCountResponse(UniversalBaseModel):
+    workspaces_spans_count: typing.Optional[typing.List[WorkspaceSpansCount]] = 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/start_multipart_upload_response.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 StartMultipartUploadResponse(UniversalBaseModel):
+    upload_id: str
+    pre_sign_urls: typing.List[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/stream_options.py

@@ -1,18 +1,16 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class StreamOptions(UniversalBaseModel):
     include_usage: typing.Optional[bool] = 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/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