feat: simplified usage to support lazyloader

Author: frankpagan

src/client.js

@@ -70,9 +70,9 @@ const componentsReference = {
 
 async function send(conversation) {
     try {
-
         let data = await crud.socket.send({
             method: 'openai.chat',
+            action: 'chat',
             chat: {
                 messages: conversation,
                 max_tokens,
@@ -100,6 +100,7 @@ async function send(conversation) {
         //         console.error('Invalid CoCreateJS API function:', component, action);
         //     }
         // }
+
         document.dispatchEvent(new CustomEvent('openAi', {
             detail: {}
         }));

src/server.js

@@ -1,91 +1,38 @@
 const OpenAI = require('openai')
 
-class CoCreateOpenAi {
-    constructor(crud) {
-        this.name = "openAi"
-        this.wsManager = crud.wsManager
-        this.crud = crud
-        this.organizations = {}
-        this.init()
-    }
-
-    init() {
-        if (this.wsManager) {
-            this.wsManager.on(this.name, (data) => this.send(data));
-            this.wsManager.on('openai.chat', (data) => this.send(data));
-        }
-    }
-
-    async send(data) {
-        try {
-            let organization = await this.organizations[data.organization_id]
-            if (!organization) {
-                if (data.chat.apiKey) {
-                    organization = new OpenAI({ apiKey: data.chat.apiKey });
-                    this.organizations[data.organization_id] = organization
-                } else {
-                    let apiKey = await this.getApiKey(data.organization_id, this.name)
-                    organization = new OpenAI({ apiKey });
-                    this.organizations[data.organization_id] = organization
-                }
-            }
-
-            const openai = organization
-            // let response
-            switch (data.method) {
-                case 'chat':
-                case 'chat.completions':
-                case 'chat.completions.create':
-                case 'openai.chat':
-                    delete data.chat.apiKey
-                    data.chat = await openai.chat.completions.create(data.chat);
-                    break;
-            }
-
-            if (data.socket) {
-                let socket = data.socket
-                delete data.socket
-                socket.send(JSON.stringify(data))
-            } else
-                return data
-        } catch (error) {
-            console.error('OpenAi Error:', error);
-            data.error = error.message
-            let socket = data.socket
-            delete data.socket
-            socket.send(JSON.stringify(data))
-        }
-    }
-
-    async getApiKey(organization_id, name) {
-        this.organizations[organization_id] = this.getOrganization(organization_id, name)
-        this.organizations[organization_id] = await this.organizations[organization_id]
-        return this.organizations[organization_id]
-    }
+const name = 'openai'
 
-    async getOrganization(organization_id, name) {
-        let organization = await this.crud.send({
-            method: 'object.read',
-            database: organization_id,
-            array: 'organizations',
-            object: [{ _id: organization_id }],
-            organization_id
-        })
+async function send(data) {
+    let openai = false;
 
-        if (organization
-            && organization.object
-            && organization.object[0]) {
-            if (organization.object[0].apis && organization.object[0].apis[name]) {
-                return organization.object[0].apis && organization.object[0].apis[name]
-            } else
-                return { [this.name]: false, error: 'An apikey could not be found' }
+    try {
+        // let organization = await this.organizations[data.organization_id]
+        // if (!organization) {
+        if (data.chat.apiKey) {
+            openai = new OpenAI({ apiKey: data.chat.apiKey });
         } else {
-            return { serverOrganization: false, error: 'An organization could not be found' }
+            let apiKey = data.apis
+            openai = new OpenAI({ apiKey });
+        }
+        // }
+
+        // let response
+        switch (data.method) {
+            case 'chat':
+            case 'chat.completions':
+            case 'chat.completions.create':
+            case 'openai.chat':
+                delete data.chat.apiKey
+                data.chat = await openai.chat.completions.create(data.chat);
+                break;
         }
 
+        return data
+    } catch (error) {
+        console.error('OpenAi Error:', error);
+        data.error = error.message
+        return data
     }
-
-
 }
 
-module.exports = CoCreateOpenAi;
+module.exports = { send };