Assistant beta v2

src/v1/thread.rs

@@ -8,6 +8,32 @@ pub struct CreateThreadRequest {
     #[serde(skip_serializing_if = "Option::is_none")]
     pub messages: Option<Vec<Message>>,
     #[serde(skip_serializing_if = "Option::is_none")]
+    pub tool_resources: Option<ToolResource>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub metadata: Option<HashMap<String, String>>,
+}
+
+#[derive(Debug, Deserialize, Serialize, Clone)]
+pub struct ToolResource {
+    pub code_interpreter: Option<CodeInterpreter>,
+    pub file_search: Option<FileSearch>,
+}
+
+#[derive(Debug, Deserialize, Serialize, Clone)]
+pub struct CodeInterpreter {
+    pub file_ids: Option<Vec<String>>,
+}
+
+#[derive(Debug, Deserialize, Serialize, Clone)]
+pub struct FileSearch {
+    pub vector_store_ids: Option<Vec<String>>,
+    pub vector_stores: Option<VectorStores>,
+}
+
+#[derive(Debug, Deserialize, Serialize, Clone)]
+pub struct VectorStores {
+    pub file_ids: Option<Vec<String>>,
+    pub chunking_strategy: Option<String>,
     pub metadata: Option<HashMap<String, String>>,
 }
 
@@ -15,6 +41,7 @@ impl CreateThreadRequest {
     pub fn new() -> Self {
         Self {
             messages: None,
+            tool_resources: None,
             metadata: None,
         }
     }
@@ -27,9 +54,9 @@ impl Default for CreateThreadRequest {
 }
 
 impl_builder_methods!(
-  CreateThreadRequest,
-  messages: Vec<Message>,
-  metadata: HashMap<String, String>
+    CreateThreadRequest,
+    messages: Vec<Message>,
+    tool_resources: ToolResource
 );
 
 #[derive(Debug, Deserialize, Serialize)]
@@ -38,17 +65,52 @@ pub struct ThreadObject {
     pub object: String,
     pub created_at: i64,
     pub metadata: HashMap<String, String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub tool_resources: Option<ToolResource>,
+    #[serde(skip_serializing_if = "Option::is_none")]
     pub headers: Option<HashMap<String, String>>,
 }
 
 #[derive(Debug, Deserialize, Serialize, Clone)]
 pub struct Message {
+    pub id: String,
+    pub object: String,
+    pub created_at: i64,
+    pub thread_id: String,
     pub role: MessageRole,
-    pub content: String,
+    pub content: Vec<Content>,
     #[serde(skip_serializing_if = "Option::is_none")]
-    pub file_ids: Option<String>,
+    pub assistant_id: Option<String>,
     #[serde(skip_serializing_if = "Option::is_none")]
+    pub run_id: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub attachments: Option<Vec<Attachment>>,
     pub metadata: Option<HashMap<String, String>>,
+    pub headers: Option<HashMap<String, String>>,
+}
+
+#[derive(Debug, Deserialize, Serialize, Clone)]
+pub struct Content {
+    #[serde(rename = "type")]
+    pub content_type: String,
+    pub text: ContentText,
+}
+
+#[derive(Debug, Deserialize, Serialize, Clone)]
+pub struct ContentText {
+    pub value: String,
+    pub annotations: Vec<String>,
+}
+
+#[derive(Serialize, Deserialize, Debug, Clone)]
+pub struct Attachment {
+    pub file_id: String,
+    pub tools: Vec<Tool>,
+}
+
+#[derive(Serialize, Deserialize, Debug, Clone)]
+pub struct Tool {
+    pub r#type: String,
 }
 
 #[derive(Debug, Deserialize, Serialize, Clone, PartialEq, Eq)]