Module llm

Module llm 

Source
Expand description

LLM integration

Provides traits and implementations for:

  • Embedding generation via external services (vLLM, OpenAI, etc.)
  • Document metadata generation
  • Query parsing
  • Reranking

Structs§

ChatMessage
Chat message for completion requests
DocumentMetadata
Generated metadata result
ExpandedQuery
Expanded query variants
HttpEmbedder
Embedder that uses external HTTP service (vLLM, OpenAI, etc.)
HttpMetadataGenerator
Metadata generator using external HTTP LLM service
HttpQueryParser
Query parser using external HTTP LLM service
LlamaEmbedder
LLaMA-based embedder
LlamaMetadataGenerator
LLaMA-based metadata generator
MetadataContext
Context information for metadata generation
MetadataFilterHint
Metadata filter hint extracted from query
MetricsSnapshot
Snapshot of API metrics
ParsedQuery
Parsed query with extracted intent and filters
QueryParser
Query parser using local LLM
RerankDocument
Document for reranking
RerankResult
Reranking result
TemporalFilter
Temporal filter for time-based queries
VLLMClient
vLLM/OpenAI-compatible client

Enums§

SearchType
Search type recommendation

Constants§

DEFAULT_EMBED_MODEL
Default embedding model (nomic-embed-text or similar)
DEFAULT_METADATA_MODEL
Default metadata generation model

Traits§

Embedder
Embedding generation trait
LLMClient
Trait for LLM service clients
MetadataGenerator
Metadata generation trait
QueryExpander
Query expansion trait
Reranker
Document reranking trait
Tokenizer
Tokenization trait

Functions§

generate_metadata_with_llm
Helper to generate metadata using LLM client