agent-discord-rs 0.2.7

A high-performance Discord Bot daemon supporting multiple AI agents (pi, opencode).
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81

use super::SlashCommand;
use async_trait::async_trait;
use serenity::all::{
    CommandInteraction, CommandOptionType, Context, CreateCommandOption, EditInteractionResponse,
};

pub struct ThinkingCommand;

#[async_trait]
impl SlashCommand for ThinkingCommand {
    fn name(&self) -> &'static str {
        "thinking"
    }

    fn description(&self, i18n: &crate::i18n::I18n) -> String {
        i18n.get("cmd_thinking_desc")
    }

    fn options(&self, i18n: &crate::i18n::I18n) -> Vec<CreateCommandOption> {
        vec![CreateCommandOption::new(
            CommandOptionType::String,
            "level",
            i18n.get("cmd_thinking_opt_level"),
        )
        .required(true)
        .add_string_choice(i18n.get("thinking_level_off"), "off")
        .add_string_choice(i18n.get("thinking_level_minimal"), "minimal")
        .add_string_choice(i18n.get("thinking_level_low"), "low")
        .add_string_choice(i18n.get("thinking_level_medium"), "medium")
        .add_string_choice(i18n.get("thinking_level_high"), "high")
        .add_string_choice(i18n.get("thinking_level_xhigh"), "xhigh")]
    }

    async fn execute(
        &self,
        ctx: &Context,
        command: &CommandInteraction,
        state: &crate::AppState,
    ) -> anyhow::Result<()> {
        command.defer_ephemeral(&ctx.http).await?;

        let level = command
            .data
            .options
            .iter()
            .find(|o| o.name == "level")
            .and_then(|o| o.value.as_str())
            .unwrap_or("medium");

        let channel_id_u64 = command.channel_id.get();
        let channel_id_str = channel_id_u64.to_string();
        let channel_config = crate::commands::agent::ChannelConfig::load()
            .await
            .unwrap_or_default();
        let agent_type = channel_config.get_agent_type(&channel_id_str);

        let (agent, _) = state
            .session_manager
            .get_or_create_session(channel_id_u64, agent_type, &state.backend_manager)
            .await?;

        let i18n = state.i18n.read().await;
        match agent.set_thinking_level(level).await {
            Ok(_) => {
                let msg = i18n.get_args("thinking_set", &[level.to_string()]);
                command
                    .edit_response(&ctx.http, EditInteractionResponse::new().content(msg))
                    .await?;
            }
            Err(e) => {
                let msg = i18n.get_args("thinking_failed", &[e.to_string()]);
                command
                    .edit_response(&ctx.http, EditInteractionResponse::new().content(msg))
                    .await?;
            }
        }
        drop(i18n);

        Ok(())
    }
}