src/parse-action.test.ts - master - githubmirror/ai-legion

src/parse-action.test.ts
231 строка5 KB
eumemic
rename filesystem module to system and implement runCommand action
29 янв 2025, 08:17
29 янв 2025, 08:170acca2b
import core from "./module/definitions/core";
import system from "./module/definitions/system";
import messaging from "./module/definitions/messaging";
import notes from "./module/definitions/notes";
import { ModuleManager } from "./module/module-manager";
import { getUsageText } from "./module/util";
import parseAction, { Action } from "./parse-action";
import { MULTILINE_DELIMITER } from "./util";

const moduleManager = new ModuleManager(
  "1",
  ["1"],
  [core, notes, messaging, system]
);

test("case 1", () => {
  assertValid(`
sendMessage
targetAgentId: 0
message:
% ff9d7713-0bb0-40d4-823c-5a66de48761b
Hello Control, this is Agent 1.
I am sending you a multi-line message.
This is another line.
% ff9d7713-0bb0-40d4-823c-5a66de48761b
  `);
});

test("case 2", () => {
  assertValid(`
sendMessage
targetAgentId: 0
message:
% ff9d7713-0bb0-40d4-823c-5a66de48761b
Hello Control, this is Agent 1.

Here is a multi-line message:
This is the first line.
This is the second line.
This is the third line.
% ff9d7713-0bb0-40d4-823c-5a66de48761b
    `);
});

test("case 3", () => {
  assertValid(`
writeFile
path: ./schema/action-dictionary.json
content:
% ff9d7713-0bb0-40d4-823c-5a66de48761b
{
  "$schema": "http://json-schema.org/draft-07/schema#",
  "type": "object",
  "title": "Action",
  "oneOf": [
    {
      ... the old content of action-dictionary.json ...
    },
    {
      "title": "NewActionDefinition",
      "description": "This is an example of a new action definition.",
      "properties": {
        "type": {
          "type": "string",
          "const": "new-action-definition"
        },
        "exampleArg": {
          "type": "string",
          "description": "An example parameter for the new action definition."
        }
      },
      "required": [
        "type",
        "exampleArg"
      ],
      "additionalProperties": false
    }
  ]
}
% ff9d7713-0bb0-40d4-823c-5a66de48761b
`);
});

test("invalid multiline with no delimiter", () => {
  expect(
    assertInvalid(`
sendMessage
targetAgentId: 0
message: Hello Control, here's a list of things,

- 9 bottles of ketchup
- 2 pairs of socks
- a very big balloon
  `)
  ).toEqual(
    `Your action could not be parsed. Did you fail to wrap the entirety of a multi-line parameter value with the multi-line delimiter (\`% ff9d7713-0bb0-40d4-823c-5a66de48761b\`)?`
  );
});

test("multiline delimiter not starting on its own line", () => {
  expect(
    assertValid(`
sendMessage
targetAgentId: 0
message: % ff9d7713-0bb0-40d4-823c-5a66de48761b
here is a
multi
line
message
% ff9d7713-0bb0-40d4-823c-5a66de48761b
`).parameters.message
  ).toBe(
    `
here is a
multi
line
message
`.trim()
  );
});

test("parameter after multiline parameter", () => {
  const { parameters, thoughts } = assertValid(`
sendMessage
targetAgentId: 0
message:
% ff9d7713-0bb0-40d4-823c-5a66de48761b
multi
line
parameter
% ff9d7713-0bb0-40d4-823c-5a66de48761b
thoughts: Trying a multi-line parameter followed by another parameter.
`);
  expect(parameters.message).toBe(
    `
multi
line
parameter
`.trim()
  );
  expect(thoughts).toBe(
    "Trying a multi-line parameter followed by another parameter."
  );
});

test("invalid command name", () => {
  expect(assertInvalid("foo")).toBe(
    "Unknown action `foo`. Please refer to the list of available actions given in the introductory message."
  );
});

test("invalid raw text", () => {
  expect(assertInvalid("Hello Control, how are you doing?")).toEqual(
    `
Your action could not be parsed. Remember to always format your entire response as an action, like this:

\`\`\`
<action name>
<arg 1 name>: <prop value>
<arg 2 name>: <prop value>
...
\`\`\`
`.trim()
  );
});

test("missing required parameter", () => {
  expect(assertInvalid("sendMessage\ntargetAgentId: 0")).toBe(
    `Missing required parameter \`message\`. ${getUsageText(
      messaging.actions.sendMessage
    )}`
  );
});

test("extra parameter", () => {
  expect(assertInvalid("noop\nfoo: bar")).toEqual(
    `Extraneous parameter \`foo\`. ${getUsageText(core.actions.noop)}`
  );
});

describe("quotes", () => {
  test("in-line parameter", () => {
    const action = assertValid(`
sendMessage
targetAgentId: 0
message: hello, "control"
`);
    expect(action.actionDef.name).toBe("sendMessage");
    expect(action.parameters).toEqual({
      targetAgentId: "0",
      message: 'hello, "control"',
    });
  });

  test("multi-line parameter", () => {
    const action = assertValid(`
sendMessage
targetAgentId: 0
message:
${MULTILINE_DELIMITER}
hello, "control"
${MULTILINE_DELIMITER}
`);
    expect(action.actionDef.name).toBe("sendMessage");
    expect(action.parameters).toEqual({
      targetAgentId: "0",
      message: 'hello, "control"',
    });
  });
});

// =============================================================================

function assertValid(text: string): Action {
  const result = parseAction(moduleManager.actions, text);
  if (result.type === "error") throw Error(`Parse failed: ${result.message}`);
  return result.action;
}

function assertInvalid(text: string): string {
  const result = parseAction(moduleManager.actions, text);
  if (result.type === "success")
    throw Error(
      `Parse succeeded when it should've failed: ${JSON.stringify(
        result.action,
        null,
        2
      )}`
    );
  return result.message;
}