xinyin025/ChatGPT-Next-Web
1
0
Fork 0
mirror of https://github.com/ChatGPTNextWeb/ChatGPT-Next-Web.git synced 2025-11-18 06:53:41 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix: allow isVisionModel utility function read env var

Browse Source
This commit is contained in:
JiangYinjin
2024-12-25 22:42:16 +08:00
parent 0c3d4462ca
commit ea1751fd4c
2 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/vision-model-checker.test.ts
View File

@@ -49,10 +49,11 @@ describe("isVisionModel", () => {
});
test("should identify models from VISION_MODELS env var", () => {
process.env.VISION_MODELS = "custom-vision-model,another-vision-model";
process.env.VISION_MODELS = "custom-vision-model,another-vision-model,OpenGVLab/InternVL2-26B";
expect(isVisionModel("custom-vision-model")).toBe(true);
expect(isVisionModel("another-vision-model")).toBe(true);
expect(isVisionModel("OpenGVLab/InternVL2-26B")).toBe(true);
expect(isVisionModel("unrelated-model")).toBe(false);
});