mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-04-26 11:48:10 +00:00
edf345cd72
- Freeze all config models (AppConfig + 15 sub-configs) with frozen=True - Purify from_file() — remove 9 load_*_from_dict() side-effect calls - Replace mtime/reload/push/pop machinery with single ContextVar + init_app_config() - Delete 10 sub-module globals and their getters/setters/loaders - Migrate 50+ consumers from get_*_config() to get_app_config().xxx - Expand DeerFlowContext: app_config + thread_id + agent_name (frozen dataclass) - Wire into Gateway runtime (worker.py) and DeerFlowClient via context= parameter - Remove sandbox_id from runtime.context — flows through ThreadState.sandbox only - Middleware/tools access runtime.context directly via Runtime[DeerFlowContext] generic - resolve_context() retained at server entry points for LangGraph Server fallback
74 lines
2.6 KiB
Python
74 lines
2.6 KiB
Python
"""Verify that all sub-config Pydantic models are frozen (immutable).
|
|
|
|
Frozen models reject attribute assignment after construction, raising
|
|
pydantic.ValidationError. This test collects every BaseModel subclass
|
|
defined in the deerflow.config package and asserts that mutation is
|
|
blocked.
|
|
"""
|
|
|
|
import inspect
|
|
import pkgutil
|
|
|
|
import pytest
|
|
from pydantic import BaseModel, ValidationError
|
|
|
|
import deerflow.config as config_pkg
|
|
|
|
|
|
def _collect_config_models() -> list[type[BaseModel]]:
|
|
"""Walk deerflow.config.* and return all concrete BaseModel subclasses."""
|
|
import importlib
|
|
|
|
models: list[type[BaseModel]] = []
|
|
package_path = config_pkg.__path__
|
|
package_prefix = config_pkg.__name__ + "."
|
|
|
|
for _importer, modname, _ispkg in pkgutil.walk_packages(package_path, prefix=package_prefix):
|
|
try:
|
|
mod = importlib.import_module(modname)
|
|
except Exception:
|
|
continue
|
|
for _name, obj in inspect.getmembers(mod, inspect.isclass):
|
|
if (
|
|
issubclass(obj, BaseModel)
|
|
and obj is not BaseModel
|
|
and obj.__module__ == mod.__name__
|
|
):
|
|
models.append(obj)
|
|
|
|
return models
|
|
|
|
|
|
_EXCLUDED: set[str] = set()
|
|
|
|
_ALL_MODELS = [m for m in _collect_config_models() if m.__name__ not in _EXCLUDED]
|
|
|
|
# Sanity: make sure we actually collected a meaningful set.
|
|
assert len(_ALL_MODELS) >= 15, f"Expected at least 15 config models, found {len(_ALL_MODELS)}: {[m.__name__ for m in _ALL_MODELS]}"
|
|
|
|
|
|
@pytest.mark.parametrize("model_cls", _ALL_MODELS, ids=lambda cls: cls.__name__)
|
|
def test_config_model_is_frozen(model_cls: type[BaseModel]):
|
|
"""Every sub-config model must have frozen=True in its model_config."""
|
|
cfg = model_cls.model_config
|
|
assert cfg.get("frozen") is True, (
|
|
f"{model_cls.__name__} is not frozen. "
|
|
f"Add `model_config = ConfigDict(frozen=True)` or add `frozen=True` to the existing ConfigDict."
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize("model_cls", _ALL_MODELS, ids=lambda cls: cls.__name__)
|
|
def test_config_model_rejects_mutation(model_cls: type[BaseModel]):
|
|
"""Constructing then mutating any field must raise ValidationError."""
|
|
# Build a minimal instance -- use model_construct to skip validation for
|
|
# required fields, then pick the first field to try mutating.
|
|
fields = list(model_cls.model_fields.keys())
|
|
if not fields:
|
|
pytest.skip(f"{model_cls.__name__} has no fields")
|
|
|
|
instance = model_cls.model_construct()
|
|
first_field = fields[0]
|
|
|
|
with pytest.raises(ValidationError):
|
|
setattr(instance, first_field, "MUTATED")
|