@elizaos/plugin-openai 1.5.18 → 1.6.0

package/dist/cjs/index.node.cjs

@@ -1,20 +1,7 @@
-var __create = Object.create;
-var __getProtoOf = Object.getPrototypeOf;
 var __defProp = Object.defineProperty;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __toESM = (mod, isNodeMode, target) => {
-  target = mod != null ? __create(__getProtoOf(mod)) : {};
-  const to = isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target;
-  for (let key of __getOwnPropNames(mod))
-    if (!__hasOwnProp.call(to, key))
-      __defProp(to, key, {
-        get: () => mod[key],
-        enumerable: true
-      });
-  return to;
-};
 var __moduleCache = /* @__PURE__ */ new WeakMap;
 var __toCommonJS = (from) => {
   var entry = __moduleCache.get(from), desc;
@@ -56,7 +43,11 @@ var import_core2 = require("@elizaos/core");
 // src/utils/config.ts
 var import_core = require("@elizaos/core");
 function getSetting(runtime, key, defaultValue) {
-  return runtime.getSetting(key) ?? process.env[key] ?? defaultValue;
+  const value = runtime.getSetting(key);
+  if (value !== undefined && value !== null) {
+    return String(value);
+  }
+  return process.env[key] ?? defaultValue;
 }
 function isBrowser() {
   return typeof globalThis !== "undefined" && "document" in globalThis && typeof globalThis.document !== "undefined";
@@ -163,10 +154,13 @@ function emitModelUsageEvent(runtime, type, prompt, usage) {
   const promptTokens = ("promptTokens" in usage ? usage.promptTokens : undefined) ?? ("inputTokens" in usage ? usage.inputTokens : undefined) ?? 0;
   const completionTokens = ("completionTokens" in usage ? usage.completionTokens : undefined) ?? ("outputTokens" in usage ? usage.outputTokens : undefined) ?? 0;
   const totalTokens = ("totalTokens" in usage ? usage.totalTokens : undefined) ?? promptTokens + completionTokens;
+  const truncatedPrompt = typeof prompt === "string" ? prompt.length > 200 ? `${prompt.slice(0, 200)}…` : prompt : "";
   runtime.emitEvent(import_core3.EventType.MODEL_USED, {
+    runtime,
+    source: "openai",
     provider: "openai",
     type,
-    prompt,
+    prompt: truncatedPrompt,
     tokens: {
       prompt: promptTokens,
       completion: completionTokens,
@@ -179,34 +173,35 @@ function emitModelUsageEvent(runtime, type, prompt, usage) {
 async function generateTextByModelType(runtime, params, modelType, getModelFn) {
   const openai = createOpenAIClient(runtime);
   const modelName = getModelFn(runtime);
-  const experimentalTelemetry = getExperimentalTelemetry(runtime);
-  import_core4.logger.log(`[OpenAI] Using ${modelType} model: ${modelName}`);
-  import_core4.logger.log(params.prompt);
-  const {
-    prompt,
-    stopSequences = [],
-    maxTokens = 8192,
-    temperature = 0.7,
-    frequencyPenalty = 0.7,
-    presencePenalty = 0.7
-  } = params;
-  const { text: openaiResponse, usage } = await import_ai.generateText({
+  import_core4.logger.debug(`[OpenAI] ${modelType} model: ${modelName}`);
+  const generateParams = {
     model: openai.languageModel(modelName),
-    prompt,
+    prompt: params.prompt,
     system: runtime.character.system ?? undefined,
-    temperature,
-    maxOutputTokens: maxTokens,
-    frequencyPenalty,
-    presencePenalty,
-    stopSequences,
-    experimental_telemetry: {
-      isEnabled: experimentalTelemetry
-    }
-  });
-  if (usage) {
-    emitModelUsageEvent(runtime, modelType, prompt, usage);
+    temperature: params.temperature ?? 0.7,
+    maxOutputTokens: params.maxTokens ?? 8192,
+    frequencyPenalty: params.frequencyPenalty ?? 0.7,
+    presencePenalty: params.presencePenalty ?? 0.7,
+    stopSequences: params.stopSequences ?? [],
+    experimental_telemetry: { isEnabled: getExperimentalTelemetry(runtime) }
+  };
+  if (params.stream) {
+    const result = import_ai.streamText(generateParams);
+    return {
+      textStream: result.textStream,
+      text: result.text,
+      usage: result.usage.then((u) => u ? {
+        promptTokens: u.inputTokens ?? 0,
+        completionTokens: u.outputTokens ?? 0,
+        totalTokens: (u.inputTokens ?? 0) + (u.outputTokens ?? 0)
+      } : undefined),
+      finishReason: result.finishReason
+    };
   }
-  return openaiResponse;
+  const { text, usage } = await import_ai.generateText(generateParams);
+  if (usage)
+    emitModelUsageEvent(runtime, modelType, params.prompt, usage);
+  return text;
 }
 async function handleTextSmall(runtime, params) {
   return generateTextByModelType(runtime, params, import_core4.ModelType.TEXT_SMALL, getSmallModel);
@@ -298,7 +293,7 @@ async function handleTextEmbedding(runtime, params) {
 // src/models/image.ts
 var import_core6 = require("@elizaos/core");
 async function handleImageGeneration(runtime, params) {
-  const n = params.n || 1;
+  const n = params.count || 1;
   const size = params.size || "1024x1024";
   const prompt = params.prompt;
   const modelName = getSetting(runtime, "OPENAI_IMAGE_MODEL", "gpt-image-1");
@@ -323,7 +318,7 @@ async function handleImageGeneration(runtime, params) {
     }
     const data = await response.json();
     const typedData = data;
-    return typedData;
+    return typedData.data;
   } catch (error) {
     const message = error instanceof Error ? error.message : String(error);
     throw error;
@@ -386,10 +381,6 @@ async function handleImageDescription(runtime, params) {
         description: "No response from API"
       };
     }
-    const isCustomPrompt = typeof params === "object" && Boolean(params.prompt) && params.prompt !== DEFAULT_PROMPT;
-    if (isCustomPrompt) {
-      return content;
-    }
     const titleMatch = content.match(/title[:\s]+(.+?)(?:\n|$)/i);
     const title = titleMatch?.[1]?.trim();
     if (!title) {
@@ -456,33 +447,6 @@ function detectAudioMimeType(buffer) {
   import_core7.logger.warn("Could not detect audio format from buffer, using generic binary type");
   return "application/octet-stream";
 }
-async function webStreamToNodeStream(webStream) {
-  try {
-    const { Readable } = await import("node:stream");
-    const reader = webStream.getReader();
-    return new Readable({
-      async read() {
-        try {
-          const { done, value } = await reader.read();
-          if (done) {
-            this.push(null);
-          } else {
-            this.push(value);
-          }
-        } catch (error) {
-          this.destroy(error);
-        }
-      },
-      destroy(error, callback) {
-        reader.cancel().finally(() => callback(error));
-      }
-    });
-  } catch (error) {
-    const message = error instanceof Error ? error.message : String(error);
-    import_core7.logger.error(`Failed to load node:stream module: ${message}`);
-    throw new Error(`Cannot convert stream: node:stream module unavailable. This feature requires a Node.js environment.`);
-  }
-}
 
 // src/models/audio.ts
 async function fetchTextToSpeech(runtime, options) {
@@ -514,13 +478,7 @@ async function fetchTextToSpeech(runtime, options) {
       const err = await res.text();
       throw new Error(`OpenAI TTS error ${res.status}: ${err}`);
     }
-    if (!res.body) {
-      throw new Error("OpenAI TTS response body is null");
-    }
-    if (!isBrowser()) {
-      return await webStreamToNodeStream(res.body);
-    }
-    return res.body;
+    return await res.arrayBuffer();
   } catch (err) {
     const message = err instanceof Error ? err.message : String(err);
     throw new Error(`Failed to fetch speech from OpenAI TTS: ${message}`);
@@ -612,8 +570,7 @@ async function handleTextToSpeech(runtime, input) {
   const resolvedModel = options.model || getSetting(runtime, "OPENAI_TTS_MODEL", "gpt-4o-mini-tts");
   import_core8.logger.log(`[OpenAI] Using TEXT_TO_SPEECH model: ${resolvedModel}`);
   try {
-    const speechStream = await fetchTextToSpeech(runtime, options);
-    return speechStream;
+    return await fetchTextToSpeech(runtime, options);
   } catch (error) {
     const message = error instanceof Error ? error.message : String(error);
     import_core8.logger.error(`Error in TEXT_TO_SPEECH: ${message}`);
@@ -843,7 +800,7 @@ var openaiPlugin = {
             try {
               const image = await runtime.useModel(import_core13.ModelType.IMAGE, {
                 prompt: "A beautiful sunset over a calm ocean",
-                n: 1,
+                count: 1,
                 size: "1024x1024"
               });
               import_core13.logger.log({ image }, "generated with test_image_generation");
@@ -896,7 +853,7 @@ var openaiPlugin = {
           name: "openai_test_text_tokenizer_encode",
           fn: async (runtime) => {
             const prompt = "Hello tokenizer encode!";
-            const tokens = await runtime.useModel(import_core13.ModelType.TEXT_TOKENIZER_ENCODE, { prompt });
+            const tokens = await runtime.useModel(import_core13.ModelType.TEXT_TOKENIZER_ENCODE, { prompt, modelType: import_core13.ModelType.TEXT_SMALL });
             if (!Array.isArray(tokens) || tokens.length === 0) {
               throw new Error("Failed to tokenize text: expected non-empty array of tokens");
             }
@@ -907,9 +864,10 @@ var openaiPlugin = {
           name: "openai_test_text_tokenizer_decode",
           fn: async (runtime) => {
             const prompt = "Hello tokenizer decode!";
-            const tokens = await runtime.useModel(import_core13.ModelType.TEXT_TOKENIZER_ENCODE, { prompt });
+            const tokens = await runtime.useModel(import_core13.ModelType.TEXT_TOKENIZER_ENCODE, { prompt, modelType: import_core13.ModelType.TEXT_SMALL });
             const decodedText = await runtime.useModel(import_core13.ModelType.TEXT_TOKENIZER_DECODE, {
-              tokens
+              tokens,
+              modelType: import_core13.ModelType.TEXT_SMALL
             });
             if (decodedText !== prompt) {
               throw new Error(`Decoded text does not match original. Expected "${prompt}", got "${decodedText}"`);
@@ -934,6 +892,49 @@ var openaiPlugin = {
               throw error;
             }
           }
+        },
+        {
+          name: "openai_test_text_generation_large",
+          fn: async (runtime) => {
+            try {
+              const result = await runtime.useModel(import_core13.ModelType.TEXT_LARGE, {
+                prompt: "Say hello in 5 words."
+              });
+              if (!result || result.length === 0) {
+                throw new Error("Text generation returned empty result");
+              }
+              import_core13.logger.log({ result }, "Text generation test completed");
+            } catch (error) {
+              const message = error instanceof Error ? error.message : String(error);
+              import_core13.logger.error(`Error in openai_test_text_generation_large: ${message}`);
+              throw error;
+            }
+          }
+        },
+        {
+          name: "openai_test_streaming",
+          fn: async (runtime) => {
+            try {
+              const chunks = [];
+              const result = await runtime.useModel(import_core13.ModelType.TEXT_LARGE, {
+                prompt: "Count from 1 to 5.",
+                onStreamChunk: (chunk) => {
+                  chunks.push(chunk);
+                }
+              });
+              if (!result || result.length === 0) {
+                throw new Error("Streaming returned empty result");
+              }
+              if (chunks.length === 0) {
+                throw new Error("No streaming chunks received");
+              }
+              import_core13.logger.log({ chunks: chunks.length, result: result.substring(0, 50) }, "Streaming test completed");
+            } catch (error) {
+              const message = error instanceof Error ? error.message : String(error);
+              import_core13.logger.error(`Error in openai_test_streaming: ${message}`);
+              throw error;
+            }
+          }
         }
       ]
     }
@@ -941,4 +942,4 @@ var openaiPlugin = {
 };
 var src_default = openaiPlugin;
 
-//# debugId=9B5BF59394995A3864756E2164756E21
+//# debugId=F03D6434951BC08A64756E2164756E21