From dacf982f607659dd46c639e4c2f9624f2a2b2261 Mon Sep 17 00:00:00 2001 From: Paul Nechifor Date: Fri, 19 Dec 2025 21:17:47 +0200 Subject: [PATCH] fix set transport --- dimos/core/module.py | 2 +- dimos/core/test_core.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/dimos/core/module.py b/dimos/core/module.py index ff4c2faa3b..c6d8a69302 100644 --- a/dimos/core/module.py +++ b/dimos/core/module.py @@ -347,7 +347,7 @@ def set_ref(self, ref) -> int: # type: ignore[no-untyped-def] def __str__(self) -> str: return f"{self.__class__.__name__}" - # called from remote + @rpc def set_transport(self, stream_name: str, transport: Transport) -> bool: # type: ignore[type-arg] stream = getattr(self, stream_name, None) if not stream: diff --git a/dimos/core/test_core.py b/dimos/core/test_core.py index f9262acd55..17ee300a44 100644 --- a/dimos/core/test_core.py +++ b/dimos/core/test_core.py @@ -87,7 +87,7 @@ def test_classmethods() -> None: # Check that we have the expected RPC methods assert "navigate_to" in class_rpcs, "navigate_to should be in rpcs" assert "start" in class_rpcs, "start should be in rpcs" - assert len(class_rpcs) == 8 + assert len(class_rpcs) == 9 # Check that the values are callable assert callable(class_rpcs["navigate_to"]), "navigate_to should be callable"