5e5e78d3e8
Expose user-added imageToText custom models as vision-capable chat
models in the default LLM selector and /v1/models, map custom service
kinds to runtime capabilities, and keep typed filtering for
/v1/models/{kind}.
Co-authored-by: Cursor <cursoragent@cursor.com>
16 lines
709 B
JavaScript
16 lines
709 B
JavaScript
import { describe, expect, it } from "vitest";
|
|
|
|
import { capabilitiesFromServiceKind } from "../../open-sse/providers/capabilities.js";
|
|
|
|
describe("capabilitiesFromServiceKind", () => {
|
|
it("maps imageToText custom models to vision-capable runtime models", () => {
|
|
expect(capabilitiesFromServiceKind("imageToText")).toMatchObject({ vision: true });
|
|
});
|
|
|
|
it("maps media output/input custom model kinds to runtime capabilities", () => {
|
|
expect(capabilitiesFromServiceKind("image")).toMatchObject({ imageOutput: true });
|
|
expect(capabilitiesFromServiceKind("stt")).toMatchObject({ audioInput: true });
|
|
expect(capabilitiesFromServiceKind("tts")).toMatchObject({ audioOutput: true });
|
|
});
|
|
});
|