Enum EmbeddingTask 

pub enum EmbeddingTask {
    NL2Code,
    Code2Code,
    Code2NL,
    Code2Completion,
    QA,
}

Embedding task types for Jina Code Embeddings 1.5B.

Each task type has separate instruction prefixes for queries and passages, enabling asymmetric retrieval (queries and documents are embedded differently).

Example

use aurora_semantic::{JinaCodeEmbedder, EmbeddingTask};

// For indexing code (use passage prefix)
let code_embedder = JinaCodeEmbedder::from_directory("./models/jina-code-1.5b")?
    .with_task(EmbeddingTask::NL2Code);
let code_embedding = code_embedder.embed_passage("fn parse_json(s: &str) -> Value { }")?;

// For search queries (use query prefix)
let query_embedding = code_embedder.embed_query("function to parse JSON")?;

Variants

NL2Code

Natural language to code retrieval (default). Use for: finding code snippets from natural language queries.

Code2Code

Code to code retrieval. Use for: finding similar code snippets.

Code2NL

Code to natural language. Use for: finding comments/documentation for code.

Code2Completion

Code completion retrieval. Use for: finding code completions.

QA

Technical question answering. Use for: finding answers to programming questions.

Implementations

impl EmbeddingTask

pub fn query_prefix(&self) -> &'static str

Get the instruction prefix for QUERY embeddings.

Use this when embedding search queries.

pub fn passage_prefix(&self) -> &'static str

Get the instruction prefix for PASSAGE/DOCUMENT embeddings.

Use this when embedding code or documents to be indexed.

pub fn instruction_prefix(&self) -> &'static str

Get the default instruction prefix (alias for query_prefix).

For backward compatibility.

pub fn name(&self) -> &'static str

Get a human-readable name for this task.

pub fn from_name(name: &str) -> Option<Self>

Parse task from string name.

Trait Implementations

impl Clone for EmbeddingTask

fn clone(&self) -> EmbeddingTask

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for EmbeddingTask

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

Formats the value using the given formatter.

impl Default for EmbeddingTask

fn default() -> EmbeddingTask

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for EmbeddingTask

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl PartialEq for EmbeddingTask

fn eq(&self, other: &EmbeddingTask) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Serialize for EmbeddingTask

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl Copy for EmbeddingTask

impl Eq for EmbeddingTask

impl StructuralPartialEq for EmbeddingTask