mod adaptive;
mod artifact;
mod generator;
mod lm;
mod rerank;
mod scorer;
#[cfg(feature = "wasm")]
pub(crate) use adaptive::{
repetition_observation_for_raw_token, session_repetition_guard_pool_limit,
AutosuggestRepetitionHistory,
};
pub use adaptive::{
AutosuggestSession, PersonalAutosuggest, PersonalAutosuggestConfig, PersonalAutosuggestContext,
PersonalAutosuggestError, PersonalAutosuggestSnapshotError, PersonalAutosuggestSuggestion,
PersonalAutosuggestTextSuggestion, DEFAULT_PERSONAL_AUTOSUGGEST_ENTRIES,
DEFAULT_PERSONAL_AUTOSUGGEST_MIN_COUNT,
};
pub use artifact::AutosuggestArtifactError;
pub use generator::{
materialize_generated_candidates_into, materialize_merged_candidates_into,
materialize_scored_union_candidates_into, merge_static_and_generated_candidates_into,
scored_union_static_and_generated_candidates_into, AutosuggestGeneratedCandidateId,
AutosuggestGeneratorCompatibility, AutosuggestGeneratorFile, AutosuggestGeneratorHandoff,
AutosuggestGeneratorHandoffError, AutosuggestGeneratorManifest,
AutosuggestGeneratorManifestError, AutosuggestGeneratorMergeOptions,
AutosuggestGeneratorMergedQuality, AutosuggestGeneratorModel, AutosuggestGeneratorNgram,
AutosuggestGeneratorQuality, AutosuggestGeneratorQualityMetrics,
AutosuggestGeneratorRuntimeContract, AutosuggestGeneratorScoredUnionPolicy,
AutosuggestGeneratorScoredUnionQuality, AutosuggestGeneratorSession,
AutosuggestMaterializedGeneratedCandidate, AutosuggestMaterializedMergedCandidate,
AutosuggestMaterializedScoredUnionCandidate, AutosuggestMergedCandidateId,
AutosuggestMergedCandidateSource, AutosuggestScoredUnionCandidateId,
AUTOSUGGEST_GENERATOR_COREML_INPUT_DTYPE, AUTOSUGGEST_GENERATOR_MANIFEST_VERSION,
AUTOSUGGEST_GENERATOR_ONNX_INPUT_DTYPE, AUTOSUGGEST_GENERATOR_PACKAGE_KIND,
AUTOSUGGEST_GENERATOR_RUNTIME_ROLE, AUTOSUGGEST_GENERATOR_SCORE_DTYPE,
AUTOSUGGEST_GENERATOR_TOKEN_ID_DTYPE, DEFAULT_AUTOSUGGEST_GENERATOR_LOCKED_STATIC_PREFIX,
};
#[cfg(not(target_arch = "wasm32"))]
pub use generator::{AutosuggestGeneratorAsset, AutosuggestGeneratorAssetReport};
#[cfg(not(target_arch = "wasm32"))]
pub use lm::AutosuggestLoadError;
pub use lm::{
scorer_candidate_i32s_for_candidates_into, scorer_candidate_ids_for_candidates_into,
AutosuggestCandidate, AutosuggestCandidateId, AutosuggestCandidatePrior, AutosuggestContext,
AutosuggestContextPriorMetadata, AutosuggestContextPriorOptions, AutosuggestLm,
AutosuggestMetadata, AutosuggestModelInfo, AutosuggestOptions, AutosuggestRerankInputMetadata,
AutosuggestResult, AutosuggestSource, AUTOSUGGEST_ARTIFACT_KIND, AUTOSUGGEST_BOS_I32,
AUTOSUGGEST_BOS_ID, AUTOSUGGEST_PAD_I32, AUTOSUGGEST_PAD_ID, AUTOSUGGEST_UNK_I32,
AUTOSUGGEST_UNK_ID, DEFAULT_AUTOSUGGEST_CANDIDATES, MAX_AUTOSUGGEST_CONTEXT_TOKENS,
MAX_AUTOSUGGEST_RERANK_CONTEXT_TOKENS,
};
pub use rerank::{
rerank_candidate_ids_with_fixed_scores_into, rerank_candidate_ids_with_scores_into,
AutosuggestRerankError, AutosuggestRerankOptions, AutosuggestScoredCandidateId,
DEFAULT_AUTOSUGGEST_RERANK_LOCKED_PREFIX, DEFAULT_AUTOSUGGEST_RERANK_RANK_PENALTY,
};
pub use scorer::{
AutosuggestMaterializedScoredCandidate, AutosuggestScorerCompatibility, AutosuggestScorerFile,
AutosuggestScorerHandoff, AutosuggestScorerHandoffError, AutosuggestScorerManifest,
AutosuggestScorerManifestError, AutosuggestScorerModel, AutosuggestScorerNgram,
AutosuggestScorerQuality, AutosuggestScorerQualityMetrics, AutosuggestScorerRuntimeContract,
AutosuggestScorerSession, AUTOSUGGEST_SCORER_COREML_INPUT_DTYPE,
AUTOSUGGEST_SCORER_MANIFEST_VERSION, AUTOSUGGEST_SCORER_ONNX_INPUT_DTYPE,
AUTOSUGGEST_SCORER_PACKAGE_KIND, AUTOSUGGEST_SCORER_RUNTIME_ROLE,
AUTOSUGGEST_SCORER_SCORE_DTYPE, AUTOSUGGEST_SCORER_TOKEN_ID_DTYPE,
};
#[cfg(not(target_arch = "wasm32"))]
pub use scorer::{AutosuggestScorerAsset, AutosuggestScorerAssetReport};