Fix skills

dimos/agents/skills/conftest.py

@@ -72,8 +72,6 @@ def google_maps_skill_container(mocker):
 @pytest.fixture
 def unitree_skills(mocker):
     container = UnitreeSkillContainer()
-    container._move = mocker.Mock()
-    container._publish_request = mocker.Mock()
     container.start()
     yield container
     container.stop()

dimos/agents/skills/test_unitree_skill_container.py

@@ -13,29 +13,19 @@
 # limitations under the License.
 
 
-def test_pounce(create_unitree_skills_agent, unitree_skills) -> None:
+def test_pounce(mocker, create_unitree_skills_agent, unitree_skills) -> None:
     agent = create_unitree_skills_agent(fixture="test_pounce.json")
+    publish_request_mock = mocker.Mock()
+    unitree_skills.get_rpc_calls = mocker.Mock(return_value=publish_request_mock)
 
     response = agent.query("pounce")
 
     assert "front pounce" in response.lower()
-    unitree_skills._publish_request.assert_called_once_with(
-        "rt/api/sport/request", {"api_id": 1032}
-    )
-
-
-def test_show_your_love(create_unitree_skills_agent, unitree_skills) -> None:
-    agent = create_unitree_skills_agent(fixture="test_show_your_love.json")
-
-    response = agent.query("show your love")
-
-    assert "finger heart" in response.lower()
-    unitree_skills._publish_request.assert_called_once_with(
-        "rt/api/sport/request", {"api_id": 1036}
-    )
+    publish_request_mock.assert_called_once_with("rt/api/sport/request", {"api_id": 1032})
 
 
-def test_did_you_mean(unitree_skills) -> None:
+def test_did_you_mean(mocker, unitree_skills) -> None:
+    unitree_skills.get_rpc_calls = mocker.Mock()
     assert (
         unitree_skills.execute_sport_command("Pounce")
         == "There's no 'Pounce' command. Did you mean: ['FrontPounce', 'Pose']"

dimos/agents/test_mock_agent.py

@@ -26,7 +26,7 @@
 from dimos.msgs.geometry_msgs import PoseStamped, Vector3
 from dimos.msgs.sensor_msgs import Image
 from dimos.protocol.skill.test_coordinator import SkillContainerTest
-from dimos.robot.unitree_webrtc.modular.connection_module import ConnectionModule
+from dimos.robot.unitree.connection.go2 import GO2Connection
 from dimos.robot.unitree_webrtc.type.lidar import LidarMessage
 
 
@@ -157,11 +157,11 @@ def test_tool_call_implicit_detections() -> None:
         system_prompt="You are a helpful robot assistant with camera capabilities.",
     )
 
-    robot_connection = dimos.deploy(ConnectionModule, connection_type="fake")
+    robot_connection = dimos.deploy(GO2Connection, connection_type="fake")
     robot_connection.lidar.transport = LCMTransport("/lidar", LidarMessage)
     robot_connection.odom.transport = LCMTransport("/odom", PoseStamped)
     robot_connection.video.transport = LCMTransport("/image", Image)
-    robot_connection.movecmd.transport = LCMTransport("/cmd_vel", Vector3)
+    robot_connection.cmd_vel.transport = LCMTransport("/cmd_vel", Vector3)
     robot_connection.camera_info.transport = LCMTransport("/camera_info", CameraInfo)
     robot_connection.start()
 

dimos/core/module.py

@@ -15,7 +15,6 @@
 from collections.abc import Callable
 from dataclasses import dataclass
 from functools import partial
-import inspect
 import sys
 import threading
 from typing import (
@@ -36,7 +35,7 @@
 
 from dimos.core import colors
 from dimos.core.core import T, rpc
-from dimos.core.introspection.module import INTERNAL_RPCS, extract_module_info, render_module_io
+from dimos.core.introspection.module import extract_module_info, render_module_io
 from dimos.core.resource import Resource
 from dimos.core.rpc_client import RpcCall
 from dimos.core.stream import In, Out, RemoteIn, RemoteOut, Transport
@@ -276,7 +275,6 @@ def __get__(
     @classmethod
     def _module_info_class(cls) -> "ModuleInfo":
         """Class-level module_info() - returns ModuleInfo from annotations."""
-        from dimos.core.introspection.module import ModuleInfo
 
         hints = get_type_hints(cls)
 

dimos/hardware/sensors/camera/module.py

@@ -19,7 +19,6 @@
 
 import reactivex as rx
 from reactivex import operators as ops
-from reactivex.observable import Observable
 
 from dimos.agents import Output, Reducer, Stream, skill
 from dimos.core import Module, ModuleConfig, Out, rpc