Anatomy of a module | Viam Documentation

Anatomy of a module

When you run viam module generate, the CLI creates a complete project with everything you need to build, test, and deploy a module. This page explains the purpose of each file and when you need to edit it.

The examples on this page use a logic module called temp-monitor that monitors a temperature sensor and logs a warning when readings exceed a threshold. It depends on one sensor and uses DoCommand to report its current state.

Directory structure

temp-monitor/
├── .github/
│   └── workflows/
│       └── deploy.yml            # CI workflow for cloud builds
├── src/
│   ├── main.py                   # Entry point
│   └── models/
│       └── temp_monitor.py       # Resource implementation
├── build.sh                      # Packages the module for upload
├── meta.json                     # Module metadata for the registry
├── requirements.txt              # Python dependencies
├── run.sh                        # Entrypoint script for viam-server
└── setup.sh                      # Installs dependencies into a virtualenv
temp-monitor/
├── .github/
│   └── workflows/
│       └── deploy.yml            # CI workflow for cloud builds
├── cmd/
│   └── module/
│       └── main.go               # Entry point
├── temp_monitor.go               # Resource implementation
├── go.mod                        # Go module definition
├── go.sum                        # Dependency checksums
├── Makefile                      # Build targets
├── meta.json                     # Module metadata for the registry
├── build.sh                      # Packages the module for upload
└── setup.sh                      # Installs build dependencies

Resource implementation

This is the main file you work in. The generator names it after the model you are implementing. A model implements a Viam API. In this example, that is src/models/temp_monitor.py in Python and temp_monitor.go in Go. The sections below walk through each section of this file.

Model definition

The model definition identifies your resource in the registry as a triplet of namespace, module name, and model name. Some examples from built-in Viam modules: viam:camera:webcam, viam:motor:gpio, viam:sensor:ultrasonic. In our example, the triplet is my-org:temp-monitor:temp-monitor.

class TempMonitor(Generic, EasyResource):
    MODEL: ClassVar[Model] = Model(
        ModelFamily("my-org", "temp-monitor"), "temp-monitor"
    )
var Model = resource.NewModel("my-org", "temp-monitor", "temp-monitor")

func init() {
    resource.RegisterService(generic.API, Model, resource.Registration[
        resource.Resource, *Config,
    ]{
        Constructor: newTempMonitor,
    })
}

In Go, the init() function registers the model with viam-server when the package is imported. The registration binds the model to the generic service API and points to the constructor function.

Config and attributes

Config attributes are the fields a user sets when they add the service this model implements to a machine. Each field maps to a key in the attributes block of the JSON config:

"attributes": {
  "sensor_name": "temp-1",
  "threshold": 40.0
}
    sensor_name: str
    threshold: float
    sensor: Sensor
    exceeded: bool

In Python, declare config attributes, resolved dependencies, and runtime state as instance variables on the class. This pattern is the same for any module you write. Only the specific variables change.

type Config struct {
    SensorName string  `json:"sensor_name"`
    Threshold  float64 `json:"threshold"`
}

type TempMonitor struct {
    resource.Named
    logger   logging.Logger
    cfg      *Config
    sensor   sensor.Sensor
    mu       sync.Mutex
    exceeded bool
    cancelFn func()
}

In Go, the Config struct defines the attributes. The json tags map each field to its key in the JSON config. The resource struct holds the parsed config, resolved dependencies, and runtime state. This pattern is the same for any module you write. Only the specific fields change.

Validation

The validation method checks that attributes are valid and declares dependencies on other resources. viam-server calls this before creating or reconfiguring the resource. It returns two lists: required dependencies and optional dependencies.

    @classmethod
    def validate_config(
        cls, config: ComponentConfig
    ) -> Tuple[Sequence[str], Sequence[str]]:
        fields = config.attributes.fields
        if "sensor_name" not in fields:
            raise Exception("sensor_name is required")
        return [fields["sensor_name"].string_value], []
func (cfg *Config) Validate(path string) ([]string, []string, error) {
    if cfg.SensorName == "" {
        return nil, nil, fmt.Errorf("sensor_name is required")
    }
    return []string{cfg.SensorName}, nil, nil
}

Constructor

viam-server calls the constructor when it creates your resource. The constructor receives the config (containing the attributes the user set), a dependencies map (containing running instances of the resources you declared in your validation method), and in Go, a context and a logger.

viam-server destroys and re-creates the resource on every config change, calling the constructor again with the new config. The constructor runs on both initial creation and every subsequent config change.

The constructor’s job is to:

  1. Parse the config into typed fields you can use (for example, extract sensor_name as a string and threshold as a float).
  2. Resolve dependencies by looking up each one by name from the dependencies map. Each entry is a ready-to-use resource instance that viam-server has already started.
  3. Store everything on the struct or instance so your API methods and background tasks can use them.
  4. Start background work if your module runs continuously (for example, a goroutine or async task that polls a sensor on an interval).

To resolve a dependency, you look it up by name from the dependencies map. In Go, every resource type in the SDK provides a FromDependencies helper that does this and returns a typed interface (for example, sensor.FromDependencies returns a sensor.Sensor). In Python, you build the key with Sensor.get_resource_name(name) and index into the dependencies map directly.

    @classmethod
    def new(cls, config, dependencies) -> Self:
        monitor = cls(config.name)
        monitor.exceeded = False

fields = config.attributes.fields
        monitor.sensor_name = fields["sensor_name"].string_value
        monitor.threshold = (
            fields["threshold"].number_value
            if "threshold" in fields
            else 100.0
        )

monitor.sensor = dependencies[
            Sensor.get_resource_name(monitor.sensor_name)
        ]

...
        return monitor
func newTempMonitor(
    ctx context.Context,
    deps resource.Dependencies,
    conf resource.Config,
    logger logging.Logger,
) (resource.Resource, error) {
    cfg, err := resource.NativeConfig[*Config](conf)
    if err != nil {
        return nil, err
    }
    if cfg.Threshold == 0 {
        cfg.Threshold = 100.0
    }

s, err := sensor.FromDependencies(deps, cfg.SensorName)
    if err != nil {
        return nil, err
    }

monitorCtx, cancelFn := context.WithCancel(context.Background())
    tm := &TempMonitor{
        Named:    conf.ResourceName().AsNamed(),
        logger:   logger,
        cfg:      cfg,
        sensor:   s,
        cancelFn: cancelFn,
    }
    go tm.monitor(monitorCtx)
    return tm, nil
}

API methods

API methods are how external code interacts with your resource. For a logic module using the generic service API, the method is DoCommand. It accepts and returns arbitrary key-value maps, so you define your own command vocabulary.

    async def do_command(self, command, **kwargs):
        if command.get("command") == "status":
            return {
                "exceeded": self.exceeded,
                "threshold": self.threshold,
            }
        return {"error": f"unknown command: {command.get('command')}"}
func (tm *TempMonitor) DoCommand(
    ctx context.Context, cmd map[string]interface{},
) (map[string]interface{}, error) {
    if cmd["command"] == "status" {
        tm.mu.Lock()
        defer tm.mu.Unlock()
        return map[string]interface{}{
            "exceeded":  tm.exceeded,
            "threshold": tm.cfg.Threshold,
        }, nil
    }
    return nil, fmt.Errorf("unknown command: %v", cmd["command"])
}

Close

viam-server calls Close when it shuts down or removes the resource. Stop background tasks and release any resources here.

    async def close(self):
        self._stop_event.set()
        if self._monitor_task is not None:
            await self._monitor_task
func (tm *TempMonitor) Close(ctx context.Context) error {
    tm.cancelFn()
    return nil
}

meta.json

Module metadata used by the Viam registry. The generator creates this file and populates it from your answers to the generator prompts.

{
  "module_id": "my-org:temp-monitor",
  "visibility": "private",
  "url": "https://github.com/my-org/temp-monitor",
  "description": "Logs a warning when a temperature sensor exceeds a threshold.",
  "models": [
    {
      "api": "rdk:service:generic",
      "model": "my-org:temp-monitor:temp-monitor"
    }
  ],
  "entrypoint": "run.sh",
  "build": {
    "setup": "./setup.sh",
    "build": "./build.sh",
    "path": "dist/archive.tar.gz",
    "arch": ["linux/amd64", "linux/arm64"]
  }
}
Field Purpose
module_id Unique ID in the registry. Format: namespace:name.
visibility Who can install the module: private, public, public_unlisted.
url Link to the source repository. Required for cloud builds.
description Shown in registry search results.
models Resource models the module provides, each with api and model.
entrypoint Command that starts the module inside the archive.
build.setup Script that installs build dependencies (runs once).
build.build Script that compiles and packages the module.
build.path Path to the packaged output archive.
build.arch Target platforms to build for.

Files you rarely edit

Entry point

The entry point starts the module server and registers your models with viam-server. You only edit this file when you add a second model to the module.

src/main.py

import asyncio
from viam.module.module import Module
from models.temp_monitor import TempMonitor  # noqa: F401

if __name__ == "__main__":
    asyncio.run(Module.run_from_registry())

cmd/module/main.go

package main

import (
    tempmonitor "my-org/temp-monitor"
    "go.viam.com/rdk/module"
    "go.viam.com/rdk/resource"
    "go.viam.com/rdk/services/generic"
)

func main() {
    module.ModularMain(
        resource.APIModel{generic.API, tempmonitor.Model},
    )
}

Build and deploy scripts

These scripts handle packaging and deployment. The generator creates working defaults. You only need to edit them if your module has unusual build requirements.

File Purpose
build.sh Compiles (Go) or packages (Python) the module into a .tar.gz archive.
setup.sh Installs build dependencies. For Python, creates a virtualenv and installs requirements.txt.
run.sh (Python only) Entrypoint script that activates the virtualenv and runs main.py.
deploy.yml GitHub Actions workflow that triggers cloud builds on tagged releases.