Skip to content

Commit 87a9bee

Browse files
committed
feat: add wss support
1 parent 95aadc6 commit 87a9bee

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

packages/openpi-client/src/openpi_client/websocket_client_policy.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,10 @@ class WebsocketClientPolicy(_base_policy.BasePolicy):
1616
"""
1717

1818
def __init__(self, host: str = "0.0.0.0", port: Optional[int] = None, api_key: Optional[str] = None) -> None:
19-
self._uri = f"ws://{host}"
19+
if host.startswith("ws"):
20+
self._uri = host
21+
else:
22+
self._uri = f"ws://{host}"
2023
if port is not None:
2124
self._uri += f":{port}"
2225
self._packer = msgpack_numpy.Packer()

0 commit comments

Comments
 (0)