From c4d6cabf53271b980c27053c064937ca46cc62ef Mon Sep 17 00:00:00 2001 From: phernandez Date: Fri, 27 Dec 2024 19:58:30 -0600 Subject: [PATCH] add type to entity_service.entity_model --- src/basic_memory/services/entity_service.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/basic_memory/services/entity_service.py b/src/basic_memory/services/entity_service.py index 1ab10528..39421474 100644 --- a/src/basic_memory/services/entity_service.py +++ b/src/basic_memory/services/entity_service.py @@ -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,