Use new chatbot

cognitive-engineering-lab · Jul 11, 2024 · 980911a · 980911a
1 parent 9917002
commit 980911a
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 9 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/server/Cargo.toml b/crates/server/Cargo.toml
@@ -8,7 +8,4 @@ anyhow = "1.0.86"
 axum = { workspace = true }
 serde = { workspace = true }
 tokio = { workspace = true }
-chatbot = { path = "../chatbot" }
-
-[dev-dependencies]
-rand = { workspace = true }
+chatbot = { path = "../chatbot" }
diff --git a/crates/server/src/main.rs b/crates/server/src/main.rs
@@ -5,6 +5,7 @@ use axum::{
     Json, Router,
 };
 use serde::{Deserialize, Serialize};
+use tokio::join;
 
 async fn root() -> Html<&'static str> {
     Html(include_str!("../index.html"))
@@ -16,10 +17,11 @@ struct Chat {
 }
 
 async fn chat(Json(mut chat): Json<Chat>) -> Json<Chat> {
-    let responses = chatbot::query_chat(&chat.messages).await;
-    let random = chatbot::gen_random_number().await;
-    chat.messages
-        .push(responses[random % responses.len()].to_string());
+    let responses_fut = chatbot::query_chat(&chat.messages);
+    let random_fut = chatbot::gen_random_number();
+    let (mut responses, random) = join!(responses_fut, random_fut);
+    let response = responses.remove(random % responses.len());
+    chat.messages.push(response);
     Json(chat)
 }