extraction

package

v1.6.0 Latest Latest Go to latest Published: Feb 19, 2026 License: MIT Imports: 5 Imported by: 0

Details

Package extraction provides entity and relationship extraction from text

This section is empty.

This section is empty.

This section is empty.

type Entity struct {
	Name        string `json:"name"`
	Type        string `json:"type"`
	Description string `json:"description"`
}

Entity represents a named entity extracted from text

type EntityExtractor struct {
	LLM llm.LLMClient
}

EntityExtractor extracts entities from text using an LLM

func NewEntityExtractor(llmClient llm.LLMClient) *EntityExtractor

NewEntityExtractor creates a new entity extractor

func (e *EntityExtractor) Extract(ctx context.Context, text string) ([]Entity, error)

Extract extracts entities from the given text

type RelationExtractor struct {
	LLM llm.LLMClient
}

RelationExtractor extracts relationships between entities from text using an LLM

func NewRelationExtractor(llmClient llm.LLMClient) *RelationExtractor

NewRelationExtractor creates a new relation extractor

func (r *RelationExtractor) Extract(ctx context.Context, text string, entities []Entity) ([]Triplet, error)

Extract extracts relationships from the given text using the provided entities

type Triplet struct {
	Subject  string `json:"subject"`
	Relation string `json:"relation"`
	Object   string `json:"object"`
}

Triplet represents a relationship between two entities