Merge pull request #56 from hytracen/main

Impl Default trait for JSONSchemaDefine
dongri · Dec 30, 2023 · d192774 · d192774
2 parents f88cc6b + bfedc7f
commit d192774
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 8 deletions.
diff --git a/examples/function_call.rs b/examples/function_call.rs
@@ -23,10 +23,7 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
         Box::new(chat_completion::JSONSchemaDefine {
             schema_type: Some(chat_completion::JSONSchemaType::String),
             description: Some("The cryptocurrency to get the price of".to_string()),
-            enum_values: None,
-            properties: None,
-            required: None,
-            items: None,
+            ..Default::default()
         }),
     );
 

diff --git a/examples/function_call_role.rs b/examples/function_call_role.rs
@@ -23,10 +23,7 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
         Box::new(chat_completion::JSONSchemaDefine {
             schema_type: Some(chat_completion::JSONSchemaType::String),
             description: Some("The cryptocurrency to get the price of".to_string()),
-            enum_values: None,
-            properties: None,
-            required: None,
-            items: None,
+            ..Default::default()
         }),
     );
 

diff --git a/src/v1/chat_completion.rs b/src/v1/chat_completion.rs
@@ -197,6 +197,19 @@ pub struct JSONSchemaDefine {
     pub items: Option<Box<JSONSchemaDefine>>,
 }
 
+impl Default for JSONSchemaDefine {
+    fn default() -> Self {
+        Self {
+            schema_type: None,
+            description: None,
+            enum_values: None,
+            properties: None,
+            required: None,
+            items: None,
+        }
+    }
+}
+
 #[derive(Debug, Serialize, Deserialize, Clone)]
 pub struct FunctionParameters {
     #[serde(rename = "type")]