fix(openai): missing type field in ToolResultContent

0xPlaygrounds · Feb 8, 2025 · 83df70d · 83df70d
1 parent 495bd69
commit 83df70d
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/rig-core/src/providers/openai.rs b/rig-core/src/providers/openai.rs
@@ -534,9 +534,18 @@ pub struct InputAudio {
 
 #[derive(Debug, Serialize, Deserialize, PartialEq, Clone)]
 pub struct ToolResultContent {
+    #[serde(default)]
+    r#type: ToolResultContentType,
     text: String,
 }
 
+#[derive(Default, Debug, Serialize, Deserialize, PartialEq, Clone)]
+#[serde(rename_all = "lowercase")]
+pub enum ToolResultContentType {
+    #[default]
+    Text,
+}
+
 impl FromStr for ToolResultContent {
     type Err = Infallible;
 
@@ -547,7 +556,10 @@ impl FromStr for ToolResultContent {
 
 impl From<String> for ToolResultContent {
     fn from(s: String) -> Self {
-        ToolResultContent { text: s }
+        ToolResultContent {
+            r#type: ToolResultContentType::default(),
+            text: s
+        }
     }
 }