3d3bd3c9f9
Files sent as m.file Matrix messages are now downloaded, MIME-detected, and forwarded to LLM providers alongside the conversation context, similar to how m.image is already handled. - OpenAI provider: sends files inline as base64 data URLs - Anthropic provider: skips files with a warning (library limitation) - OpenAI-compat provider: skips files with a warning (library limitation) Controller routing respects the existing prefix requirement setting. MIME detection expanded to cover PDF, text, code, and document formats. Docs updated to reflect file support and known limitations. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
65 lines
2.0 KiB
Rust
65 lines
2.0 KiB
Rust
use chrono::{DateTime, Utc};
|
|
use regex::Regex;
|
|
|
|
use mxlink::matrix_sdk::ruma::OwnedUserId;
|
|
use mxlink::matrix_sdk::ruma::events::room::message::{
|
|
FileMessageEventContent, ImageMessageEventContent,
|
|
};
|
|
use mxlink::mime::Mime;
|
|
|
|
#[derive(Clone)]
|
|
pub struct MatrixMessage {
|
|
pub sender_id: OwnedUserId,
|
|
pub content: MatrixMessageContent,
|
|
pub mentioned_users: Vec<OwnedUserId>,
|
|
pub timestamp: DateTime<Utc>,
|
|
}
|
|
|
|
#[derive(Clone)]
|
|
pub enum MatrixMessageContent {
|
|
Text(String),
|
|
Notice(String),
|
|
Image(ImageMessageEventContent, Mime, Vec<u8>),
|
|
File(FileMessageEventContent, Mime, Vec<u8>),
|
|
}
|
|
|
|
#[derive(Clone)]
|
|
pub struct MatrixMessageProcessingParams {
|
|
pub(crate) bot_user_id: OwnedUserId,
|
|
|
|
/// The prefixes that will be stripped when processing the messages in the context (thread or reply chain),
|
|
/// which are found to be mentioning the bot user (`bot_user_id`).
|
|
pub(crate) bot_user_prefixes_to_strip: Vec<String>,
|
|
|
|
/// The prefixes that will be stripped when processing the 1st message in the context (thread or reply chain).
|
|
pub(crate) first_message_prefixes_to_strip: Vec<String>,
|
|
|
|
/// A list of users whose messages are allowed.
|
|
/// If None, all messages are allowed.
|
|
/// If Some, only messages from the allowed users (and the bot itself, `bot_user_id`) are allowed.
|
|
pub(crate) allowed_users: Option<Vec<Regex>>,
|
|
}
|
|
|
|
impl MatrixMessageProcessingParams {
|
|
pub fn new(bot_user_id: OwnedUserId, allowed_users: Option<Vec<Regex>>) -> Self {
|
|
Self {
|
|
bot_user_id,
|
|
bot_user_prefixes_to_strip: vec![],
|
|
|
|
first_message_prefixes_to_strip: vec![],
|
|
|
|
allowed_users,
|
|
}
|
|
}
|
|
|
|
pub fn with_bot_user_prefixes_to_strip(mut self, value: Vec<String>) -> Self {
|
|
self.bot_user_prefixes_to_strip = value;
|
|
self
|
|
}
|
|
|
|
pub fn with_first_message_prefixes_to_strip(mut self, value: Vec<String>) -> Self {
|
|
self.first_message_prefixes_to_strip = value;
|
|
self
|
|
}
|
|
}
|