Support structured outputs

64bit · Aug 12, 2024 · feb270e · feb270e
1 parent 6d70a33
commit feb270e
Show file tree

Hide file tree

Showing 2 changed files with 675 additions and 665 deletions.
diff --git a/async-openai/src/types/assistant.rs b/async-openai/src/types/assistant.rs
@@ -129,8 +129,11 @@ pub enum AssistantsApiResponseFormatOption {
 /// An object describing the expected output of the model. If `json_object` only `function` type `tools` are allowed to be passed to the Run. If `text` the model can return text or any value needed.
 #[derive(Clone, Serialize, Debug, Deserialize, PartialEq, Default)]
 pub struct AssistantsApiResponseFormat {
-    /// Must be one of `text` or `json_object`.
+    /// Must be one of `text`, `json_object` or `json_schema`.
     pub r#type: AssistantsApiResponseFormatType,
+
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub json_schema: Option<serde_json::Value>,
 }
 
 #[derive(Clone, Serialize, Debug, Deserialize, PartialEq, Default)]
@@ -139,6 +142,7 @@ pub enum AssistantsApiResponseFormatType {
     #[default]
     Text,
     JsonObject,
+    JsonSchema,
 }
 
 /// Retrieval tool