venice-e2ee-proxy 0.1.3

OpenAI-compatible proxy for Venice.ai E2EE models
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152

//! OpenAI-compatible request and response formatting.
//!
//! Includes the typed model-list response used by `GET /v1/models` and the
//! shared OpenAI-style error envelope used by fail-closed validation responses.

use serde::{Deserialize, Serialize};

pub mod chat;

/// OpenAI-compatible model-list response envelope.
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
pub struct ModelListResponse {
    pub object: String,
    pub data: Vec<ModelObject>,
}

impl ModelListResponse {
    /// Builds a model-list response from already-normalized model objects.
    pub fn new(data: Vec<ModelObject>) -> Self {
        Self {
            object: "list".to_owned(),
            data,
        }
    }
}

/// OpenAI-compatible model object with Venice metadata preserved for clients
/// that need it.
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
pub struct ModelObject {
    pub id: String,
    pub object: String,
    pub created: i64,
    pub owned_by: String,
    pub name: String,
    pub info: ModelInfo,
    pub venice: VeniceModelMetadata,
}

impl ModelObject {
    /// Builds an OpenAI-compatible model object with mirrored `id` and `name` fields.
    pub fn new(
        id: impl Into<String>,
        created: i64,
        owned_by: impl Into<String>,
        capabilities: ModelCapabilities,
        venice: VeniceModelMetadata,
    ) -> Self {
        let id = id.into();

        Self {
            name: id.clone(),
            id,
            object: "model".to_owned(),
            created,
            owned_by: owned_by.into(),
            info: ModelInfo {
                meta: ModelMeta { capabilities },
            },
            venice,
        }
    }
}

/// OpenAI-compatible model information wrapper containing model metadata.
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
pub struct ModelInfo {
    pub meta: ModelMeta,
}

/// OpenAI-compatible model metadata containing capability flags.
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
pub struct ModelMeta {
    pub capabilities: ModelCapabilities,
}

/// Capability flags exposed in the OpenAI-compatible model metadata.
#[derive(Debug, Clone, Copy, Default, PartialEq, Eq, Serialize, Deserialize)]
pub struct ModelCapabilities {
    pub function_calling: bool,
    pub builtin_tools: bool,
    pub web_search: bool,
    pub code_interpreter: bool,
    pub vision: bool,
    pub reasoning: bool,
    pub reasoning_effort: bool,
}

/// Venice-specific model metadata preserved alongside the OpenAI-compatible model object.
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
pub struct VeniceModelMetadata {
    pub id: String,
    #[serde(rename = "supportsE2EE")]
    pub supports_e2ee: bool,
    #[serde(rename = "supportsTeeAttestation")]
    pub supports_tee_attestation: bool,
    #[serde(rename = "supportsReasoning")]
    pub supports_reasoning: bool,
    #[serde(rename = "supportsReasoningEffort")]
    pub supports_reasoning_effort: bool,
}

impl VeniceModelMetadata {
    /// Builds Venice metadata from the upstream model id and supported feature flags.
    pub fn new(
        id: impl Into<String>,
        supports_e2ee: bool,
        supports_tee_attestation: bool,
        supports_reasoning: bool,
        supports_reasoning_effort: bool,
    ) -> Self {
        Self {
            id: id.into(),
            supports_e2ee,
            supports_tee_attestation,
            supports_reasoning,
            supports_reasoning_effort,
        }
    }
}

/// OpenAI-compatible error response envelope.
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
pub struct ErrorResponse {
    pub error: ErrorObject,
}

impl ErrorResponse {
    /// Builds an OpenAI-compatible error response from message, type, and code strings.
    pub fn new(
        message: impl Into<String>,
        error_type: impl Into<String>,
        code: impl Into<String>,
    ) -> Self {
        Self {
            error: ErrorObject {
                message: message.into(),
                kind: error_type.into(),
                code: code.into(),
            },
        }
    }
}

/// OpenAI-compatible error object containing the client-facing message, type, and code.
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
pub struct ErrorObject {
    pub message: String,
    #[serde(rename = "type")]
    pub kind: String,
    pub code: String,
}