Add unit test

Author: mme

sdks/typescript/packages/client/src/agent/agent.ts

@@ -37,7 +37,10 @@ export abstract class AbstractAgent {
   public subscribers: AgentSubscriber[] = [];
   public isRunning: boolean = false;
   private middlewares: Middleware[] = [];
-  public maxVersion = packageJson.version;
+
+  get maxVersion() {
+    return packageJson.version;
+  }
 
   constructor({
     agentId,

sdks/typescript/packages/client/src/middleware/__tests__/backward-compatibility-0-0-39.test.ts

@@ -0,0 +1,66 @@
+import { AbstractAgent } from "@/agent";
+import { BaseEvent, EventType, Message, RunAgentInput } from "@ag-ui/core";
+import { Observable, of } from "rxjs";
+
+class LegacyAgent extends AbstractAgent {
+  public receivedInput?: RunAgentInput;
+
+  constructor(initialMessages: Message[]) {
+    super({ initialMessages });
+  }
+
+  override get maxVersion(): string {
+    return "0.0.39";
+  }
+
+  override run(input: RunAgentInput): Observable<BaseEvent> {
+    this.receivedInput = input;
+    return of({
+      type: EventType.RUN_STARTED,
+      threadId: input.threadId,
+      runId: input.runId,
+    } as BaseEvent);
+  }
+
+  protected override prepareRunAgentInput(
+    parameters?: Parameters<AbstractAgent["prepareRunAgentInput"]>[0],
+  ): RunAgentInput {
+    const prepared = super.prepareRunAgentInput(parameters);
+    return { ...prepared, parentRunId: "legacy-parent" };
+  }
+}
+
+describe("BackwardCompatibility_0_0_39 middleware (auto insertion)", () => {
+  it("automatically strips parentRunId and flattens array message content when maxVersion <= 0.0.39", async () => {
+    const initialMessages: Message[] = [
+      {
+        id: "msg-1",
+        role: "user",
+        content: [
+          { type: "text", text: "Hello " },
+          { type: "text", text: "world!" },
+          { type: "binary", mimeType: "text/plain", data: "ignored" },
+        ] as unknown as Message["content"],
+      } as Message,
+      {
+        id: "msg-2",
+        role: "assistant",
+        content: undefined,
+      } as Message,
+    ];
+
+    const agent = new LegacyAgent(initialMessages);
+
+    await agent.runAgent({
+      runId: "run-1",
+      tools: [],
+      context: [],
+      forwardedProps: {},
+    });
+
+    expect(agent.receivedInput).toBeDefined();
+    expect(agent.receivedInput?.parentRunId).toBeUndefined();
+    expect(agent.receivedInput?.messages[0].content).toBe("Hello world!");
+    expect(agent.receivedInput?.messages[1].content).toBe("");
+  });
+});