diff --git a/src/main/java/com/google/genai/AsyncLive.java b/src/main/java/com/google/genai/AsyncLive.java index 1a7915ae812..c764a713997 100644 --- a/src/main/java/com/google/genai/AsyncLive.java +++ b/src/main/java/com/google/genai/AsyncLive.java @@ -92,9 +92,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(),