@aj-archipelago/cortex/pathways/system/workspaces/run_workspace_prompt.js

Cortex is a GraphQL API for AI. It provides a simple, extensible interface for using AI services from OpenAI, Azure and others.

import { config } from '../../../config.js';
import { chatArgsHasImageUrl, chatArgsHasType, getAvailableFiles, removeOldImageAndFileContent } from '../../../lib/util.js';
import { loadEntityConfig } from '../../../pathways/system/entity/tools/shared/sys_entity_tools.js';
import { Prompt } from '../../../server/prompt.js';

export default {
    emulateOpenAIChatModel: 'cortex-agent',
    useInputChunking: false,
    enableDuplicateRequests: false,
    useSingleTokenStream: false,

    inputParameters: {
        prompt: "",
        systemPrompt: "",
        chatHistory: [{role: '', content: []}],
        text: "",
        entityId: "labeeb",
        aiName: "Jarvis",
        language: "English",
        model: "oai-gpt41", // Allow user to specify model
    },

    timeout: 600,

    executePathway: async ({args, runAllPrompts, resolver}) => {
        let pathwayResolver = resolver;

        // Load input parameters and information into args
        const { entityId, aiName, language, model } = { ...pathwayResolver.pathway.inputParameters, ...args };
        
        const entityConfig = loadEntityConfig(entityId);

        // Initialize chat history if needed
        if (!args.chatHistory || args.chatHistory.length === 0) {
            args.chatHistory = [];
        }

        // Limit the chat history to 20 messages to speed up processing
        args.chatHistory = args.chatHistory.slice(-20);

        // Add entity constants for template rendering
        args = {
            ...args,
            ...config.get('entityConstants'),
            entityId,
            aiName,
            language,
            model
        };

        // Extract available files from chat history
        const availableFiles = getAvailableFiles(args.chatHistory);

        // Check for both image and file content (CSV files have type 'file', not 'image_url')
        const hasImageContent = chatArgsHasImageUrl(args);
        const hasFileContent = chatArgsHasType(args, 'file');
        const visionContentPresent = hasImageContent || hasFileContent;

        // Remove old image and file content while preserving the latest uploads
        visionContentPresent && (args.chatHistory = removeOldImageAndFileContent(args.chatHistory));

        const promptMessages = [
            {"role": "system", "content": `${args.systemPrompt || "Assistant is an expert journalist's assistant for Al Jazeera Media Network. When a user posts a request, Assistant will come up with the best response while upholding the highest journalistic standards."}\n\n{{renderTemplate AI_TOOLS}}\n\n{{renderTemplate AI_AVAILABLE_FILES}}\n\n{{renderTemplate AI_DATETIME}}`},
            "{{chatHistory}}"
        ];

        // Only add a user message if there's actual text or prompt content
        const userContent = `${args.text || ""}\n\n${args.prompt || ""}`.trim();
        if (userContent) {
            promptMessages.push({"role": "user", "content": userContent});
        }

        pathwayResolver.pathwayPrompt = [
            new Prompt({ messages: promptMessages }),
        ];

        pathwayResolver.args = {...args};

        try {
            let currentMessages = JSON.parse(JSON.stringify(args.chatHistory));

            let response = await runAllPrompts({
                ...args,
                chatHistory: currentMessages,
                availableFiles,
                model: args.model // Pass the model from args
            });

            return response;

        } catch (e) {
            pathwayResolver.logError(e);
            throw e;
        }
    }
};