@aj-archipelago/cortex 1.2.0 → 1.2.1

package/config.js

@@ -244,6 +244,15 @@ var config = convict({
                     "Content-Type": "application/json"
                 },
             },
+            "replicate-recraft-v3": {
+                "type": "REPLICATE-API",
+                "url": "https://api.replicate.com/v1/models/recraft-ai/recraft-v3/predictions",
+                "headers": {
+                    "Prefer": "wait",
+                    "Authorization": "Token {{REPLICATE_API_KEY}}",
+                    "Content-Type": "application/json"
+                },
+            },
             "azure-video-translate": {
                 "type": "AZURE-VIDEO-TRANSLATE",
                 "headers": {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aj-archipelago/cortex",
-  "version": "1.2.0",
+  "version": "1.2.1",
   "description": "Cortex is a GraphQL API for AI. It provides a simple, extensible interface for using AI services from OpenAI, Azure and others.",
   "private": false,
   "repository": {

package/pathways/{flux_image.js → image_flux.js}

@@ -5,9 +5,13 @@ export default {
   inputParameters: {
     model: "runware-flux-schnell",
     negativePrompt: "",
-    width: 512,
-    height: 512,
+    width: 1024,
+    height: 1024,
+    aspectRatio: "custom",
     numberResults: 1,
+    safety_tolerance: 5,
+    output_format: "webp",
+    output_quality: 80,
     steps: 4,
   },
 };

package/pathways/image_recraft.js

@@ -0,0 +1,10 @@
+export default {
+  prompt: ["{{text}}"],
+
+  enableDuplicateRequests: false,
+  inputParameters: {
+    model: "replicate-recraft-v3",
+    size: "1024x1024",
+    style: "realistic_image",
+  },
+};

package/pathways/index.js

@@ -16,7 +16,6 @@ import edit from './edit.js';
 import embeddings from './embeddings.js';
 import entities from './entities.js';
 import expand_story from './expand_story.js';
-import flux_image from './flux_image.js';
 import format_paragraph_turbo from './format_paragraph_turbo.js';
 import gemini_15_vision from './gemini_15_vision.js';
 import gemini_vision from './gemini_vision.js';
@@ -26,6 +25,8 @@ import headline from './headline.js';
 import headline_custom from './headline_custom.js';
 import highlights from './highlights.js';
 import image from './image.js';
+import image_flux from './image_flux.js';
+import image_recraft from './image_recraft.js';
 import jira_story from './jira_story.js';
 import keywords from './keywords.js';
 import language from './language.js';
@@ -100,13 +101,14 @@ export {
     embeddings,
     entities,
     expand_story,
-    flux_image,
     format_paragraph_turbo,
     gemini_15_vision,
     gemini_vision,
     grammar,
     hashtags, headline, headline_custom, highlights,
     image,
+    image_flux,
+    image_recraft,
     jira_story,
     keywords,
     language,

package/server/parser.js

@@ -40,7 +40,12 @@ const isNumberedList = (data) => {
 
 async function parseJson(str) {
     try {
-        JSON.parse(str); // Validate JSON
+        // check for the common error case that the JSON is surrounded by markdown
+        const match = str.match(/```\s*(?:json)?(.*?)```/s);
+        if (match) {
+            str = match[1].trim();
+        }
+        JSON.parse(str);
         return str;
     } catch (error) {
         try {

package/server/plugins/replicateApiPlugin.js

@@ -18,15 +18,19 @@ class ReplicateApiPlugin extends ModelPlugin {
 
     const requestParameters = {
       input: {
-        aspect_ratio: "1:1",
-        output_format: "webp",
-        output_quality: 80,
+        aspect_ratio: combinedParameters.aspectRatio || "1:1",
+        output_format: combinedParameters.outputFormat || "webp",
+        output_quality: combinedParameters.outputQuality || 80,
         prompt: modelPromptText,
-        //prompt_upsampling: false,
-        //safety_tolerance: 5,
+        prompt_upsampling: combinedParameters.promptUpsampling || false,
+        safety_tolerance: combinedParameters.safety_tolerance || 3,
         go_fast: true,
         megapixels: "1",
         num_outputs: combinedParameters.numberResults,
+        width: combinedParameters.width,
+        height: combinedParameters.height,
+        size: combinedParameters.size || "1024x1024",
+        style: combinedParameters.style || "realistic_image",
       },
     };