fg-lib-gpt 1.0.0 → 1.1.1

package/main.js

@@ -1,3 +1,84 @@
+const axios = require('axios')
+
+const OPENAI_API_KEY = process.env.OPENAI_API_KEY
+
+const DEFAULT_MODEL = process.env.GPT_DEFAULT_MODEL || 'gpt-4o-mini'
+const DEFAULT_TIMEOUT = process.env.GPT_TIMEOUT || 180000
+const DEFAULT_TEMPERATURE = process.env.GPT_TEMPERATURE || 0.7
+
+
+const doGpt = async ({
+    messages,
+    model = DEFAULT_MODEL,
+    timeout = DEFAULT_TIMEOUT,
+    temperature = DEFAULT_TEMPERATURE,
+    validate,
+    response_format = { 'type': 'text' }
+}) => {
+
+    try {
+        const response = await Promise.race([
+            getGptResponse({ messages, model, temperature, response_format }),
+            new Promise((_, reject) =>
+                setTimeout(() => reject(new Error('Custom timeout')), timeout))
+        ])
+
+        if (typeof validate === 'function')
+            validate(response)
+
+
+        const json = response_format.type === 'json' || response_format.type === 'json_schema'
+
+        // JSON.parse might also fail
+        return {
+            success: true,
+            content: json ? JSON.parse(response) : response
+        }
+    }
+    catch (e) {
+        console.error('**** ERROR ***** OpenAI Error (1): ' + e.message)
+        try {
+            const response = await getGptResponse({ messages, model, temperature, response_format })
+
+            return {
+                success: true,
+                content: json ? JSON.parse(response) : response
+            }
+        }
+        catch (e2) {
+            console.error('**** ERROR ***** OpenAI Error (2): ' + e2.message)
+            return {
+                success: false,
+                content: '🚫 An error occurred. Please try again. If the problem persists, please contact @fgalkov.'
+            }
+        }
+    }
+}
+
+
+
+const getGptResponse = async ({ messages, model, response_format, temperature }) => {
+
+    const res = await axios.post(
+        'https://api.openai.com/v1/chat/completions',
+        {
+            messages,
+            model,
+            response_format,
+            temperature
+        },
+        {
+            headers: {
+                'Authorization': 'Bearer ' + OPENAI_API_KEY,
+                'Content-Type': 'application/json',
+            }
+        }
+    )
+    return res.data.choices[0].message.content.trim()
+}
+
+
+
 const generateSchema = (schemaObj, name = 'schema') => {
 
     const schema = obj => {
@@ -52,5 +133,6 @@ const generateSchema = (schemaObj, name = 'schema') => {
 
 
 module.exports = {
-    generateSchema: generateSchema
+    doGpt: doGpt,
+    generateSchema: generateSchema,
 }

package/package.json

@@ -1,11 +1,14 @@
 {
   "name": "fg-lib-gpt",
-  "version": "1.0.0",
+  "version": "1.1.1",
   "main": "main.js",
   "scripts": {
     "test": "echo \"Error: no test specified\" && exit 1"
   },
   "author": "",
   "license": "ISC",
-  "description": ""
+  "description": "",
+  "dependencies": {
+    "axios": "^1.7.7"
+  }
 }