import { describe, expect, test } from "vitest";
import { buildGenerateNodePayload } from "./venice-api";

describe("venice api", () => {
  test("builds branch requests with parent context and uploaded image data", () => {
    const payload = buildGenerateNodePayload({
      query: "AMD GPU visual search",
      branchPrompt: "compare model serving options",
      imageBase64: "data:image/png;base64,abc",
      parent: {
        id: "node_parent",
        title: "AMD GPU visual search",
        query: "AMD GPU visual search",
        imagePrompt: "visual board",
      },
    });

    expect(payload).toEqual({
      query: "AMD GPU visual search",
      branchPrompt: "compare model serving options",
      imageBase64: "data:image/png;base64,abc",
      parentNodeId: "node_parent",
      parentContext: {
        title: "AMD GPU visual search",
        query: "AMD GPU visual search",
        imagePrompt: "visual board",
      },
    });
  });
});