mirror of
https://github.com/mii443/openai-api-rs.git
synced 2025-08-22 23:25:39 +00:00
Add dimensions parameter
This commit is contained in:
@ -6,7 +6,9 @@ use std::env;
|
|||||||
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
let client = Client::new(env::var("OPENAI_API_KEY").unwrap().to_string());
|
let client = Client::new(env::var("OPENAI_API_KEY").unwrap().to_string());
|
||||||
|
|
||||||
let req = EmbeddingRequest::new(TEXT_EMBEDDING_3_SMALL.to_string(), "story time".to_string());
|
let mut req =
|
||||||
|
EmbeddingRequest::new(TEXT_EMBEDDING_3_SMALL.to_string(), "story time".to_string());
|
||||||
|
req.dimensions = Some(10);
|
||||||
|
|
||||||
let result = client.embedding(req)?;
|
let result = client.embedding(req)?;
|
||||||
println!("{:?}", result.data);
|
println!("{:?}", result.data);
|
||||||
|
@ -15,6 +15,8 @@ pub struct EmbeddingRequest {
|
|||||||
pub model: String,
|
pub model: String,
|
||||||
pub input: String,
|
pub input: String,
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
|
pub dimensions: Option<i32>,
|
||||||
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
pub user: Option<String>,
|
pub user: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -23,6 +25,7 @@ impl EmbeddingRequest {
|
|||||||
Self {
|
Self {
|
||||||
model,
|
model,
|
||||||
input,
|
input,
|
||||||
|
dimensions: None,
|
||||||
user: None,
|
user: None,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user