@langchain/community/dist/chat_models/tests/chatfireworks.standard.test.js

Third-party integrations for LangChain.js

/* eslint-disable no-process-env */
import { test, expect } from "@jest/globals";
import { ChatModelUnitTests } from "@langchain/standard-tests";
import { ChatFireworks } from "../fireworks.js";
class ChatFireworksStandardUnitTests extends ChatModelUnitTests {
    constructor() {
        super({
            Cls: ChatFireworks,
            chatModelHasToolCalling: true,
            chatModelHasStructuredOutput: true,
            constructorArgs: {},
        });
        process.env.FIREWORKS_API_KEY = "test";
    }
    testChatModelInitApiKey() {
        // Unset the API key env var here so this test can properly check
        // the API key class arg.
        process.env.FIREWORKS_API_KEY = "";
        super.testChatModelInitApiKey();
        // Re-set the API key env var here so other tests can run properly.
        process.env.FIREWORKS_API_KEY = "test";
    }
}
const testClass = new ChatFireworksStandardUnitTests();
test("ChatFireworksStandardUnitTests", () => {
    const testResults = testClass.runTests();
    expect(testResults).toBe(true);
});