Function updatePastAssistantMessage

lib/edge-runtime/angelaUtils.ts:11–47

(
  command: FunctionCall,
  nonSystemMessages: GPTMessageInclSummary[],
)

import _ from "lodash";

export function updatePastAssistantMessage(
  command: FunctionCall,
  nonSystemMessages: GPTMessageInclSummary[],
) {
  /** Updates the history of the assistant message to include the filled-in parameters in the command
   *
   * E.g. from
   * test(a=b)
   *
   * where requiredParam was missing to
   * test(a=b, requiredParam=value)
   * **/
  console.log("Updating past assistant message with command:", command);
  const newCommandLine = `${command.name}(${Object.entries(command.args)
    .map(([key, value]) => `${key}=${JSON.stringify(value)}`)
    .join(", ")})`;

  // Update the last assistant message in the nonSystemMessages array
  const lastAssistantMessageIndex =
    nonSystemMessages.length -
    1 -
    nonSystemMessages.reverse().findIndex((m) => m.role === "assistant");
  nonSystemMessages.reverse(); // Reverse back to original order

  nonSystemMessages[lastAssistantMessageIndex].content = nonSystemMessages[
    lastAssistantMessageIndex
  ].content
    .split("\n") // Split into lines
    .map((line) => {
      // Below checks if it's the same function call as the one we're updating
      if (line.startsWith(command.name + "(") && line.trim().endsWith(")")) {
        return newCommandLine;
      }
      return line;
    })
    .join("\n"); // Re-join lines
}

export async function streamResponseToUser(
  res: ReadableStream,