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

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

import { parseNumberedList, parseNumberedObjectList, parseCommaSeparatedList, isCommaSeparatedList, isNumberedList, parseJson } from './parser.js';

class PathwayResponseParser {
    constructor(pathway) {
        this.pathway = pathway;
    }

    async parse(data) {
        if (this.pathway.parser) {
            return this.pathway.parser(data);
        }

        if (this.pathway.list) {
            if (isNumberedList(data)) {
                if (this.pathway.format) {
                    return await parseNumberedObjectList(data, this.pathway.format);
                }
                return parseNumberedList(data);
            } else if (isCommaSeparatedList(data)) {
                return parseCommaSeparatedList(data);
            }
            return [data];
        }

        if (this.pathway.json) {
            return await parseJson(data);
        }

        return data;
    }
}

export { PathwayResponseParser };