@elizaos/plugin-openai 1.5.18 → 1.6.0

package/dist/node/index.node.js

@@ -1,22 +1,3 @@
-import { createRequire } from "node:module";
-var __create = Object.create;
-var __getProtoOf = Object.getPrototypeOf;
-var __defProp = Object.defineProperty;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-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 __require = /* @__PURE__ */ createRequire(import.meta.url);
-
 // src/index.ts
 import { logger as logger10, ModelType as ModelType7 } from "@elizaos/core";
 
@@ -26,7 +7,11 @@ import { logger as logger2 } from "@elizaos/core";
 // src/utils/config.ts
 import { logger } from "@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";
@@ -117,7 +102,7 @@ function initializeOpenAI(_config, runtime) {
 
 // src/models/text.ts
 import { logger as logger3, ModelType } from "@elizaos/core";
-import { generateText } from "ai";
+import { generateText, streamText } from "ai";
 
 // src/providers/openai.ts
 import { createOpenAI } from "@ai-sdk/openai";
@@ -133,10 +118,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(EventType.MODEL_USED, {
+    runtime,
+    source: "openai",
     provider: "openai",
     type,
-    prompt,
+    prompt: truncatedPrompt,
     tokens: {
       prompt: promptTokens,
       completion: completionTokens,
@@ -149,34 +137,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);
-  logger3.log(`[OpenAI] Using ${modelType} model: ${modelName}`);
-  logger3.log(params.prompt);
-  const {
-    prompt,
-    stopSequences = [],
-    maxTokens = 8192,
-    temperature = 0.7,
-    frequencyPenalty = 0.7,
-    presencePenalty = 0.7
-  } = params;
-  const { text: openaiResponse, usage } = await generateText({
+  logger3.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 = 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 generateText(generateParams);
+  if (usage)
+    emitModelUsageEvent(runtime, modelType, params.prompt, usage);
+  return text;
 }
 async function handleTextSmall(runtime, params) {
   return generateTextByModelType(runtime, params, ModelType.TEXT_SMALL, getSmallModel);
@@ -268,7 +257,7 @@ async function handleTextEmbedding(runtime, params) {
 // src/models/image.ts
 import { logger as logger5, ModelType as ModelType3 } from "@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");
@@ -293,7 +282,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;
@@ -356,10 +345,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) {
@@ -426,33 +411,6 @@ function detectAudioMimeType(buffer) {
   logger6.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);
-    logger6.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) {
@@ -484,13 +442,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}`);
@@ -582,8 +534,7 @@ async function handleTextToSpeech(runtime, input) {
   const resolvedModel = options.model || getSetting(runtime, "OPENAI_TTS_MODEL", "gpt-4o-mini-tts");
   logger7.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);
     logger7.error(`Error in TEXT_TO_SPEECH: ${message}`);
@@ -813,7 +764,7 @@ var openaiPlugin = {
             try {
               const image = await runtime.useModel(ModelType7.IMAGE, {
                 prompt: "A beautiful sunset over a calm ocean",
-                n: 1,
+                count: 1,
                 size: "1024x1024"
               });
               logger10.log({ image }, "generated with test_image_generation");
@@ -866,7 +817,7 @@ var openaiPlugin = {
           name: "openai_test_text_tokenizer_encode",
           fn: async (runtime) => {
             const prompt = "Hello tokenizer encode!";
-            const tokens = await runtime.useModel(ModelType7.TEXT_TOKENIZER_ENCODE, { prompt });
+            const tokens = await runtime.useModel(ModelType7.TEXT_TOKENIZER_ENCODE, { prompt, modelType: ModelType7.TEXT_SMALL });
             if (!Array.isArray(tokens) || tokens.length === 0) {
               throw new Error("Failed to tokenize text: expected non-empty array of tokens");
             }
@@ -877,9 +828,10 @@ var openaiPlugin = {
           name: "openai_test_text_tokenizer_decode",
           fn: async (runtime) => {
             const prompt = "Hello tokenizer decode!";
-            const tokens = await runtime.useModel(ModelType7.TEXT_TOKENIZER_ENCODE, { prompt });
+            const tokens = await runtime.useModel(ModelType7.TEXT_TOKENIZER_ENCODE, { prompt, modelType: ModelType7.TEXT_SMALL });
             const decodedText = await runtime.useModel(ModelType7.TEXT_TOKENIZER_DECODE, {
-              tokens
+              tokens,
+              modelType: ModelType7.TEXT_SMALL
             });
             if (decodedText !== prompt) {
               throw new Error(`Decoded text does not match original. Expected "${prompt}", got "${decodedText}"`);
@@ -904,6 +856,49 @@ var openaiPlugin = {
               throw error;
             }
           }
+        },
+        {
+          name: "openai_test_text_generation_large",
+          fn: async (runtime) => {
+            try {
+              const result = await runtime.useModel(ModelType7.TEXT_LARGE, {
+                prompt: "Say hello in 5 words."
+              });
+              if (!result || result.length === 0) {
+                throw new Error("Text generation returned empty result");
+              }
+              logger10.log({ result }, "Text generation test completed");
+            } catch (error) {
+              const message = error instanceof Error ? error.message : String(error);
+              logger10.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(ModelType7.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");
+              }
+              logger10.log({ chunks: chunks.length, result: result.substring(0, 50) }, "Streaming test completed");
+            } catch (error) {
+              const message = error instanceof Error ? error.message : String(error);
+              logger10.error(`Error in openai_test_streaming: ${message}`);
+              throw error;
+            }
+          }
         }
       ]
     }
@@ -915,4 +910,4 @@ export {
   src_default as default
 };
 
-//# debugId=967FB48261E5AE3064756E2164756E21
+//# debugId=382E411BFB8DEF2564756E2164756E21