fleet-memory/memora-clients/python/models/update_personality_request.py
Nicolò Boschi c82fff949b add profiles
2025-11-13 13:22:27 +01:00

69 lines
2 KiB
Python

from __future__ import annotations
from collections.abc import Mapping
from typing import TYPE_CHECKING, Any, TypeVar
from attrs import define as _attrs_define
from attrs import field as _attrs_field
if TYPE_CHECKING:
from ..models.personality_traits import PersonalityTraits
T = TypeVar("T", bound="UpdatePersonalityRequest")
@_attrs_define
class UpdatePersonalityRequest:
"""Request model for updating personality traits.
Attributes:
personality (PersonalityTraits): Personality traits based on Big Five model. Example: {'agreeableness': 0.7,
'bias_strength': 0.7, 'conscientiousness': 0.6, 'extraversion': 0.5, 'neuroticism': 0.3, 'openness': 0.8}.
"""
personality: PersonalityTraits
additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
def to_dict(self) -> dict[str, Any]:
personality = self.personality.to_dict()
field_dict: dict[str, Any] = {}
field_dict.update(self.additional_properties)
field_dict.update(
{
"personality": personality,
}
)
return field_dict
@classmethod
def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
from ..models.personality_traits import PersonalityTraits
d = dict(src_dict)
personality = PersonalityTraits.from_dict(d.pop("personality"))
update_personality_request = cls(
personality=personality,
)
update_personality_request.additional_properties = d
return update_personality_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