@aj-archipelago/cortex 1.4.21 → 1.4.23

package/server/resolver.js

@@ -2,6 +2,7 @@ import { fulfillWithTimeout } from '../lib/promiser.js';
 import { PathwayResolver } from './pathwayResolver.js';
 import CortexResponse from '../lib/cortexResponse.js';
 import { withRequestLoggingDisabled } from '../lib/logger.js';
+import { sanitizeBase64 } from '../lib/util.js';
 
 // This resolver uses standard parameters required by Apollo server:
 // (parent, args, contextValue, info)
@@ -41,9 +42,37 @@ const rootResolver = async (parent, args, contextValue, info) => {
     let resultData = pathwayResolver.pathwayResultData ? JSON.stringify(pathwayResolver.pathwayResultData) : null;
     
     const { warnings, errors, previousResult, savedContextId, tool } = pathwayResolver;    
-
-    // Add request parameters back as debug
-    const debug = pathwayResolver.prompts.map(prompt => prompt.debugInfo || '').join('\n').trim();
+    
+    // Add request parameters back as debug - sanitize base64 data before returning
+    const debug = pathwayResolver.prompts.map(prompt => {
+        if (!prompt.debugInfo) return '';
+        try {
+            // Try to parse entire debugInfo as JSON first (for single JSON object)
+            try {
+                const parsed = JSON.parse(prompt.debugInfo);
+                return JSON.stringify(sanitizeBase64(parsed));
+            } catch (e) {
+                // Not a single JSON object, try line-by-line
+                const lines = prompt.debugInfo.split('\n');
+                return lines.map(line => {
+                    const trimmed = line.trim();
+                    if (trimmed.startsWith('{') || trimmed.startsWith('[')) {
+                        try {
+                            const parsed = JSON.parse(line);
+                            return JSON.stringify(sanitizeBase64(parsed));
+                        } catch (e) {
+                            // Not valid JSON on this line, return as-is
+                            return line;
+                        }
+                    }
+                    return line;
+                }).join('\n');
+            }
+        } catch (e) {
+            // If sanitization fails, return original
+            return prompt.debugInfo;
+        }
+    }).join('\n').trim();
     
     return { 
         debug, 

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) => {