gproxy_protocol/protocol/openai/
models.rs

1use std::collections::BTreeMap;
2
3use serde::{Deserialize, Serialize};
4
5use super::common::*;
6
7pub type ModelsWireModel = OpenAiWireModel<(), ModelListResponse>;
8pub type ModelRetrieveWireModel = OpenAiWireModel<(), Model>;
9
10#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, gproxy_protocol_macros::WireBuilder)]
11#[non_exhaustive]
12pub struct ModelListResponse {
13    pub data: Vec<Model>,
14    pub object: ListObjectType,
15    #[serde(default, flatten, skip_serializing_if = "BTreeMap::is_empty")]
16    pub extra: Extra,
17}
18
19#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, gproxy_protocol_macros::WireBuilder)]
20#[non_exhaustive]
21pub struct Model {
22    pub id: OpenAiModelId,
23    // OpenAI-compatible providers (e.g. DeepSeek) omit `created`; keep it
24    // optional so decoding their model list for a response transform doesn't
25    // fail on the missing field.
26    #[serde(default, skip_serializing_if = "Option::is_none")]
27    pub created: Option<u64>,
28    pub object: ModelObjectType,
29    pub owned_by: String,
30    #[serde(default, flatten, skip_serializing_if = "BTreeMap::is_empty")]
31    pub extra: Extra,
32}
33
34#[cfg(test)]
35mod tests {
36    use super::*;
37
38    /// DeepSeek (and other OpenAI-compatible providers) return model entries
39    /// without `created`; decoding for a response transform must not fail.
40    #[test]
41    fn model_decodes_without_created() {
42        let m: Model = serde_json::from_str(
43            r#"{"id":"deepseek-chat","object":"model","owned_by":"deepseek"}"#,
44        )
45        .expect("decode without created");
46        assert_eq!(m.created, None);
47        // …and a missing `created` is omitted on re-encode, not fabricated.
48        let s = serde_json::to_string(&m).unwrap();
49        assert!(!s.contains("created"), "{s}");
50    }
51}
gproxy_protocol/protocol/openai/models.rs

gproxy_protocol/protocol/openai/
models.rs
