diff --git a/src/client.rs b/src/client.rs
index d4845fc..79c51ed 100644
--- a/src/client.rs
+++ b/src/client.rs
@@ -1,8 +1,6 @@
-use crate::dialogue::Message;
 use crate::error::{Error as GeminiError, Result as GeminiResult};
 use crate::network::event_source::{EventSource, ServerSentEvent};
 use crate::prelude::*;
-use crate::types::{PredictImageRequest, PredictImageResponse, Role};
 
 use tokio_stream::{Stream, StreamExt};
 use tokio_util::codec::LinesCodecError;
diff --git a/src/dialogue.rs b/src/dialogue.rs
index d82b8ff..8117124 100644
--- a/src/dialogue.rs
+++ b/src/dialogue.rs
@@ -1,6 +1,6 @@
 use serde::{Deserialize, Serialize};
 
-use crate::{client::GeminiClient, error::Result, types::Role};
+use crate::{error::Result, prelude::*};
 
 #[derive(Clone, Debug, Serialize, Deserialize)]
 pub struct Message {