@contentgrowth/llm-service 0.6.5 → 0.6.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@contentgrowth/llm-service",
-  "version": "0.6.5",
+  "version": "0.6.7",
   "description": "Unified LLM Service for Content Growth",
   "main": "src/index.js",
   "type": "module",

package/src/llm/providers/gemini-provider.js

@@ -314,10 +314,23 @@ export class GeminiProvider extends BaseLLMProvider {
             };
         }
 
+        const parts = [{ text: prompt }];
+
+        if (options.images && options.images.length > 0) {
+            options.images.forEach(img => {
+                parts.push({
+                    inlineData: {
+                        data: img.data,
+                        mimeType: img.mimeType
+                    }
+                });
+            });
+        }
+
         const result = await model.generateContent({
             contents: [{
                 role: "user",
-                parts: [{ text: prompt }]
+                parts: parts
             }],
             generationConfig
         });

package/src/llm-service.js

@@ -159,18 +159,12 @@ export class LLMService {
         }
     }
 
-    /**
-     * Wrap of chatCompletion to handle toolcalls from LLM.
-     * @param {Array} messages - Conversation messages
-     * @param {string} tenantId - Tenant identifier
-     * @param {string} systemPrompt - System instructions
-     * @param {Array} tools - Tools array
-     * @returns {Object} Response with content, tool_calls, and optionally parsedContent
-     */
     async chatWithTools(messages, tenantId, systemPrompt, tools = []) {
         const provider = await this._getProvider(tenantId);
 
         let currentMessages = [...messages];
+        const MAX_ITERATIONS = 10; // Prevent infinite loops
+        let iteration = 0;
 
         // Initial call - no options since tools + JSON mode are incompatible
         const initialResponse = await provider.chatCompletion(
@@ -181,9 +175,10 @@ export class LLMService {
 
         let { content, tool_calls, parsedContent } = initialResponse;
 
-        // Tool execution loop
-        while (tool_calls) {
-            console.log('[Tool Call] Assistant wants to use tools:', tool_calls);
+        // Tool execution loop with safety limit
+        while (tool_calls && iteration < MAX_ITERATIONS) {
+            iteration++;
+            console.log(`[Tool Call] Iteration ${iteration}/${MAX_ITERATIONS}: Assistant wants to use tools:`, tool_calls);
             currentMessages.push({ role: 'assistant', content: content || '', tool_calls });
 
             // Execute tools using the provider's helper (which formats results for that provider)
@@ -201,6 +196,10 @@ export class LLMService {
             parsedContent = nextResponse.parsedContent; // Preserve parsedContent from final response
         }
 
+        if (iteration >= MAX_ITERATIONS) {
+            console.warn(`[Tool Call] Reached maximum iterations (${MAX_ITERATIONS}). Forcing completion.`);
+        }
+
         // Return both content and parsedContent (if available)
         return { content, parsedContent, toolCalls: tool_calls };
     }