orca-sdk 0.0.95__py3-none-any.whl → 0.0.97__py3-none-any.whl

orca_sdk/_generated_api_client/models/constraint_violation_error_response_status_code.py

@@ -0,0 +1,8 @@
+from enum import IntEnum
+
+
+class ConstraintViolationErrorResponseStatusCode(IntEnum):
+    VALUE_409 = 409
+
+    def __str__(self) -> str:
+        return str(self.value)

orca_sdk/_generated_api_client/models/create_classification_model_request.py

@@ -28,6 +28,8 @@ class CreateClassificationModelRequest:
         name (str):
         memoryset_id (str):
         description (Union[None, Unset, str]):
+        notes (Union[None, Unset, str]):
+        memoryset_name (Union[None, Unset, str]):
         memory_lookup_count (Union[None, Unset, int]):
         head_type (Union[Unset, RACHeadType]):
         weigh_memories (Union[None, Unset, bool]):
@@ -38,6 +40,8 @@ class CreateClassificationModelRequest:
     name: str
     memoryset_id: str
     description: Union[None, Unset, str] = UNSET
+    notes: Union[None, Unset, str] = UNSET
+    memoryset_name: Union[None, Unset, str] = UNSET
     memory_lookup_count: Union[None, Unset, int] = UNSET
     head_type: Union[Unset, RACHeadType] = UNSET
     weigh_memories: Union[None, Unset, bool] = UNSET
@@ -56,6 +60,18 @@ class CreateClassificationModelRequest:
         else:
             description = self.description
 
+        notes: Union[None, Unset, str]
+        if isinstance(self.notes, Unset):
+            notes = UNSET
+        else:
+            notes = self.notes
+
+        memoryset_name: Union[None, Unset, str]
+        if isinstance(self.memoryset_name, Unset):
+            memoryset_name = UNSET
+        else:
+            memoryset_name = self.memoryset_name
+
         memory_lookup_count: Union[None, Unset, int]
         if isinstance(self.memory_lookup_count, Unset):
             memory_lookup_count = UNSET
@@ -94,6 +110,10 @@ class CreateClassificationModelRequest:
         )
         if description is not UNSET:
             field_dict["description"] = description
+        if notes is not UNSET:
+            field_dict["notes"] = notes
+        if memoryset_name is not UNSET:
+            field_dict["memoryset_name"] = memoryset_name
         if memory_lookup_count is not UNSET:
             field_dict["memory_lookup_count"] = memory_lookup_count
         if head_type is not UNSET:
@@ -123,6 +143,24 @@ class CreateClassificationModelRequest:
 
         description = _parse_description(d.pop("description", UNSET))
 
+        def _parse_notes(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        notes = _parse_notes(d.pop("notes", UNSET))
+
+        def _parse_memoryset_name(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        memoryset_name = _parse_memoryset_name(d.pop("memoryset_name", UNSET))
+
         def _parse_memory_lookup_count(data: object) -> Union[None, Unset, int]:
             if data is None:
                 return data
@@ -170,6 +208,8 @@ class CreateClassificationModelRequest:
             name=name,
             memoryset_id=memoryset_id,
             description=description,
+            notes=notes,
+            memoryset_name=memoryset_name,
             memory_lookup_count=memory_lookup_count,
             head_type=head_type,
             weigh_memories=weigh_memories,

orca_sdk/_generated_api_client/models/create_datasource_from_content_request.py

@@ -0,0 +1,101 @@
+"""
+This file is generated by the openapi-python-client tool via the generate_api_client.py script
+
+It is a customized template from the openapi-python-client tool's default template:
+https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
+
+The main change is:
+- Fix typing issues
+"""
+
+# flake8: noqa: C901
+
+from typing import Any, Type, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="CreateDatasourceFromContentRequest")
+
+
+@_attrs_define
+class CreateDatasourceFromContentRequest:
+    """Request model for creating a datasource from JSON content.
+
+    Attributes:
+        name (str):
+        content (Any):
+        description (Union[None, Unset, str]):
+    """
+
+    name: str
+    content: Any
+    description: Union[None, Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        name = self.name
+
+        content = self.content
+
+        description: Union[None, Unset, str]
+        if isinstance(self.description, Unset):
+            description = UNSET
+        else:
+            description = self.description
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "name": name,
+                "content": content,
+            }
+        )
+        if description is not UNSET:
+            field_dict["description"] = description
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        name = d.pop("name")
+
+        content = d.pop("content")
+
+        def _parse_description(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        description = _parse_description(d.pop("description", UNSET))
+
+        create_datasource_from_content_request = cls(
+            name=name,
+            content=content,
+            description=description,
+        )
+
+        create_datasource_from_content_request.additional_properties = d
+        return create_datasource_from_content_request
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

orca_sdk/_generated_api_client/models/create_memoryset_request.py

@@ -35,6 +35,7 @@ class CreateMemorysetRequest:
         datasource_id (str):
         datasource_value_column (str):
         description (Union[None, Unset, str]):
+        notes (Union[None, Unset, str]):
         datasource_label_column (Union[None, Unset, str]):
         datasource_score_column (Union[None, Unset, str]):
         datasource_source_id_column (Union[None, Unset, str]):
@@ -45,12 +46,14 @@ class CreateMemorysetRequest:
         label_names (Union[List[str], None, Unset]):
         index_type (Union[Unset, CreateMemorysetRequestIndexType]):  Default: CreateMemorysetRequestIndexType.FLAT.
         index_params (Union[Unset, CreateMemorysetRequestIndexParams]):
+        prompt (Union[None, Unset, str]):
     """
 
     name: str
     datasource_id: str
     datasource_value_column: str
     description: Union[None, Unset, str] = UNSET
+    notes: Union[None, Unset, str] = UNSET
     datasource_label_column: Union[None, Unset, str] = UNSET
     datasource_score_column: Union[None, Unset, str] = UNSET
     datasource_source_id_column: Union[None, Unset, str] = UNSET
@@ -61,6 +64,7 @@ class CreateMemorysetRequest:
     label_names: Union[List[str], None, Unset] = UNSET
     index_type: Union[Unset, CreateMemorysetRequestIndexType] = CreateMemorysetRequestIndexType.FLAT
     index_params: Union[Unset, "CreateMemorysetRequestIndexParams"] = UNSET
+    prompt: Union[None, Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -76,6 +80,12 @@ class CreateMemorysetRequest:
         else:
             description = self.description
 
+        notes: Union[None, Unset, str]
+        if isinstance(self.notes, Unset):
+            notes = UNSET
+        else:
+            notes = self.notes
+
         datasource_label_column: Union[None, Unset, str]
         if isinstance(self.datasource_label_column, Unset):
             datasource_label_column = UNSET
@@ -137,6 +147,12 @@ class CreateMemorysetRequest:
         if not isinstance(self.index_params, Unset):
             index_params = self.index_params.to_dict()
 
+        prompt: Union[None, Unset, str]
+        if isinstance(self.prompt, Unset):
+            prompt = UNSET
+        else:
+            prompt = self.prompt
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update(
@@ -148,6 +164,8 @@ class CreateMemorysetRequest:
         )
         if description is not UNSET:
             field_dict["description"] = description
+        if notes is not UNSET:
+            field_dict["notes"] = notes
         if datasource_label_column is not UNSET:
             field_dict["datasource_label_column"] = datasource_label_column
         if datasource_score_column is not UNSET:
@@ -168,6 +186,8 @@ class CreateMemorysetRequest:
             field_dict["index_type"] = index_type
         if index_params is not UNSET:
             field_dict["index_params"] = index_params
+        if prompt is not UNSET:
+            field_dict["prompt"] = prompt
 
         return field_dict
 
@@ -191,6 +211,15 @@ class CreateMemorysetRequest:
 
         description = _parse_description(d.pop("description", UNSET))
 
+        def _parse_notes(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        notes = _parse_notes(d.pop("notes", UNSET))
+
         def _parse_datasource_label_column(data: object) -> Union[None, Unset, str]:
             if data is None:
                 return data
@@ -288,11 +317,21 @@ class CreateMemorysetRequest:
         else:
             index_params = CreateMemorysetRequestIndexParams.from_dict(_index_params)
 
+        def _parse_prompt(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        prompt = _parse_prompt(d.pop("prompt", UNSET))
+
         create_memoryset_request = cls(
             name=name,
             datasource_id=datasource_id,
             datasource_value_column=datasource_value_column,
             description=description,
+            notes=notes,
             datasource_label_column=datasource_label_column,
             datasource_score_column=datasource_score_column,
             datasource_source_id_column=datasource_source_id_column,
@@ -303,6 +342,7 @@ class CreateMemorysetRequest:
             label_names=label_names,
             index_type=index_type,
             index_params=index_params,
+            prompt=prompt,
         )
 
         create_memoryset_request.additional_properties = d

orca_sdk/_generated_api_client/models/create_org_plan_request.py

@@ -0,0 +1,73 @@
+"""
+This file is generated by the openapi-python-client tool via the generate_api_client.py script
+
+It is a customized template from the openapi-python-client tool's default template:
+https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
+
+The main change is:
+- Fix typing issues
+"""
+
+# flake8: noqa: C901
+
+from enum import Enum
+from typing import Any, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.create_org_plan_request_tier import CreateOrgPlanRequestTier
+
+T = TypeVar("T", bound="CreateOrgPlanRequest")
+
+
+@_attrs_define
+class CreateOrgPlanRequest:
+    """
+    Attributes:
+        tier (CreateOrgPlanRequestTier):
+    """
+
+    tier: CreateOrgPlanRequestTier
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        tier = self.tier.value if isinstance(self.tier, Enum) else self.tier
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "tier": tier,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        tier = CreateOrgPlanRequestTier(d.pop("tier"))
+
+        create_org_plan_request = cls(
+            tier=tier,
+        )
+
+        create_org_plan_request.additional_properties = d
+        return create_org_plan_request
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

orca_sdk/_generated_api_client/models/create_org_plan_request_tier.py

@@ -0,0 +1,11 @@
+from enum import Enum
+
+
+class CreateOrgPlanRequestTier(str, Enum):
+    CANCELLED = "CANCELLED"
+    ENTERPRISE = "ENTERPRISE"
+    FREE = "FREE"
+    PRO = "PRO"
+
+    def __str__(self) -> str:
+        return str(self.value)

orca_sdk/_generated_api_client/models/create_regression_model_request.py

@@ -28,6 +28,7 @@ class CreateRegressionModelRequest:
         name (str):
         memoryset_id (str):
         description (Union[None, Unset, str]):
+        notes (Union[None, Unset, str]):
         memory_lookup_count (Union[None, Unset, int]):
         head_type (Union[Unset, RARHeadType]):
     """
@@ -35,6 +36,7 @@ class CreateRegressionModelRequest:
     name: str
     memoryset_id: str
     description: Union[None, Unset, str] = UNSET
+    notes: Union[None, Unset, str] = UNSET
     memory_lookup_count: Union[None, Unset, int] = UNSET
     head_type: Union[Unset, RARHeadType] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
@@ -50,6 +52,12 @@ class CreateRegressionModelRequest:
         else:
             description = self.description
 
+        notes: Union[None, Unset, str]
+        if isinstance(self.notes, Unset):
+            notes = UNSET
+        else:
+            notes = self.notes
+
         memory_lookup_count: Union[None, Unset, int]
         if isinstance(self.memory_lookup_count, Unset):
             memory_lookup_count = UNSET
@@ -70,6 +78,8 @@ class CreateRegressionModelRequest:
         )
         if description is not UNSET:
             field_dict["description"] = description
+        if notes is not UNSET:
+            field_dict["notes"] = notes
         if memory_lookup_count is not UNSET:
             field_dict["memory_lookup_count"] = memory_lookup_count
         if head_type is not UNSET:
@@ -93,6 +103,15 @@ class CreateRegressionModelRequest:
 
         description = _parse_description(d.pop("description", UNSET))
 
+        def _parse_notes(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        notes = _parse_notes(d.pop("notes", UNSET))
+
         def _parse_memory_lookup_count(data: object) -> Union[None, Unset, int]:
             if data is None:
                 return data
@@ -113,6 +132,7 @@ class CreateRegressionModelRequest:
             name=name,
             memoryset_id=memoryset_id,
             description=description,
+            notes=notes,
             memory_lookup_count=memory_lookup_count,
             head_type=head_type,
         )

orca_sdk/_generated_api_client/models/embed_request.py

@@ -26,10 +26,12 @@ class EmbedRequest:
     Attributes:
         values (List[str]):
         max_seq_length (Union[None, Unset, int]):
+        prompt (Union[None, Unset, str]):
     """
 
     values: List[str]
     max_seq_length: Union[None, Unset, int] = UNSET
+    prompt: Union[None, Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -43,6 +45,12 @@ class EmbedRequest:
         else:
             max_seq_length = self.max_seq_length
 
+        prompt: Union[None, Unset, str]
+        if isinstance(self.prompt, Unset):
+            prompt = UNSET
+        else:
+            prompt = self.prompt
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update(
@@ -52,6 +60,8 @@ class EmbedRequest:
         )
         if max_seq_length is not UNSET:
             field_dict["max_seq_length"] = max_seq_length
+        if prompt is not UNSET:
+            field_dict["prompt"] = prompt
 
         return field_dict
 
@@ -110,9 +120,19 @@ class EmbedRequest:
 
         max_seq_length = _parse_max_seq_length(d.pop("max_seq_length", UNSET))
 
+        def _parse_prompt(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        prompt = _parse_prompt(d.pop("prompt", UNSET))
+
         embed_request = cls(
             values=values,
             max_seq_length=max_seq_length,
+            prompt=prompt,
         )
 
         embed_request.additional_properties = d

orca_sdk/_generated_api_client/models/embedding_evaluation_payload.py

@@ -32,7 +32,7 @@ class EmbeddingEvaluationPayload:
         datasource_id (str):
         neighbor_count (Union[Unset, int]):  Default: 5.
         label_names (Union[List[str], None, Unset]):
-        embedding_models (Union[List[PretrainedEmbeddingModelName], None, Unset]):
+        embedding_models (Union[List[Union[PretrainedEmbeddingModelName, str]], None, Unset]):
     """
 
     value_column: str
@@ -41,7 +41,7 @@ class EmbeddingEvaluationPayload:
     datasource_id: str
     neighbor_count: Union[Unset, int] = 5
     label_names: Union[List[str], None, Unset] = UNSET
-    embedding_models: Union[List[PretrainedEmbeddingModelName], None, Unset] = UNSET
+    embedding_models: Union[List[Union[PretrainedEmbeddingModelName, str]], None, Unset] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -71,11 +71,15 @@ class EmbeddingEvaluationPayload:
         elif isinstance(self.embedding_models, list):
             embedding_models = []
             for embedding_models_type_0_item_data in self.embedding_models:
-                embedding_models_type_0_item = (
-                    embedding_models_type_0_item_data.value
-                    if isinstance(embedding_models_type_0_item_data, Enum)
-                    else embedding_models_type_0_item_data
-                )
+                embedding_models_type_0_item: str
+                if isinstance(embedding_models_type_0_item_data, PretrainedEmbeddingModelName):
+                    embedding_models_type_0_item = (
+                        embedding_models_type_0_item_data.value
+                        if isinstance(embedding_models_type_0_item_data, Enum)
+                        else embedding_models_type_0_item_data
+                    )
+                else:
+                    embedding_models_type_0_item = embedding_models_type_0_item_data
                 embedding_models.append(embedding_models_type_0_item)
 
         else:
@@ -135,7 +139,7 @@ class EmbeddingEvaluationPayload:
 
         label_names = _parse_label_names(d.pop("label_names", UNSET))
 
-        def _parse_embedding_models(data: object) -> Union[List[PretrainedEmbeddingModelName], None, Unset]:
+        def _parse_embedding_models(data: object) -> Union[List[Union[PretrainedEmbeddingModelName, str]], None, Unset]:
             if data is None:
                 return data
             if isinstance(data, Unset):
@@ -146,14 +150,28 @@ class EmbeddingEvaluationPayload:
                 embedding_models_type_0 = []
                 _embedding_models_type_0 = data
                 for embedding_models_type_0_item_data in _embedding_models_type_0:
-                    embedding_models_type_0_item = PretrainedEmbeddingModelName(embedding_models_type_0_item_data)
+
+                    def _parse_embedding_models_type_0_item(data: object) -> Union[PretrainedEmbeddingModelName, str]:
+                        try:
+                            if not isinstance(data, str):
+                                raise TypeError()
+                            embedding_models_type_0_item_type_0 = PretrainedEmbeddingModelName(data)
+
+                            return embedding_models_type_0_item_type_0
+                        except:  # noqa: E722
+                            pass
+                        return cast(Union[PretrainedEmbeddingModelName, str], data)
+
+                    embedding_models_type_0_item = _parse_embedding_models_type_0_item(
+                        embedding_models_type_0_item_data
+                    )
 
                     embedding_models_type_0.append(embedding_models_type_0_item)
 
                 return embedding_models_type_0
             except:  # noqa: E722
                 pass
-            return cast(Union[List[PretrainedEmbeddingModelName], None, Unset], data)
+            return cast(Union[List[Union[PretrainedEmbeddingModelName, str]], None, Unset], data)
 
         embedding_models = _parse_embedding_models(d.pop("embedding_models", UNSET))
 

orca_sdk/_generated_api_client/models/embedding_evaluation_request.py

@@ -31,7 +31,7 @@ class EmbeddingEvaluationRequest:
         source_id_column (Union[None, str]):
         neighbor_count (Union[Unset, int]):  Default: 5.
         label_names (Union[List[str], None, Unset]):
-        embedding_models (Union[List[PretrainedEmbeddingModelName], None, Unset]):
+        embedding_models (Union[List[Union[PretrainedEmbeddingModelName, str]], None, Unset]):
     """
 
     value_column: str
@@ -39,7 +39,7 @@ class EmbeddingEvaluationRequest:
     source_id_column: Union[None, str]
     neighbor_count: Union[Unset, int] = 5
     label_names: Union[List[str], None, Unset] = UNSET
-    embedding_models: Union[List[PretrainedEmbeddingModelName], None, Unset] = UNSET
+    embedding_models: Union[List[Union[PretrainedEmbeddingModelName, str]], None, Unset] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -67,11 +67,15 @@ class EmbeddingEvaluationRequest:
         elif isinstance(self.embedding_models, list):
             embedding_models = []
             for embedding_models_type_0_item_data in self.embedding_models:
-                embedding_models_type_0_item = (
-                    embedding_models_type_0_item_data.value
-                    if isinstance(embedding_models_type_0_item_data, Enum)
-                    else embedding_models_type_0_item_data
-                )
+                embedding_models_type_0_item: str
+                if isinstance(embedding_models_type_0_item_data, PretrainedEmbeddingModelName):
+                    embedding_models_type_0_item = (
+                        embedding_models_type_0_item_data.value
+                        if isinstance(embedding_models_type_0_item_data, Enum)
+                        else embedding_models_type_0_item_data
+                    )
+                else:
+                    embedding_models_type_0_item = embedding_models_type_0_item_data
                 embedding_models.append(embedding_models_type_0_item)
 
         else:
@@ -128,7 +132,7 @@ class EmbeddingEvaluationRequest:
 
         label_names = _parse_label_names(d.pop("label_names", UNSET))
 
-        def _parse_embedding_models(data: object) -> Union[List[PretrainedEmbeddingModelName], None, Unset]:
+        def _parse_embedding_models(data: object) -> Union[List[Union[PretrainedEmbeddingModelName, str]], None, Unset]:
             if data is None:
                 return data
             if isinstance(data, Unset):
@@ -139,14 +143,28 @@ class EmbeddingEvaluationRequest:
                 embedding_models_type_0 = []
                 _embedding_models_type_0 = data
                 for embedding_models_type_0_item_data in _embedding_models_type_0:
-                    embedding_models_type_0_item = PretrainedEmbeddingModelName(embedding_models_type_0_item_data)
+
+                    def _parse_embedding_models_type_0_item(data: object) -> Union[PretrainedEmbeddingModelName, str]:
+                        try:
+                            if not isinstance(data, str):
+                                raise TypeError()
+                            embedding_models_type_0_item_type_0 = PretrainedEmbeddingModelName(data)
+
+                            return embedding_models_type_0_item_type_0
+                        except:  # noqa: E722
+                            pass
+                        return cast(Union[PretrainedEmbeddingModelName, str], data)
+
+                    embedding_models_type_0_item = _parse_embedding_models_type_0_item(
+                        embedding_models_type_0_item_data
+                    )
 
                     embedding_models_type_0.append(embedding_models_type_0_item)
 
                 return embedding_models_type_0
             except:  # noqa: E722
                 pass
-            return cast(Union[List[PretrainedEmbeddingModelName], None, Unset], data)
+            return cast(Union[List[Union[PretrainedEmbeddingModelName, str]], None, Unset], data)
 
         embedding_models = _parse_embedding_models(d.pop("embedding_models", UNSET))