rssume 0.2.1

RSS middleware with AI-powered translation and summarization
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

use super::{StreamResult, chat_stream};
use crate::config::LlmProviderConfig;

const SYSTEM_PROMPT: &str = r#"You are a professional translator. Translate the following text accurately into the target language.
Preserve all formatting, HTML tags, code blocks, and technical terms.
Only output the translated text, nothing else."#;

pub async fn translate(
    config: &LlmProviderConfig,
    text: &str,
    target_lang: &str,
    on_token: impl FnMut(&str),
) -> Result<StreamResult, crate::error::AppError> {
    let prompt = format!(
        "Translate the following text to {}:\n\n{}",
        target_lang, text
    );
    let append = config.prompt_append.clone().unwrap_or_default();
    let full = if append.is_empty() {
        prompt
    } else {
        format!("{}\n{}", prompt, append)
    };
    chat_stream(config, SYSTEM_PROMPT, &full, on_token).await
}