@aj-archipelago/cortex 1.4.22 → 1.4.23

package/pathways/system/entity/files/sys_update_file_metadata.js

@@ -2,31 +2,40 @@
 // GraphQL pathway for updating file metadata (replaces sys_save_memory for renames and metadata updates)
 // Only updates Cortex-managed fields (displayFilename, tags, notes, etc.), not CFH fields (url, gcs, hash, filename)
 
-import { updateFileMetadata } from '../../../../lib/fileUtils.js';
+import { updateFileMetadata, getDefaultContext } from '../../../../lib/fileUtils.js';
 
 export default {
     inputParameters: {
-        contextId: ``,
+        agentContext: [{ contextId: ``, contextKey: ``, default: true }],
         hash: ``,
         displayFilename: { type: 'string' }, // Optional - no default
         tags: { type: 'array', items: { type: 'string' } }, // Optional - no default
         notes: { type: 'string' }, // Optional - no default
         mimeType: { type: 'string' }, // Optional - no default
         permanent: { type: 'boolean' }, // Optional - no default
-        inCollection: { type: 'array', items: { type: 'string' } }, // Optional - array of chat IDs, or can be boolean true/false (normalized to ['*'] or removed)
-        contextKey: `` // Optional - context key for encryption
+        inCollection: { type: 'array', items: { type: 'string' } } // Optional - array of chat IDs, or can be boolean true/false (normalized to ['*'] or removed)
     },
     model: 'oai-gpt4o',
     isMutation: true, // Declaratively mark this as a Mutation
 
     resolver: async (_parent, args, _contextValue, _info) => {
-        const { contextId, hash, displayFilename, tags, notes, mimeType, permanent, inCollection, contextKey } = args;
+        const { agentContext, hash, displayFilename, tags, notes, mimeType, permanent, inCollection } = args;
+        
+        const defaultCtx = getDefaultContext(agentContext);
+        if (!defaultCtx) {
+            return JSON.stringify({ 
+                success: false, 
+                error: 'agentContext with at least one default context is required' 
+            });
+        }
+        const contextId = defaultCtx.contextId;
+        const contextKey = defaultCtx.contextKey || null;
         
         // Validate required parameters
-        if (!contextId || !hash) {
+        if (!hash) {
             return JSON.stringify({ 
                 success: false, 
-                error: 'contextId and hash are required' 
+                error: 'hash is required' 
             });
         }
         

package/pathways/system/entity/sys_entity_agent.js

@@ -63,7 +63,9 @@ export default {
     inputParameters: {
         privateData: false,    
         chatHistory: [{role: '', content: []}],
-        contextId: ``,
+        agentContext: [
+            { contextId: ``, contextKey: ``, default: true }
+        ],
         chatId: ``,
         language: "English",
         aiName: "Jarvis",
@@ -76,8 +78,7 @@ export default {
         entityId: ``,
         researchMode: false,
         userInfo: '',
-        model: 'oai-gpt41',
-        contextKey: ``
+        model: 'oai-gpt41'
     },
     timeout: 600,
 
@@ -547,10 +548,11 @@ export default {
             args.chatHistory = args.chatHistory.slice(-20);
         }
 
-        // Sync files from chat history to collection and strip file content
-        // Files are accessible via tools (AnalyzeFile, ReadTextFile, etc.)
+        // Process files in chat history:
+        // - Files in collection (all agentContext contexts): stripped, accessible via tools
+        // - Files not in collection: left in message for model to see directly
         const { chatHistory: strippedHistory, availableFiles } = await syncAndStripFilesFromChatHistory(
-            args.chatHistory, args.contextId, args.contextKey
+            args.chatHistory, args.agentContext
         );
         args.chatHistory = strippedHistory;
 

package/pathways/system/entity/tools/sys_tool_codingagent.js

@@ -81,12 +81,12 @@ export default {
 
     executePathway: async ({args, resolver}) => {
         try {
-            const { codingTask, userMessage, inputFiles, codingTaskKeywords, contextId, contextKey } = args;
+            const { codingTask, userMessage, inputFiles, codingTaskKeywords } = args;
 
             let taskSuffix = "";
             if (inputFiles) {
-                if (!contextId) {
-                    throw new Error("contextId is required when using the 'inputFiles' parameter. Use ListFileCollection or SearchFileCollection to find available files.");
+                if (!args.agentContext || !Array.isArray(args.agentContext) || args.agentContext.length === 0) {
+                    throw new Error("agentContext is required when using the 'inputFiles' parameter. Use ListFileCollection or SearchFileCollection to find available files.");
                 }
                 
                 // Resolve file parameters to URLs
@@ -100,7 +100,7 @@ export default {
                 
                 for (const fileRef of fileReferences) {
                     // Try to resolve each file reference
-                    const resolvedUrl = await resolveFileParameter(fileRef, contextId, contextKey);
+                    const resolvedUrl = await resolveFileParameter(fileRef, args.agentContext);
                     if (resolvedUrl) {
                         resolvedUrls.push(resolvedUrl);
                     } else {

package/pathways/system/entity/tools/sys_tool_editfile.js

@@ -2,7 +2,7 @@
 // Entity tool that modifies existing files by replacing line ranges or exact string matches
 import logger from '../../../../lib/logger.js';
 import { axios } from '../../../../lib/requestExecutor.js';
-import { uploadFileToCloud, findFileInCollection, loadFileCollection, getMimeTypeFromFilename, deleteFileByHash, isTextMimeType, updateFileMetadata, writeFileDataToRedis, invalidateFileCollectionCache, getActualContentMimeType } from '../../../../lib/fileUtils.js';
+import { uploadFileToCloud, findFileInCollection, loadMergedFileCollection, getDefaultContext, getMimeTypeFromFilename, deleteFileByHash, isTextMimeType, updateFileMetadata, writeFileDataToRedis, invalidateFileCollectionCache, getActualContentMimeType } from '../../../../lib/fileUtils.js';
 
 // Maximum file size for editing (50MB) - prevents memory blowup on huge files
 const MAX_EDITABLE_FILE_SIZE = 50 * 1024 * 1024;
@@ -145,7 +145,19 @@ export default {
     ],
 
     executePathway: async ({args, runAllPrompts, resolver}) => {
-        const { file, startLine, endLine, content, oldString, newString, replaceAll = false, contextId, contextKey } = args;
+        const { file, startLine, endLine, content, oldString, newString, replaceAll = false, agentContext } = args;
+        
+        const defaultCtx = getDefaultContext(agentContext);
+        if (!defaultCtx) {
+            const errorResult = {
+                success: false,
+                error: "agentContext with at least one default context is required"
+            };
+            resolver.tool = JSON.stringify({ toolUsed: "EditFile" });
+            return JSON.stringify(errorResult);
+        }
+        const contextId = defaultCtx.contextId;
+        const contextKey = defaultCtx.contextKey || null;
         
         // Determine which tool was called based on parameters
         const isSearchReplace = oldString !== undefined && newString !== undefined;
@@ -162,14 +174,6 @@ export default {
             return JSON.stringify(errorResult);
         }
 
-        if (!contextId) {
-            const errorResult = {
-                success: false,
-                error: "contextId is required for file modification"
-            };
-            resolver.tool = JSON.stringify({ toolUsed: toolName });
-            return JSON.stringify(errorResult);
-        }
 
         // Validate that we have the right parameters for the tool being used
         if (!isSearchReplace && !isEditByLine) {
@@ -243,7 +247,7 @@ export default {
 
         try {
             // Resolve file ID first (needed for serialization)
-            const collection = await loadFileCollection(contextId, contextKey, false);
+            const collection = await loadMergedFileCollection(agentContext);
             const foundFile = findFileInCollection(file, collection);
             
             if (!foundFile) {
@@ -281,7 +285,7 @@ export default {
                     logger.info(`Using cached content for: ${currentFile.displayFilename || file}`);
                 } else {
                     // First edit in session: load collection and download file
-                    const currentCollection = await loadFileCollection(contextId, contextKey, false);
+                    const currentCollection = await loadMergedFileCollection(agentContext);
                     currentFile = findFileInCollection(file, currentCollection);
 
                     if (!currentFile) {
@@ -467,7 +471,7 @@ export default {
             if (editResult._isLastOperation) {
                 // Flush: upload the final content and update metadata
                 const { modifiedContent, currentFile, fileIdToUpdate: initialFileId, filename, mimeType, 
-                        modificationInfo, message, contextId: ctxId, contextKey: ctxKey, resolver: res,
+                        modificationInfo, message, resolver: res,
                         file: fileParam, isEditByLine: isByLine, isSearchReplace: isSR, replaceAll: repAll,
                         startLine: sLine, endLine: eLine } = editResult;
                 
@@ -482,7 +486,7 @@ export default {
                     mimeType,
                     filename,
                     res,
-                    ctxId
+                    contextId
                 );
 
                 if (!uploadResult || !uploadResult.url) {
@@ -490,7 +494,8 @@ export default {
                 }
 
                 // Update the file collection entry directly (atomic operation)
-                const latestCollection = await loadFileCollection(ctxId, ctxKey, false);
+                // Use default context from agentContext for consistency
+                const latestCollection = await loadMergedFileCollection(agentContext);
                 let fileToUpdate = latestCollection.find(f => f.id === fileIdToUpdate);
                 
                 // If not found by ID, try to find by the original file parameter
@@ -512,7 +517,7 @@ export default {
                     const { getRedisClient } = await import('../../../../lib/fileUtils.js');
                     const redisClient = await getRedisClient();
                     if (redisClient) {
-                        const contextMapKey = `FileStoreMap:ctx:${ctxId}`;
+                        const contextMapKey = `FileStoreMap:ctx:${contextId}`;
                         
                         const existingDataStr = await redisClient.hget(contextMapKey, uploadResult.hash);
                         let existingData = {};
@@ -541,21 +546,21 @@ export default {
                             permanent: fileToUpdate.permanent || false
                         };
                         
-                        await writeFileDataToRedis(redisClient, contextMapKey, uploadResult.hash, fileData, ctxKey);
+                        await writeFileDataToRedis(redisClient, contextMapKey, uploadResult.hash, fileData, contextKey);
                         
                         if (oldHashToDelete && oldHashToDelete !== uploadResult.hash) {
                             await redisClient.hdel(contextMapKey, oldHashToDelete);
                         }
                         
-                        invalidateFileCollectionCache(ctxId, ctxKey);
+                        invalidateFileCollectionCache(contextId, contextKey);
                     }
                 } else if (fileToUpdate.hash) {
-                    await updateFileMetadata(ctxId, fileToUpdate.hash, {
+                    await updateFileMetadata(contextId, fileToUpdate.hash, {
                         filename: filename,
                         lastAccessed: new Date().toISOString()
-                    }, ctxKey);
+                    }, contextKey);
                     
-                    invalidateFileCollectionCache(ctxId, ctxKey);
+                    invalidateFileCollectionCache(contextId, contextKey);
                 }
 
                 // Delete old file version (fire-and-forget)
@@ -563,7 +568,7 @@ export default {
                     (async () => {
                         try {
                             logger.info(`Deleting old file version with hash ${oldHashToDelete} (background task)`);
-                            await deleteFileByHash(oldHashToDelete, res, ctxId);
+                            await deleteFileByHash(oldHashToDelete, res, contextId);
                         } catch (cleanupError) {
                             logger.warn(`Failed to cleanup old file version: ${cleanupError.message}`);
                         }

package/pathways/system/entity/tools/sys_tool_file_collection.js

@@ -3,7 +3,7 @@
 // Uses Redis hash maps (FileStoreMap:ctx:<contextId>) for storage
 // Supports atomic rename/tag/notes updates via UpdateFileMetadata
 import logger from '../../../../lib/logger.js';
-import { addFileToCollection, loadFileCollection, findFileInCollection, deleteFileByHash, updateFileMetadata, invalidateFileCollectionCache } from '../../../../lib/fileUtils.js';
+import { addFileToCollection, loadFileCollection, loadMergedFileCollection, findFileInCollection, deleteFileByHash, updateFileMetadata, invalidateFileCollectionCache, getDefaultContext } from '../../../../lib/fileUtils.js';
 
 export default {
     prompt: [],
@@ -198,7 +198,12 @@ export default {
     ],
 
     executePathway: async ({args, runAllPrompts, resolver}) => {
-        const { contextId, contextKey } = args;
+        const defaultCtx = getDefaultContext(args.agentContext);
+        if (!defaultCtx) {
+            throw new Error("agentContext with at least one default context is required");
+        }
+        const contextId = defaultCtx.contextId;
+        const contextKey = defaultCtx.contextKey || null;
 
         // Determine which function was called based on which parameters are present
         // Order matters: check most specific operations first
@@ -359,12 +364,12 @@ export default {
                 const safeFilterTags = Array.isArray(filterTags) ? filterTags : [];
                 const queryLower = query.toLowerCase();
                 
-                // Update lastAccessed for matching files directly (atomic operations)
-                const allFiles = await loadFileCollection(contextId, contextKey, false);
+                // Load primary collection for lastAccessed updates (only update files in primary context)
+                const primaryFiles = await loadFileCollection(contextId, contextKey, false);
                 const now = new Date().toISOString();
                 
-                // Find matching files and update lastAccessed directly
-                for (const file of allFiles) {
+                // Find matching files in primary collection and update lastAccessed directly
+                for (const file of primaryFiles) {
                     if (!file.hash) continue;
                     
                     // Fallback to filename if displayFilename is not set (for files uploaded before displayFilename was added)
@@ -387,8 +392,8 @@ export default {
                     }
                 }
                 
-                // Reload collection to get results (after update)
-                const updatedFiles = await loadFileCollection(contextId, contextKey, false);
+                // Load merged collection for search results (includes all agentContext files)
+                const updatedFiles = await loadMergedFileCollection(args.agentContext);
                 
                 // Filter and sort results (for display only, not modifying)
                 let results = updatedFiles.filter(file => {
@@ -560,8 +565,8 @@ export default {
                 // List collection (read-only, no locking needed)
                 const { tags: filterTags = [], sortBy = 'date', limit = 50 } = args;
                 
-                // Use useCache: false to ensure we get the latest file data (important after edits)
-                const collection = await loadFileCollection(contextId, contextKey, false);
+                // Use merged collection to include files from all agentContext contexts
+                const collection = await loadMergedFileCollection(args.agentContext);
                 let results = collection;
 
                 // Filter by tags if provided

package/pathways/system/entity/tools/sys_tool_image.js

@@ -113,7 +113,7 @@ export default {
                 
                 for (let i = 0; i < imagesToProcess.length; i++) {
                     const imageRef = imagesToProcess[i];
-                    const resolved = await resolveFileParameter(imageRef, args.contextId, args.contextKey);
+                    const resolved = await resolveFileParameter(imageRef, args.agentContext);
                     if (!resolved) {
                         throw new Error(`File not found: "${imageRef}". Use ListFileCollection or SearchFileCollection to find available files.`);
                     }

package/pathways/system/entity/tools/sys_tool_image_gemini.js

@@ -109,7 +109,7 @@ export default {
                 
                 for (let i = 0; i < imagesToProcess.length; i++) {
                     const imageRef = imagesToProcess[i];
-                    const resolved = await resolveFileParameter(imageRef, args.contextId, args.contextKey, { preferGcs: true });
+                    const resolved = await resolveFileParameter(imageRef, args.agentContext, { preferGcs: true });
                     if (!resolved) {
                         throw new Error(`File not found: "${imageRef}". Use ListFileCollection or SearchFileCollection to find available files.`);
                     }

package/pathways/system/entity/tools/sys_tool_readfile.js

@@ -135,20 +135,20 @@ export default {
             const MAX_CHARS = 100000;
             const MAX_LINES = 1000;
             
-            let { cloudUrl, file, startChar, endChar, startLine, endLine, contextId, contextKey } = args;
+            let { cloudUrl, file, startChar, endChar, startLine, endLine } = args;
             
             // If file parameter is provided, resolve it to a URL using the common utility
             if (file) {
-                if (!contextId) {
+                if (!args.agentContext || !Array.isArray(args.agentContext) || args.agentContext.length === 0) {
                     const errorResult = {
                         success: false,
-                        error: "contextId is required when using the 'file' parameter. Use ListFileCollection or SearchFileCollection to find available files."
+                        error: "agentContext is required when using the 'file' parameter. Use ListFileCollection or SearchFileCollection to find available files."
                     };
                     resolver.tool = JSON.stringify({ toolUsed: "ReadFile" });
                     return JSON.stringify(errorResult);
                 }
                 // Use useCache: false to ensure we get the latest file data (important after edits)
-                const resolvedUrl = await resolveFileParameter(file, contextId, contextKey, { useCache: false });
+                const resolvedUrl = await resolveFileParameter(file, args.agentContext, { useCache: false });
                 if (!resolvedUrl) {
                     const errorResult = {
                         success: false,

package/pathways/system/entity/tools/sys_tool_slides_gemini.js

@@ -80,7 +80,7 @@ export default {
                 
                 for (let i = 0; i < imagesToProcess.length; i++) {
                     const imageRef = imagesToProcess[i];
-                    const resolved = await resolveFileParameter(imageRef, args.contextId, args.contextKey, { preferGcs: true });
+                    const resolved = await resolveFileParameter(imageRef, args.agentContext, { preferGcs: true });
                     if (!resolved) {
                         throw new Error(`File not found: "${imageRef}". Use ListFileCollection or SearchFileCollection to find available files.`);
                     }

package/pathways/system/entity/tools/sys_tool_video_veo.js

@@ -128,7 +128,7 @@ export default {
                     throw new Error("contextId is required when using the 'inputImage' parameter. Use ListFileCollection or SearchFileCollection to find available files.");
                 }
                 
-                const resolved = await resolveFileParameter(args.inputImage, args.contextId, args.contextKey, { preferGcs: true });
+                const resolved = await resolveFileParameter(args.inputImage, args.agentContext, { preferGcs: true });
                 if (!resolved) {
                     throw new Error(`File not found: "${args.inputImage}". Use ListFileCollection or SearchFileCollection to find available files.`);
                 }

package/pathways/system/entity/tools/sys_tool_view_image.js

@@ -1,7 +1,7 @@
 // sys_tool_view_image.js
 // Tool pathway that allows agents to view image files from the file collection
 import logger from '../../../../lib/logger.js';
-import { loadFileCollection, findFileInCollection, ensureShortLivedUrl } from '../../../../lib/fileUtils.js';
+import { loadMergedFileCollection, findFileInCollection, ensureShortLivedUrl, getDefaultContext } from '../../../../lib/fileUtils.js';
 import { config } from '../../../../config.js';
 
 export default {
@@ -34,15 +34,19 @@ export default {
     },
 
     executePathway: async ({args, runAllPrompts, resolver}) => {
-        const { files, contextId, contextKey } = args;
+        const { files } = args;
 
         if (!files || !Array.isArray(files) || files.length === 0) {
             throw new Error("Files parameter is required and must be a non-empty array");
         }
 
+        if (!args.agentContext || !Array.isArray(args.agentContext) || args.agentContext.length === 0) {
+            throw new Error("agentContext is required");
+        }
+
         try {
-            // Load the file collection
-            const collection = await loadFileCollection(contextId, contextKey, true);
+            // Load the file collection (merged from all agentContext contexts)
+            const collection = await loadMergedFileCollection(args.agentContext);
             
             const imageUrls = [];
             const errors = [];
@@ -70,7 +74,8 @@ export default {
 
                 // Resolve to short-lived URL if possible
                 const fileHandlerUrl = config.get('whisperMediaApiUrl');
-                const fileWithShortLivedUrl = await ensureShortLivedUrl(foundFile, fileHandlerUrl, contextId);
+                const defaultCtx = getDefaultContext(args.agentContext);
+                const fileWithShortLivedUrl = await ensureShortLivedUrl(foundFile, fileHandlerUrl, defaultCtx?.contextId || null);
 
                 // Add to imageUrls array
                 imageUrls.push({

package/pathways/system/workspaces/run_workspace_agent.js

@@ -6,8 +6,11 @@ export default {
 
     inputParameters: {
         model: "oai-gpt41",
-        aiStyle: "OpenAI",
         chatHistory: [{role: '', content: []}],
+        researchMode: false,
+        agentContext: [
+            { contextId: "", contextKey: "", default: true }
+        ]
     },
     timeout: 600,
 

package/pathways/video_seedance.js

@@ -11,6 +11,8 @@ export default {
     image: "",
     camera_fixed: false,
     seed: -1,
+    generate_audio: false,
+    last_frame_image: "",
   },
 
   timeout: 60 * 30, // 30 minutes

package/server/executeWorkspace.js

@@ -87,14 +87,57 @@ const executePathwayWithFallback = async (pathway, pathwayArgs, contextValue, in
     if (cortexPathwayName) {
         // Use the specific cortex pathway
         // Transform parameters for cortex pathway
-        // Spread all pathway args first (including contextId, contextKey, etc.), then override specific fields
+        // Spread all pathway args first, then override specific fields
         const cortexArgs = {
-            ...pathwayArgs, // Spread all pathway args (including contextId, contextKey, etc.)
+            ...pathwayArgs, // Spread all pathway args
             model: pathway.model || pathwayArgs.model || "labeeb-agent", // Use pathway model or default
             chatHistory: pathwayArgs.chatHistory ? JSON.parse(JSON.stringify(pathwayArgs.chatHistory)) : [],
             systemPrompt: pathway.systemPrompt || pathwayArgs.systemPrompt
         };
         
+        // Transform old parameters to new format for run_workspace_agent
+        if (cortexPathwayName === 'run_workspace_agent') {
+            // Remove old aiStyle parameter (no longer used)
+            delete cortexArgs.aiStyle;
+            
+            // Transform context parameters to agentContext array format (only if agentContext not already provided)
+            if (!cortexArgs.agentContext && (cortexArgs.contextId || cortexArgs.contextKey || cortexArgs.altContextId || cortexArgs.altContextKey)) {
+                const agentContext = [];
+                
+                // Add primary context if present
+                if (cortexArgs.contextId) {
+                    agentContext.push({
+                        contextId: cortexArgs.contextId,
+                        contextKey: cortexArgs.contextKey || null,
+                        default: true
+                    });
+                }
+                
+                // Add alternate context if present
+                if (cortexArgs.altContextId) {
+                    agentContext.push({
+                        contextId: cortexArgs.altContextId,
+                        contextKey: cortexArgs.altContextKey || null,
+                        default: false
+                    });
+                }
+                
+                // If we have at least one context, set agentContext and remove old params
+                if (agentContext.length > 0) {
+                    cortexArgs.agentContext = agentContext;
+                    delete cortexArgs.contextId;
+                    delete cortexArgs.contextKey;
+                    delete cortexArgs.altContextId;
+                    delete cortexArgs.altContextKey;
+                }
+            }
+            
+            // Ensure researchMode defaults to false if not provided
+            if (cortexArgs.researchMode === undefined) {
+                cortexArgs.researchMode = false;
+            }
+        }
+        
         // If we have text parameter, we need to add it to the chatHistory
         if (pathwayArgs.text) {
             // Find the last user message or create a new one

package/server/pathwayResolver.js

@@ -367,6 +367,24 @@ class PathwayResolver {
     }
 
     async executePathway(args) {
+        // Bidirectional context transformation for backward compatibility:
+        // 1. If agentContext provided: extract contextId/contextKey for legacy pathways
+        // 2. If contextId provided without agentContext: create agentContext for new pathways
+        if (args.agentContext && Array.isArray(args.agentContext) && args.agentContext.length > 0) {
+            const defaultCtx = args.agentContext.find(ctx => ctx.default) || args.agentContext[0];
+            if (defaultCtx) {
+                args.contextId = defaultCtx.contextId;
+                args.contextKey = defaultCtx.contextKey || null;
+            }
+        } else if (args.contextId && !args.agentContext) {
+            // Backward compat: create agentContext from legacy contextId/contextKey
+            args.agentContext = [{ 
+                contextId: args.contextId, 
+                contextKey: args.contextKey || null, 
+                default: true 
+            }];
+        }
+        
         if (this.pathway.executePathway && typeof this.pathway.executePathway === 'function') {
             return await this.pathway.executePathway({ args, runAllPrompts: this.promptAndParse.bind(this), resolver: this });
         }

package/server/plugins/replicateApiPlugin.js

@@ -321,6 +321,24 @@ class ReplicateApiPlugin extends ModelPlugin {
         };
         break;
       }
+      case "replicate-seedance-1.5-pro": {
+        const validRatios = ["16:9", "4:3", "1:1", "3:4", "9:16", "21:9", "9:21"];
+
+        requestParameters = {
+          input: {
+            prompt: modelPromptText,
+            aspect_ratio: validRatios.includes(combinedParameters.aspectRatio) ? combinedParameters.aspectRatio : "16:9",
+            duration: Math.min(12, Math.max(2, combinedParameters.duration || 5)),
+            fps: 24,
+            camera_fixed: combinedParameters.camera_fixed || false,
+            generate_audio: combinedParameters.generate_audio || false,
+            ...(combinedParameters.seed && Number.isInteger(combinedParameters.seed) && combinedParameters.seed > 0 ? { seed: combinedParameters.seed } : {}),
+            ...(combinedParameters.image ? { image: combinedParameters.image } : {}),
+            ...(combinedParameters.image && combinedParameters.last_frame_image ? { last_frame_image: combinedParameters.last_frame_image } : {}),
+          },
+        };
+        break;
+      }
       case "replicate-seedream-4": {
         const validSizes = ["1K", "2K", "4K", "custom"];
         const validRatios = ["1:1", "4:3", "3:4", "16:9", "9:16", "match_input_image"];

package/server/typeDef.js

@@ -58,6 +58,10 @@ const getGraphQlType = (value) => {
       const items = schema.items || {};
       const def = schema.default;
       const defaultArray = Array.isArray(def) ? JSON.stringify(def) : '[]';
+      // Support explicit object type name (e.g., items: { objType: 'AgentContextInput' })
+      if (items.objType) {
+        return { type: `[${items.objType}]`, defaultValue: `"${defaultArray.replace(/"/g, '\\"')}"` };
+      }
       if (items.type === 'string') {
         return { type: '[String]', defaultValue: defaultArray };
       }
@@ -103,6 +107,10 @@ const getGraphQlType = (value) => {
           if (Array.isArray(value[0]?.content)) {
             return {type: '[MultiMessage]', defaultValue: `"${JSON.stringify(value).replace(/"/g, '\\"')}"`};
           }
+          // Check if it's AgentContextInput (has contextId and default properties)
+          else if (value[0] && typeof value[0] === 'object' && 'contextId' in value[0] && 'default' in value[0]) {
+            return {type: '[AgentContextInput]', defaultValue: `"${JSON.stringify(value).replace(/"/g, '\\"')}"`};
+          }
           else {
             return {type: '[Message]', defaultValue: `"${JSON.stringify(value).replace(/"/g, '\\"')}"`};
           }
@@ -123,8 +131,9 @@ const getGraphQlType = (value) => {
 const getMessageTypeDefs = () => {
   const messageType = `input Message { role: String, content: String, name: String }`;
   const multiMessageType = `input MultiMessage { role: String, content: [String], name: String, tool_calls: [String], tool_call_id: String }`;
+  const agentContextType = `input AgentContextInput { contextId: String, contextKey: String, default: Boolean }`;
   
-  return `${messageType}\n\n${multiMessageType}`;
+  return `${messageType}\n\n${multiMessageType}\n\n${agentContextType}`;
 };
 
 const getPathwayTypeDef = (name, returnType) => {