mii/openai-api-rs
1
0
Fork 0
mirror of https://github.com/mii443/openai-api-rs.git synced 2025-08-23 15:48:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add embedding models

Browse Source
This commit is contained in:
Dongri Jin
2024-01-26 11:13:24 +09:00
parent 91987abed7
commit 4e6fbcedd5
3 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/embedding.rs
View File

@ -1,14 +1,12 @@
use openai_api_rs::v1::api::Client; use openai_api_rs::v1::api::Client;
use openai_api_rs::v1::common::TEXT_EMBEDDING_3_SMALL;
use openai_api_rs::v1::embedding::EmbeddingRequest; use openai_api_rs::v1::embedding::EmbeddingRequest;
use std::env; 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( let req = EmbeddingRequest::new(TEXT_EMBEDDING_3_SMALL.to_string(), "story time".to_string());
"text-embedding-ada-002".to_string(),
"story time".to_string(),
);
let result = client.embedding(req)?; let result = client.embedding(req)?;
println!("{:?}", result.data); println!("{:?}", result.data);

4
src/v1/chat_completion.rs
View File

@ -158,7 +158,7 @@ pub struct ChatCompletionMessageForResponse {
pub tool_calls: Option<Vec<ToolCall>>, pub tool_calls: Option<Vec<ToolCall>>,
} }
#[derive(Debug, Deserialize)] #[derive(Debug, Deserialize, Serialize)]
pub struct ChatCompletionChoice { pub struct ChatCompletionChoice {
pub index: i64, pub index: i64,
pub message: ChatCompletionMessageForResponse, pub message: ChatCompletionMessageForResponse,
@ -166,7 +166,7 @@ pub struct ChatCompletionChoice {
pub finish_details: Option<FinishDetails>, pub finish_details: Option<FinishDetails>,
} }
#[derive(Debug, Deserialize)] #[derive(Debug, Deserialize, Serialize)]
pub struct ChatCompletionResponse { pub struct ChatCompletionResponse {
pub id: String, pub id: String,
pub object: String, pub object: String,

9
src/v1/common.rs
View File

@ -1,6 +1,6 @@
use serde::Deserialize; use serde::{Deserialize, Serialize};
#[derive(Debug, Deserialize)] #[derive(Debug, Deserialize, Serialize)]
pub struct Usage { pub struct Usage {
pub prompt_tokens: i32, pub prompt_tokens: i32,
pub completion_tokens: i32, pub completion_tokens: i32,
@ -45,3 +45,8 @@ pub const GPT4_32K_0314: &str = "gpt-4-32k-0314";
// https://platform.openai.com/docs/api-reference/images/object // https://platform.openai.com/docs/api-reference/images/object
pub const DALL_E_2: &str = "dall-e-2"; pub const DALL_E_2: &str = "dall-e-2";
pub const DALL_E_3: &str = "dall-e-3"; pub const DALL_E_3: &str = "dall-e-3";
// https://platform.openai.com/docs/guides/embeddings/embedding-models
pub const TEXT_EMBEDDING_3_SMALL: &str = "text-embedding-3-small";
pub const TEXT_EMBEDDING_3_LARGE: &str = "text-embedding-3-large";
pub const TEXT_EMBEDDING_ADA_002: &str = "text-embedding-ada-002";