diff --git a/example/lib/main.dart b/example/lib/main.dart
index d20c816..a8b86d8 100644
--- a/example/lib/main.dart
+++ b/example/lib/main.dart
@@ -3,7 +3,6 @@ import 'dart:developer';
 import 'dart:io';
 
 import 'package:openai_client/openai_client.dart';
-import 'package:openai_client/src/model/openai_chat/chat_message.dart';
 
 Future<void> main() async {
   // Load app credentials from environment variables or file.
diff --git a/lib/openai_client.dart b/lib/openai_client.dart
index 5a00ea3..5727807 100644
--- a/lib/openai_client.dart
+++ b/lib/openai_client.dart
@@ -5,4 +5,5 @@ export 'src/completions.dart';
 export 'src/configuration.dart';
 export 'src/logger/logger.dart';
 export 'src/model/model.dart';
+export 'src/model/openai_chat/openai_chat.dart';
 export 'src/network/network.dart';
diff --git a/test/chat_test.dart b/test/chat_test.dart
index 5ac6b51..2ce3a9a 100644
--- a/test/chat_test.dart
+++ b/test/chat_test.dart
@@ -1,5 +1,4 @@
 import 'package:openai_client/openai_client.dart';
-import 'package:openai_client/src/model/openai_chat/openai_chat.dart';
 import 'package:test/test.dart';
 
 import 'test_utils.dart';