Bases: BaseModel
Provide YAML-based validation and serialization helpers.
classmethod
model_validate_yaml(yaml_data)
Validate a model from a YAML string.
Source code in swarmauri_base/YamlMixin.py
11
12
13
14
15
16
17
18
19
20
21
22
23 | @classmethod
def model_validate_yaml(cls, yaml_data: str):
"""Validate a model from a YAML string."""
try:
# Parse YAML into a Python dictionary
yaml_content = yaml.safe_load(yaml_data)
# Convert the dictionary to JSON and validate using Pydantic
return cls.model_validate_json(json.dumps(yaml_content))
except yaml.YAMLError as e:
raise ValueError(f"Invalid YAML data: {e}")
except ValidationError as e:
raise ValueError(f"Validation failed: {e}")
|
model_dump_yaml(
fields_to_exclude=None, api_key_placeholder=None
)
Return a YAML representation of the model.
Source code in swarmauri_base/YamlMixin.py
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55 | def model_dump_yaml(self, fields_to_exclude=None, api_key_placeholder=None):
"""Return a YAML representation of the model."""
if fields_to_exclude is None:
fields_to_exclude = []
# Load the JSON string into a Python dictionary
json_data = json.loads(self.model_dump_json())
# Function to recursively remove specific keys and handle api_key placeholders
def process_fields(data, fields_to_exclude):
"""Recursively filter fields and apply placeholders."""
if isinstance(data, dict):
return {
key: (
api_key_placeholder
if key == "api_key" and api_key_placeholder is not None
else process_fields(value, fields_to_exclude)
)
for key, value in data.items()
if key not in fields_to_exclude
}
elif isinstance(data, list):
return [process_fields(item, fields_to_exclude) for item in data]
else:
return data
# Filter the JSON data
filtered_data = process_fields(json_data, fields_to_exclude)
# Convert the filtered data into YAML using safe mode
return yaml.safe_dump(filtered_data, default_flow_style=False)
|