fix mujoco menagerie

dimos/agents/vlm_agent.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
+from langchain_core.messages import AIMessage, HumanMessage
 
 from dimos.agents.llm_init import build_llm, build_system_message
 from dimos.agents.spec import AgentSpec, AnyMessage

dimos/protocol/mcp/test_mcp_module.py

@@ -21,7 +21,6 @@
 import socket
 import subprocess
 import sys
-import threading
 
 import pytest
 

dimos/robot/unitree_webrtc/mujoco_connection.py

@@ -61,9 +61,11 @@ def __init__(self, global_config: GlobalConfig) -> None:
         # Pre-download the mujoco_sim data.
         get_data("mujoco_sim")
 
-        # Trigger the download of the mujoco_menajerie package. This is so it
+        # Trigger the download of the mujoco_menagerie package. This is so it
         # doesn't trigger in the mujoco process where it can time out.
-        import mujoco_playground
+        from mujoco_playground._src import mjx_env
+
+        mjx_env.ensure_menagerie_exists()
 
         self.global_config = global_config
         self.process: subprocess.Popen[bytes] | None = None

dimos/stream/video_operators.py

@@ -16,7 +16,7 @@
 from collections.abc import Callable
 from datetime import datetime, timedelta
 from enum import Enum
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING
 
 import cv2
 import numpy as np