add type to entity_service.entity_model

2026-06-21 13:47:35 +00:00 · 2024-12-27 19:58:30 -06:00
parent 99e7673fc8
commit c4d6cabf53
1 changed files with 1 additions and 1 deletions
@@ -11,7 +11,7 @@ from basic_memory.services.exceptions import EntityNotFoundError
 from .service import BaseService


-def entity_model(entity):
+def entity_model(entity: EntitySchema):
    model = EntityModel(
        name=entity.name,
        entity_type=entity.entity_type,