diff --git a/clients/ts-sdk/openapi.json b/clients/ts-sdk/openapi.json
index 2697e4a68c..b8e6403926 100644
--- a/clients/ts-sdk/openapi.json
+++ b/clients/ts-sdk/openapi.json
@@ -5620,6 +5620,13 @@
                 },
                 "description": "Query ID that is used for tracking analytics"
               }
+            },
+            "content": {
+              "application/json": {
+                "schema": {
+                  "$ref": "#/components/schemas/ImageEditResponse"
+                }
+              }
             }
           },
           "400": {
diff --git a/clients/ts-sdk/package.json b/clients/ts-sdk/package.json
index ba5f6e2152..4e0ec149bb 100644
--- a/clients/ts-sdk/package.json
+++ b/clients/ts-sdk/package.json
@@ -17,7 +17,7 @@
   "files": [
     "dist"
   ],
-  "version": "0.0.94",
+  "version": "0.0.95",
   "license": "MIT",
   "scripts": {
     "lint": "eslint 'src/**/*.ts'",
diff --git a/clients/ts-sdk/src/types.gen.ts b/clients/ts-sdk/src/types.gen.ts
index 3cff346831..f823c6b1a9 100644
--- a/clients/ts-sdk/src/types.gen.ts
+++ b/clients/ts-sdk/src/types.gen.ts
@@ -5849,7 +5849,7 @@ export type EditImageData = {
     trDataset: string;
 };
 
-export type EditImageResponse = (string);
+export type EditImageResponse = (ImageEditResponse);
 
 export type GetToolFunctionParamsData = {
     /**
@@ -7542,7 +7542,7 @@ export type $OpenApiTs = {
                 /**
                  * A list of base64 encoded images
                  */
-                200: string;
+                200: ImageEditResponse;
                 /**
                  * Service error relating to editing the image
                  */
diff --git a/server/src/handlers/message_handler.rs b/server/src/handlers/message_handler.rs
index 7308c42fc7..8fec30a3bc 100644
--- a/server/src/handlers/message_handler.rs
+++ b/server/src/handlers/message_handler.rs
@@ -1237,7 +1237,7 @@ pub struct ImageResponseData {
     tag = "Message",
     request_body(content = EditImageReqPayload, description = "JSON request payload to edit an image", content_type = "application/json"),
     responses(
-        (status = 200, description = "A list of base64 encoded images",
+        (status = 200, description = "A list of base64 encoded images", body = ImageEditResponse,
             headers(
                 ("TR-QueryID" = uuid::Uuid, description = "Query ID that is used for tracking analytics")
             )
@@ -1258,7 +1258,6 @@ pub async fn edit_image(
 ) -> Result<HttpResponse, ServiceError> {
     let dataset_config =
         DatasetConfiguration::from_json(dataset_org_plan_sub.dataset.clone().server_configuration);
-    log::info!("Dataset config: {dataset_config:#?}");
 
     let llm_api_key = get_llm_api_key(&dataset_config);
     let base_url = dataset_config.LLM_BASE_URL.clone();
diff --git a/yarn.lock b/yarn.lock
index 02463c71a1..0dded50aeb 100644
--- a/yarn.lock
+++ b/yarn.lock
@@ -5240,11 +5240,6 @@ mdast-util-gfm-task-list-item@^2.0.0:
   version "2.0.0"
   resolved "https://registry.yarnpkg.com/mdast-util-gfm-task-list-item/-/mdast-util-gfm-task-list-item-2.0.0.tgz#e68095d2f8a4303ef24094ab642e1047b991a936"
   integrity sha512-IrtvNvjxC1o06taBAVJznEnkiHxLFTzgonUdy8hzFVeDun0uTjxxrRGVaNFqkU1wJR3RBPEfsxmU6jDWPofrTQ==
-  dependencies:
-    "@types/mdast" "^4.0.0"
-    devlop "^1.0.0"
-    mdast-util-from-markdown "^2.0.0"
-    mdast-util-to-markdown "^2.0.0"
 
 mdast-util-gfm@^2.0.0:
   version "2.0.2"
@@ -7505,7 +7500,16 @@ std-env@^3.8.0:
   resolved "https://registry.npmjs.org/std-env/-/std-env-3.8.1.tgz"
   integrity sha512-vj5lIj3Mwf9D79hBkltk5qmkFI+biIKWS2IBxEyEU3AX1tUf7AoL8nSazCOiiqQsGKIq01SClsKEzweu34uwvA==
 
-"string-width-cjs@npm:string-width@^4.2.0", string-width@^4.1.0, string-width@^4.2.0, string-width@^4.2.3:
+"string-width-cjs@npm:string-width@^4.2.0":
+  version "4.2.3"
+  resolved "https://registry.npmjs.org/string-width/-/string-width-4.2.3.tgz"
+  integrity sha512-wKyQRQpjJ0sIp62ErSZdGsjMJWsap5oRNihHhu6G7JVO/9jIB6UyevL+tXuOqrng8j/cxKTWyWUwvSTriiZz/g==
+  dependencies:
+    emoji-regex "^8.0.0"
+    is-fullwidth-code-point "^3.0.0"
+    strip-ansi "^6.0.1"
+
+string-width@^4.1.0, string-width@^4.2.0, string-width@^4.2.3:
   version "4.2.3"
   resolved "https://registry.npmjs.org/string-width/-/string-width-4.2.3.tgz"
   integrity sha512-wKyQRQpjJ0sIp62ErSZdGsjMJWsap5oRNihHhu6G7JVO/9jIB6UyevL+tXuOqrng8j/cxKTWyWUwvSTriiZz/g==
@@ -7600,7 +7604,14 @@ stringify-entities@^4.0.0:
     character-entities-html4 "^2.0.0"
     character-entities-legacy "^3.0.0"
 
-"strip-ansi-cjs@npm:strip-ansi@^6.0.1", strip-ansi@^6.0.0, strip-ansi@^6.0.1:
+"strip-ansi-cjs@npm:strip-ansi@^6.0.1":
+  version "6.0.1"
+  resolved "https://registry.npmjs.org/strip-ansi/-/strip-ansi-6.0.1.tgz"
+  integrity sha512-Y38VPSHcqkFrCpFnQ9vuSXmquuv5oXOKpGeT6aGrr3o3Gc9AlVa6JBfUSOCnbxGGZF+/0ooI7KrPuUSztUdU5A==
+  dependencies:
+    ansi-regex "^5.0.1"
+
+strip-ansi@^6.0.0, strip-ansi@^6.0.1:
   version "6.0.1"
   resolved "https://registry.npmjs.org/strip-ansi/-/strip-ansi-6.0.1.tgz"
   integrity sha512-Y38VPSHcqkFrCpFnQ9vuSXmquuv5oXOKpGeT6aGrr3o3Gc9AlVa6JBfUSOCnbxGGZF+/0ooI7KrPuUSztUdU5A==
@@ -7854,6 +7865,11 @@ trieve-ts-sdk@^0.0.92:
   resolved "https://registry.yarnpkg.com/trieve-ts-sdk/-/trieve-ts-sdk-0.0.92.tgz#d290d81e9961eb2f9d87100f0c9f648f7789ef5c"
   integrity sha512-V0gMzr/o0Zv7MWptqItmM5Q1oGwzS452S0WSUBH2C8k1q63bz2UcBNm743pbTHLkZ0Bw54oM3yct+hLPGJQ6Cg==
 
+trieve-ts-sdk@^0.0.93:
+  version "0.0.93"
+  resolved "https://registry.yarnpkg.com/trieve-ts-sdk/-/trieve-ts-sdk-0.0.93.tgz#57168b4d4d5822e0e3947ebef86cec2371123ac0"
+  integrity sha512-LLshhGlRRStPU5FfFW2I1AVt7/aHKm4cCO0Bd8PMfpHVJv7vAOkn1KDu/NVoI4seY3K3DJg3NyEUwcKoymoauQ==
+
 trim-lines@^3.0.0:
   version "3.0.1"
   resolved "https://registry.npmjs.org/trim-lines/-/trim-lines-3.0.1.tgz"
@@ -8491,7 +8507,16 @@ wordwrap@^1.0.0:
   resolved "https://registry.npmjs.org/wordwrap/-/wordwrap-1.0.0.tgz"
   integrity sha512-gvVzJFlPycKc5dZN4yPkP8w7Dc37BtP1yczEneOb4uq34pXZcvrtRTmWV8W+Ume+XCxKgbjM+nevkyFPMybd4Q==
 
-"wrap-ansi-cjs@npm:wrap-ansi@^7.0.0", wrap-ansi@^7.0.0:
+"wrap-ansi-cjs@npm:wrap-ansi@^7.0.0":
+  version "7.0.0"
+  resolved "https://registry.npmjs.org/wrap-ansi/-/wrap-ansi-7.0.0.tgz"
+  integrity sha512-YVGIj2kamLSTxw6NsZjoBxfSwsn0ycdesmc4p+Q21c5zPuZ1pl+NfxVdxPtdHvmNVOQ6XSYG4AUtyt/Fi7D16Q==
+  dependencies:
+    ansi-styles "^4.0.0"
+    string-width "^4.1.0"
+    strip-ansi "^6.0.0"
+
+wrap-ansi@^7.0.0:
   version "7.0.0"
   resolved "https://registry.npmjs.org/wrap-ansi/-/wrap-ansi-7.0.0.tgz"
   integrity sha512-YVGIj2kamLSTxw6NsZjoBxfSwsn0ycdesmc4p+Q21c5zPuZ1pl+NfxVdxPtdHvmNVOQ6XSYG4AUtyt/Fi7D16Q==