entity

type Enricher struct {
	// contains filtered or unexported fields
}

type Entity struct {
	Text  string
	Label string
	Score float32
	Start int
	End   int
}

type EntityRecord struct {
	ID           string
	Name         string
	Label        string
	MentionCount int
}

type ExtractOptions struct {
	EntityLabels   []string
	RelationLabels []string
}

type Extraction struct {
	Entities  []Entity
	Relations []Relation
}

type Extractor interface {
	Extract(ctx context.Context, texts []string, opts ExtractOptions) ([]Extraction, error)
}

type Option func(*Enricher)

type Relation struct {
	Head  Entity
	Label string
	Score float32
	Tail  Entity
}

type RelationRecord struct {
	ID           string
	Label        string
	HeadEntity   string
	TailEntity   string
	HeadName     string
	TailName     string
	HeadLabel    string
	TailLabel    string
	Weight       float64
	MentionCount int
}

type Result struct {
	EntityRecords       map[string]EntityRecord
	SectionEntityKeys   map[string][]string
	RelationRecords     map[string]RelationRecord
	SectionRelationKeys map[string][]string
}