jevr 0.1.1

Semantic code search for coding agents: find the code or document files matching a concept, described in plain language
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

//! Serde mirror of the TypeSafe System One wire format.
//!
//! Why a hand-rolled mirror: the tool speaks raw HTTP (no Rust SDK exists), and the
//! format was captured byte-for-byte from the official Python SDK's debug log, so
//! these types are the contract — tests/wire.rs replays the captured bodies.

use serde::{Deserialize, Serialize};
use std::collections::BTreeMap;

/// One POST body: a shared `state`, the pinned `model`, and named typed
/// questions evaluated independently against that state.
#[derive(Serialize)]
pub struct JevRequest<S: Serialize> {
    pub state: S,
    pub model: String,
    pub questions: BTreeMap<String, Question>,
}

/// Only the two primitives the pipeline uses are serializable (KISS): Noul for
/// membership verification, Choice for the listwise rerank. Score exists on the
/// wire but no stage asks one.
#[derive(Serialize)]
#[serde(tag = "type", rename_all = "lowercase")]
pub enum Question {
    Noul {
        instructions: String,
        #[serde(skip_serializing_if = "Option::is_none")]
        criteria: Option<NoulCriteria>,
    },
    Choice {
        instructions: String,
        /// Option key → description; `None` when the key text is self-describing
        /// (wire: {"py": "Python", "rs": null}).
        criteria: BTreeMap<String, Option<String>>,
    },
}

/// Boundary clarifications attached to a Noul: what a true and a false
/// answer are each supposed to mean.
#[derive(Serialize)]
pub struct NoulCriteria {
    /// Wire keys are the literal strings "true"/"false".
    #[serde(rename = "true")]
    pub true_means: String,
    #[serde(rename = "false")]
    pub false_means: String,
}

/// Top-level response: answers keyed by question name, plus billed usage.
#[derive(Deserialize)]
pub struct JevResponse {
    pub model: String,
    pub answers: BTreeMap<String, Answer>,
    pub usage: Usage,
}

/// All three answer shapes are parsed so an unexpected primitive in a response
/// never fails deserialization of the whole batch. Noul answers never carry
/// confidence; choice/score do, inline. Score's "legend" map is ignored.
#[derive(Deserialize)]
#[serde(tag = "type", rename_all = "lowercase")]
pub enum Answer {
    Noul {
        noul: f64,
    },
    Choice {
        choice: String,
        confidence: Option<f64>,
        probabilities: BTreeMap<String, f64>,
    },
    Score {
        score: f64,
        confidence: Option<f64>,
        probabilities: BTreeMap<String, f64>,
    },
}

/// Billed token counts for one request.
#[derive(Deserialize)]
pub struct Usage {
    pub input_tokens: u64,
    pub output_tokens: u64,
}