Update TeapotAI.js

TeapotAI.js

@@ -1,15 +1,23 @@
 import { pipeline, env } from '@huggingface/transformers';
 
+env.cacheDir = './.cache';
+
 class TeapotAI {
   /**
   * Initializes the TeapotAI class.
   * @param {object} options - Configuration options.
   * @param {string} [options.modelId='teapotai/teapotllm'] - The Hugging Face model ID.
   * @param {boolean} [options.verbose=true] - Whether to print status messages.
+  * @param {object} [options.pipelineOptions={}] - Additional pipeline options passed to the transformer.
   */
-  constructor({ modelId = 'teapotai/teapotllm', verbose = false } = {}) {
+  constructor({
+    modelId = 'teapotai/teapotllm',
+    verbose = false,
+    pipelineOptions = {}
+  } = {}) {
     this.modelId = modelId;
     this.verbose = verbose;
+    this.pipelineOptions = pipelineOptions;
     this.generator = null;
     this.isInitialized = false;
     
@@ -29,7 +37,10 @@ class TeapotAI {
     }
     try {
       if (this.verbose) console.log(`Initializing generator pipeline for model: ${this.modelId}...`);
-      const pipelineOptions = { model: this.modelId };
+      const pipelineOptions = {
+        model: this.modelId,
+        ...this.pipelineOptions
+      };
       
       this.generator = await pipeline('text2text-generation', pipelineOptions.model, pipelineOptions);
       this.isInitialized = true;
@@ -65,7 +76,7 @@ class TeapotAI {
         max_new_tokens: 512,
       });
       
-      const generatedText = output[0]?.generated_text ?? "Error: Could not generate text.";
+      const generatedText = output[0]?.generated_text?.trim() ?? "Error: Could not generate text.";
       if (this.verbose) console.log("Text generation complete.");
       return generatedText;