strapi-llm-translator 0.9.1 → 0.9.3

package/dist/server/index.js

@@ -128,6 +128,26 @@ const routes = {
 };
 const DEFAULT_SYSTEM_PROMPT = "You are a professional translator. Your task is to translate the provided content accurately while preserving the original meaning and tone.";
 const SYSTEM_PROMPT_APPENDIX = `The user asks you to translate the text to a specific language, the language is provided via short code like "en", "fr", "de", etc.`;
+const SYSTEM_PROMPT_FIX = `You are a JSON correction assistant. Only return valid, corrected JSON.`;
+const USER_PROMPT_FIX_PREFIX = "Fix this invalid JSON and return ONLY the corrected JSON. No explanations allowed. The JSON is:";
+const cleanJSONString = (content) => {
+  return content.replace(/^```json\s*\n/, "").replace(/^```\s*\n/, "").replace(/\n\s*```$/, "").replace(/\u200B/g, "").replace(/[\u2018\u2019]/g, "'").replace(/[\u201C\u201D]/g, '"').trim();
+};
+const balanceJSONBraces = (content) => {
+  const openBraces = (content.match(/{/g) || []).length;
+  const closeBraces = (content.match(/}/g) || []).length;
+  if (openBraces > closeBraces) {
+    return content + "}".repeat(openBraces - closeBraces);
+  }
+  return content;
+};
+const safeJSONParse = (content) => {
+  const parsed = JSON.parse(content);
+  if (typeof parsed === "object" && parsed !== null) {
+    return parsed;
+  }
+  throw new Error("Invalid response format - not an object");
+};
 const openai = new openai$1.OpenAI({
   baseURL: process.env.STRAPI_ADMIN_LLM_TRANSLATOR_LLM_BASE_URL,
   apiKey: process.env.LLM_TRANSLATOR_LLM_API_KEY
@@ -262,7 +282,7 @@ const llmService = ({ strapi: strapi2 }) => ({
       const prompt = buildPrompt(translationPayload, config2.targetLanguage);
       const systemPrompt = await buildSystemPrompt(userConfig);
       const response = await callLLMProvider(prompt, systemPrompt, model, userConfig);
-      const translatedData = parseLLMResponse(response);
+      const translatedData = await parseLLMResponse(response);
       const mergedContent = mergeTranslatedContent(fields, translatedData, translatableFields);
       const uidFields = findUIDFields(contentType);
       const translatedUIDs = await generateUIDsForTranslatedFields(
@@ -306,8 +326,9 @@ IMPORTANT RULES:
 4. Keep HTML tags intact if present
 5. Preserve any special characters or placeholders
 6. Return ONLY the translated JSON object
-7. Do not add any explanations or comments
-8. Ensure professional and culturally appropriate translations
+7. Ensure the JSON is valid and well-formed, all values must be strings
+8. Do not add any explanations or comments
+9. Ensure professional and culturally appropriate translations
 
 SOURCE JSON:
 ${JSON.stringify(fields, null, 2)}`;
@@ -324,10 +345,16 @@ const getUserConfig = async () => {
 const buildSystemPrompt = async (config2) => {
   return `${config2.systemPrompt || DEFAULT_SYSTEM_PROMPT} ${SYSTEM_PROMPT_APPENDIX}`;
 };
+const createLLMRequest = (messages, temperature = 0.1) => {
+  return openai.chat.completions.create({
+    model,
+    messages,
+    temperature
+  });
+};
 const callLLMProvider = async (prompt, systemPrompt, model2, config2) => {
-  const response = await openai.chat.completions.create({
-    model: model2,
-    messages: [
+  return createLLMRequest(
+    [
       {
         role: "system",
         content: systemPrompt
@@ -337,37 +364,39 @@ const callLLMProvider = async (prompt, systemPrompt, model2, config2) => {
         content: prompt
       }
     ],
-    temperature: config2.temperature
-  });
-  return response;
+    config2.temperature
+  );
+};
+const requestJSONCorrection = async (invalidJson) => {
+  const response = await createLLMRequest([
+    {
+      role: "system",
+      content: SYSTEM_PROMPT_FIX
+    },
+    {
+      role: "user",
+      content: `${USER_PROMPT_FIX_PREFIX} ${invalidJson}`
+    }
+  ]);
+  const correctedContent = response.choices[0]?.message?.content;
+  if (!correctedContent) throw new Error("No content in correction response");
+  return safeJSONParse(correctedContent.trim());
 };
-const parseLLMResponse = (response) => {
+const parseLLMResponse = async (response) => {
   try {
     const content = response.choices[0]?.message?.content;
     if (!content) throw new Error("No content in response");
-    const cleanContent = content.replace(/^```json\s*\n/, "").replace(/^```\s*\n/, "").replace(/\n\s*```$/, "").replace(/\u200B/g, "").replace(/[\u2018\u2019]/g, "'").replace(/[\u201C\u201D]/g, '"').trim();
+    const cleanContent = cleanJSONString(content);
     try {
-      const parsed = JSON.parse(cleanContent);
-      if (typeof parsed === "object" && parsed !== null) {
-        return parsed;
-      }
-      throw new Error("Invalid response format - not an object");
+      return safeJSONParse(cleanContent);
     } catch (parseError) {
-      const openBraces = (cleanContent.match(/{/g) || []).length;
-      const closeBraces = (cleanContent.match(/}/g) || []).length;
-      if (openBraces > closeBraces) {
-        const missingBraces = openBraces - closeBraces;
-        const fixedContent = cleanContent + "}".repeat(missingBraces);
-        try {
-          const parsed = JSON.parse(fixedContent);
-          if (typeof parsed === "object" && parsed !== null) {
-            return parsed;
-          }
-        } catch (secondError) {
-          console.error("Second parse attempt failed:", secondError);
-        }
+      const balancedContent = balanceJSONBraces(cleanContent);
+      try {
+        return safeJSONParse(balancedContent);
+      } catch (secondError) {
+        console.error("Second parse attempt failed:", secondError);
+        return await requestJSONCorrection(cleanContent);
       }
-      throw new Error(`JSON parsing failed: ${parseError.message}`);
     }
   } catch (error) {
     const errorMessage = error instanceof Error ? error.message : "Unknown error";

package/dist/server/index.mjs

@@ -127,6 +127,26 @@ const routes = {
 };
 const DEFAULT_SYSTEM_PROMPT = "You are a professional translator. Your task is to translate the provided content accurately while preserving the original meaning and tone.";
 const SYSTEM_PROMPT_APPENDIX = `The user asks you to translate the text to a specific language, the language is provided via short code like "en", "fr", "de", etc.`;
+const SYSTEM_PROMPT_FIX = `You are a JSON correction assistant. Only return valid, corrected JSON.`;
+const USER_PROMPT_FIX_PREFIX = "Fix this invalid JSON and return ONLY the corrected JSON. No explanations allowed. The JSON is:";
+const cleanJSONString = (content) => {
+  return content.replace(/^```json\s*\n/, "").replace(/^```\s*\n/, "").replace(/\n\s*```$/, "").replace(/\u200B/g, "").replace(/[\u2018\u2019]/g, "'").replace(/[\u201C\u201D]/g, '"').trim();
+};
+const balanceJSONBraces = (content) => {
+  const openBraces = (content.match(/{/g) || []).length;
+  const closeBraces = (content.match(/}/g) || []).length;
+  if (openBraces > closeBraces) {
+    return content + "}".repeat(openBraces - closeBraces);
+  }
+  return content;
+};
+const safeJSONParse = (content) => {
+  const parsed = JSON.parse(content);
+  if (typeof parsed === "object" && parsed !== null) {
+    return parsed;
+  }
+  throw new Error("Invalid response format - not an object");
+};
 const openai = new OpenAI({
   baseURL: process.env.STRAPI_ADMIN_LLM_TRANSLATOR_LLM_BASE_URL,
   apiKey: process.env.LLM_TRANSLATOR_LLM_API_KEY
@@ -261,7 +281,7 @@ const llmService = ({ strapi: strapi2 }) => ({
       const prompt = buildPrompt(translationPayload, config2.targetLanguage);
       const systemPrompt = await buildSystemPrompt(userConfig);
       const response = await callLLMProvider(prompt, systemPrompt, model, userConfig);
-      const translatedData = parseLLMResponse(response);
+      const translatedData = await parseLLMResponse(response);
       const mergedContent = mergeTranslatedContent(fields, translatedData, translatableFields);
       const uidFields = findUIDFields(contentType);
       const translatedUIDs = await generateUIDsForTranslatedFields(
@@ -305,8 +325,9 @@ IMPORTANT RULES:
 4. Keep HTML tags intact if present
 5. Preserve any special characters or placeholders
 6. Return ONLY the translated JSON object
-7. Do not add any explanations or comments
-8. Ensure professional and culturally appropriate translations
+7. Ensure the JSON is valid and well-formed, all values must be strings
+8. Do not add any explanations or comments
+9. Ensure professional and culturally appropriate translations
 
 SOURCE JSON:
 ${JSON.stringify(fields, null, 2)}`;
@@ -323,10 +344,16 @@ const getUserConfig = async () => {
 const buildSystemPrompt = async (config2) => {
   return `${config2.systemPrompt || DEFAULT_SYSTEM_PROMPT} ${SYSTEM_PROMPT_APPENDIX}`;
 };
+const createLLMRequest = (messages, temperature = 0.1) => {
+  return openai.chat.completions.create({
+    model,
+    messages,
+    temperature
+  });
+};
 const callLLMProvider = async (prompt, systemPrompt, model2, config2) => {
-  const response = await openai.chat.completions.create({
-    model: model2,
-    messages: [
+  return createLLMRequest(
+    [
       {
         role: "system",
         content: systemPrompt
@@ -336,37 +363,39 @@ const callLLMProvider = async (prompt, systemPrompt, model2, config2) => {
         content: prompt
       }
     ],
-    temperature: config2.temperature
-  });
-  return response;
+    config2.temperature
+  );
+};
+const requestJSONCorrection = async (invalidJson) => {
+  const response = await createLLMRequest([
+    {
+      role: "system",
+      content: SYSTEM_PROMPT_FIX
+    },
+    {
+      role: "user",
+      content: `${USER_PROMPT_FIX_PREFIX} ${invalidJson}`
+    }
+  ]);
+  const correctedContent = response.choices[0]?.message?.content;
+  if (!correctedContent) throw new Error("No content in correction response");
+  return safeJSONParse(correctedContent.trim());
 };
-const parseLLMResponse = (response) => {
+const parseLLMResponse = async (response) => {
   try {
     const content = response.choices[0]?.message?.content;
     if (!content) throw new Error("No content in response");
-    const cleanContent = content.replace(/^```json\s*\n/, "").replace(/^```\s*\n/, "").replace(/\n\s*```$/, "").replace(/\u200B/g, "").replace(/[\u2018\u2019]/g, "'").replace(/[\u201C\u201D]/g, '"').trim();
+    const cleanContent = cleanJSONString(content);
     try {
-      const parsed = JSON.parse(cleanContent);
-      if (typeof parsed === "object" && parsed !== null) {
-        return parsed;
-      }
-      throw new Error("Invalid response format - not an object");
+      return safeJSONParse(cleanContent);
     } catch (parseError) {
-      const openBraces = (cleanContent.match(/{/g) || []).length;
-      const closeBraces = (cleanContent.match(/}/g) || []).length;
-      if (openBraces > closeBraces) {
-        const missingBraces = openBraces - closeBraces;
-        const fixedContent = cleanContent + "}".repeat(missingBraces);
-        try {
-          const parsed = JSON.parse(fixedContent);
-          if (typeof parsed === "object" && parsed !== null) {
-            return parsed;
-          }
-        } catch (secondError) {
-          console.error("Second parse attempt failed:", secondError);
-        }
+      const balancedContent = balanceJSONBraces(cleanContent);
+      try {
+        return safeJSONParse(balancedContent);
+      } catch (secondError) {
+        console.error("Second parse attempt failed:", secondError);
+        return await requestJSONCorrection(cleanContent);
       }
-      throw new Error(`JSON parsing failed: ${parseError.message}`);
     }
   } catch (error) {
     const errorMessage = error instanceof Error ? error.message : "Unknown error";

package/dist/server/src/config/constants.d.ts

@@ -1,2 +1,4 @@
 export declare const DEFAULT_SYSTEM_PROMPT = "You are a professional translator. Your task is to translate the provided content accurately while preserving the original meaning and tone.";
 export declare const SYSTEM_PROMPT_APPENDIX = "The user asks you to translate the text to a specific language, the language is provided via short code like \"en\", \"fr\", \"de\", etc.";
+export declare const SYSTEM_PROMPT_FIX = "You are a JSON correction assistant. Only return valid, corrected JSON.";
+export declare const USER_PROMPT_FIX_PREFIX = "Fix this invalid JSON and return ONLY the corrected JSON. No explanations allowed. The JSON is:";

package/dist/server/src/utils/json-utils.d.ts

@@ -0,0 +1,3 @@
+export declare const cleanJSONString: (content: string) => string;
+export declare const balanceJSONBraces: (content: string) => string;
+export declare const safeJSONParse: (content: string) => Record<string, any>;

package/package.json

@@ -1,5 +1,5 @@
 {
-  "version": "0.9.1",
+  "version": "0.9.3",
   "keywords": [
     "strapi",
     "plugin",
@@ -72,5 +72,13 @@
   "name": "strapi-llm-translator",
   "description": "AI-Powered Content Translation for Strapi",
   "license": "MIT",
-  "author": "grenzbotin <franziska@vulpis.dev>"
+  "author": "grenzbotin <franziska@vulpis.dev>",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/grenzbotin/strapi-llm-translator.git"
+  },
+  "bugs": {
+    "url": "https://github.com/grenzbotin/strapi-llm-translator/issues"
+  },
+  "homepage": "https://github.com/grenzbotin/strapi-llm-translator#readme"
 }