Spaces:

Zhitomir
/

chat

Runtime error

chat / api /server /middleware /enforceModelSpec.js

chat init

aa2e6af over 2 years ago

1.71 kB

const interchangeableKeys = new Map([
  ['chatGptLabel', ['modelLabel']],
  ['modelLabel', ['chatGptLabel']],
]);

/**
 * Middleware to enforce the model spec for a conversation
 * @param {TModelSpec} modelSpec - The model spec to enforce
 * @param {TConversation} parsedBody - The parsed body of the conversation
 * @returns {boolean} - Whether the model spec is enforced
 */
const enforceModelSpec = (modelSpec, parsedBody) => {
  for (const [key, value] of Object.entries(modelSpec.preset)) {
    if (key === 'endpoint') {
      continue;
    }

    if (!checkMatch(key, value, parsedBody)) {
      return false;
    }
  }
  return true;
};

/**
 * Checks if there is a match for the given key and value in the parsed body
 * or any of its interchangeable keys, including deep comparison for objects and arrays.
 * @param {string} key
 * @param {any} value
 * @param {object} parsedBody
 * @returns {boolean}
 */
const checkMatch = (key, value, parsedBody) => {
  const isEqual = (a, b) => {
    if (Array.isArray(a) && Array.isArray(b)) {
      return a.length === b.length && a.every((val, index) => isEqual(val, b[index]));
    } else if (typeof a === 'object' && typeof b === 'object' && a !== null && b !== null) {
      const keysA = Object.keys(a);
      const keysB = Object.keys(b);
      return keysA.length === keysB.length && keysA.every((k) => isEqual(a[k], b[k]));
    }
    return a === b;
  };

  if (isEqual(parsedBody[key], value)) {
    return true;
  }

  if (interchangeableKeys.has(key)) {
    return interchangeableKeys
      .get(key)
      .some((interchangeableKey) => isEqual(parsedBody[interchangeableKey], value));
  }

  return false;
};

module.exports = enforceModelSpec;

	const interchangeableKeys = new Map([
	['chatGptLabel', ['modelLabel']],
	['modelLabel', ['chatGptLabel']],
	]);

	/**
	* Middleware to enforce the model spec for a conversation
	* @param {TModelSpec} modelSpec - The model spec to enforce
	* @param {TConversation} parsedBody - The parsed body of the conversation
	* @returns {boolean} - Whether the model spec is enforced
	*/
	const enforceModelSpec = (modelSpec, parsedBody) => {
	for (const [key, value] of Object.entries(modelSpec.preset)) {
	if (key === 'endpoint') {
	continue;
	}

	if (!checkMatch(key, value, parsedBody)) {
	return false;
	}
	}
	return true;
	};

	/**
	* Checks if there is a match for the given key and value in the parsed body
	* or any of its interchangeable keys, including deep comparison for objects and arrays.
	* @param {string} key
	* @param {any} value
	* @param {object} parsedBody
	* @returns {boolean}
	*/
	const checkMatch = (key, value, parsedBody) => {
	const isEqual = (a, b) => {
	if (Array.isArray(a) && Array.isArray(b)) {
	return a.length === b.length && a.every((val, index) => isEqual(val, b[index]));
	} else if (typeof a === 'object' && typeof b === 'object' && a !== null && b !== null) {
	const keysA = Object.keys(a);
	const keysB = Object.keys(b);
	return keysA.length === keysB.length && keysA.every((k) => isEqual(a[k], b[k]));
	}
	return a === b;
	};

	if (isEqual(parsedBody[key], value)) {
	return true;
	}

	if (interchangeableKeys.has(key)) {
	return interchangeableKeys
	.get(key)
	.some((interchangeableKey) => isEqual(parsedBody[interchangeableKey], value));
	}

	return false;
	};

	module.exports = enforceModelSpec;