Merge pull request #60 from OnlyFerris/main

Make fields of Tool public. Rename a field.
dongri · Jan 6, 2024 · f0973ea · f0973ea
2 parents baae003 + b666bc0
commit f0973ea
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/src/v1/chat_completion.rs b/src/v1/chat_completion.rs
@@ -263,7 +263,7 @@ where
         Some(ToolChoiceType::Auto) => serializer.serialize_str("auto"),
         Some(ToolChoiceType::ToolChoice { tool }) => {
             let mut map = serializer.serialize_map(Some(2))?;
-            map.serialize_entry("type", &tool.tool_type)?;
+            map.serialize_entry("type", &tool.r#type)?;
             map.serialize_entry("function", &tool.function)?;
             map.end()
         }
@@ -273,7 +273,12 @@ where
 
 #[derive(Debug, Serialize, Deserialize, Clone)]
 pub struct Tool {
-    #[serde(rename = "type")]
-    tool_type: String,
-    function: Function,
+    pub r#type: ToolType,
+    pub function: Function,
+}
+
+#[derive(Debug, Serialize, Deserialize, Copy, Clone)]
+#[serde(rename_all = "snake_case")]
+pub enum ToolType {
+    Function,
 }