Skip to content

Commit 99bb01d

Browse files
committed
fix bidi web socket channel connect for chrome
1 parent d5d535b commit 99bb01d

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

packages/firebase_ai/firebase_ai/lib/src/base_model.dart

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,9 +18,11 @@ import 'dart:convert';
1818
import 'package:firebase_app_check/firebase_app_check.dart';
1919
import 'package:firebase_auth/firebase_auth.dart';
2020
import 'package:firebase_core/firebase_core.dart';
21+
import 'package:flutter/foundation.dart';
2122
import 'package:http/http.dart' as http;
2223
import 'package:meta/meta.dart';
2324
import 'package:web_socket_channel/io.dart';
25+
import 'package:web_socket_channel/web_socket_channel.dart';
2426

2527
import 'api.dart';
2628
import 'client.dart';

packages/firebase_ai/firebase_ai/lib/src/live_model.dart

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,10 @@ final class LiveGenerativeModel extends BaseModel {
8989

9090
final request = jsonEncode(setupJson);
9191
final headers = await BaseModel.firebaseTokens(_appCheck, _auth, _app)();
92-
var ws = IOWebSocketChannel.connect(Uri.parse(uri), headers: headers);
92+
93+
var ws = kIsWeb
94+
? WebSocketChannel.connect(Uri.parse(uri))
95+
: IOWebSocketChannel.connect(Uri.parse(uri), headers: headers);
9396
await ws.ready;
9497

9598
ws.sink.add(request);

0 commit comments

Comments
 (0)