feat(ts-sdk): add Together LLM provider (#6049)
Co-authored-by: zhangsixiao <zhangsixiao@bytedance.com> Co-authored-by: kartik-mem0 <kartik.labhshetwar@mem0.ai>
This commit is contained in:
co-authored by
zhangsixiao
kartik-mem0
parent
803ff13bb8
commit
9b04509433
@@ -117,6 +117,11 @@ jest.mock("../src/llms/minimax", () => ({
|
||||
.fn()
|
||||
.mockImplementation((config) => ({ type: "minimax-llm", config })),
|
||||
}));
|
||||
jest.mock("../src/llms/together", () => ({
|
||||
TogetherLLM: jest
|
||||
.fn()
|
||||
.mockImplementation((config) => ({ type: "together-llm", config })),
|
||||
}));
|
||||
jest.mock("../src/llms/vllm", () => ({
|
||||
VllmLLM: jest
|
||||
.fn()
|
||||
@@ -266,6 +271,7 @@ describe("LLMFactory", () => {
|
||||
["xai"],
|
||||
["litellm"],
|
||||
["minimax"],
|
||||
["together"],
|
||||
["vllm"],
|
||||
])("creates LLM for provider '%s'", (provider) => {
|
||||
expect(() => LLMFactory.create(provider, dummyLLMConfig)).not.toThrow();
|
||||
@@ -297,6 +303,7 @@ describe("VectorStoreFactory", () => {
|
||||
VectorStoreFactory.create("memory", {
|
||||
collectionName: "test",
|
||||
dimension: 4,
|
||||
dbPath: ":memory:",
|
||||
}),
|
||||
).not.toThrow();
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user