Skip to content

Commit 767796e

Browse files
Fix Ollama tests
1 parent 98968f9 commit 767796e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

packages/modules/ollama/src/ollama-container.test.ts

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import { ImageName } from "testcontainers";
12
import { getImage } from "../../../testcontainers/src/utils/test-helper";
23
import { OllamaContainer } from "./ollama-container";
34

@@ -11,7 +12,7 @@ describe("OllamaContainer", { timeout: 180_000 }, () => {
1112
const response = await fetch(`${container.getEndpoint()}/api/version`);
1213
expect(response.status).toEqual(200);
1314
const body = (await response.json()) as { version: string };
14-
expect(body.version).toEqual("0.1.44");
15+
expect(body.version).toEqual(ImageName.fromString(IMAGE).tag);
1516
await container.stop();
1617
});
1718

0 commit comments

Comments
 (0)