diff --git a/src/basic_memory/service.py b/src/basic_memory/service.py index 76138eb6..9fca90f7 100644 --- a/src/basic_memory/service.py +++ b/src/basic_memory/service.py @@ -121,14 +121,14 @@ class MemoryService: except Exception as e: raise DatabaseSyncError(f"Failed to update database index: {str(e)}") from e - async def create_entity(self, name: str, type: str, context: Optional[str] = None) -> Entity: + async def create_entity(self, name: str, type: str, description: Optional[str] = None) -> Entity: """Create a new entity.""" entity_id = self._generate_id(name) entity_data = { "id": entity_id, "name": name, "entity_type": type, - "context": context, + "description": description, "created_at": datetime.now(UTC) } diff --git a/tests/test_service_create.py b/tests/test_service_create.py index d5ff11ba..f69967c4 100644 --- a/tests/test_service_create.py +++ b/tests/test_service_create.py @@ -16,14 +16,14 @@ async def test_create_entity_success(memory_service): entity = await memory_service.create_entity( name="Test Entity", type="test", - context="test context" + description="test description" ) # Assert assert isinstance(entity, Entity) assert entity.name == "Test Entity" assert entity.entity_type == "test" - assert entity.context == "test context" + assert entity.description == "test description" # Verify file was created entity_file = memory_service.project_path / "entities" / f"{entity.id}.md"