use http::StatusCode;
use serde::{Deserialize, Serialize};
use crate::openai::count_tokens::types::{OpenAiApiErrorResponse, OpenAiResponseHeaders};
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
pub struct ResponseBody {
pub input_tokens: u64,
pub object: OpenAiCountTokensObject,
}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
pub enum OpenAiCountTokensObject {
#[serde(rename = "response.input_tokens")]
ResponseInputTokens,
}
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
#[serde(untagged)]
pub enum OpenAiCountTokensResponse {
Success {
#[serde(with = "crate::openai::types::status_code_serde")]
stats_code: StatusCode,
headers: OpenAiResponseHeaders,
body: ResponseBody,
},
Error {
#[serde(with = "crate::openai::types::status_code_serde")]
stats_code: StatusCode,
headers: OpenAiResponseHeaders,
body: OpenAiApiErrorResponse,
},
}