Rodrigo Rodriguez (Pragmatismo)
26c09b5695
- Added LLM server readiness check in AutomationService before starting tasks - Renamed `user` parameter to `user_session` in execute_talk for clarity - Updated BotResponse fields to use user_session data instead of hardcoded values - Improved Redis key generation in execute_talk to use user_session fields - Removed commented Redis code in set_current_context_keyword The changes ensure proper initialization of automation tasks by checking LLM server availability first, and improve code clarity by using more descriptive variable names for user session data.
59 lines
1.5 KiB
Rust
59 lines
1.5 KiB
Rust
use diesel::prelude::*;
|
|
use log::info;
|
|
use serde_json::{json, Value};
|
|
use uuid::Uuid;
|
|
|
|
use crate::shared::models::TriggerKind;
|
|
|
|
|
|
pub fn execute_set_schedule(
|
|
conn: &mut diesel::PgConnection,
|
|
cron: &str,
|
|
script_name: &str,
|
|
bot_uuid: Uuid,
|
|
) -> Result<Value, Box<dyn std::error::Error>> {
|
|
info!(
|
|
"Scheduling SET SCHEDULE cron: {}, script: {}, bot_id: {:?}",
|
|
cron, script_name, bot_uuid
|
|
);
|
|
|
|
// First check if bot exists
|
|
use crate::shared::models::bots::dsl::bots;
|
|
let bot_exists: bool = diesel::select(diesel::dsl::exists(
|
|
bots.filter(crate::shared::models::bots::dsl::id.eq(bot_uuid))
|
|
))
|
|
.get_result(conn)?;
|
|
|
|
if !bot_exists {
|
|
return Err(format!("Bot with id {} does not exist", bot_uuid).into());
|
|
}
|
|
|
|
use crate::shared::models::system_automations::dsl::*;
|
|
|
|
let new_automation = (
|
|
bot_id.eq(bot_uuid),
|
|
kind.eq(TriggerKind::Scheduled as i32),
|
|
schedule.eq(cron),
|
|
param.eq(script_name),
|
|
is_active.eq(true),
|
|
);
|
|
|
|
let result = diesel::insert_into(system_automations)
|
|
.values(&new_automation)
|
|
.on_conflict((bot_id, kind, param))
|
|
.do_update()
|
|
.set((
|
|
schedule.eq(cron),
|
|
is_active.eq(true),
|
|
last_triggered.eq(None::<chrono::DateTime<chrono::Utc>>),
|
|
))
|
|
.execute(&mut *conn)?;
|
|
|
|
Ok(json!({
|
|
"command": "set_schedule",
|
|
"schedule": cron,
|
|
"script": script_name,
|
|
"bot_id": bot_uuid.to_string(),
|
|
"rows_affected": result
|
|
}))
|
|
}
|