rssume 0.2.0

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
26

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

const SYSTEM_PROMPT: &str = r#"You are a skilled content summarizer. Write a concise one-sentence summary of the following article in Chinese.
Capture the key point. Output only the summary, nothing else."#;

pub async fn summarize(
    config: &LlmProviderConfig,
    title: &str,
    content: &str,
    on_token: impl FnMut(&str),
) -> Result<StreamResult, crate::error::AppError> {
    let text = if content.len() > 4000 {
        &content[..content.floor_char_boundary(4000)]
    } else {
        content
    };
    let prompt = format!("Title: {}\n\nContent:\n{}", title, 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
}