viam-sdk 0.41.1__py3-none-linux_armv6l.whl → 0.66.0__py3-none-linux_armv6l.whl

viam/media/utils/pil/__init__.py

@@ -39,6 +39,10 @@ def pil_to_viam_image(image: Image.Image, mime_type: CameraMimeType) -> ViamImag
     Returns:
         ViamImage: The resulting ViamImage
     """
+    # Make sure at runtime the mime_type string is actually a CameraMimeType
+    if not isinstance(mime_type, CameraMimeType):
+        raise ValueError(f"Cannot encode to unsupported mimetype: {mime_type}")
+
     if mime_type.name in LIBRARY_SUPPORTED_FORMATS:
         buf = BytesIO()
         if image.mode == "RGBA" and mime_type == CameraMimeType.JPEG:
@@ -46,6 +50,6 @@ def pil_to_viam_image(image: Image.Image, mime_type: CameraMimeType) -> ViamImag
         image.save(buf, format=mime_type.name)
         data = buf.getvalue()
     else:
-        raise ValueError(f"Cannot encode image to {mime_type}")
+        raise ValueError(f"Cannot encode to unsupported mimetype: {mime_type}")
 
     return ViamImage(data, mime_type)

viam/media/video.py

@@ -1,21 +1,60 @@
 from array import array
-from enum import Enum
-from typing import List, Optional, Tuple
+from typing import Any, List, Optional, Tuple
 
-from typing_extensions import Self
+from typing_extensions import ClassVar, Self
 
 from viam.errors import NotSupportedError
-from viam.proto.component.camera import Format
 
 from .viam_rgba import RGBA_HEADER_LENGTH, RGBA_MAGIC_NUMBER
 
 
-class CameraMimeType(str, Enum):
-    VIAM_RGBA = "image/vnd.viam.rgba"
-    VIAM_RAW_DEPTH = "image/vnd.viam.dep"
-    JPEG = "image/jpeg"
-    PNG = "image/png"
-    PCD = "pointcloud/pcd"
+class _FrozenClassAttributesMeta(type):
+    """
+    A metaclass that prevents the reassignment of existing class attributes.
+    """
+
+    def __setattr__(cls, name: str, value: Any):
+        # Check if the attribute `name` already exists on the class
+        if name in cls.__dict__:
+            # If it exists, raise an error to prevent overwriting
+            raise AttributeError(f"Cannot reassign constant '{name}'")
+        # If it's a new attribute, allow it to be set
+        super().__setattr__(name, value)
+
+
+class CameraMimeType(str, metaclass=_FrozenClassAttributesMeta):
+    """
+    The compatible mime-types for cameras and vision services.
+
+    You can use the `CameraMimeType.CUSTOM(...)` method to use an unlisted mime-type.
+    """
+
+    VIAM_RGBA: ClassVar[Self]
+    VIAM_RAW_DEPTH: ClassVar[Self]
+    JPEG: ClassVar[Self]
+    PNG: ClassVar[Self]
+    PCD: ClassVar[Self]
+
+    @property
+    def name(self) -> str:
+        for key, value in self.__class__.__dict__.items():
+            if value == self:
+                return key
+        return "CUSTOM"
+
+    @property
+    def value(self) -> str:
+        return self
+
+    @classmethod
+    def CUSTOM(cls, mime_type: str) -> Self:
+        """
+        Create a custom mime type.
+
+        Args:
+            mime_type (str): The mimetype as a string
+        """
+        return cls.from_string(mime_type)
 
     @classmethod
     def from_string(cls, value: str) -> Self:
@@ -30,37 +69,12 @@ class CameraMimeType(str, Enum):
         value_mime = value[:-5] if value.endswith("+lazy") else value  # ViamImage lazy encodes by default
         return cls(value_mime)
 
-    @classmethod
-    def from_proto(cls, format: Format.ValueType) -> "CameraMimeType":
-        """Returns the mimetype from a proto enum.
-
-        Args:
-            format (Format.ValueType): The mimetype in a proto enum.
-
-        Returns:
-            Self: The mimetype.
-        """
-        mimetypes = {
-            Format.FORMAT_RAW_RGBA: CameraMimeType.VIAM_RGBA,
-            Format.FORMAT_RAW_DEPTH: CameraMimeType.VIAM_RAW_DEPTH,
-            Format.FORMAT_JPEG: CameraMimeType.JPEG,
-            Format.FORMAT_PNG: CameraMimeType.PNG,
-        }
-        return mimetypes.get(format, CameraMimeType.JPEG)
 
-    def to_proto(self) -> Format.ValueType:
-        """Returns the mimetype in a proto enum.
-
-        Returns:
-            Format.ValueType: The mimetype in a proto enum.
-        """
-        formats = {
-            self.VIAM_RGBA: Format.FORMAT_RAW_RGBA,
-            self.VIAM_RAW_DEPTH: Format.FORMAT_RAW_DEPTH,
-            self.JPEG: Format.FORMAT_JPEG,
-            self.PNG: Format.FORMAT_PNG,
-        }
-        return formats.get(self, Format.FORMAT_UNSPECIFIED)
+CameraMimeType.VIAM_RGBA = CameraMimeType.from_string("image/vnd.viam.rgba")
+CameraMimeType.VIAM_RAW_DEPTH = CameraMimeType.from_string("image/vnd.viam.dep")
+CameraMimeType.JPEG = CameraMimeType.from_string("image/jpeg")
+CameraMimeType.PNG = CameraMimeType.from_string("image/png")
+CameraMimeType.PCD = CameraMimeType.from_string("pointcloud/pcd")
 
 
 class ViamImage:

viam/module/module.py

@@ -1,11 +1,14 @@
 import argparse
 import io
 import logging as pylogging
+import os
 import sys
+from collections.abc import Iterable
 from inspect import iscoroutinefunction
 from threading import Lock
 from typing import List, Mapping, Optional, Sequence, Tuple
 
+from grpclib.metadata import Deadline
 from grpclib.utils import _service_name
 from typing_extensions import Self
 
@@ -29,13 +32,37 @@ from viam.resource.base import ResourceBase
 from viam.resource.registry import Registry
 from viam.resource.types import API, RESOURCE_TYPE_COMPONENT, RESOURCE_TYPE_SERVICE, Model, ResourceName, resource_name_from_string
 from viam.robot.client import RobotClient
-from viam.rpc.dial import DialOptions
+from viam.rpc.dial import DialOptions, _host_port_from_url
 from viam.rpc.server import Server
 
+# These imports are required to register built-in resources with the registry
+from ..components.arm import Arm  # noqa: F401
+from ..components.base import Base  # noqa: F401
+from ..components.board import Board  # noqa: F401
+from ..components.button import Button  # noqa: F401
+from ..components.camera import Camera  # noqa: F401
+from ..components.encoder import Encoder  # noqa: F401
+from ..components.gantry import Gantry  # noqa: F401
+from ..components.generic import Generic as GenericComponent  # noqa: F401
+from ..components.gripper import Gripper  # noqa: F401
+from ..components.input import Controller  # noqa: F401
+from ..components.motor import Motor  # noqa: F401
+from ..components.movement_sensor import MovementSensor  # noqa: F401
+from ..components.pose_tracker import PoseTracker  # noqa: F401
+from ..components.power_sensor import PowerSensor  # noqa: F401
+from ..components.sensor import Sensor  # noqa: F401
+from ..components.servo import Servo  # noqa: F401
+from ..components.switch import Switch  # noqa: F401
+from ..services.discovery import Discovery  # noqa: F401
+from ..services.generic import Generic as GenericService  # noqa: F401
+from ..services.motion import Motion  # noqa: F401
+from ..services.navigation import Navigation  # noqa: F401
+from ..services.slam import SLAM  # noqa: F401
+from ..services.vision import Vision  # noqa: F401
 from .service import ModuleRPCService
 from .types import Reconfigurable, Stoppable
 
-LOGGER = logging.getLogger(__name__)
+NO_MODULE_PARENT = os.environ.get("VIAM_NO_MODULE_PARENT", "").lower() == "true"
 
 
 def _parse_module_args() -> argparse.Namespace:
@@ -45,7 +72,8 @@ def _parse_module_args() -> argparse.Namespace:
     p = argparse.ArgumentParser(description="Start this viam python module")
     p.add_argument("socket_path", help="path where this module will serve a unix socket")
     p.add_argument("--log-level", type=lambda name: pylogging._nameToLevel[name.upper()], default=logging.INFO)
-    return p.parse_args()
+    p.add_argument("--tcp-mode", action="store_true")
+    return p.parse_known_args()[0]
 
 
 class Module:
@@ -54,8 +82,10 @@ class Module:
     _ready: bool
     _log_level: int
     _lock: Lock
+    _tcp_mode: bool
     parent: Optional[RobotClient] = None
     server: Server
+    logger: pylogging.Logger
 
     @classmethod
     def from_args(cls) -> Self:
@@ -71,7 +101,7 @@ class Module:
             Module: a new Module instance
         """
         args = _parse_module_args()
-        return cls(args.socket_path, log_level=args.log_level)
+        return cls(args.socket_path, log_level=args.log_level, tcp_mode=args.tcp_mode)
 
     @classmethod
     async def run_with_models(cls, *models: ResourceBase):
@@ -103,7 +133,7 @@ class Module:
             module.add_model_from_registry(*key.split("/"))  # pyright: ignore [reportArgumentType]
         await module.start()
 
-    def __init__(self, address: str, *, log_level: int = logging.INFO) -> None:
+    def __init__(self, address: str, *, log_level: int = logging.INFO, tcp_mode: bool = False) -> None:
         # When a module is launched by viam-server, its stdout is not connected to a tty.  In
         # response, python disables line buffering, which prevents `print` statements from being
         # immediately flushed to viam-server. This behavior can be confusing, interfere with
@@ -114,12 +144,21 @@ class Module:
         if isinstance(sys.stderr, io.TextIOWrapper):
             sys.stderr.reconfigure(line_buffering=True)
         self._address = address
+        self._tcp_mode = tcp_mode
         self.server = Server(resources=[], module_service=ModuleRPCService(self))
         self._log_level = log_level
+
+        module_name = os.environ.get("VIAM_MODULE_NAME")
+        # this can happen if the user is running an old version of viam-server that doesn't set `VIAM_MODULE_NAME`
+        if module_name is None:
+            module_name = __name__
+        self.logger = logging.getLogger(module_name)
         self._ready = True
         self._lock = Lock()
 
     async def _connect_to_parent(self):
+        if NO_MODULE_PARENT:
+            return
         if self.parent is None:
             if self._parent_address is None:
                 raise ValueError("Parent address not found")
@@ -130,7 +169,7 @@ class Module:
                     log_level=self._log_level,
                 ),
             )
-            LOGGER.debug("Starting module logging")
+            self.logger.debug("Starting module logging")
             logging.setParent(self.parent)
 
     async def _get_resource(self, name: ResourceName) -> ResourceBase:
@@ -153,19 +192,23 @@ class Module:
     async def start(self):
         """Start the module service and gRPC server"""
         try:
-            await self.server.serve(log_level=self._log_level, path=self._address)
+            if self._tcp_mode:
+                host, port = _host_port_from_url(self._address)
+                await self.server.serve(log_level=self._log_level, host=host, port=port)
+            else:
+                await self.server.serve(log_level=self._log_level, path=self._address)
         finally:
             await self.stop()
 
     async def stop(self):
         """Stop the module service and gRPC server"""
-        LOGGER.debug("Shutting down module")
+        self.logger.debug("Shutting down module")
         try:
             logging.shutdown()
             if self.parent is not None:
                 await self.parent.close()
         except Exception as e:
-            LOGGER.error("Encountered error while shutting down module", exc_info=e)
+            self.logger.error("Encountered error while shutting down module", exc_info=e)
 
     def set_ready(self, ready: bool):
         """Set the module's ready state. The module automatically sets to READY on load. Setting to False can be useful
@@ -177,13 +220,15 @@ class Module:
         with self._lock:
             self._ready = ready
 
-    async def add_resource(self, request: AddResourceRequest):
+    async def add_resource(self, request: AddResourceRequest, *, deadline: Optional[Deadline] = None):
         dependencies = await self._get_dependencies(request.dependencies)
         config: ComponentConfig = request.config
         api = API.from_string(config.api)
         model = Model.from_string(config.model, ignore_errors=True)
         creator = Registry.lookup_resource_creator(api, model)
         resource = creator(config, dependencies)
+        if deadline is not None and deadline.time_remaining() <= 0:
+            raise TimeoutError("Deadline expired")
         update_log_level(resource.logger, config.log_configuration.level.upper())
         self.server.register(resource)
 
@@ -210,10 +255,13 @@ class Module:
         rn = resource_name_from_string(request.name)
         resource = self.server.get_resource(ResourceBase, rn)
         if isinstance(resource, Stoppable):
-            if iscoroutinefunction(resource.stop):
-                await resource.stop()
-            else:
-                resource.stop()
+            try:
+                if iscoroutinefunction(resource.stop):
+                    await resource.stop()
+                else:
+                    resource.stop()
+            except Exception as e:
+                self.logger.warning(f"Could not remove resource named {resource.name}", exc_info=e)
         await self.server.remove_resource(rn)
 
     async def ready(self, request: ReadyRequest) -> ReadyResponse:
@@ -266,7 +314,28 @@ class Module:
         model = Model.from_string(config.model)
         validator = Registry.lookup_validator(api, model)
         try:
-            dependencies = validator(config)
-            return ValidateConfigResponse(dependencies=dependencies)
+            # backwards compatibility. Support both ([], []) or [] with deprecation warning.
+            # If user's validate returns [str], it will be treated as required dependencies only.
+            # Incorect formats, e.g. int, will raise ValidationError.
+            _validator_return_test = validator(config)
+            if not (isinstance(_validator_return_test, tuple) and len(_validator_return_test) == 2):
+                msg = f"Your validate function {validator.__name__} did not return \
+                        type tuple[Sequence[str], Sequence[str]]. Got {_validator_return_test}."
+                self.logger.warning(msg)
+                if (isinstance(_validator_return_test, Iterable) and not isinstance(_validator_return_test, str)) and all(
+                    isinstance(e, str)
+                    for e in _validator_return_test  # type: ignore
+                ):
+                    self.logger.warning(
+                        f"Detected deprecated validate function signature. \
+                                        Treating all dependencies {_validator_return_test} as required dependencies. \
+                                        Please update to new signature Tuple[Sequence[str], Sequence[str]] soon."
+                    )
+                    return ValidateConfigResponse(dependencies=_validator_return_test)
+                else:
+                    raise ValidationError(msg)
+
+            dependencies, optional_dependencies = _validator_return_test
+            return ValidateConfigResponse(dependencies=dependencies, optional_dependencies=optional_dependencies)
         except Exception as e:
             raise ValidationError(f"{type(Exception)}: {e}").grpc_error

viam/module/resource_data_consumer.py

@@ -0,0 +1,41 @@
+import datetime
+import os
+from typing import Any, Dict, List, Optional
+
+from viam.app.viam_client import ViamClient
+
+
+class ResourceDataConsumer:
+    """Client for retrieving historical module data from app.
+
+    Inherit from this class in a module to get access to historical module data.
+    """
+
+    @classmethod
+    def construct_query(cls, part_id: str, resource_name: str, time_back: datetime.timedelta) -> List[Dict[str, Any]]:
+        return [
+            {
+                "$match": {
+                    "part_id": part_id,
+                    "component_name": resource_name,
+                    "time_received": {"$gte": datetime.datetime.now() - time_back},
+                }
+            }
+        ]
+
+    @classmethod
+    async def query_tabular_data(
+        cls, resource_name: str, time_back: datetime.timedelta, additional_stages: Optional[List[Dict[str, Any]]] = None, **kwargs
+    ) -> List[Dict[str, Any]]:
+        """Return historical data for this module, queried with MQL."""
+        viam_client = await ViamClient.create_from_env_vars()
+
+        org_id = os.environ["VIAM_PRIMARY_ORG_ID"]
+        part_id = os.environ["VIAM_MACHINE_PART_ID"]
+
+        query = cls.construct_query(part_id=part_id, resource_name=resource_name, time_back=time_back)
+
+        if additional_stages is not None:
+            query += additional_stages
+
+        return await viam_client.data_client.tabular_data_by_mql(org_id, query)

viam/module/service.py

@@ -1,7 +1,9 @@
 from typing import TYPE_CHECKING
 
+from grpclib import Status
 from grpclib.server import Stream
 
+from viam.errors import ViamGRPCError
 from viam.proto.module import (
     AddResourceRequest,
     AddResourceResponse,
@@ -29,7 +31,13 @@ class ModuleRPCService(ModuleServiceBase):
     async def AddResource(self, stream: Stream[AddResourceRequest, AddResourceResponse]) -> None:
         request = await stream.recv_message()
         assert request is not None
-        await self._module.add_resource(request)
+        try:
+            await self._module.add_resource(request, deadline=stream.deadline)
+        except TimeoutError:
+            raise ViamGRPCError(
+                message="Timeout while adding resource",
+                grpc_code=Status.DEADLINE_EXCEEDED,
+            )
         await stream.send_message(AddResourceResponse())
 
     async def ReconfigureResource(self, stream: Stream[ReconfigureResourceRequest, ReconfigureResourceResponse]) -> None:

viam/proto/app/__init__.py

@@ -11,6 +11,9 @@ from ...gen.app.v1.app_pb2 import (
     AddRoleResponse,
     APIKey,
     APIKeyWithAuthorizations,
+    App,
+    AppCustomizations,
+    AppType,
     AuthenticationType,
     AuthenticatorInfo,
     Authorization,
@@ -83,10 +86,17 @@ from ...gen.app.v1.app_pb2 import (
     FragmentError,
     FragmentErrorType,
     FragmentHistoryEntry,
+    FragmentImport,
+    FragmentImportList,
     FragmentRevision,
+    FragmentSummary,
     FragmentTag,
     FragmentUsage,
     FragmentVisibility,
+    GetAppBrandingRequest,
+    GetAppBrandingResponse,
+    GetAppContentRequest,
+    GetAppContentResponse,
     GetBillingServiceConfigRequest,
     GetBillingServiceConfigResponse,
     GetFragmentHistoryRequest,
@@ -115,6 +125,8 @@ from ...gen.app.v1.app_pb2 import (
     GetRobotAPIKeysResponse,
     GetRobotMetadataRequest,
     GetRobotMetadataResponse,
+    GetRobotPartByNameAndLocationRequest,
+    GetRobotPartByNameAndLocationResponse,
     GetRobotPartHistoryRequest,
     GetRobotPartHistoryResponse,
     GetRobotPartLogsRequest,
@@ -141,8 +153,12 @@ from ...gen.app.v1.app_pb2 import (
     ListLocationsResponse,
     ListMachineFragmentsRequest,
     ListMachineFragmentsResponse,
+    ListMachineSummariesRequest,
+    ListMachineSummariesResponse,
     ListModulesRequest,
     ListModulesResponse,
+    ListNestedFragmentsRequest,
+    ListNestedFragmentsResponse,
     ListOAuthAppsRequest,
     ListOAuthAppsResponse,
     ListOrganizationMembersRequest,
@@ -153,6 +169,10 @@ from ...gen.app.v1.app_pb2 import (
     ListOrganizationsResponse,
     ListRegistryItemsRequest,
     ListRegistryItemsResponse,
+    ListRobotsForLocationsRequest,
+    ListRobotsForLocationsResponse,
+    ListRobotsForOrgRequest,
+    ListRobotsForOrgResponse,
     ListRobotsRequest,
     ListRobotsResponse,
     Location,
@@ -160,6 +180,9 @@ from ...gen.app.v1.app_pb2 import (
     LocationAuthRequest,
     LocationAuthResponse,
     LocationOrganization,
+    LocationSummary,
+    MachinePickerCustomizations,
+    MachineSummary,
     MarkPartAsMainRequest,
     MarkPartAsMainResponse,
     MarkPartForRestartRequest,
@@ -170,13 +193,16 @@ from ...gen.app.v1.app_pb2 import (
     Model,
     Module,
     ModuleFileInfo,
+    ModuleLanguage,
     ModuleMetadata,
+    ModuleSourceType,
     ModuleVersion,
     NewRobotPartRequest,
     NewRobotPartResponse,
     NewRobotRequest,
     NewRobotResponse,
     OAuthConfig,
+    OnlineState,
     Organization,
     OrganizationGetLogoRequest,
     OrganizationGetLogoResponse,
@@ -190,6 +216,7 @@ from ...gen.app.v1.app_pb2 import (
     OrganizationSetSupportEmailRequest,
     OrganizationSetSupportEmailResponse,
     OrgDetails,
+    PartSummary,
     ReadOAuthAppRequest,
     ReadOAuthAppResponse,
     RegistryItem,
@@ -198,6 +225,8 @@ from ...gen.app.v1.app_pb2 import (
     RemoveRoleResponse,
     RenameKeyRequest,
     RenameKeyResponse,
+    RenameRegistryItemRequest,
+    RenameRegistryItemResponse,
     ResendOrganizationInviteRequest,
     ResendOrganizationInviteResponse,
     ResolvedFragment,
@@ -217,6 +246,7 @@ from ...gen.app.v1.app_pb2 import (
     StorageConfig,
     TailRobotPartLogsRequest,
     TailRobotPartLogsResponse,
+    TextOverrides,
     TransferRegistryItemRequest,
     TransferRegistryItemResponse,
     UnshareLocationRequest,
@@ -240,6 +270,8 @@ from ...gen.app.v1.app_pb2 import (
     UpdateOrganizationInviteAuthorizationsResponse,
     UpdateOrganizationMetadataRequest,
     UpdateOrganizationMetadataResponse,
+    UpdateOrganizationNamespaceRequest,
+    UpdateOrganizationNamespaceResponse,
     UpdateOrganizationRequest,
     UpdateOrganizationResponse,
     UpdateRegistryItemRequest,
@@ -257,6 +289,8 @@ from ...gen.app.v1.app_pb2 import (
     Uploads,
     URLValidation,
     VersionHistory,
+    ViamAgentVersion,
+    ViamServerVersion,
     Visibility,
 )
 
@@ -269,6 +303,9 @@ __all__ = [
     "AddRoleRequest",
     "AddRoleResponse",
     "AdditionalFragment",
+    "App",
+    "AppCustomizations",
+    "AppType",
     "AuthenticationType",
     "AuthenticatorInfo",
     "Authorization",
@@ -341,10 +378,17 @@ __all__ = [
     "FragmentError",
     "FragmentErrorType",
     "FragmentHistoryEntry",
+    "FragmentImport",
+    "FragmentImportList",
     "FragmentRevision",
+    "FragmentSummary",
     "FragmentTag",
     "FragmentUsage",
     "FragmentVisibility",
+    "GetAppBrandingRequest",
+    "GetAppBrandingResponse",
+    "GetAppContentRequest",
+    "GetAppContentResponse",
     "GetBillingServiceConfigRequest",
     "GetBillingServiceConfigResponse",
     "GetFragmentHistoryRequest",
@@ -373,6 +417,8 @@ __all__ = [
     "GetRobotAPIKeysResponse",
     "GetRobotMetadataRequest",
     "GetRobotMetadataResponse",
+    "GetRobotPartByNameAndLocationRequest",
+    "GetRobotPartByNameAndLocationResponse",
     "GetRobotPartHistoryRequest",
     "GetRobotPartHistoryResponse",
     "GetRobotPartLogsRequest",
@@ -399,8 +445,12 @@ __all__ = [
     "ListLocationsResponse",
     "ListMachineFragmentsRequest",
     "ListMachineFragmentsResponse",
+    "ListMachineSummariesRequest",
+    "ListMachineSummariesResponse",
     "ListModulesRequest",
     "ListModulesResponse",
+    "ListNestedFragmentsRequest",
+    "ListNestedFragmentsResponse",
     "ListOAuthAppsRequest",
     "ListOAuthAppsResponse",
     "ListOrganizationMembersRequest",
@@ -411,6 +461,10 @@ __all__ = [
     "ListOrganizationsResponse",
     "ListRegistryItemsRequest",
     "ListRegistryItemsResponse",
+    "ListRobotsForLocationsRequest",
+    "ListRobotsForLocationsResponse",
+    "ListRobotsForOrgRequest",
+    "ListRobotsForOrgResponse",
     "ListRobotsRequest",
     "ListRobotsResponse",
     "Location",
@@ -418,9 +472,12 @@ __all__ = [
     "LocationAuthRequest",
     "LocationAuthResponse",
     "LocationOrganization",
+    "LocationSummary",
     "MLModelMetadata",
     "MLTrainingMetadata",
     "MLTrainingVersion",
+    "MachinePickerCustomizations",
+    "MachineSummary",
     "MarkPartAsMainRequest",
     "MarkPartAsMainResponse",
     "MarkPartForRestartRequest",
@@ -428,13 +485,16 @@ __all__ = [
     "Model",
     "Module",
     "ModuleFileInfo",
+    "ModuleLanguage",
     "ModuleMetadata",
+    "ModuleSourceType",
     "ModuleVersion",
     "NewRobotPartRequest",
     "NewRobotPartResponse",
     "NewRobotRequest",
     "NewRobotResponse",
     "OAuthConfig",
+    "OnlineState",
     "OrgDetails",
     "Organization",
     "OrganizationGetLogoRequest",
@@ -449,6 +509,7 @@ __all__ = [
     "OrganizationSetSupportEmailRequest",
     "OrganizationSetSupportEmailResponse",
     "PKCE",
+    "PartSummary",
     "ReadOAuthAppRequest",
     "ReadOAuthAppResponse",
     "RegistryItem",
@@ -457,6 +518,8 @@ __all__ = [
     "RemoveRoleResponse",
     "RenameKeyRequest",
     "RenameKeyResponse",
+    "RenameRegistryItemRequest",
+    "RenameRegistryItemResponse",
     "ResendOrganizationInviteRequest",
     "ResendOrganizationInviteResponse",
     "ResolvedFragment",
@@ -476,6 +539,7 @@ __all__ = [
     "StorageConfig",
     "TailRobotPartLogsRequest",
     "TailRobotPartLogsResponse",
+    "TextOverrides",
     "TransferRegistryItemRequest",
     "TransferRegistryItemResponse",
     "URLValidation",
@@ -500,6 +564,8 @@ __all__ = [
     "UpdateOrganizationInviteAuthorizationsResponse",
     "UpdateOrganizationMetadataRequest",
     "UpdateOrganizationMetadataResponse",
+    "UpdateOrganizationNamespaceRequest",
+    "UpdateOrganizationNamespaceResponse",
     "UpdateOrganizationRequest",
     "UpdateOrganizationResponse",
     "UpdateRegistryItemRequest",
@@ -516,5 +582,7 @@ __all__ = [
     "UploadModuleFileResponse",
     "Uploads",
     "VersionHistory",
+    "ViamAgentVersion",
+    "ViamServerVersion",
     "Visibility",
 ]