70 lines
1.9 KiB
Python
70 lines
1.9 KiB
Python
"""Helpers for validating LLM-generated ontology structures."""
|
|
|
|
from typing import Any, Dict, List, Optional
|
|
|
|
|
|
MAX_ONTOLOGY_TYPES = 10
|
|
MAX_ONTOLOGY_ATTRIBUTES = 10
|
|
RESERVED_ONTOLOGY_ATTRIBUTE_NAMES = frozenset({
|
|
"uuid",
|
|
"name",
|
|
"group_id",
|
|
"graph_id",
|
|
"name_embedding",
|
|
"summary",
|
|
"created_at",
|
|
})
|
|
|
|
_FALLBACK_ATTRIBUTE = {
|
|
"name": "details",
|
|
"type": "text",
|
|
"description": "Additional details about this ontology type.",
|
|
}
|
|
|
|
|
|
def normalize_ontology_attribute(attribute: Any) -> Optional[Dict[str, Any]]:
|
|
"""Return a safe attribute definition, or ``None`` for unusable values."""
|
|
|
|
if isinstance(attribute, str):
|
|
if not attribute.strip():
|
|
return None
|
|
return {
|
|
"name": attribute,
|
|
"type": "text",
|
|
"description": attribute,
|
|
}
|
|
|
|
if not isinstance(attribute, dict):
|
|
return None
|
|
|
|
name = attribute.get("name")
|
|
if not isinstance(name, str) or not name.strip():
|
|
return None
|
|
|
|
normalized = dict(attribute)
|
|
description = normalized.get("description")
|
|
if not isinstance(description, str) or not description:
|
|
normalized["description"] = name
|
|
return normalized
|
|
|
|
|
|
def normalize_ontology_attributes(attributes: Any) -> List[Dict[str, Any]]:
|
|
"""Return a non-empty Zep-compatible attribute list within service limits."""
|
|
|
|
if not isinstance(attributes, list):
|
|
attributes = []
|
|
|
|
normalized_attributes: List[Dict[str, Any]] = []
|
|
for attribute in attributes:
|
|
normalized = normalize_ontology_attribute(attribute)
|
|
if normalized is None:
|
|
continue
|
|
normalized_attributes.append(normalized)
|
|
if len(normalized_attributes) == MAX_ONTOLOGY_ATTRIBUTES:
|
|
break
|
|
|
|
if not normalized_attributes:
|
|
normalized_attributes.append(dict(_FALLBACK_ATTRIBUTE))
|
|
|
|
return normalized_attributes
|