llmjs2 1.3.7 → 1.3.9

package/index.js

@@ -138,7 +138,7 @@ class LLMJS2 {
       return {
         model: null,
         messages: [{ role: 'user', content: input }],
-        options: {}
+        options: { final: true }
       };
     }
 
@@ -152,13 +152,15 @@ class LLMJS2 {
         throw new Error('messages array cannot be empty');
       }
 
+      
+
       // Validate message format
       for (const msg of input.messages) {
         if (!msg.role || !msg.content) {
           throw new Error('Each message must have role and content properties');
         }
-        if (!['system', 'user', 'assistant'].includes(msg.role)) {
-          throw new Error('Message role must be system, user, or assistant');
+        if (!['system', 'user', 'assistant', 'tool'].includes(msg.role)) {
+          throw new Error('Message role must be system, user, tool or assistant');
         }
       }
 
@@ -175,7 +177,8 @@ class LLMJS2 {
           tools: input.tools,
           toolChoice: input.tool_choice || input.toolChoice,
           apiKey: input.apiKey,
-          timeout: input.timeout
+          timeout: input.timeout,
+          final: input.final ?? true
         }
       };
     }
@@ -189,6 +192,7 @@ class LLMJS2 {
   async completion(input) {
     try {
       const { model, messages, options } = this.validateInput(input);
+      const { final, ...providerOptions } = options;
 
       let provider, finalModel;
 
@@ -224,7 +228,7 @@ class LLMJS2 {
         messages: messages
       });
 
-      const result = await provider.createCompletion(messages, { ...options, model: finalModel });
+      const result = await provider.createCompletion(messages, { ...providerOptions, model: finalModel });
 
       // Log response information
       logger.info('LLMJS2 📥 Received from LLM provider', {
@@ -232,7 +236,7 @@ class LLMJS2 {
         ...result
       });
 
-      return result.content;
+      return final ? result.content : result;
 
     } catch (error) {
       // Sanitize error message to avoid leaking sensitive information

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "llmjs2",
-  "version": "1.3.7",
+  "version": "1.3.9",
   "description": "A unified Node.js library for connecting to multiple LLM providers: OpenAI, Ollama, and OpenRouter",
   "main": "index.js",
   "type": "commonjs",