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
// SPDX-FileCopyrightText: Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
// SPDX-License-Identifier: Apache-2.0
use serde::{Deserialize, Serialize};
use strum::Display;
#[derive(Copy, Debug, Clone, Display, Serialize, Deserialize, Eq, PartialEq, Hash)]
pub enum EndpointType {
// Chat Completions API
Chat,
/// Older completions API
Completion,
/// Embeddings API
Embedding,
/// Images API (Diffusion/DALL-E)
Images,
/// Audios API (speech/audio generation)
Audios,
/// Videos API (video generation)
Videos,
/// Responses API
Responses,
/// Anthropic Messages API
AnthropicMessages,
}
impl EndpointType {
pub fn as_str(&self) -> &str {
match self {
Self::Chat => "chat",
Self::Completion => "completion",
Self::Embedding => "embedding",
Self::Images => "images",
Self::Audios => "audios",
Self::Videos => "videos",
Self::Responses => "responses",
Self::AnthropicMessages => "anthropic_messages",
}
}
pub fn all() -> Vec<Self> {
vec![
Self::Chat,
Self::Completion,
Self::Embedding,
Self::Images,
Self::Audios,
Self::Videos,
Self::Responses,
Self::AnthropicMessages,
]
}
}