issues (#87)

Author: jamesrochabrun

Sources/OpenAI/AIProxy/AIProxyService.swift

@@ -406,6 +406,15 @@ struct AIProxyService: OpenAIService {
       let request = try await OpenAIAPI.message(.modify(threadID: threadID, messageID: messageID)).request(aiproxyPartialKey: partialKey, serviceURL: serviceURL, clientID: clientID, organizationID: organizationID, method: .post, params: parameters, betaHeaderField: Self.assistantsBetaV2)
       return try await fetch(debugEnabled: debugEnabled, type: MessageObject.self, with: request)
    }
+   
+   func deleteMessage(
+      threadID: String,
+      messageID: String)
+      async throws -> DeletionStatus
+   {
+      let request = try await OpenAIAPI.message(.delete(threadID: threadID, messageID: messageID)).request(aiproxyPartialKey: partialKey, serviceURL: serviceURL, clientID: clientID, organizationID: organizationID, method: .delete, betaHeaderField: Self.assistantsBetaV2)
+      return try await fetch(debugEnabled: debugEnabled, type: DeletionStatus.self, with: request)
+   }
 
    func listMessages(
       threadID: String,

Sources/OpenAI/Azure/AzureOpenAIAPI.swift

@@ -43,6 +43,7 @@ enum AzureOpenAIAPI {
       case create(threadID: String)
       case retrieve(threadID: String, messageID: String)
       case modify(threadID: String, messageID: String)
+      case delete(threadID: String, messageID: String)
       case list(threadID: String)
    }
 
@@ -104,7 +105,7 @@ extension AzureOpenAIAPI: Endpoint {
       case .message(let category):
          switch category {
          case .create(let threadID), .list(let threadID): return "/openai/threads/\(threadID)/messages"
-         case .retrieve(let threadID, let messageID), .modify(let threadID, let messageID): return "/openai/threads/\(threadID)/messages/\(messageID)"
+         case .retrieve(let threadID, let messageID), .modify(let threadID, let messageID), .delete(let threadID, let messageID): return "/openai/threads/\(threadID)/messages/\(messageID)"
          }
       case .run(let category):
          switch category {

Sources/OpenAI/Azure/DefaultOpenAIAzureService.swift

@@ -304,6 +304,21 @@ final public class DefaultOpenAIAzureService: OpenAIService {
       return try await fetch(debugEnabled: debugEnabled, type: MessageObject.self, with: request)
    }
 
+   public func deleteMessage(
+      threadID: String,
+      messageID: String)
+      async throws -> DeletionStatus
+   {
+      let request = try AzureOpenAIAPI.message(.delete(threadID: threadID, messageID: messageID)).request(
+         apiKey: apiKey,
+         organizationID: nil,
+         method: .delete,
+         queryItems: initialQueryItems,
+         betaHeaderField: Self.assistantsBetaV2,
+         extraHeaders: extraHeaders)
+      return try await fetch(debugEnabled: debugEnabled, type: DeletionStatus.self, with: request)
+   }
+   
    public func listMessages(threadID: String, limit: Int?, order: String?, after: String?, before: String?, runID: String?) async throws -> OpenAIResponse<MessageObject> {
       var queryItems: [URLQueryItem] = initialQueryItems
       if let limit {

Sources/OpenAI/LocalModelService/LocalModelService.swift

@@ -187,6 +187,10 @@ struct LocalModelService: OpenAIService {
       fatalError("Currently, this API is not supported. We welcome and encourage contributions to our open-source project. Please consider opening an issue or submitting a pull request to add support for this feature.")
    }
 
+   func deleteMessage(threadID: String, messageID: String) async throws -> DeletionStatus {
+      fatalError("Currently, this API is not supported. We welcome and encourage contributions to our open-source project. Please consider opening an issue or submitting a pull request to add support for this feature.")
+   }
+   
    func listMessages(threadID: String, limit: Int?, order: String?, after: String?, before: String?, runID: String?) async throws -> OpenAIResponse<MessageObject> {
       fatalError("Currently, this API is not supported. We welcome and encourage contributions to our open-source project. Please consider opening an issue or submitting a pull request to add support for this feature.")
    }

Sources/OpenAI/Private/Networking/OpenAIAPI.swift

@@ -71,6 +71,7 @@ enum OpenAIAPI {
       case create(threadID: String)
       case retrieve(threadID: String, messageID: String)
       case modify(threadID: String, messageID: String)
+      case delete(threadID: String, messageID: String)
       case list(threadID: String)
    }
 
@@ -174,7 +175,7 @@ extension OpenAIAPI: Endpoint {
       case .message(let category):
          switch category {
          case .create(let threadID), .list(let threadID): return "/v1/threads/\(threadID)/messages"
-         case .retrieve(let threadID, let messageID), .modify(let threadID, let messageID): return "/v1/threads/\(threadID)/messages/\(messageID)"
+         case .retrieve(let threadID, let messageID), .modify(let threadID, let messageID), .delete(let threadID, let messageID): return "/v1/threads/\(threadID)/messages/\(messageID)"
          }
       case .model(let category):
          switch category {

Sources/OpenAI/Public/Service/DefaultOpenAIService.swift

@@ -393,6 +393,15 @@ struct DefaultOpenAIService: OpenAIService {
       return try await fetch(debugEnabled: debugEnabled, type: MessageObject.self, with: request)
    }
 
+   func deleteMessage(
+      threadID: String,
+      messageID: String)
+      async throws -> DeletionStatus
+   {
+      let request = try OpenAIAPI.message(.delete(threadID: threadID, messageID: messageID)).request(apiKey: apiKey, organizationID: organizationID, method: .delete, betaHeaderField: Self.assistantsBetaV2)
+      return try await fetch(debugEnabled: debugEnabled, type: DeletionStatus.self, with: request)
+   }
+   
    func listMessages(
       threadID: String,
       limit: Int? = nil,

Sources/OpenAI/Public/Service/OpenAIService.swift

@@ -479,6 +479,18 @@ public protocol OpenAIService {
       parameters: ModifyMessageParameters)
       async throws -> MessageObject
 
+   /// Deletes a message.
+   ///
+   /// - Parameter threadID: The ID of the thread to which this message belongs.
+   /// - Parameter messageID: The ID of the message to modify.
+   /// - Throws: An error if the request fails.
+   ///
+   /// For more information, refer to [OpenAI's Message API documentation](https://platform.openai.com/docs/api-reference/messages/deleteMessage).
+   func deleteMessage(
+      threadID: String,
+      messageID: String)
+      async throws -> DeletionStatus
+
    /// Returns a list of messages for a given thread.
    ///
    /// - Parameter threadID: The ID of the [thread](https://platform.openai.com/docs/api-reference/threads) the messages belong to.

Sources/OpenAI/Public/Shared/MessageAttachment.swift

@@ -18,4 +18,9 @@ public struct MessageAttachment: Codable {
       case fileID = "file_id"
       case tools
    }
+   
+   public init(fileID: String, tools: [AssistantObject.Tool]) {
+       self.fileID = fileID
+       self.tools = tools
+   }
 }