Merge pull request #78 from ryanolson/deserialize_completion_reqs

Adding ability to deserialize completion and chat completion requests
dongri · Feb 12, 2024 · 9e3dcb9 · 9e3dcb9
2 parents 7fe9f4e + bd57dbb
commit 9e3dcb9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/v1/chat_completion.rs b/src/v1/chat_completion.rs
@@ -6,14 +6,14 @@ use std::collections::HashMap;
 use crate::impl_builder_methods;
 use crate::v1::common;
 
-#[derive(Debug, Serialize, Clone, PartialEq, Eq)]
+#[derive(Debug, Serialize, Deserialize, Clone, PartialEq, Eq)]
 pub enum ToolChoiceType {
     None,
     Auto,
     ToolChoice { tool: Tool },
 }
 
-#[derive(Debug, Serialize, Clone)]
+#[derive(Debug, Serialize, Deserialize, Clone)]
 pub struct ChatCompletionRequest {
     pub model: String,
     pub messages: Vec<ChatCompletionMessage>,

diff --git a/src/v1/completion.rs b/src/v1/completion.rs
@@ -18,7 +18,7 @@ pub const GPT3_CURIE: &str = "curie";
 pub const GPT3_ADA: &str = "ada";
 pub const GPT3_BABBAGE: &str = "babbage";
 
-#[derive(Debug, Serialize, Clone)]
+#[derive(Debug, Serialize, Deserialize, Clone)]
 pub struct CompletionRequest {
     pub model: String,
     pub prompt: String,