Skip to main content

Module entity_extraction

Module entity_extraction 

Source
Expand description

Entity extraction service for extracting entities from text using LLM.

Provides LLM-powered entity extraction with graceful degradation when LLM is unavailable.

ยงExample

โ“˜
use subcog::services::EntityExtractorService;
use subcog::llm::AnthropicClient;
use subcog::models::Domain;

let llm = AnthropicClient::new();
let service = EntityExtractorService::new(Box::new(llm), Domain::for_user());

let result = service.extract("Alice from Acme Corp uses Rust")?;
println!("Extracted {} entities", result.entities.len());

Structsยง

EntityExtractorService
Service for extracting entities from text content.
ExtractedEntity
An entity extracted from text.
ExtractedRelationship
A relationship extracted from text.
ExtractionResult
Result of entity extraction from text.
InferenceResult
Result of relationship inference between entities.
InferredRelationship
A relationship inferred between existing entities.
LlmExtractionResponse ๐Ÿ”’
LLM response structure for entity extraction.
LlmInferenceResponse ๐Ÿ”’
LLM response structure for relationship inference.

Staticsยง

TECH_PATTERNS ๐Ÿ”’
Technology patterns for fallback entity extraction.

Functionsยง

default_confidence ๐Ÿ”’
parse_entity_type ๐Ÿ”’
Parses entity type string to EntityType enum.
parse_relationship_type ๐Ÿ”’
Parses relationship type string to RelationshipType enum.