@aj-archipelago/cortex/server/resolver.js

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

import { fulfillWithTimeout } from '../lib/promiser.js';
import { PathwayResolver } from './pathwayResolver.js';

// This resolver uses standard parameters required by Apollo server:
// (parent, args, contextValue, info)
const rootResolver = async (parent, args, contextValue, info) => {
    const { config, pathway } = contextValue;
    const { temperature, enableGraphqlCache } = pathway;

    // Turn on graphql caching if enableGraphqlCache true and temperature is 0
    if (enableGraphqlCache && temperature == 0) { // || 
        info.cacheControl.setCacheHint({ maxAge: 60 * 60 * 24, scope: 'PUBLIC' });
    }

    const pathwayResolver = new PathwayResolver({ config, pathway, args });
    contextValue.pathwayResolver = pathwayResolver;

    // Execute the request with timeout
    let result = null;

    try {
        result = await fulfillWithTimeout(pathway.resolver(parent, args, contextValue, info), pathway.timeout);
    } catch (error) {
        pathwayResolver.logError(error);
        result = error.message || error.toString();
    }
    
    const { warnings, errors, previousResult, savedContextId, tool } = pathwayResolver;
    
    // Add request parameters back as debug
    const debug = pathwayResolver.prompts.map(prompt => prompt.debugInfo || '').join('\n').trim();
    
    return { debug, result, warnings, errors, previousResult, tool, contextId: savedContextId }
}

// This resolver is used by the root resolver to process the request
const resolver = async (parent, args, contextValue, _info) => {
    const { pathwayResolver } = contextValue;
    return await pathwayResolver.resolve(args);
}

const cancelRequestResolver = (parent, args, contextValue, _info) => {
    const { requestId } = args;
    const { requestState } = contextValue;
    requestState[requestId] = { canceled: true };
    return true
}

export {
    resolver, rootResolver, cancelRequestResolver
};