Config adjustments

[Dreamsorcerer]

Problem

Working towards #1082, this helps standardise setting in modules which can then be expanded upon to enable further things such as CLI arguments to be injected. They are also now type checked at runtime via Pydantic.

Solution

Primary change to developers is that the __init__() method now uses a fixed signature and all configurable options appear in the config. That config is now a pydantic model, so it gets verified when the module is instantiated.

A lot of typing fixes were also made by changing incorrect references to Module to ModuleBase and similar adjustments.

Breaking Changes

Subclasses of Module must now match the __init__() signature of the Module class.

[Dreamsorcerer]

As an alternative to asking subclasses (that need a custom __init__()) to include global_config here, I could move it into the ModuleConfig itself? Then subclasses will just pass kwargs alone.

[greptile-apps]

Greptile Summary

This PR standardizes module configuration by migrating from dataclasses to Pydantic models and enforcing a fixed __init__() signature across all Module subclasses.

Key Changes:

• All ModuleConfig classes now inherit from BaseConfig (Pydantic BaseModel) instead of using @dataclass
• Module __init__() signature standardized to __init__(self, global_config: GlobalConfig = global_config, **kwargs: Any)
• Configuration parameters moved from constructor arguments to config class fields
• Runtime type checking now provided via Pydantic validation
• Type annotations updated throughout (Module → ModuleBase, added generic TypeVars)
• Infrastructure updated (Worker, ModuleCoordinator, Blueprint) to pass configs correctly

Issues Found:

• TemporalMemory.__init__() doesn't follow the new pattern - still uses custom parameters instead of the standardized signature
• Detection2DModule.camera_info changed from optional to required (documented breaking change, tests updated)

Impact:
Most modules properly migrated to the new pattern. The standardization will enable CLI argument injection and better configuration management going forward.

Confidence Score: 3/5

• This PR is mostly safe but has one module that doesn't follow the new pattern
• The refactoring is well-executed across 59 of 60 files with consistent pattern application and proper test updates. However, TemporalMemory module wasn't fully migrated and still uses the old __init__ signature, which violates the documented breaking change requirement. This will cause runtime errors when the module is instantiated through the standard deployment pipeline.
• Pay close attention to dimos/perception/experimental/temporal_memory/temporal_memory.py - the __init__ method needs to be updated to match the new standardized signature before merging

Important Files Changed

Filename	Overview
dimos/core/module.py	Core module refactored to use fixed __init__ signature with Pydantic config, TypeVar updates for Python 3.13 compatibility
dimos/core/native_module.py	Migrated from dataclass to Pydantic BaseModel, updated compute_cli_args() to use model_fields
dimos/protocol/service/spec.py	Added BaseConfig as Pydantic BaseModel with arbitrary_types_allowed, updated Configurable class
dimos/core/worker.py	Updated Worker to pass global_config and kwargs instead of args and kwargs, type annotations updated
dimos/core/module_coordinator.py	Updated deploy methods to use new ModuleSpec tuple format, removed signature inspection logic
dimos/core/blueprints.py	Removed args from _BlueprintAtom, removed signature inspection for cfg parameter
dimos/perception/experimental/temporal_memory/temporal_memory.py	Type annotations updated but __init__ signature not updated to match new pattern - potential bug
dimos/perception/detection/module2D.py	Migrated to Pydantic, camera_info now required (was optional), uses Pydantic validation for filter field
dimos/robot/unitree/b1/connection.py	Properly migrated with new B1ConnectionConfig, __init__ updated correctly
dimos/models/base.py	Converted LocalModelConfig and HuggingFaceModelConfig from dataclass to Pydantic BaseConfig
dimos/protocol/service/lcmservice.py	Migrated LCMConfig to Pydantic, removed __post_init__, added TypeVar for generic config
dimos/core/worker_manager.py	Updated deploy methods to pass global_config and kwargs instead of args tuple

Class Diagram

%%{init: {'theme': 'neutral'}}%%
classDiagram
    class BaseConfig {
        <<Pydantic BaseModel>>
        +model_config: arbitrary_types_allowed
    }
    
    class ModuleConfig {
        +rpc_transport: RPCSpec
        +tf_transport: TFSpec
        +frame_id_prefix: str | None
        +frame_id: str | None
    }
    
    class NativeModuleConfig {
        +executable: str
        +build_command: str | None
        +extra_args: list
        +extra_env: dict
        +compute_cli_args() list
    }
    
    class Configurable~ConfigT~ {
        <<Generic>>
        +default_config: type[ConfigT]
        +config: ConfigT
        +__init__(**kwargs)
    }
    
    class ModuleBase~ModuleConfigT~ {
        +default_config: ModuleConfig
        -_rpc: RPCSpec | None
        -_tf: TFSpec | None
        +__init__(config_args: dict, global_config: GlobalConfig)
    }
    
    class Module~ModuleConfigT~ {
        +__init__(global_config: GlobalConfig, **kwargs)
        +set_ref(ref)
        +start()
        +stop()
    }
    
    class NativeModule~NativeConfig~ {
        +default_config: NativeModuleConfig
        -_process: Popen | None
        +__init__(global_config: GlobalConfig, **kwargs)
    }
    
    BaseConfig <|-- ModuleConfig : inherits
    ModuleConfig <|-- NativeModuleConfig : inherits
    Configurable <|-- ModuleBase : inherits
    ModuleBase <|-- Module : inherits
    Module <|-- NativeModule : inherits
    
    Configurable --> BaseConfig : uses ConfigT
    ModuleBase --> ModuleConfig : uses ModuleConfigT
    
    note for ModuleBase "Fixed __init__ signature:\nconfig_args + global_config"
    note for Module "Standardized signature:\nglobal_config + **kwargs"
    note for BaseConfig "Runtime validation\nvia Pydantic"

Loading

_{Last reviewed commit: 28a43f7}

[greptile-apps]

_{60 files reviewed, 1 comment}

_{Edit Code Review Agent Settings | Greptile}

[paul-nechifor]

The tests are still failing.

Note that you have to cherry-pick this to fix protobuf: https://github.com/dimensionalOS/dimos/pull/1372/changes (No longer needed, you just need to rebased on dev now.)