Struct Corpus 

pub struct Corpus { /* private fields */ }

A corpus of grounded documents for cross-document operations.

Enables inter-document coreference resolution and entity linking across multiple documents.

Implementations

impl Corpus

pub fn new() -> Corpus

Create a new empty corpus.

pub fn identities(&self) -> &HashMap<IdentityId, Identity>

Get all identities in the corpus.

pub fn get_identity(&self, id: IdentityId) -> Option<&Identity>

Get an identity by ID.

pub fn add_identity(&mut self, identity: Identity) -> IdentityId

Add an identity to the corpus and return its ID.

This method assigns the next available identity ID and inserts the identity. Used by coalescing operations to create cross-document identities.

pub fn next_identity_id(&self) -> IdentityId

Get the next identity ID that would be assigned.

This is used by coalescing operations to reserve identity IDs.

pub fn documents(&self) -> impl Iterator<Item = &GroundedDocument>

Get all documents in the corpus.

Returns an iterator over all documents.

pub fn get_document(&self, doc_id: &str) -> Option<&GroundedDocument>

Get a document by ID.

Returns None if the document doesn’t exist.

pub fn get_document_mut( &mut self, doc_id: &str, ) -> Option<&mut GroundedDocument>

Get a mutable reference to a document by ID.

Returns None if the document doesn’t exist.

pub fn add_document(&mut self, document: GroundedDocument) -> String

Add a document to the corpus.

If a document with the same ID already exists, it will be replaced. Returns the document ID.

pub fn link_track_to_kb( &mut self, track_ref: &TrackRef, kb_name: impl Into<String>, kb_id: impl Into<String>, canonical_name: impl Into<String>, ) -> Result<IdentityId, Error>

Link a track to a knowledge base entity.

This is the entity linking (NED) operation. It creates or updates an identity with KB information.

Parameters

• track_ref - Reference to the track to link
• kb_name - Knowledge base name (e.g., “wikidata”)
• kb_id - Knowledge base entity ID (e.g., “Q7186”)
• canonical_name - Canonical name from KB

Returns

The identity ID (new or existing), or an error if the track reference is invalid.

Errors

Returns Error::TrackRef if:

• The document ID doesn’t exist in the corpus
• The track ID doesn’t exist in the document

Trait Implementations

impl Clone for Corpus

fn clone(&self) -> Corpus

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Corpus

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

Formats the value using the given formatter.

impl Default for Corpus

fn default() -> Corpus

Returns the “default value” for a type.