From 32cecb6033b81ea849d2587dfcfc1428646e519b Mon Sep 17 00:00:00 2001 From: Dongri Jin Date: Tue, 7 Nov 2023 09:24:34 +0900 Subject: [PATCH] Add new models --- src/v1/chat_completion.rs | 16 +++++++++++++--- 1 file changed, 13 insertions(+), 3 deletions(-) diff --git a/src/v1/chat_completion.rs b/src/v1/chat_completion.rs index fe5e7ec..ff321b1 100644 --- a/src/v1/chat_completion.rs +++ b/src/v1/chat_completion.rs @@ -5,16 +5,26 @@ use std::collections::HashMap; use crate::impl_builder_methods; use crate::v1::common; +// https://platform.openai.com/docs/models/gpt-3-5 +pub const GPT3_5_TURBO_1106: &str = "gpt-3.5-turbo-1106"; pub const GPT3_5_TURBO: &str = "gpt-3.5-turbo"; pub const GPT3_5_TURBO_16K: &str = "gpt-3.5-turbo-16k"; -pub const GPT3_5_TURBO_0301: &str = "gpt-3.5-turbo-0301"; +pub const GPT3_5_TURBO_INSTRUCT: &str = "gpt-3.5-turbo-instruct"; +// - legacy pub const GPT3_5_TURBO_0613: &str = "gpt-3.5-turbo-0613"; +pub const GPT3_5_TURBO_16K_0613: &str = "gpt-3.5-turbo-16k-0613"; +pub const GPT3_5_TURBO_0301: &str = "gpt-3.5-turbo-0301"; +// https://platform.openai.com/docs/models/gpt-4-and-gpt-4-turbo +pub const GPT4_1106_PREVIEW: &str = "gpt-4-1106-preview"; +pub const GPT4_VISION_PREVIEW: &str = "gpt-4-vision-preview"; pub const GPT4: &str = "gpt-4"; -pub const GPT4_0314: &str = "gpt-4-0314"; pub const GPT4_32K: &str = "gpt-4-32k"; -pub const GPT4_32K_0314: &str = "gpt-4-32k-0314"; pub const GPT4_0613: &str = "gpt-4-0613"; +pub const GPT4_32K_0613: &str = "gpt-4-32k-0613"; +// - legacy +pub const GPT4_0314: &str = "gpt-4-0314"; +pub const GPT4_32K_0314: &str = "gpt-4-32k-0314"; #[derive(Debug, Serialize)] pub enum FunctionCallType {