feat: Implement Vertex AI support for Anthropic and Google models

This commit is contained in:
ryanhex53
2024-11-09 23:11:53 +08:00
parent 108069a0c6
commit 5222c2b6a7
11 changed files with 399 additions and 17 deletions

View File

@@ -1,4 +1,4 @@
import { getModelProvider } from "../app/utils/model";
import { getModelProvider, isModelAvailableInServer } from "../app/utils/model";
describe("getModelProvider", () => {
test("should return model and provider when input contains '@'", () => {
@@ -29,3 +29,15 @@ describe("getModelProvider", () => {
expect(provider).toBeUndefined();
});
});
describe("isModelAvailableInServer", () => {
test("works when model null", () => {
const jsonBody = JSON.parse("{}") as { model?: string };
const result = isModelAvailableInServer(
"gpt-3.5-turbo@OpenAI",
jsonBody.model as string,
"OpenAI",
);
expect(result).toBe(false);
});
});