ai-openai 1.0.14 → 1.0.16

package/README.md

@@ -13,49 +13,51 @@ Simplifies calling AI models for chat responses, learning advice, and more.
 - Easy integration with websites, Discord bots, or other apps
 
 ---
+
 ## Setup
 
-1. Get your free HuggingFace token at https://huggingface.co/settings/tokens
-2. Create a .env file and add your HuggingFace token:
+1. Get your free HuggingFace token at https://huggingface.co/settings/tokens  
+2. Create a `.env` file in your project root and add your HuggingFace token:
 HUGGINGFACE=your_token_here
 3. Then run your project.
 
+---
+
 ## Usage
+
 ```js
 const { chat } = require("ai-openai"); // import from npm package
 require("dotenv").config();
 
 (async () => {
   try {
-    const reply = await chat("What is x if y = 2x + 3 and y = 7?"); // call function directly
-    console.log("AI reply:", reply);
+    // Default usage
+    const reply = await chat("What is x if y = 2x + 3 and y = 7?");
+    console.log("Default model reply:", reply);
+
+    // Using a custom model
+    const reply2 = await chat("Explain black holes briefly.", {
+      model: "mistralai/Mistral-7B-Instruct-v0.2"
+    });
+    console.log("Custom model reply:", reply2);
+
+    // Using a custom system prompt
+    const reply3 = await chat("Explain black holes briefly.", {
+      system: "You are a physics professor explaining to a high school student."
+    });
+    console.log("Custom system prompt reply:", reply3);
+
   } catch (err) {
     console.error(err);
   }
 })();
 ```
-
-The default model used is openai/gpt-oss-120b
+- **Note:** The default model is openai/gpt-oss-120b.
 
 ---
 
-## Using a Custom Model (optional)
-```js
-const reply = await chat("Explain black holes", {
-  model: "mistralai/Mistral-7B-Instruct-v0.2"
-});
-
-```
-
-## Using a custom token (optional)
-If you don't want to use a .env file:
-```js
-const reply = await chat("Hello!", {
-  token: "hf_your_token_here"
-});
-
-```
 ## Installation
 
 ```bash
 npm install ai-openai
+

package/index.js

@@ -10,33 +10,37 @@ require("dotenv").config();
  * @returns {Promise<string>} - AI reply
  */
 async function chat(message, options = {}) {
-  const model = options.model || "openai/gpt-oss-120b";
-  const system = options.system || "You are a helpful assistant.";
-  const token = process.env.HUGGINGFACE;
+  if (!message) throw new Error("Message is required");
 
+  const token = process.env.HUGGINGFACE;
   if (!token) throw new Error("HuggingFace token required in .env as HUGGINGFACE");
-  if (!message) throw new Error("Message is required");
+
+  // Ensure model & system prompt are set correctly
+  const model = options.model ? String(options.model).trim() : "openai/gpt-oss-120b";
+  const system = options.system ? String(options.system).trim() : "You are a helpful assistant.";
 
   try {
-    const response = await axios.post(
-      "https://router.huggingface.co/v1/chat/completions",
-      {
-        model,
-        messages: [
-          { role: "system", content: system },
-          { role: "user", content: message }
-        ],
-      },
-      {
-        headers: {
-          Authorization: `Bearer ${token}`,
-          "Content-Type": "application/json"
-        },
-      }
-    );
-
-    // Extract reply text from router response
-    const aiText = response.data?.choices?.[0]?.message?.content || "No response";
+    const payload = {
+      model,
+      messages: [
+        { role: "system", content: system },
+        { role: "user", content: message }
+      ]
+    };
+
+    const headers = {
+      Authorization: `Bearer ${token}`,
+      "Content-Type": "application/json"
+    };
+
+    const response = await axios.post("https://router.huggingface.co/v1/chat/completions", payload, { headers });
+
+    // HuggingFace sometimes wraps message content differently; fallback robust extraction
+    const aiText =
+      response.data?.choices?.[0]?.message?.content ??
+      response.data?.choices?.[0]?.text ??
+      "No response";
+
     return aiText;
 
   } catch (err) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-openai",
-  "version": "1.0.14",
+  "version": "1.0.16",
   "description": "Unofficial helper for OpenAI-like models via HuggingFace",
   "main": "index.js",
   "homepage": "https://ai-openai.bolt.host/",