nodebb-plugin-integration 0.4.2 → 0.4.4

package/src/ai/openai.js

@@ -1,331 +1,75 @@
-"use strict";
-
-
-/*
- * NodeBB Integration Layer
- *
- * OpenAI Adapter
- *
- * Provides:
- *
- * - Chat completion
- * - Embeddings
- *
- * Safe modes:
- *
- * - Development
- * - Test
- * - Production
- */
-
-
-
-const Config =
-    require("../config");
-
-
-
-let client = null;
-
-
-
-
-
-
-
-
-/*
- * Check API availability
- */
-
-
-function isAvailable() {
-
-
-    return Boolean(
-
-        Config.openai.apiKey
-
-    );
-
-
-}
-
-
-
-
-
-
-
-
-
-/*
- * Lazy OpenAI client creation
- */
-
-
-async function getClient() {
-
-
-
-    if(client) {
-
-
-        return client;
-
-
-    }
-
-
-
-
-
-
-
-    if(!isAvailable()) {
-
-
-        console.log(
-
-            "[OpenAI] API key not configured"
-
-        );
-
-
-        return null;
-
-
-    }
-
-
-
-
-
-
-
-
-    try {
-
-
-
-        const OpenAI =
-            require("openai");
-
-
-
-
-
-        client =
-            new OpenAI({
-
-                apiKey:
-                    Config.openai.apiKey
-
-            });
-
-
-
-
-
-
-        return client;
-
-
-    }
-
-
-    catch(error) {
-
-
-
-        console.error(
-
-            "[OpenAI] initialization failed",
-
-            error
-
-        );
-
-
-
-        return null;
-
-
-    }
-
-
-}
-
-
-
-
-
-
-
-
-
-/*
- * Create embedding vector
- */
-
-
-async function embeddings(text) {
-
-
-
-    const openai =
-        await getClient();
-
-
-
-
-
-    if(!openai) {
-
-
-
-        return null;
-
-
-    }
-
-
-
-
-
-
-
-
-    if(!text || !text.trim()) {
-
-
-
-        return null;
-
-
-    }
-
-
-
-
-
-
-
-
-    const response =
-        await openai.embeddings.create({
-
-            model:
-                Config.openai.embeddingModel,
-
-
-            input:
-                text
-
-
-        });
-
-
-
-
-
-
-    return response
-        .data[0]
-        .embedding;
-
-
-}
-
-
-
-
-
-
-
-
-
-/*
- * Chat completion
- */
-
-
-async function chat(messages) {
-
-
-
-    const openai =
-        await getClient();
-
-
-
-
-
-    if(!openai) {
-
-
-        return null;
-
-
-    }
-
-
-
-
-
-
-
-    const response =
-        await openai.chat.completions.create({
-
-
-            model:
-                Config.openai.model,
-
-
-
-            messages
-
-
-
-        });
-
-
-
-
-
-
-
-    return response
-        .choices[0]
-        .message
-        .content;
-
-
-}
-
-
-
-
-
-
-
-
-
-module.exports = {
-
-
-    isAvailable,
-
-
-    getClient,
-
-
-    embeddings,
-
-
-    chat
-
-
-
+"use strict";
+
+const OpenAI = require("openai");
+const Config = require("../config");
+
+let client = null;
+let clientKey = null;
+
+async function getClient() {
+    const settings = await Config.get();
+
+    if (settings.aiProvider !== "openai") {
+        const error = new Error(
+            "OpenAI is disabled because AI provider is set to fake"
+        );
+        error.code = "AI_DISABLED";
+        throw error;
+    }
+
+    if (!settings.openaiApiKey) {
+        throw new Error("OpenAI API key is not configured");
+    }
+
+    if (client && clientKey === settings.openaiApiKey) {
+        return client;
+    }
+
+    client = new OpenAI({
+        apiKey: settings.openaiApiKey,
+        timeout: 60000,
+        maxRetries: 2
+    });
+
+    clientKey = settings.openaiApiKey;
+
+    return client;
+}
+
+async function createEmbedding(input) {
+    if (typeof input !== "string" || !input.trim()) {
+        throw new TypeError("Embedding input must be a non-empty string");
+    }
+
+    const settings = await Config.get();
+    const openai = await getClient();
+
+    const response = await openai.embeddings.create({
+        model: settings.openaiModel,
+        input: input.trim(),
+        dimensions: settings.embeddingSize,
+        encoding_format: "float"
+    });
+
+    const embedding = response.data?.[0]?.embedding;
+
+    if (!Array.isArray(embedding)) {
+        throw new Error("OpenAI returned an invalid embedding");
+    }
+
+    if (embedding.length !== settings.embeddingSize) {
+        throw new Error(
+            `Embedding dimension mismatch: expected ${settings.embeddingSize}, received ${embedding.length}`
+        );
+    }
+
+    if (!embedding.every(Number.isFinite)) {
+        throw new Error("OpenAI returned non-finite embedding values");
+    }
+
+    return embedding;
+}
+
+module.exports = {
+    createEmbedding
 };