alith_client/workflows/classify/
mod.rs

1use alith_interface::{llms::LLMBackend, requests::completion::CompletionRequest};
2use std::sync::Arc;
3use subject_of_text::ClassifySubjectOfText;
4
5pub mod hierarchical_classification;
6pub mod subject_of_text;
7
8pub struct Classify {
9    backend: Arc<LLMBackend>,
10}
11
12impl Classify {
13    pub fn new(backend: Arc<LLMBackend>) -> Self {
14        Self { backend }
15    }
16
17    pub fn subject_of_text<T: AsRef<str>>(self, content: T) -> ClassifySubjectOfText {
18        ClassifySubjectOfText::new(CompletionRequest::new(self.backend), content)
19    }
20}