Add finbert sentiment analysis

Signed-off-by: Nikolaos Karaolidis <nick@karaolidis.com>

src/threads/data/websocket.rs

@@ -2,7 +2,7 @@ use super::{backfill, Guard, ThreadType};
 use crate::{
     config::Config,
     database,
-    types::{alpaca::websocket, Bar, News, Subset},
+    types::{alpaca::websocket, news::Prediction, Bar, News, Subset},
 };
 use futures_util::{
     stream::{SplitSink, SplitStream},
@@ -19,6 +19,7 @@ use tokio::{
     net::TcpStream,
     spawn,
     sync::{mpsc, Mutex, RwLock},
+    task::spawn_blocking,
 };
 use tokio_tungstenite::{tungstenite, MaybeTlsStream, WebSocketStream};
 
@@ -93,6 +94,7 @@ async fn handle_websocket_message(
 }
 
 #[allow(clippy::significant_drop_tightening)]
+#[allow(clippy::too_many_lines)]
 async fn handle_parsed_websocket_message(
     app_config: Arc<Config>,
     thread_type: ThreadType,
@@ -195,6 +197,28 @@ async fn handle_parsed_websocket_message(
                 "{:?} - Received news for {:?}: {}.",
                 thread_type, news.symbols, news.time_created
             );
+
+            let app_config_clone = app_config.clone();
+            let input = format!("{}\n\n{}", news.headline, news.content);
+
+            let prediction = spawn_blocking(move || {
+                app_config_clone
+                    .sequence_classifier
+                    .lock()
+                    .unwrap()
+                    .predict(vec![input.as_str()])
+                    .into_iter()
+                    .map(|label| Prediction::try_from(label).unwrap())
+                    .collect::<Vec<_>>()[0]
+            })
+            .await
+            .unwrap();
+
+            let news = News {
+                sentiment: prediction.sentiment,
+                confidence: prediction.confidence,
+                ..news
+            };
             database::news::upsert(&app_config.clickhouse_client, &news).await;
         }
         websocket::incoming::Message::Success(_) => {}