def validate_config(
config: str,
) -> Tuple[Dict[str, EntitySchema], List[RelationSchema], str, bool]:
user_config = get_config_dict_from_module(config)
# validate entites and relations config
entities_config = user_config.get("entities")
relations_config = user_config.get("relations")
entity_path = user_config.get("entity_path")
dynamic_relations = user_config.get("dynamic_relations", False)
if not isinstance(entities_config, dict):
raise TypeError("Config entities is not of type dict")
if not isinstance(relations_config, list):
raise TypeError("Config relations is not of type list")
if not isinstance(entity_path, str):
raise TypeError("Config entity_path is not of type str")
if not isinstance(dynamic_relations, bool):
raise TypeError("Config dynamic_relations is not of type bool")
entities = {}
relations = []
for entity, entity_config in entities_config.items():
entities[entity] = EntitySchema.from_dict(entity_config)
for relation in relations_config:
relations.append(RelationSchema.from_dict(relation))
return entities, relations, entity_path, dynamic_relations
def parse_config_partial(
config_dict: Any,
) -> Tuple[Dict[str, EntitySchema], List[RelationSchema], str, List[str], bool]:
entities_config = config_dict.get("entities")
relations_config = config_dict.get("relations")
entity_path = config_dict.get("entity_path")
edge_paths = config_dict.get("edge_paths")
dynamic_relations = config_dict.get("dynamic_relations", False)
if not isinstance(entities_config, dict):
raise TypeError("Config entities is not of type dict")
if any(not isinstance(k, str) for k in entities_config.keys()):
raise TypeError("Config entities has some keys that are not of type str")
if not isinstance(relations_config, list):
raise TypeError("Config relations is not of type list")
if not isinstance(entity_path, str):
raise TypeError("Config entity_path is not of type str")
if not isinstance(edge_paths, list):
raise TypeError("Config edge_paths is not of type list")
if any(not isinstance(p, str) for p in edge_paths):
raise TypeError("Config edge_paths has some items that are not of type str")
if not isinstance(dynamic_relations, bool):
raise TypeError("Config dynamic_relations is not of type bool")
entities: Dict[str, EntitySchema] = {}
relations: List[RelationSchema] = []
for entity, entity_config in entities_config.items():
entities[entity] = EntitySchema.from_dict(entity_config)
for relation in relations_config:
relations.append(RelationSchema.from_dict(relation))
return entities, relations, entity_path, edge_paths, dynamic_relations