Function _validate_state_entry

src/google/adk/sessions/state.py:28–58

Validates a single state key-value pair against a Pydantic schema. Raises StateSchemaError if the key is not in the schema or the value does not match the field's type annotation. Prefixed keys (any key containing ``:``) bypass validation.

(
    schema: type[BaseModel],
    key: str,
    value: Any,
)

def _validate_state_entry(
    schema: type[BaseModel],
    key: str,
    value: Any,
) -> None:
  """Validates a single state key-value pair against a Pydantic schema.

  Raises StateSchemaError if the key is not in the schema or the value
  does not match the field's type annotation.  Prefixed keys (any key
  containing ``:``) bypass validation.
  """
  if ":" in key:
    return

  fields = schema.model_fields
  if key not in fields:
    raise StateSchemaError(
        f"Key '{key}' is not declared in state schema "
        f"'{schema.__name__}'. Declared fields: {sorted(fields.keys())}"
    )

  from pydantic import TypeAdapter
  from pydantic import ValidationError as PydanticValidationError

  try:
    TypeAdapter(fields[key].annotation).validate_python(value)
  except PydanticValidationError as e:
    raise StateSchemaError(
        f"Value for '{key}' does not match type "
        f"'{fields[key].annotation}' in '{schema.__name__}': {e}"
    ) from e


class State: