label-studio-sdk 1.0.5__py3-none-any.whl → 1.0.8__py3-none-any.whl

label_studio_sdk/types/__init__.py

@@ -13,6 +13,8 @@ from .base_task import BaseTask
 from .base_task_file_upload import BaseTaskFileUpload
 from .base_task_updated_by import BaseTaskUpdatedBy
 from .base_user import BaseUser
+from .comment import Comment
+from .comment_created_by import CommentCreatedBy
 from .converted_format import ConvertedFormat
 from .converted_format_status import ConvertedFormatStatus
 from .data_manager_task_serializer import DataManagerTaskSerializer
@@ -31,6 +33,17 @@ from .gcs_export_storage import GcsExportStorage
 from .gcs_export_storage_status import GcsExportStorageStatus
 from .gcs_import_storage import GcsImportStorage
 from .gcs_import_storage_status import GcsImportStorageStatus
+from .inference_run import InferenceRun
+from .inference_run_cost_estimate import InferenceRunCostEstimate
+from .inference_run_created_by import InferenceRunCreatedBy
+from .inference_run_organization import InferenceRunOrganization
+from .inference_run_project_subset import InferenceRunProjectSubset
+from .inference_run_status import InferenceRunStatus
+from .key_indicator_value import KeyIndicatorValue
+from .key_indicators import KeyIndicators
+from .key_indicators_item import KeyIndicatorsItem
+from .key_indicators_item_additional_kpis_item import KeyIndicatorsItemAdditionalKpisItem
+from .key_indicators_item_extra_kpis_item import KeyIndicatorsItemExtraKpisItem
 from .local_files_export_storage import LocalFilesExportStorage
 from .local_files_export_storage_status import LocalFilesExportStorageStatus
 from .local_files_import_storage import LocalFilesImportStorage
@@ -38,6 +51,12 @@ from .local_files_import_storage_status import LocalFilesImportStorageStatus
 from .ml_backend import MlBackend
 from .ml_backend_auth_method import MlBackendAuthMethod
 from .ml_backend_state import MlBackendState
+from .model_provider_connection import ModelProviderConnection
+from .model_provider_connection_budget_reset_period import ModelProviderConnectionBudgetResetPeriod
+from .model_provider_connection_created_by import ModelProviderConnectionCreatedBy
+from .model_provider_connection_organization import ModelProviderConnectionOrganization
+from .model_provider_connection_provider import ModelProviderConnectionProvider
+from .model_provider_connection_scope import ModelProviderConnectionScope
 from .prediction import Prediction
 from .project import Project
 from .project_import import ProjectImport
@@ -45,10 +64,19 @@ from .project_import_status import ProjectImportStatus
 from .project_label_config import ProjectLabelConfig
 from .project_sampling import ProjectSampling
 from .project_skip_queue import ProjectSkipQueue
+from .prompt import Prompt
+from .prompt_created_by import PromptCreatedBy
+from .prompt_organization import PromptOrganization
+from .prompt_version import PromptVersion
+from .prompt_version_created_by import PromptVersionCreatedBy
+from .prompt_version_organization import PromptVersionOrganization
+from .prompt_version_provider import PromptVersionProvider
 from .redis_export_storage import RedisExportStorage
 from .redis_export_storage_status import RedisExportStorageStatus
 from .redis_import_storage import RedisImportStorage
 from .redis_import_storage_status import RedisImportStorageStatus
+from .refined_prompt_response import RefinedPromptResponse
+from .refined_prompt_response_refinement_status import RefinedPromptResponseRefinementStatus
 from .s3export_storage import S3ExportStorage
 from .s3export_storage_status import S3ExportStorageStatus
 from .s3import_storage import S3ImportStorage
@@ -60,6 +88,7 @@ from .serialization_option import SerializationOption
 from .serialization_options import SerializationOptions
 from .task import Task
 from .task_annotators_item import TaskAnnotatorsItem
+from .task_comment_authors_item import TaskCommentAuthorsItem
 from .task_filter_options import TaskFilterOptions
 from .user_simple import UserSimple
 from .view import View
@@ -83,6 +112,8 @@ __all__ = [
     "BaseTaskFileUpload",
     "BaseTaskUpdatedBy",
     "BaseUser",
+    "Comment",
+    "CommentCreatedBy",
     "ConvertedFormat",
     "ConvertedFormatStatus",
     "DataManagerTaskSerializer",
@@ -101,6 +132,17 @@ __all__ = [
     "GcsExportStorageStatus",
     "GcsImportStorage",
     "GcsImportStorageStatus",
+    "InferenceRun",
+    "InferenceRunCostEstimate",
+    "InferenceRunCreatedBy",
+    "InferenceRunOrganization",
+    "InferenceRunProjectSubset",
+    "InferenceRunStatus",
+    "KeyIndicatorValue",
+    "KeyIndicators",
+    "KeyIndicatorsItem",
+    "KeyIndicatorsItemAdditionalKpisItem",
+    "KeyIndicatorsItemExtraKpisItem",
     "LocalFilesExportStorage",
     "LocalFilesExportStorageStatus",
     "LocalFilesImportStorage",
@@ -108,6 +150,12 @@ __all__ = [
     "MlBackend",
     "MlBackendAuthMethod",
     "MlBackendState",
+    "ModelProviderConnection",
+    "ModelProviderConnectionBudgetResetPeriod",
+    "ModelProviderConnectionCreatedBy",
+    "ModelProviderConnectionOrganization",
+    "ModelProviderConnectionProvider",
+    "ModelProviderConnectionScope",
     "Prediction",
     "Project",
     "ProjectImport",
@@ -115,10 +163,19 @@ __all__ = [
     "ProjectLabelConfig",
     "ProjectSampling",
     "ProjectSkipQueue",
+    "Prompt",
+    "PromptCreatedBy",
+    "PromptOrganization",
+    "PromptVersion",
+    "PromptVersionCreatedBy",
+    "PromptVersionOrganization",
+    "PromptVersionProvider",
     "RedisExportStorage",
     "RedisExportStorageStatus",
     "RedisImportStorage",
     "RedisImportStorageStatus",
+    "RefinedPromptResponse",
+    "RefinedPromptResponseRefinementStatus",
     "S3ExportStorage",
     "S3ExportStorageStatus",
     "S3ImportStorage",
@@ -130,6 +187,7 @@ __all__ = [
     "SerializationOptions",
     "Task",
     "TaskAnnotatorsItem",
+    "TaskCommentAuthorsItem",
     "TaskFilterOptions",
     "UserSimple",
     "View",

label_studio_sdk/types/comment.py

@@ -0,0 +1,39 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+from .comment_created_by import CommentCreatedBy
+
+
+class Comment(pydantic_v1.BaseModel):
+    id: int
+    text: str
+    project: int
+    task: int
+    annotation: int
+    created_by: CommentCreatedBy
+    created_at: dt.datetime
+    updated_at: dt.datetime
+    is_resolved: typing.Optional[bool] = None
+    resolved_at: typing.Optional[dt.datetime] = None
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

label_studio_sdk/types/comment_created_by.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+CommentCreatedBy = typing.Union[int, typing.Dict[str, typing.Any]]

label_studio_sdk/types/inference_run.py

@@ -0,0 +1,43 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+from .inference_run_created_by import InferenceRunCreatedBy
+from .inference_run_organization import InferenceRunOrganization
+from .inference_run_project_subset import InferenceRunProjectSubset
+from .inference_run_status import InferenceRunStatus
+
+
+class InferenceRun(pydantic_v1.BaseModel):
+    organization: typing.Optional[InferenceRunOrganization] = None
+    project: int
+    model_version: typing.Optional[str] = None
+    created_by: typing.Optional[InferenceRunCreatedBy] = None
+    project_subset: InferenceRunProjectSubset
+    status: typing.Optional[InferenceRunStatus] = None
+    job_id: typing.Optional[str] = None
+    created_at: typing.Optional[dt.datetime] = None
+    triggered_at: typing.Optional[dt.datetime] = None
+    predictions_updated_at: typing.Optional[dt.datetime] = None
+    completed_at: typing.Optional[dt.datetime] = None
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

label_studio_sdk/types/inference_run_cost_estimate.py

@@ -0,0 +1,57 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+
+
+class InferenceRunCostEstimate(pydantic_v1.BaseModel):
+    prompt_cost_usd: typing.Optional[str] = pydantic_v1.Field(default=None)
+    """
+    Cost of the prompt (in USD)
+    """
+
+    completion_cost_usd: typing.Optional[str] = pydantic_v1.Field(default=None)
+    """
+    Cost of the completion (in USD)
+    """
+
+    total_cost_usd: typing.Optional[str] = pydantic_v1.Field(default=None)
+    """
+    Total cost of the inference (in USD)
+    """
+
+    is_error: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    """
+    Whether an error occurred or not
+    """
+
+    error_type: typing.Optional[str] = pydantic_v1.Field(default=None)
+    """
+    Type of error (e.g. "Timeout", "Rate Limit", etc)
+    """
+
+    error_message: typing.Optional[str] = pydantic_v1.Field(default=None)
+    """
+    Error message details
+    """
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

label_studio_sdk/types/inference_run_created_by.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+InferenceRunCreatedBy = typing.Union[int, typing.Dict[str, typing.Any]]

label_studio_sdk/types/inference_run_organization.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+InferenceRunOrganization = typing.Union[int, typing.Dict[str, typing.Any]]

label_studio_sdk/types/inference_run_project_subset.py

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

label_studio_sdk/types/inference_run_status.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+InferenceRunStatus = typing.Union[
+    typing.Literal["Pending", "InProgress", "Completed", "Failed", "Canceled"], typing.Any
+]

label_studio_sdk/types/key_indicator_value.py

@@ -0,0 +1,30 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+
+
+class KeyIndicatorValue(pydantic_v1.BaseModel):
+    title: typing.Optional[str] = None
+    values: typing.Optional[typing.Dict[str, typing.Any]] = None
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

label_studio_sdk/types/key_indicators.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .key_indicators_item import KeyIndicatorsItem
+
+KeyIndicators = typing.List[KeyIndicatorsItem]

label_studio_sdk/types/key_indicators_item.py

@@ -0,0 +1,51 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+from .key_indicators_item_additional_kpis_item import KeyIndicatorsItemAdditionalKpisItem
+from .key_indicators_item_extra_kpis_item import KeyIndicatorsItemExtraKpisItem
+
+
+class KeyIndicatorsItem(pydantic_v1.BaseModel):
+    key: str = pydantic_v1.Field()
+    """
+    The key for this KPI, where you can find the value from inside main_kpi
+    """
+
+    title: str = pydantic_v1.Field()
+    """
+    The title for this metric, to be displayed to the user
+    """
+
+    main_kpi: str
+    secondary_kpi: typing.Optional[str] = None
+    additional_kpis: typing.Optional[typing.List[KeyIndicatorsItemAdditionalKpisItem]] = pydantic_v1.Field(default=None)
+    """
+    Additional KPIs to be displayed at the bottom of the box
+    """
+
+    extra_kpis: typing.Optional[typing.List[KeyIndicatorsItemExtraKpisItem]] = pydantic_v1.Field(default=None)
+    """
+    Extra KPIs to be displayed in the hover-tootip for that indicator
+    """
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

label_studio_sdk/types/key_indicators_item_additional_kpis_item.py

@@ -0,0 +1,37 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+
+
+class KeyIndicatorsItemAdditionalKpisItem(pydantic_v1.BaseModel):
+    key: typing.Optional[str] = pydantic_v1.Field(default=None)
+    """
+    The key for this KPI, where you can find the value from inside main_kpi
+    """
+
+    label: typing.Optional[str] = pydantic_v1.Field(default=None)
+    """
+    The label for this KPI, to be displayed to the user
+    """
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

label_studio_sdk/types/key_indicators_item_extra_kpis_item.py

@@ -0,0 +1,37 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+
+
+class KeyIndicatorsItemExtraKpisItem(pydantic_v1.BaseModel):
+    key: typing.Optional[str] = pydantic_v1.Field(default=None)
+    """
+    The key for this KPI, where you can find the value from inside main_kpi
+    """
+
+    label: typing.Optional[str] = pydantic_v1.Field(default=None)
+    """
+    The label for this KPI, to be displayed to the user
+    """
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

label_studio_sdk/types/model_provider_connection.py

@@ -0,0 +1,71 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+from .model_provider_connection_budget_reset_period import ModelProviderConnectionBudgetResetPeriod
+from .model_provider_connection_created_by import ModelProviderConnectionCreatedBy
+from .model_provider_connection_organization import ModelProviderConnectionOrganization
+from .model_provider_connection_provider import ModelProviderConnectionProvider
+from .model_provider_connection_scope import ModelProviderConnectionScope
+
+
+class ModelProviderConnection(pydantic_v1.BaseModel):
+    provider: ModelProviderConnectionProvider
+    api_key: typing.Optional[str] = None
+    deployment_name: typing.Optional[str] = None
+    endpoint: typing.Optional[str] = None
+    scope: typing.Optional[ModelProviderConnectionScope] = None
+    organization: typing.Optional[ModelProviderConnectionOrganization] = None
+    created_by: typing.Optional[ModelProviderConnectionCreatedBy] = None
+    created_at: typing.Optional[dt.datetime] = None
+    updated_at: typing.Optional[dt.datetime] = None
+    is_internal: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    """
+    Whether the model provider connection is internal, not visible to the user.
+    """
+
+    budget_limit: typing.Optional[float] = pydantic_v1.Field(default=None)
+    """
+    Budget limit for the model provider connection (null if unlimited)
+    """
+
+    budget_last_reset_date: typing.Optional[dt.datetime] = pydantic_v1.Field(default=None)
+    """
+    Date and time the budget was last reset
+    """
+
+    budget_reset_period: typing.Optional[ModelProviderConnectionBudgetResetPeriod] = pydantic_v1.Field(default=None)
+    """
+    Budget reset period for the model provider connection (null if not reset)
+    """
+
+    budget_total_spent: typing.Optional[float] = pydantic_v1.Field(default=None)
+    """
+    Tracked total budget spent for the given provider connection within the current budget period
+    """
+
+    budget_alert_threshold: typing.Optional[float] = pydantic_v1.Field(default=None)
+    """
+    Budget alert threshold for the given provider connection
+    """
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

label_studio_sdk/types/model_provider_connection_budget_reset_period.py

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

label_studio_sdk/types/model_provider_connection_created_by.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+ModelProviderConnectionCreatedBy = typing.Union[int, typing.Dict[str, typing.Any]]

label_studio_sdk/types/model_provider_connection_organization.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+ModelProviderConnectionOrganization = typing.Union[int, typing.Dict[str, typing.Any]]

label_studio_sdk/types/model_provider_connection_provider.py

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

label_studio_sdk/types/model_provider_connection_scope.py

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

label_studio_sdk/types/prompt.py

@@ -0,0 +1,79 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+from .prompt_created_by import PromptCreatedBy
+from .prompt_organization import PromptOrganization
+
+
+class Prompt(pydantic_v1.BaseModel):
+    title: str = pydantic_v1.Field()
+    """
+    Title of the prompt
+    """
+
+    description: typing.Optional[str] = pydantic_v1.Field(default=None)
+    """
+    Description of the prompt
+    """
+
+    created_by: typing.Optional[PromptCreatedBy] = pydantic_v1.Field(default=None)
+    """
+    User ID of the creator of the prompt
+    """
+
+    created_at: typing.Optional[dt.datetime] = pydantic_v1.Field(default=None)
+    """
+    Date and time the prompt was created
+    """
+
+    updated_at: typing.Optional[dt.datetime] = pydantic_v1.Field(default=None)
+    """
+    Date and time the prompt was last updated
+    """
+
+    organization: typing.Optional[PromptOrganization] = pydantic_v1.Field(default=None)
+    """
+    Organization ID of the prompt
+    """
+
+    input_fields: typing.List[str] = pydantic_v1.Field()
+    """
+    List of input fields
+    """
+
+    output_classes: typing.List[str] = pydantic_v1.Field()
+    """
+    List of output classes
+    """
+
+    associated_projects: typing.Optional[typing.List[int]] = pydantic_v1.Field(default=None)
+    """
+    List of associated projects IDs
+    """
+
+    skill_name: typing.Optional[str] = pydantic_v1.Field(default=None)
+    """
+    Name of the skill
+    """
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

label_studio_sdk/types/prompt_created_by.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+PromptCreatedBy = typing.Union[int, typing.Dict[str, typing.Any]]