diff --git a/lib/crewai/src/crewai/state/runtime.py b/lib/crewai/src/crewai/state/runtime.py
index c5655398e..29fc5d57f 100644
--- a/lib/crewai/src/crewai/state/runtime.py
+++ b/lib/crewai/src/crewai/state/runtime.py
@@ -9,7 +9,7 @@ via ``RuntimeState.model_rebuild()``.
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, TypedDict
+from typing import TYPE_CHECKING, Any
 
 from pydantic import (
     ModelWrapValidatorHandler,
@@ -29,19 +29,6 @@ if TYPE_CHECKING:
     from crewai import Entity
 
 
-class CheckpointPayload(TypedDict):
-    entities: list[Entity]
-    event_record: dict[str, Any]
-
-
-def _entity_discriminator(v: dict[str, Any] | object) -> str:
-    if isinstance(v, dict):
-        raw = v.get("entity_type", "agent")
-    else:
-        raw = getattr(v, "entity_type", "agent")
-    return str(raw)
-
-
 def _sync_checkpoint_fields(entity: object) -> None:
     """Copy private runtime attrs into checkpoint fields before serializing.