fix null in tool type response (#687)

* fix null in tool type response * remove debug test * add back trait we dropped :)
EricLBuehler · Aug 15, 2024 · a09a68a · a09a68a
1 parent 376e9d8
commit a09a68a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/mistralrs-core/src/tools/response.rs b/mistralrs-core/src/tools/response.rs
@@ -1,9 +1,8 @@
 #[cfg_attr(feature = "pyo3_macros", pyo3::pyclass(eq, eq_int))]
 #[cfg_attr(feature = "pyo3_macros", pyo3(get_all))]
 #[derive(Clone, Debug, serde::Serialize, PartialEq)]
-#[serde(untagged)]
+#[serde(rename_all = "snake_case")]
 pub enum ToolCallType {
-    #[serde(rename = "function")]
     Function,
 }