Add Together embedder to TS SDK (#5989)

Co-authored-by: kartik-mem0 <kartik.labhshetwar@mem0.ai>

mem0-ts/src/oss/tests/factory.unit.test.ts

@@ -35,6 +35,11 @@ jest.mock("../src/embeddings/lmstudio", () => ({
     .fn()
     .mockImplementation((config) => ({ type: "lmstudio-embedder", config })),
 }));
+jest.mock("../src/embeddings/together", () => ({
+  TogetherEmbedder: jest
+    .fn()
+    .mockImplementation((config) => ({ type: "together-embedder", config })),
+}));
 
 jest.mock("../src/llms/openai", () => ({
   OpenAILLM: jest
@@ -190,6 +195,7 @@ describe("EmbedderFactory", () => {
     ["azure_openai"],
     ["langchain"],
     ["lmstudio"],
+    ["together"],
   ])("creates embedder for provider '%s'", (provider) => {
     expect(() =>
       EmbedderFactory.create(provider, dummyEmbedConfig),

mem0-ts/src/oss/tests/together-embedder.test.ts

@@ -0,0 +1,119 @@
+/// <reference types="jest" />
+
+const mockEmbeddingsCreate = jest.fn();
+const mockOpenAI = jest.fn().mockImplementation(() => ({
+  embeddings: { create: mockEmbeddingsCreate },
+}));
+
+jest.mock("openai", () => ({
+  __esModule: true,
+  default: mockOpenAI,
+}));
+
+import { TogetherEmbedder } from "../src/embeddings/together";
+
+const mockEmbedding = [0.1, 0.2, 0.3];
+const originalEnv = process.env;
+
+describe("TogetherEmbedder (unit)", () => {
+  beforeEach(() => {
+    jest.resetModules();
+    process.env = { ...originalEnv };
+    delete process.env.TOGETHER_API_KEY;
+    mockOpenAI.mockClear();
+    mockEmbeddingsCreate.mockReset();
+    mockEmbeddingsCreate.mockResolvedValue({
+      data: [{ index: 0, embedding: mockEmbedding }],
+    });
+  });
+
+  afterAll(() => {
+    process.env = originalEnv;
+  });
+
+  it("uses Together defaults with an API key from config", async () => {
+    const embedder = new TogetherEmbedder({ apiKey: "test-key" });
+
+    await embedder.embed("hello");
+
+    expect(mockOpenAI).toHaveBeenCalledWith({
+      apiKey: "test-key",
+      baseURL: "https://api.together.ai/v1",
+    });
+    expect(mockEmbeddingsCreate).toHaveBeenCalledWith({
+      model: "intfloat/multilingual-e5-large-instruct",
+      input: "hello",
+      encoding_format: "float",
+    });
+  });
+
+  it("uses TOGETHER_API_KEY when config apiKey is not provided", async () => {
+    process.env.TOGETHER_API_KEY = "env-key";
+
+    const embedder = new TogetherEmbedder({});
+
+    await embedder.embed("hello");
+
+    expect(mockOpenAI).toHaveBeenCalledWith({
+      apiKey: "env-key",
+      baseURL: "https://api.together.ai/v1",
+    });
+  });
+
+  it("supports custom model and baseURL without forwarding embeddingDims", async () => {
+    const embedder = new TogetherEmbedder({
+      apiKey: "test-key",
+      model: "custom-together-embed",
+      baseURL: "https://proxy.example.com/v1",
+      embeddingDims: 512,
+    });
+
+    await embedder.embed("hello");
+
+    expect(mockOpenAI).toHaveBeenCalledWith({
+      apiKey: "test-key",
+      baseURL: "https://proxy.example.com/v1",
+    });
+    expect(mockEmbeddingsCreate).toHaveBeenCalledWith({
+      model: "custom-together-embed",
+      input: "hello",
+      encoding_format: "float",
+    });
+  });
+
+  it("uses url as a baseURL fallback", async () => {
+    const embedder = new TogetherEmbedder({
+      apiKey: "test-key",
+      url: "https://url-fallback.example.com/v1",
+    });
+
+    await embedder.embed("hello");
+
+    expect(mockOpenAI).toHaveBeenCalledWith({
+      apiKey: "test-key",
+      baseURL: "https://url-fallback.example.com/v1",
+    });
+  });
+
+  it("sorts batch embeddings by response index", async () => {
+    mockEmbeddingsCreate.mockResolvedValueOnce({
+      data: [
+        { index: 1, embedding: [0.3, 0.4] },
+        { index: 0, embedding: [0.1, 0.2] },
+      ],
+    });
+
+    const embedder = new TogetherEmbedder({ apiKey: "test-key" });
+
+    await expect(embedder.embedBatch(["first", "second"])).resolves.toEqual([
+      [0.1, 0.2],
+      [0.3, 0.4],
+    ]);
+  });
+
+  it("throws when no API key is available", () => {
+    expect(() => new TogetherEmbedder({})).toThrow(
+      "Together API key is required",
+    );
+  });
+});