@langchain/community
Version:
Third-party integrations for LangChain.js
105 lines (104 loc) • 3.11 kB
JavaScript
import { test, expect } from "@jest/globals";
import { SystemMessage, HumanMessage } from "@langchain/core/messages";
import { ChatMoonshot } from "../moonshot.js";
const runTest = async ({ modelName, config, system = "", message = "Hello!", shouldThrow = false, }) => {
const description = `Test ChatMoonshot ${modelName || "default model"} ${config.description || ""}`.trim();
let nrNewTokens = 0;
let streamedCompletion = "";
const passedConfig = { ...config };
if (passedConfig.streaming) {
passedConfig.callbacks = [
{
async handleLLMNewToken(token) {
nrNewTokens += 1;
streamedCompletion += token;
},
},
];
}
test.skip(description, async () => {
const chat = new ChatMoonshot({
modelName,
...config,
});
const messages = [];
if (system) {
messages.push(new SystemMessage(system));
}
messages.push(new HumanMessage(message));
if (shouldThrow) {
await expect(chat.invoke(messages)).rejects.toThrow();
return;
}
const res = await chat.invoke(messages, passedConfig);
console.log({ res });
if (passedConfig.streaming) {
expect(nrNewTokens > 0).toBe(true);
expect(res.content).toBe(streamedCompletion);
}
});
};
const testConfigs = [
{ modelName: undefined, config: {} },
{ modelName: "moonshot-v1-8k", config: {} },
{
modelName: "moonshot-v1-8k",
config: { description: "with temperature", temperature: 1 },
},
{
modelName: "moonshot-v1-8k",
config: { description: "with topP", topP: 1 },
},
{
modelName: "moonshot-v1-8k",
config: { description: "with repetitionPenalty" },
},
{
modelName: "moonshot-v1-8k",
config: {
description: "in streaming mode",
streaming: true,
},
message: "您好，请讲个长笑话",
},
{
modelName: "moonshot-v1-8k",
config: {
description: "illegal input should throw an error",
temperature: 0,
},
shouldThrow: true,
},
{
modelName: "moonshot-v1-8k",
config: {
description: "illegal input in streaming mode should throw an error",
streaming: true,
temperature: 0,
},
shouldThrow: true,
},
{ modelName: "moonshot-v1-128k", config: {} },
{
modelName: "moonshot-v1-128k",
config: {
description: "in streaming mode",
streaming: true,
},
message: "您好，请讲个长笑话",
},
{
modelName: "moonshot-v1-128k",
config: {
description: "with system message",
},
system: "你是一个说文言文的人",
},
{
modelName: "moonshot-v1-128k",
config: {},
},
];
testConfigs.forEach((testConfig) => {
void runTest(testConfig);
});