From 408cd045e4d2606f5e05229ffc633c445d6d3ce0 Mon Sep 17 00:00:00 2001 From: Greyson LaLonde Date: Mon, 6 Apr 2026 20:47:41 +0800 Subject: [PATCH] cleanup: remove dead _entity_discriminator and CheckpointPayload --- lib/crewai/src/crewai/state/runtime.py | 15 +-------------- 1 file changed, 1 insertion(+), 14 deletions(-) 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.