mirror of
https://github.com/katanemo/plano.git
synced 2026-06-17 15:25:17 +02:00
3bd2ffe9fb
* split wasm filter * fix int and unit tests * rename public_types => common and move common code there * rename * fix int test
59 lines
2.2 KiB
Rust
59 lines
2.2 KiB
Rust
/*
|
|
* OMF Embeddings
|
|
*
|
|
* No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
|
|
*
|
|
* The version of the OpenAPI document: 1.0.0
|
|
*
|
|
* Generated by: https://openapi-generator.tech
|
|
*/
|
|
|
|
use crate::embeddings;
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize)]
|
|
pub struct CreateEmbeddingRequest {
|
|
#[serde(rename = "input")]
|
|
pub input: Box<embeddings::CreateEmbeddingRequestInput>,
|
|
/// ID of the model to use. You can use the [List models](/docs/api-reference/models/list) API to see all of your available models, or see our [Model overview](/docs/models/overview) for descriptions of them.
|
|
#[serde(rename = "model")]
|
|
pub model: String,
|
|
/// The format to return the embeddings in. Can be either `float` or [`base64`](https://pypi.org/project/pybase64/).
|
|
#[serde(rename = "encoding_format", skip_serializing_if = "Option::is_none")]
|
|
pub encoding_format: Option<EncodingFormat>,
|
|
/// The number of dimensions the resulting output embeddings should have. Only supported in `text-embedding-3` and later models.
|
|
#[serde(rename = "dimensions", skip_serializing_if = "Option::is_none")]
|
|
pub dimensions: Option<i32>,
|
|
/// A unique identifier representing your end-user, which can help to monitor and detect abuse. [Learn more](/docs/guides/safety-best-practices/end-user-ids).
|
|
#[serde(rename = "user", skip_serializing_if = "Option::is_none")]
|
|
pub user: Option<String>,
|
|
}
|
|
|
|
impl CreateEmbeddingRequest {
|
|
pub fn new(
|
|
input: embeddings::CreateEmbeddingRequestInput,
|
|
model: String,
|
|
) -> CreateEmbeddingRequest {
|
|
CreateEmbeddingRequest {
|
|
input: Box::new(input),
|
|
model,
|
|
encoding_format: None,
|
|
dimensions: None,
|
|
user: None,
|
|
}
|
|
}
|
|
}
|
|
/// The format to return the embeddings in. Can be either `float` or [`base64`](https://pypi.org/project/pybase64/).
|
|
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
|
|
pub enum EncodingFormat {
|
|
#[serde(rename = "float")]
|
|
Float,
|
|
#[serde(rename = "base64")]
|
|
Base64,
|
|
}
|
|
|
|
impl Default for EncodingFormat {
|
|
fn default() -> EncodingFormat {
|
|
Self::Float
|
|
}
|
|
}
|