# 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.llm_generation_run_config import LlmGenerationRunConfig
from arize._generated.api_client.models.template_evaluation_run_config import TemplateEvaluationRunConfig
from pydantic import StrictStr, Field
from typing import Union, List, Set, Optional, Dict
from typing_extensions import Literal, Self
RUNCONFIGURATION_ONE_OF_SCHEMAS = ["LlmGenerationRunConfig", "TemplateEvaluationRunConfig"]
[docs]
class RunConfiguration(BaseModel):
"""
Experiment execution configuration for a `run_experiment` task. Exactly one variant must be supplied, identified by `experiment_type`. All fields sit at the top level alongside `experiment_type` (flat — no wrapper sub-object).
"""
# data type: LlmGenerationRunConfig
oneof_schema_1_validator: Optional[LlmGenerationRunConfig] = None
# data type: TemplateEvaluationRunConfig
oneof_schema_2_validator: Optional[TemplateEvaluationRunConfig] = None
actual_instance: Optional[Union[LlmGenerationRunConfig, TemplateEvaluationRunConfig]] = None
one_of_schemas: Set[str] = { "LlmGenerationRunConfig", "TemplateEvaluationRunConfig" }
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 = RunConfiguration.model_construct()
error_messages = []
match = 0
# validate data type: LlmGenerationRunConfig
if not isinstance(v, LlmGenerationRunConfig):
error_messages.append(f"Error! Input type `{type(v)}` is not `LlmGenerationRunConfig`")
else:
match += 1
# validate data type: TemplateEvaluationRunConfig
if not isinstance(v, TemplateEvaluationRunConfig):
error_messages.append(f"Error! Input type `{type(v)}` is not `TemplateEvaluationRunConfig`")
else:
match += 1
if match > 1:
# more than 1 match
raise ValueError("Multiple matches found when setting `actual_instance` in RunConfiguration with oneOf schemas: LlmGenerationRunConfig, TemplateEvaluationRunConfig. Details: " + ", ".join(error_messages))
elif match == 0:
# no match
raise ValueError("No match found when setting `actual_instance` in RunConfiguration with oneOf schemas: LlmGenerationRunConfig, TemplateEvaluationRunConfig. 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 LlmGenerationRunConfig
try:
instance.actual_instance = LlmGenerationRunConfig.from_json(json_str)
match += 1
except (ValidationError, ValueError) as e:
error_messages.append(str(e))
# deserialize data into TemplateEvaluationRunConfig
try:
instance.actual_instance = TemplateEvaluationRunConfig.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 RunConfiguration with oneOf schemas: LlmGenerationRunConfig, TemplateEvaluationRunConfig. Details: " + ", ".join(error_messages))
elif match == 0:
# no match
raise ValueError("No match found when deserializing the JSON string into RunConfiguration with oneOf schemas: LlmGenerationRunConfig, TemplateEvaluationRunConfig. 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], LlmGenerationRunConfig, TemplateEvaluationRunConfig]]:
"""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())
