From 8048e0b32f3b842f778720d91b5f0eca85867178 Mon Sep 17 00:00:00 2001
From: Saket Aryan <saketaryan2002@gmail.com>
Date: Thu, 22 May 2025 21:56:45 +0530
Subject: [PATCH] fix(ts-sdk): Fixed Types from Message Interface (#2763)

---
 docs/changelog.mdx                     |  5 +++++
 mem0-ts/package.json                   |  2 +-
 mem0-ts/src/client/mem0.ts             | 10 +++-------
 mem0-ts/src/client/mem0.types.ts       |  2 +-
 mem0-ts/src/client/telemetry.ts        |  2 +-
 mem0-ts/src/oss/src/utils/telemetry.ts |  2 +-
 6 files changed, 12 insertions(+), 11 deletions(-)

diff --git a/docs/changelog.mdx b/docs/changelog.mdx
index 9d55dde1..eb512ffd 100644
--- a/docs/changelog.mdx
+++ b/docs/changelog.mdx
@@ -288,6 +288,11 @@ mode: "wide"
 
 <Tab title="TypeScript">
 
+<Update label="2025-05-08" description="v2.1.25">
+**Improvements:**
+- **Client:** Removed type `string` from `messages` interface
+</Update>
+
 <Update label="2025-05-08" description="v2.1.25">
 **Improvements:**
 - **Client:** Improved error handling in client.
diff --git a/mem0-ts/package.json b/mem0-ts/package.json
index 5f3f7b75..6c629a9d 100644
--- a/mem0-ts/package.json
+++ b/mem0-ts/package.json
@@ -1,6 +1,6 @@
 {
   "name": "mem0ai",
-  "version": "2.1.25",
+  "version": "2.1.26",
   "description": "The Memory Layer For Your AI Apps",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",
diff --git a/mem0-ts/src/client/mem0.ts b/mem0-ts/src/client/mem0.ts
index e4aece97..7de835c6 100644
--- a/mem0-ts/src/client/mem0.ts
+++ b/mem0-ts/src/client/mem0.ts
@@ -160,15 +160,11 @@ export default class MemoryClient {
   }
 
   _preparePayload(
-    messages: string | Array<Message>,
+    messages: Array<Message>,
     options: MemoryOptions,
   ): object {
     const payload: any = {};
-    if (typeof messages === "string") {
-      payload.messages = [{ role: "user", content: messages }];
-    } else if (Array.isArray(messages)) {
-      payload.messages = messages;
-    }
+    payload.messages = messages;
     return { ...payload, ...options };
   }
 
@@ -217,7 +213,7 @@ export default class MemoryClient {
   }
 
   async add(
-    messages: string | Array<Message>,
+    messages: Array<Message>,
     options: MemoryOptions = {},
   ): Promise<Array<Memory>> {
     if (this.telemetryId === "") await this.ping();
diff --git a/mem0-ts/src/client/mem0.types.ts b/mem0-ts/src/client/mem0.types.ts
index ebbbe4cb..88d4dd97 100644
--- a/mem0-ts/src/client/mem0.types.ts
+++ b/mem0-ts/src/client/mem0.types.ts
@@ -53,7 +53,7 @@ export interface MultiModalMessages {
 }
 
 export interface Messages {
-  role: string;
+  role: "user" | "assistant";
   content: string | MultiModalMessages;
 }
 
diff --git a/mem0-ts/src/client/telemetry.ts b/mem0-ts/src/client/telemetry.ts
index 3691a201..f7b5cd44 100644
--- a/mem0-ts/src/client/telemetry.ts
+++ b/mem0-ts/src/client/telemetry.ts
@@ -1,7 +1,7 @@
 // @ts-nocheck
 import type { TelemetryClient, TelemetryOptions } from "./telemetry.types";
 
-let version = "2.1.16";
+let version = "2.1.26";
 
 // Safely check for process.env in different environments
 let MEM0_TELEMETRY = true;
diff --git a/mem0-ts/src/oss/src/utils/telemetry.ts b/mem0-ts/src/oss/src/utils/telemetry.ts
index 7edff41d..2facbdd6 100644
--- a/mem0-ts/src/oss/src/utils/telemetry.ts
+++ b/mem0-ts/src/oss/src/utils/telemetry.ts
@@ -4,7 +4,7 @@ import type {
   TelemetryEventData,
 } from "./telemetry.types";
 
-let version = "2.1.16";
+let version = "2.1.26";
 
 // Safely check for process.env in different environments
 let MEM0_TELEMETRY = true;