Added Mem0 TS Library (#2270)

mem0-ts/src/oss/src/llms/openai.ts

@@ -0,0 +1,43 @@
+import OpenAI from "openai";
+import { LLM, LLMResponse } from "./base";
+import { LLMConfig, Message } from "../types";
+
+export class OpenAILLM implements LLM {
+  private openai: OpenAI;
+  private model: string;
+
+  constructor(config: LLMConfig) {
+    this.openai = new OpenAI({ apiKey: config.apiKey });
+    this.model = config.model || "gpt-4-turbo-preview";
+  }
+
+  async generateResponse(
+    messages: Message[],
+    responseFormat?: { type: string },
+  ): Promise<string> {
+    const completion = await this.openai.chat.completions.create({
+      messages: messages.map((msg) => ({
+        role: msg.role as "system" | "user" | "assistant",
+        content: msg.content,
+      })),
+      model: this.model,
+      response_format: responseFormat as { type: "text" | "json_object" },
+    });
+    return completion.choices[0].message.content || "";
+  }
+
+  async generateChat(messages: Message[]): Promise<LLMResponse> {
+    const completion = await this.openai.chat.completions.create({
+      messages: messages.map((msg) => ({
+        role: msg.role as "system" | "user" | "assistant",
+        content: msg.content,
+      })),
+      model: this.model,
+    });
+    const response = completion.choices[0].message;
+    return {
+      content: response.content || "",
+      role: response.role,
+    };
+  }
+}