add module models

viamrobotics · Jan 2, 2025 · ca1dfb3 · ca1dfb3
1 parent d20195c
commit ca1dfb3
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/lib/src/robot/client.dart b/lib/src/robot/client.dart
@@ -349,4 +349,14 @@ class RobotClient {
     final response = await _client.discoverComponents(request);
     return response.discovery.map((d) => Discovery.fromProto(d)).toList();
   }
+
+  /// GetModelsFromModules returns the list of models supported in modules on the machine.
+  ///
+  /// ```
+  /// var modelsFromModules = await machine.getModelsFromModules();
+  /// ```
+  Future<List<ModuleModel>> getModelsFromModules() async {
+    final response = await _client.getModelsFromModules(request);
+    return response.model.map((d) => ModuleModel.fromProto(d)).toList();
+  }
 }