Tweak imports

gabotechs · Jan 11, 2025 · 3c87207 · 3c87207
1 parent 6a6a97d
commit 3c87207
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/backend/server.rs b/src/backend/server.rs
@@ -79,7 +79,7 @@ async fn web_app() -> Html<&'static str> {
 mod tests {
     use async_trait::async_trait;
     use std::sync::atomic::{AtomicU16, Ordering};
-
+    use std::time::Duration;
     use futures_util::{SinkExt, StreamExt};
     use serde::de::DeserializeOwned;
     use serde::Serialize;
@@ -89,9 +89,7 @@ mod tests {
 
     use crate::backend::_test_utils::DummyJobProcessor;
     use crate::backend::music_gpt_chat::{AiChatEntry, ChatEntry, UserChatEntry};
-    use crate::backend::music_gpt_ws_handler::{
-        ChatRequest, GenerateAudioRequest, InboundMsg, OutboundMsg,
-    };
+    use crate::backend::music_gpt_ws_handler::{AbortGenerationRequest, ChatRequest, GenerateAudioRequest, InboundMsg, OutboundMsg};
 
     use super::*;