diff --git a/src/main/java/com/google/genai/AsyncLive.java b/src/main/java/com/google/genai/AsyncLive.java index 1681b39f7e6..0ca7f0020b7 100644 --- a/src/main/java/com/google/genai/AsyncLive.java +++ b/src/main/java/com/google/genai/AsyncLive.java @@ -93,9 +93,10 @@ private URI getWebSocketUri() { } try { URI baseUri = new URI(baseUrl); + String wsScheme = "https".equalsIgnoreCase(baseUri.getScheme()) ? "wss" : "ws"; String wsBaseUrl = new URI( - "wss", + wsScheme, baseUri.getAuthority(), baseUri.getPath(), baseUri.getQuery(),