from basic_memory.schemas.response import RelationResponse
def test_relation_response_resolves_from_to_from_dict_fallbacks():
data = {
"permalink": "rel/1",
"relation_type": "relates_to",
"context": "ctx",
"to_name": None,
"from_entity": {"permalink": None, "file_path": "From.md"},
"to_entity": {"permalink": None, "file_path": "To.md", "title": "To Title"},
}
rel = RelationResponse.model_validate(data)
assert rel.from_id == "From.md"
assert rel.to_id == "To.md"
assert rel.to_name == "To Title"
def test_relation_response_resolves_from_to_from_orm_like_object_fallbacks():
class EntityLike:
def __init__(self, permalink, file_path, title=None):
self.permalink = permalink
self.file_path = file_path
self.title = title
class RelationLike:
def __init__(self):
self.permalink = "rel/2"
self.relation_type = "relates_to"
self.context = "ctx"
self.to_name = None
self.from_entity = EntityLike(permalink=None, file_path="From2.md")
self.to_entity = EntityLike(permalink=None, file_path="To2.md", title="To2 Title")
rel = RelationResponse.model_validate(RelationLike())
assert rel.from_id == "From2.md"
assert rel.to_id == "To2.md"
assert rel.to_name == "To2 Title"
