Struct EntityExtractor 

pub struct EntityExtractor<P: LLMProvider> { /* private fields */ }

Entity extractor using LLM.

TigerStyle: Generic over provider for sim/production flexibility.

Example

use umi_memory::extraction::{EntityExtractor, ExtractionOptions};
use umi_memory::llm::SimLLMProvider;

#[tokio::main]
async fn main() {
    // Simulation provider for testing
    let provider = SimLLMProvider::with_seed(42);
    let extractor = EntityExtractor::new(provider);

    let result = extractor
        .extract("Alice works at Acme Corp", ExtractionOptions::default())
        .await
        .unwrap();

    // Result contains entities and relations
    assert!(!result.is_empty());
}

Implementations

impl<P: LLMProvider> EntityExtractor<P>

pub fn new(provider: P) -> Self

Create a new entity extractor with the given LLM provider.

pub async fn extract( &self, text: &str, options: ExtractionOptions, ) -> Result<ExtractionResult, ExtractionError>

Extract entities and relations from text.

Arguments

• text - Text to extract from
• options - Extraction options (existing entities, min confidence)

Returns

ExtractionResult with entities, relations, and raw text.

Errors

Returns ExtractionError if text is empty or too long.

Graceful Degradation

If the LLM fails or returns invalid JSON, a fallback “note” entity is created from the input text. This ensures extraction never fails due to LLM issues.

pub async fn extract_entities_only( &self, text: &str, ) -> Result<Vec<ExtractedEntity>, ExtractionError>

Extract only entities (convenience method).

Arguments

• text - Text to extract from

Returns

Vector of extracted entities.

pub fn provider(&self) -> &P

Get a reference to the underlying provider.

Trait Implementations

impl<P: Debug + LLMProvider> Debug for EntityExtractor<P>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.