GeneralBots/botserver
2
1
Fork 0
Code Issues Pull requests Projects Releases 323 Packages Wiki Activity Actions

- Answering in Web using streaming.

Browse source
This commit is contained in:
Rodrigo Rodriguez (Pragmatismo) 2025-10-12 15:06:16 -03:00
parent efa8ff5c80
commit 386b55bf34
3 changed files with 10 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
src/bot/mod.rs
View file

@ -144,7 +144,7 @@ impl BotOrchestrator {
session_manager.save_message( session_manager.save_message(
session.id, session.id,
user_id, user_id,
"user", 1,
&message.content, &message.content,
message.message_type, message.message_type,
)?; )?;
@ -154,7 +154,7 @@ impl BotOrchestrator {
{ {
let mut session_manager = self.session_manager.lock().await; let mut session_manager = self.session_manager.lock().await;
session_manager.save_message(session.id, user_id, "assistant", &response_content, 1)?; session_manager.save_message(session.id, user_id, 2, &response_content, 1)?;
} }
let bot_response = BotResponse { let bot_response = BotResponse {
@ -244,7 +244,7 @@ impl BotOrchestrator {
sm.save_message( sm.save_message(
session.id, session.id,
user_id, user_id,
"user", 1,
&message.content, &message.content,
message.message_type, message.message_type,
)?; )?;
@ -301,7 +301,7 @@ impl BotOrchestrator {
// Save the complete assistant reply. // Save the complete assistant reply.
{ {
let mut sm = self.session_manager.lock().await; let mut sm = self.session_manager.lock().await;
sm.save_message(session.id, user_id, "assistant", &full_response, 1)?; sm.save_message(session.id, user_id, 2, &full_response, 1)?;
} }
// Notify the client that the stream is finished. // Notify the client that the stream is finished.
@ -363,7 +363,7 @@ impl BotOrchestrator {
session_manager.save_message( session_manager.save_message(
session.id, session.id,
user_id, user_id,
"user", 1,
&message.content, &message.content,
message.message_type, message.message_type,
)?; )?;
@ -412,13 +412,7 @@ impl BotOrchestrator {
{ {
Ok(tool_result) => { Ok(tool_result) => {
let mut session_manager = self.session_manager.lock().await; let mut session_manager = self.session_manager.lock().await;
session_manager.save_message( session_manager.save_message(session.id, user_id, 2, &tool_result.output, 2)?;
session.id,
user_id,
"assistant",
&tool_result.output,
2,
)?;
tool_result.output tool_result.output
} }
@ -443,7 +437,7 @@ impl BotOrchestrator {
{ {
let mut session_manager = self.session_manager.lock().await; let mut session_manager = self.session_manager.lock().await;
session_manager.save_message(session.id, user_id, "assistant", &response, 1)?; session_manager.save_message(session.id, user_id, 2, &response, 1)?;
} }
let bot_response = BotResponse { let bot_response = BotResponse {

4
src/session/mod.rs
View file

@ -130,7 +130,7 @@ impl SessionManager {
&mut self, &mut self,
sess_id: Uuid, sess_id: Uuid,
uid: Uuid, uid: Uuid,
_role_str: &str, ro: i32,
content: &str, content: &str,
msg_type: i32, msg_type: i32,
) -> Result<(), Box<dyn Error + Send + Sync>> { ) -> Result<(), Box<dyn Error + Send + Sync>> {
@ -146,7 +146,7 @@ impl SessionManager {
id.eq(Uuid::new_v4()), id.eq(Uuid::new_v4()),
session_id.eq(sess_id), session_id.eq(sess_id),
user_id.eq(uid), user_id.eq(uid),
// role.eq(role_str), role.eq(ro),
content_encrypted.eq(content), content_encrypted.eq(content),
message_type.eq(msg_type), message_type.eq(msg_type),
message_index.eq(next_index), message_index.eq(next_index),

2
src/shared/models.rs
View file

@ -188,7 +188,7 @@ diesel::table! {
id -> Uuid, id -> Uuid,
session_id -> Uuid, session_id -> Uuid,
user_id -> Uuid, user_id -> Uuid,
role -> Text, role -> Int4,
content_encrypted -> Text, content_encrypted -> Text,
message_type -> Int4, message_type -> Int4,
message_index -> Int8, message_index -> Int8,