lariv-rs 0.1.0

Compile-time plugin web application framework built on Axum, SeaORM, Maud, and HTMX
Documentation 
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

//! Assistant plugin configuration (`[llm_assistant]` in TOML).
//!
//! The Gemini API key, selected chat model, and Google CSE credentials are stored in
//! [`crate::plugins::llm_assistant::preferences`] (DB). [`LlmAssistantConfig::chat_model`]
//! is the default when preferences have no model yet.

use serde::Deserialize;

use crate::config::ConfigSection;

/// Config HList tag for [`LlmAssistantConfig`].
pub struct LlmAssistantConfigTag;

impl ConfigSection for LlmAssistantConfigTag {
    const KEY: Option<&'static str> = Some("llm_assistant");
}

pub const DEFAULT_CHAT_MODEL: &str = "gemini-2.5-flash";

/// Hard-coded app limits.
pub const CHAT_MAX_OUTPUT_TOKENS: i32 = 4096;
pub const ASSISTANT_TOOL_ROUNDS: i32 = 128;
pub const GOOGLE_SEARCH_RESULT_LIMIT_CAP: i32 = 20;
pub const WEBPAGE_TEXT_CHAR_LIMIT: usize = 50_000;

/// Max attachment parts kept per inbound email.
pub const EMAIL_MAX_ATTACHMENTS: usize = 10;
/// Max bytes per attachment part.
pub const EMAIL_MAX_ATTACHMENT_BYTES: usize = 10 * 1024 * 1024;
/// Max total attachment bytes per inbound email.
pub const EMAIL_MAX_TOTAL_ATTACHMENT_BYTES: usize = 25 * 1024 * 1024;

#[derive(Debug, Clone, Deserialize)]
pub struct LlmAssistantConfig {
    #[serde(default = "default_chat_model", rename = "chatModel")]
    pub chat_model: String,
}

fn default_chat_model() -> String {
    DEFAULT_CHAT_MODEL.into()
}

impl Default for LlmAssistantConfig {
    fn default() -> Self {
        Self {
            chat_model: default_chat_model(),
        }
    }
}