tuitbot-core 0.1.47

Core library for Tuitbot autonomous X growth assistant
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127

//! Thread plan step: generate and analyze a multi-tweet thread via LLM.
//!
//! Produces a structured thread plan with hook analysis and performance estimates.

use std::sync::Arc;

use crate::config::Config;
use crate::content::frameworks::ThreadStructure;
use crate::llm::LlmProvider;

use super::{make_content_gen, WorkflowError};

/// Input for the thread plan step.
#[derive(Debug, Clone)]
pub struct ThreadPlanInput {
    /// The topic to write a thread about.
    pub topic: String,
    /// Optional objective (e.g., "establish expertise").
    pub objective: Option<String>,
    /// Optional target audience description.
    pub target_audience: Option<String>,
    /// Optional structure override (e.g., "transformation", "framework").
    pub structure: Option<String>,
}

/// Output from the thread plan step.
#[derive(Debug, Clone, serde::Serialize)]
pub struct ThreadPlanOutput {
    /// The generated tweets in thread order.
    pub thread_tweets: Vec<String>,
    /// Number of tweets in the thread.
    pub tweet_count: usize,
    /// The structure that was used.
    pub structure_used: String,
    /// Hook analysis for the first tweet.
    pub hook_type: String,
    /// First tweet preview.
    pub first_tweet_preview: String,
    /// Estimated performance based on topic relevance.
    pub estimated_performance: String,
    /// Objective alignment description.
    pub objective_alignment: String,
    /// Target audience description.
    pub target_audience: String,
    /// Topic relevance assessment.
    pub topic_relevance: String,
}

/// Parse a structure string into a `ThreadStructure`.
fn parse_structure(s: &str) -> Option<ThreadStructure> {
    match s.to_lowercase().as_str() {
        "transformation" => Some(ThreadStructure::Transformation),
        "framework" => Some(ThreadStructure::Framework),
        "mistakes" => Some(ThreadStructure::Mistakes),
        "analysis" => Some(ThreadStructure::Analysis),
        _ => None,
    }
}

/// Analyze the hook type of the first tweet.
fn analyze_hook(first_tweet: &str) -> &'static str {
    let trimmed = first_tweet.trim();
    if trimmed.ends_with('?') {
        "question"
    } else if trimmed.starts_with("Most people")
        || trimmed.starts_with("Everyone")
        || trimmed.starts_with("most people")
    {
        "contrarian"
    } else if trimmed.starts_with("I ") || trimmed.starts_with("I'") {
        "story"
    } else {
        "statement"
    }
}

/// Execute the thread plan step.
pub async fn execute(
    llm: &Arc<dyn LlmProvider>,
    config: &Config,
    input: ThreadPlanInput,
) -> Result<ThreadPlanOutput, WorkflowError> {
    let structure_override = input.structure.as_deref().and_then(parse_structure);

    let gen = make_content_gen(llm, &config.business);

    let thread = gen
        .generate_thread_with_structure(&input.topic, structure_override)
        .await?;

    let tweet_count = thread.tweets.len();
    let hook_type = thread
        .tweets
        .first()
        .map(|t| analyze_hook(t))
        .unwrap_or("unknown");

    // Relevance heuristic: check if topic overlaps with configured industry topics
    let topic_lower = input.topic.to_lowercase();
    let relevance = config.business.effective_industry_topics().iter().any(|t| {
        topic_lower.contains(&t.to_lowercase()) || t.to_lowercase().contains(&topic_lower)
    });

    let estimated_performance = if relevance { "high" } else { "medium" };
    let structure_used = input.structure.as_deref().unwrap_or("auto_selected");

    Ok(ThreadPlanOutput {
        first_tweet_preview: thread.tweets.first().cloned().unwrap_or_default(),
        thread_tweets: thread.tweets,
        tweet_count,
        structure_used: structure_used.to_string(),
        hook_type: hook_type.to_string(),
        estimated_performance: estimated_performance.to_string(),
        objective_alignment: input
            .objective
            .unwrap_or_else(|| "general engagement".to_string()),
        target_audience: input
            .target_audience
            .unwrap_or_else(|| "general".to_string()),
        topic_relevance: if relevance {
            "matches_industry_topics"
        } else {
            "novel_topic"
        }
        .to_string(),
    })
}