content-extractor-rl 0.1.1

RL-based article extraction from HTML using Deep Q-Networks and heuristic fallback
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
128
129
130
131
132
133
134
135

//! Content Extractor RL - RL-based HTML article extraction library
//!
//! This library provides functionality for extracting article content from HTML
//! using reinforcement learning with fallback to heuristic-based extraction.

// ============================================================================
// FILE: crates/content-extractor-rl/src/lib.rs
// ============================================================================

pub mod config;
pub mod text_utils;
pub mod html_parser;
pub mod site_profile;
pub mod baseline_extractor;
pub mod evaluation;

pub use evaluation::{
    GroundTruthData, GroundTruthEvaluator, EvaluationMetrics,
    algorithm_comparison::{AlgorithmComparator, ComparisonReport},
};
pub mod models;
pub use models::ModelMetadata;

pub mod environment;
pub mod replay_buffer;
pub mod reward;
pub mod curriculum;

pub mod agents;
pub use agents::{AgentFactory, AlgorithmType, RLAgent};
pub use cli_utils::*;
pub use checkpoint::{Checkpoint, CheckpointManager};

pub mod training;
pub mod hyperparameter_tuner;
pub mod plotting;
pub mod device;

// Optional MLflow integration
#[cfg(feature = "mlflow-rs")]
pub mod mlflow;

// Re-exports
pub use config::Config;
pub use site_profile::{SiteProfile, SiteProfileMemory};
pub use baseline_extractor::BaselineExtractor;
pub use environment::ArticleExtractionEnvironment;
pub use training::{train_standard, train_with_improvements, TrainingMetrics};
pub use hyperparameter_tuner::{TPEOptimizer, Hyperparameters, HyperparameterSpace, TrialResult};

pub use plotting::{TrainingPlotter, PlotConfig};
pub use device::{get_device, cuda_is_available, get_device_info, print_device_info};

pub mod checkpoint;
pub mod cli_utils;



#[cfg(feature = "mlflow-rs")]
pub use mlflow::{MlflowTracker, create_tracker};

/// Result type for article extraction operations
pub type Result<T> = std::result::Result<T, ExtractionError>;

/// Errors that can occur during article extraction
#[derive(Debug)]
pub enum ExtractionError {
    IoError(std::io::Error),
    ParseError(String),
    NetworkError(String),
    ModelError(String),
    ExtractionFailed(String),
    CandleError(String),
    RuntimeError(String),
    MlflowError(String),
}

impl From<anyhow::Error> for ExtractionError {
    fn from(err: anyhow::Error) -> Self {
        ExtractionError::MlflowError(format!("{}", err))
    }
}

impl std::fmt::Display for ExtractionError {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            ExtractionError::IoError(e) => write!(f, "IO error: {}", e),
            ExtractionError::ParseError(e) => write!(f, "Parse error: {}", e),
            ExtractionError::NetworkError(e) => write!(f, "Network error: {}", e),
            ExtractionError::ModelError(e) => write!(f, "Model error: {}", e),
            ExtractionError::ExtractionFailed(e) => write!(f, "Extraction failed: {}", e),
            ExtractionError::CandleError(e) => write!(f, "Candle error: {}", e),
            ExtractionError::RuntimeError(e) => write!(f, "Runtime error: {}", e),
            ExtractionError::MlflowError(e) => write!(f, "MLFlow error: {}", e),
        }
    }
}

impl std::error::Error for ExtractionError {}

impl From<std::io::Error> for ExtractionError {
    fn from(err: std::io::Error) -> Self {
        ExtractionError::IoError(err)
    }
}

impl From<serde_json::Error> for ExtractionError {
    fn from(err: serde_json::Error) -> Self {
        ExtractionError::ParseError(err.to_string())
    }
}

impl From<candle_core::Error> for ExtractionError {
    fn from(err: candle_core::Error) -> Self {
        ExtractionError::CandleError(err.to_string())
    }
}

/// Extracted article result
#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
pub struct ExtractedArticle {
    pub url: String,
    pub title: Option<String>,
    pub date: Option<String>,
    pub content: String,
    pub quality_score: f32,
    pub method: String,
    pub xpath: Option<String>,
}

/// Batch extraction result
#[derive(Debug, serde::Serialize, serde::Deserialize)]
pub struct BatchExtractionResult {
    pub articles: Vec<ExtractedArticle>,
}