Function generate_title

src-tauri/src/services/chat_service.rs:865–929

Generate a short title for a conversation using the same provider/model.

(
    db: &SqlitePool,
    session_id: &str,
    provider_id: Option<&str>,
    model_id: Option<&str>,
)

/// Generate a short title for a conversation using the same provider/model.
pub async fn generate_title(
    db: &SqlitePool,
    session_id: &str,
    provider_id: Option<&str>,
    model_id: Option<&str>,
) -> AppResult<String> {
    let provider = provider_service::get_provider_for_chat(db, provider_id).await?;
    let model = model_id.unwrap_or(&provider.model);
    let history = get_messages(db, session_id).await?;

    if history.is_empty() {
        return Ok("New Chat".to_string());
    }

    // Build a compact summary of the conversation (max first 2 exchanges)
    let snippet: Vec<Value> = history
        .iter()
        .filter(|m| m.role != "system")
        .take(4)
        .map(|m| {
            let content = if m.content.len() > 200 {
                format!("{}…", &m.content[..200])
            } else {
                m.content.clone()
            };
            serde_json::json!({ "role": m.role, "content": content })
        })
        .collect();

    let mut messages = vec![serde_json::json!({
        "role": "system",
        "content": "Generate a short title (2-5 words) for this conversation. Reply with ONLY the title, nothing else. No quotes, no punctuation at the end. Examples: 'React Auth Setup', 'Greeting', 'Python Bug Fix', 'Database Migration Help'."
    })];
    messages.extend(snippet);
    messages.push(serde_json::json!({
        "role": "user",
        "content": "Generate a short title for the conversation above."
    }));

    let title = if provider.uses_anthropic_format() {
        generate_title_anthropic(&provider, model, &messages).await?
    } else {
        generate_title_openai(&provider, model, &messages).await?
    };

    // Clean up: remove quotes, trim, cap length
    let cleaned = title
        .trim()
        .trim_matches('"')
        .trim_matches('\'')
        .trim_end_matches('.')
        .trim();

    let final_title = if cleaned.is_empty() {
        "New Chat".to_string()
    } else if cleaned.len() > 50 {
        let cut = &cleaned[..50];
        let last_space = cut.rfind(' ').unwrap_or(50);