protocol

uberduck-ai · Feb 22, 2024 · 866474a · 866474a
1 parent f9029ec
commit 866474a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/clients/simple/simple_bot.py b/clients/simple/simple_bot.py
@@ -112,7 +112,8 @@ async def run_websocket(uri, audio_queue):
 
 
 if __name__ == "__main__":
-    uri = f"ws://{UBERDUCK_API_HOST}?session_id={session}"
+    ws_proto = "ws" if "localhost" in UBERDUCK_API_HOST else "wss"
+    uri = f"{ws_proto}://{UBERDUCK_API_HOST}?session_id={session}"
     play_startup_sound()
 
     recorder = AudioRecorder()