# coding: utf-8
"""
Arize REST API
API specification for the backend data server. The API is hosted globally at https://api.arize.com/v2 or in your own environment.
The version of the OpenAPI document: 2.0.0
Generated by OpenAPI Generator (https://openapi-generator.tech)
Do not edit the class manually.
""" # noqa: E501
from __future__ import annotations
import json
import pprint
from pydantic import BaseModel, ConfigDict, Field, StrictStr, ValidationError, field_validator
from typing import Any, List, Optional
from arize._generated.api_client.models.custom_code_config import CustomCodeConfig
from arize._generated.api_client.models.managed_code_config import ManagedCodeConfig
from pydantic import StrictStr, Field
from typing import Union, List, Set, Optional, Dict
from typing_extensions import Literal, Self
CODECONFIG_ONE_OF_SCHEMAS = ["CustomCodeConfig", "ManagedCodeConfig"]
[docs]
class CodeConfig(BaseModel):
"""
Discriminated union representing either a managed (built-in) or custom (user-supplied Python) code evaluator configuration, resolved by the nested `type` field (`managed` -> `ManagedCodeConfig`, `custom` -> `CustomCodeConfig`). This inner `type` is independent of the parent evaluator version's `type` (which is always `code` here).
"""
# data type: ManagedCodeConfig
oneof_schema_1_validator: Optional[ManagedCodeConfig] = None
# data type: CustomCodeConfig
oneof_schema_2_validator: Optional[CustomCodeConfig] = None
actual_instance: Optional[Union[CustomCodeConfig, ManagedCodeConfig]] = None
one_of_schemas: Set[str] = { "CustomCodeConfig", "ManagedCodeConfig" }
model_config = ConfigDict(
validate_assignment=True,
protected_namespaces=(),
)
discriminator_value_class_map: Dict[str, str] = {
}
def __init__(self, *args, **kwargs) -> None:
if args:
if len(args) > 1:
raise ValueError("If a position argument is used, only 1 is allowed to set `actual_instance`")
if kwargs:
raise ValueError("If a position argument is used, keyword arguments cannot be used.")
super().__init__(actual_instance=args[0])
else:
super().__init__(**kwargs)
[docs]
@field_validator('actual_instance')
def actual_instance_must_validate_oneof(cls, v):
instance = CodeConfig.model_construct()
error_messages = []
match = 0
# validate data type: ManagedCodeConfig
if not isinstance(v, ManagedCodeConfig):
error_messages.append(f"Error! Input type `{type(v)}` is not `ManagedCodeConfig`")
else:
match += 1
# validate data type: CustomCodeConfig
if not isinstance(v, CustomCodeConfig):
error_messages.append(f"Error! Input type `{type(v)}` is not `CustomCodeConfig`")
else:
match += 1
if match > 1:
# more than 1 match
raise ValueError("Multiple matches found when setting `actual_instance` in CodeConfig with oneOf schemas: CustomCodeConfig, ManagedCodeConfig. Details: " + ", ".join(error_messages))
elif match == 0:
# no match
raise ValueError("No match found when setting `actual_instance` in CodeConfig with oneOf schemas: CustomCodeConfig, ManagedCodeConfig. Details: " + ", ".join(error_messages))
else:
return v
[docs]
@classmethod
def from_dict(cls, obj: Union[str, Dict[str, Any]]) -> Self:
return cls.from_json(json.dumps(obj))
[docs]
@classmethod
def from_json(cls, json_str: str) -> Self:
"""Returns the object represented by the json string"""
instance = cls.model_construct()
error_messages = []
match = 0
# deserialize data into ManagedCodeConfig
try:
instance.actual_instance = ManagedCodeConfig.from_json(json_str)
match += 1
except (ValidationError, ValueError) as e:
error_messages.append(str(e))
# deserialize data into CustomCodeConfig
try:
instance.actual_instance = CustomCodeConfig.from_json(json_str)
match += 1
except (ValidationError, ValueError) as e:
error_messages.append(str(e))
if match > 1:
# more than 1 match
raise ValueError("Multiple matches found when deserializing the JSON string into CodeConfig with oneOf schemas: CustomCodeConfig, ManagedCodeConfig. Details: " + ", ".join(error_messages))
elif match == 0:
# no match
raise ValueError("No match found when deserializing the JSON string into CodeConfig with oneOf schemas: CustomCodeConfig, ManagedCodeConfig. Details: " + ", ".join(error_messages))
else:
return instance
[docs]
def to_json(self) -> str:
"""Returns the JSON representation of the actual instance"""
if self.actual_instance is None:
return "null"
if hasattr(self.actual_instance, "to_json") and callable(self.actual_instance.to_json):
return self.actual_instance.to_json()
else:
return json.dumps(self.actual_instance)
[docs]
def to_dict(self) -> Optional[Union[Dict[str, Any], CustomCodeConfig, ManagedCodeConfig]]:
"""Returns the dict representation of the actual instance"""
if self.actual_instance is None:
return None
if hasattr(self.actual_instance, "to_dict") and callable(self.actual_instance.to_dict):
return self.actual_instance.to_dict()
else:
# primitive type
return self.actual_instance
[docs]
def to_str(self) -> str:
"""Returns the string representation of the actual instance"""
return pprint.pformat(self.model_dump())
