@memberjunction/ai-openai 0.9.9

package/dist/index.d.ts

@@ -0,0 +1 @@
+export * from './models/openAI';

package/dist/index.js

@@ -0,0 +1,18 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./models/openAI"), exports);
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,kDAAgC"}

package/dist/models/openAI.d.ts

@@ -0,0 +1,11 @@
+import { BaseLLM, ChatMessage, ChatParams, ChatResult, ClassifyParams, ClassifyResult, SummarizeParams, SummarizeResult } from "@memberjunction/ai";
+import { ChatCompletionRequestMessage } from "openai";
+export declare class OpenAILLM extends BaseLLM {
+    static _openAI: any;
+    constructor(apiKey: string);
+    ChatCompletion(params: ChatParams): Promise<ChatResult>;
+    SummarizeText(params: SummarizeParams): Promise<SummarizeResult>;
+    ClassifyText(params: ClassifyParams): Promise<ClassifyResult>;
+    ConvertMJToOpenAIChatMessages(messages: ChatMessage[]): ChatCompletionRequestMessage[];
+    ConvertMJToOpenAIRole(role: string): "user" | "system" | "assistant";
+}

package/dist/models/openAI.js

@@ -0,0 +1,99 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var OpenAILLM_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OpenAILLM = void 0;
+const ai_1 = require("@memberjunction/ai");
+const openai_1 = require("openai");
+const global_1 = require("@memberjunction/global");
+let OpenAILLM = OpenAILLM_1 = class OpenAILLM extends ai_1.BaseLLM {
+    constructor(apiKey) {
+        super(apiKey);
+        const configuration = new openai_1.Configuration({
+            apiKey: apiKey
+        });
+        if (!OpenAILLM_1._openAI)
+            OpenAILLM_1._openAI = new openai_1.OpenAIApi(configuration);
+    }
+    async ChatCompletion(params) {
+        const messages = this.ConvertMJToOpenAIChatMessages(params.messages);
+        const startTime = new Date();
+        const result = await OpenAILLM_1._openAI.createChatCompletion({
+            model: params.model,
+            messages: messages,
+            temperature: params.temperature,
+        });
+        const endTime = new Date();
+        const timeElapsed = endTime.getTime() - startTime.getTime();
+        return {
+            data: {
+                choices: result.data.choices.map((c) => {
+                    return {
+                        message: {
+                            role: openai_1.ChatCompletionRequestMessageRoleEnum[c.message.role],
+                            content: c.message.content
+                        },
+                        finish_reason: c.finish_reason,
+                        index: c.index
+                    };
+                }),
+                usage: new ai_1.ModelUsage(result.data.usage.prompt_tokens, result.data.usage.completion_tokens)
+            },
+            success: result.status === 200,
+            statusText: result.statusText,
+            startTime: startTime,
+            endTime: endTime,
+            timeElapsed: timeElapsed,
+            errorMessage: null,
+            exception: null
+        };
+    }
+    async SummarizeText(params) {
+        const messages = this.ConvertMJToOpenAIChatMessages(params.messages);
+        const startTime = new Date();
+        const result = await OpenAILLM_1._openAI.createChatCompletion({
+            model: params.model,
+            messages: messages
+        });
+        const endTime = new Date();
+        const success = result.data && result.data.choices && result.data.choices.length > 0;
+        let summaryText = null;
+        if (success)
+            summaryText = result.data.choices[0].message.content;
+        return new ai_1.SummarizeResult((0, ai_1.GetUserMessageFromChatParams)(params), summaryText, success, startTime, endTime);
+    }
+    async ClassifyText(params) {
+        throw new Error("Method not implemented.");
+    }
+    ConvertMJToOpenAIChatMessages(messages) {
+        // add user messages - using types OpenAI likes
+        return messages.map(m => {
+            return {
+                role: this.ConvertMJToOpenAIRole(m.role),
+                content: m.content
+            };
+        });
+    }
+    ConvertMJToOpenAIRole(role) {
+        switch (role.trim().toLowerCase()) {
+            case 'system':
+                return openai_1.ChatCompletionRequestMessageRoleEnum.System;
+            case 'user':
+                return openai_1.ChatCompletionRequestMessageRoleEnum.User;
+            case 'assistant':
+                return openai_1.ChatCompletionRequestMessageRoleEnum.Assistant;
+            default:
+                throw new Error(`Unknown role ${role}`);
+        }
+    }
+};
+exports.OpenAILLM = OpenAILLM;
+exports.OpenAILLM = OpenAILLM = OpenAILLM_1 = __decorate([
+    (0, global_1.RegisterClass)(ai_1.BaseLLM, 'OpenAILLM')
+], OpenAILLM);
+//# sourceMappingURL=openAI.js.map

package/dist/models/openAI.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openAI.js","sourceRoot":"","sources":["../../src/models/openAI.ts"],"names":[],"mappings":";;;;;;;;;;AAAA,2CAA8L;AAC9L,mCAAsH;AACtH,mDAAuD;AAGhD,IAAM,SAAS,iBAAf,MAAM,SAAU,SAAQ,YAAO;IAGlC,YAAY,MAAc;QACtB,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,MAAM,aAAa,GAAG,IAAI,sBAAa,CAAC;YACpC,MAAM,EAAE,MAAM;SACjB,CAAC,CAAC;QACH,IAAI,CAAC,WAAS,CAAC,OAAO;YAClB,WAAS,CAAC,OAAO,GAAG,IAAI,kBAAS,CAAC,aAAa,CAAC,CAAC;IACzD,CAAC;IAEM,KAAK,CAAC,cAAc,CAAC,MAAkB;QAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,6BAA6B,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAGrE,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,MAAM,WAAS,CAAC,OAAO,CAAC,oBAAoB,CAAC;YACxD,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,QAAQ,EAAE,QAAQ;YAClB,WAAW,EAAE,MAAM,CAAC,WAAW;SAClC,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,IAAI,EAAE,CAAC;QAC3B,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,EAAE,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC;QAE5D,OAAO;YACH,IAAI,EAAE;gBACF,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAyF,EAAE,EAAE;oBAC3H,OAAO;wBACH,OAAO,EAAE;4BACL,IAAI,EAAE,6CAAoC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC;4BAC1D,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO;yBAC7B;wBACD,aAAa,EAAE,CAAC,CAAC,aAAa;wBAC9B,KAAK,EAAE,CAAC,CAAC,KAAK;qBACjB,CAAA;gBACL,CAAC,CAAC;gBACF,KAAK,EAAE,IAAI,eAAU,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC;aAC9F;YACD,OAAO,EAAE,MAAM,CAAC,MAAM,KAAK,GAAG;YAC9B,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,SAAS,EAAE,SAAS;YACpB,OAAO,EAAE,OAAO;YAChB,WAAW,EAAE,WAAW;YACxB,YAAY,EAAE,IAAI;YAClB,SAAS,EAAE,IAAI;SAClB,CAAA;IACL,CAAC;IAGM,KAAK,CAAC,aAAa,CAAC,MAAuB;QAC9C,MAAM,QAAQ,GAAG,IAAI,CAAC,6BAA6B,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAGrE,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,MAAM,WAAS,CAAC,OAAO,CAAC,oBAAoB,CAAC;YACxD,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,QAAQ,EAAE,QAAQ;SACrB,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,IAAI,EAAE,CAAC;QAE3B,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;QACrF,IAAI,WAAW,GAAG,IAAI,CAAC;QACvB,IAAI,OAAO;YACP,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;QAEzD,OAAO,IAAI,oBAAe,CAAC,IAAA,iCAA4B,EAAC,MAAM,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;IAC/G,CAAC;IAEM,KAAK,CAAC,YAAY,CAAC,MAAsB;QAC5C,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;IAEM,6BAA6B,CAAC,QAAuB;QACxD,+CAA+C;QAC/C,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;YACpB,OAAO;gBACH,IAAI,EAAE,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC;gBACxC,OAAO,EAAE,CAAC,CAAC,OAAO;aACrB,CAAA;QACL,CAAC,CAAC,CAAC;IACP,CAAC;IAEM,qBAAqB,CAAC,IAAY;QACrC,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC;YAChC,KAAK,QAAQ;gBACT,OAAO,6CAAoC,CAAC,MAAM,CAAA;YACtD,KAAK,MAAM;gBACP,OAAO,6CAAoC,CAAC,IAAI,CAAA;YACpD,KAAK,WAAW;gBACZ,OAAO,6CAAoC,CAAC,SAAS,CAAA;YACzD;gBACI,MAAM,IAAI,KAAK,CAAC,gBAAgB,IAAI,EAAE,CAAC,CAAA;QAC/C,CAAC;IACL,CAAC;CACJ,CAAA;AA/FY,8BAAS;oBAAT,SAAS;IADrB,IAAA,sBAAa,EAAC,YAAO,EAAE,WAAW,CAAC;GACvB,SAAS,CA+FrB"}

package/package.json

@@ -0,0 +1,27 @@
+{
+  "name": "@memberjunction/ai-openai",
+  "version": "0.9.9",
+  "description": "MemberJunction Wrapper for OpenAI AI Models",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "files": [
+    "/dist"
+  ],
+  "scripts": {
+    "start": "ts-node-dev src/index.ts",
+    "build": "tsc",
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
+  "author": "MemberJunction.com",
+  "license": "ISC",
+  "devDependencies": {
+    "ts-node-dev": "^2.0.0",
+    "typescript": "^5.3.3",
+    "openai": "^3.2.1"
+  },
+  "dependencies": {
+    "@memberjunction/ai": "^0.9.152",
+    "@memberjunction/global": "^0.9.146",
+    "@google/generative-ai": "~0.2.1"
+  }
+}

package/readme.md

@@ -0,0 +1,2 @@
+# @memberjunction/ai-gemini
+Simple wrapper class for OpenAI's AI Models to use with the MemberJunction framework.