add support for assistant vision model, partial implementation of https://platform.openai.com/docs/assistants/migration

Author: Loris Mendiondo

assistant.go

@@ -14,25 +14,39 @@ const (
 )
 
 type Assistant struct {
-	ID           string          `json:"id"`
-	Object       string          `json:"object"`
-	CreatedAt    int64           `json:"created_at"`
-	Name         *string         `json:"name,omitempty"`
-	Description  *string         `json:"description,omitempty"`
-	Model        string          `json:"model"`
-	Instructions *string         `json:"instructions,omitempty"`
-	Tools        []AssistantTool `json:"tools"`
-	FileIDs      []string        `json:"file_ids,omitempty"`
-	Metadata     map[string]any  `json:"metadata,omitempty"`
+	ID            string          `json:"id"`
+	Object        string          `json:"object"`
+	CreatedAt     int64           `json:"created_at"`
+	Name          *string         `json:"name,omitempty"`
+	Description   *string         `json:"description,omitempty"`
+	Model         string          `json:"model"`
+	Instructions  *string         `json:"instructions,omitempty"`
+	Tools         []AssistantTool `json:"tools"`
+	FileIDs       []string        `json:"file_ids,omitempty"`
+	Metadata      map[string]any  `json:"metadata,omitempty"`
+	ToolResources ToolResources   `json:"tool_resources,omitempty"`
 
 	httpHeader
 }
 
+type ToolResources struct {
+	FileSearch      *FileSearchResources      `json:"file_search,omitempty"`
+	CodeInterpreter *CodeInterpreterResources `json:"code_interpreter,omitempty"`
+}
+
+type FileSearchResources struct {
+	VectorStoreIDs []string `json:"vector_store_ids"`
+}
+
+type CodeInterpreterResources struct {
+	FileIDs []string `json:"file_ids"`
+}
+
 type AssistantToolType string
 
 const (
 	AssistantToolTypeCodeInterpreter AssistantToolType = "code_interpreter"
-	AssistantToolTypeRetrieval       AssistantToolType = "retrieval"
+	AssistantToolTypeFileSearch      AssistantToolType = "file_search"
 	AssistantToolTypeFunction        AssistantToolType = "function"
 )
 

messages.go

@@ -12,20 +12,29 @@ const (
 )
 
 type Message struct {
-	ID          string           `json:"id"`
-	Object      string           `json:"object"`
-	CreatedAt   int              `json:"created_at"`
-	ThreadID    string           `json:"thread_id"`
-	Role        string           `json:"role"`
-	Content     []MessageContent `json:"content"`
-	FileIds     []string         `json:"file_ids"` //nolint:revive //backwards-compatibility
-	AssistantID *string          `json:"assistant_id,omitempty"`
-	RunID       *string          `json:"run_id,omitempty"`
-	Metadata    map[string]any   `json:"metadata"`
+	ID          string              `json:"id"`
+	Object      string              `json:"object"`
+	CreatedAt   int                 `json:"created_at"`
+	ThreadID    string              `json:"thread_id"`
+	Role        string              `json:"role"`
+	Content     []MessageContent    `json:"content"`
+	AssistantID *string             `json:"assistant_id,omitempty"`
+	RunID       *string             `json:"run_id,omitempty"`
+	Metadata    map[string]any      `json:"metadata"`
+	Attachments []MessageAttachment `json:"attachments,omitempty"`
 
 	httpHeader
 }
 
+type MessageAttachment struct {
+	FileID string        `json:"file_id"`
+	Tools  []MessageTool `json:"tools"`
+}
+
+type MessageTool struct {
+	Type string `json:"type"`
+}
+
 type MessagesList struct {
 	Messages []Message `json:"data"`
 
@@ -42,6 +51,7 @@ type MessageContent struct {
 	Text      *MessageText `json:"text,omitempty"`
 	ImageFile *ImageFile   `json:"image_file,omitempty"`
 }
+
 type MessageText struct {
 	Value       string `json:"value"`
 	Annotations []any  `json:"annotations"`

thread.go

@@ -33,11 +33,18 @@ const (
 	ThreadMessageRoleUser ThreadMessageRole = "user"
 )
 
+type ContentInterface interface{}
+
 type ThreadMessage struct {
-	Role     ThreadMessageRole `json:"role"`
-	Content  string            `json:"content"`
-	FileIDs  []string          `json:"file_ids,omitempty"`
-	Metadata map[string]any    `json:"metadata,omitempty"`
+	Role        ThreadMessageRole  `json:"role"`
+	Content     ContentInterface   `json:"content"`
+	Metadata    map[string]any     `json:"metadata,omitempty"`
+	Attachments []ThreadAttachment `json:"attachments,omitempty"`
+}
+
+type ThreadAttachment struct {
+	FileID string        `json:"file_id"`
+	Tools  []MessageTool `json:"tools,omitempty"`
 }
 
 type ThreadDeleteResponse struct {