From 01b53c30dfaedd818fe09a9361dc7ef7b0b0da1d Mon Sep 17 00:00:00 2001
From: borislav nikolov <borislav@fastmail.com>
Date: Tue, 9 Apr 2024 17:37:35 +0200
Subject: [PATCH] Revert "[BREAKING_CHANGES] Fix update message payload (#699)"

This reverts commit 187f4169f8898d78716f7944d87e5d95aa9a7c41.
---
 messages.go      | 4 ++--
 messages_test.go | 9 ++-------
 2 files changed, 4 insertions(+), 9 deletions(-)

diff --git a/messages.go b/messages.go
index 6fd0adbc9..861463235 100644
--- a/messages.go
+++ b/messages.go
@@ -139,11 +139,11 @@ func (c *Client) RetrieveMessage(
 func (c *Client) ModifyMessage(
 	ctx context.Context,
 	threadID, messageID string,
-	metadata map[string]string,
+	metadata map[string]any,
 ) (msg Message, err error) {
 	urlSuffix := fmt.Sprintf("/threads/%s/%s/%s", threadID, messagesSuffix, messageID)
 	req, err := c.newRequest(ctx, http.MethodPost, c.fullURL(urlSuffix),
-		withBody(map[string]any{"metadata": metadata}), withBetaAssistantV1())
+		withBody(metadata), withBetaAssistantV1())
 	if err != nil {
 		return
 	}
diff --git a/messages_test.go b/messages_test.go
index 40858bb0c..6fdf90ed4 100644
--- a/messages_test.go
+++ b/messages_test.go
@@ -68,10 +68,6 @@ func TestMessages(t *testing.T) {
 				metadata := map[string]any{}
 				err := json.NewDecoder(r.Body).Decode(&metadata)
 				checks.NoError(t, err, "unable to decode metadata in modify message call")
-				payload, ok := metadata["metadata"].(map[string]any)
-				if !ok {
-					t.Fatalf("metadata payload improperly wrapped %+v", metadata)
-				}
 
 				resBytes, _ := json.Marshal(
 					openai.Message{
@@ -90,9 +86,8 @@ func TestMessages(t *testing.T) {
 						FileIds:     nil,
 						AssistantID: &emptyStr,
 						RunID:       &emptyStr,
-						Metadata:    payload,
+						Metadata:    metadata,
 					})
-
 				fmt.Fprintln(w, string(resBytes))
 			case http.MethodGet:
 				resBytes, _ := json.Marshal(
@@ -217,7 +212,7 @@ func TestMessages(t *testing.T) {
 	}
 
 	msg, err = client.ModifyMessage(ctx, threadID, messageID,
-		map[string]string{
+		map[string]any{
 			"foo": "bar",
 		})
 	checks.NoError(t, err, "ModifyMessage error")