cr

langchain-core/src/caches/tests/in_memory_cache.test.ts

@@ -1,3 +1,4 @@
+import { test, expect } from "@jest/globals";
 import { MessageContentComplex } from "../../messages/base.js";
 import { InMemoryCache } from "../base.js";
 

langchain-core/src/callbacks/tests/manager.int.test.ts

@@ -1,5 +1,5 @@
 /* eslint-disable no-process-env */
-import { test } from "@jest/globals";
+import { test, expect } from "@jest/globals";
 
 import { PromptTemplate } from "../../prompts/prompt.js";
 import { FakeLLM } from "../../utils/testing/index.js";

langchain-core/src/callbacks/tests/run_collector.test.ts

@@ -9,6 +9,7 @@ import { BaseLLM } from "../../language_models/llms.js";
 import { StringOutputParser } from "../../output_parsers/string.js";
 import type { LLMResult } from "../../outputs.js";
 import { RunCollectorCallbackHandler } from "../../tracers/run_collector.js";
+import { describe, it, expect } from "@jest/globals";
 
 class FakeLLM extends BaseLLM {
   nrMapCalls = 0;

langchain-core/src/language_models/tests/chat_models.test.ts

@@ -1,6 +1,6 @@
 /* eslint-disable no-promise-executor-return */
 
-import { test } from "@jest/globals";
+import { test, expect } from "@jest/globals";
 import { z } from "zod";
 import { zodToJsonSchema } from "zod-to-json-schema";
 import { FakeChatModel, FakeListChatModel } from "../../utils/testing/index.js";

langchain-core/src/language_models/tests/llms.test.ts

@@ -1,6 +1,6 @@
 /* eslint-disable no-promise-executor-return */
 
-import { test } from "@jest/globals";
+import { test, expect } from "@jest/globals";
 import { FakeLLM, FakeStreamingLLM } from "../../utils/testing/index.js";
 import { HumanMessagePromptTemplate } from "../../prompts/chat.js";
 

langchain-core/src/messages/tests/message_utils.test.ts

@@ -1,4 +1,4 @@
-import { it, expect } from "@jest/globals";
+import { it, describe, test, expect } from "@jest/globals";
 import {
   filterMessages,
   mergeMessageRuns,

langchain-core/src/output_parsers/openai_tools/tests/json_output_tools_parser.test.ts

@@ -1,4 +1,4 @@
-import { test } from "@jest/globals";
+import { test, expect } from "@jest/globals";
 import { z } from "zod";
 import { JsonOutputKeyToolsParser } from "../json_output_tools_parsers.js";
 import { AIMessage } from "../../../messages/index.js";

langchain-core/src/output_parsers/tests/json.test.ts

@@ -3,6 +3,7 @@ import { RunnableSequence } from "../../runnables/base.js";
 import { RunnablePassthrough } from "../../runnables/passthrough.js";
 import { FakeStreamingLLM } from "../../utils/testing/index.js";
 import { JsonOutputParser } from "../json.js";
+import { test, expect } from "@jest/globals";
 
 const STREAMED_TOKENS = `
 {

langchain-core/src/output_parsers/tests/output_parser.test.ts

@@ -1,7 +1,7 @@
 /* eslint-disable no-loop-func */
 /* eslint-disable no-promise-executor-return */
 
-import { test } from "@jest/globals";
+import { test, expect } from "@jest/globals";
 import { FakeStreamingLLM } from "../../utils/testing/index.js";
 import { BytesOutputParser } from "../bytes.js";
 import {

langchain-core/src/output_parsers/tests/string.test.ts

@@ -1,4 +1,4 @@
-import { describe } from "@jest/globals";
+import { describe, test, expect } from "@jest/globals";
 import { StringOutputParser } from "../string.js";
 import {
   AIMessage,

langchain-core/src/output_parsers/tests/xml.test.ts

@@ -1,5 +1,6 @@
 import { FakeStreamingLLM } from "../../utils/testing/index.js";
 import { XMLOutputParser } from "../xml.js";
+import { test, expect } from "@jest/globals";
 
 const XML_EXAMPLE = `<?xml version="1.0" encoding="UTF-8"?>
 <userProfile>

langchain-core/src/prompts/tests/few_shot.test.ts

@@ -1,4 +1,4 @@
-import { expect, test } from "@jest/globals";
+import { expect, describe, test } from "@jest/globals";
 import {
   FewShotChatMessagePromptTemplate,
   FewShotPromptTemplate,

langchain-core/src/prompts/tests/structured.test.ts

@@ -11,6 +11,7 @@ import { RunnableConfig } from "../../runnables/config.js";
 import { FakeListChatModel } from "../../utils/testing/index.js";
 import { StructuredPrompt } from "../structured.js";
 import { load } from "../../load/index.js";
+import { test, expect } from "@jest/globals";
 
 class FakeStructuredChatModel extends FakeListChatModel {
   withStructuredOutput<

langchain-core/src/runnables/tests/runnable.test.ts

@@ -28,6 +28,7 @@ import { RunnableSequence, RunnableLambda } from "../base.js";
 import { RouterRunnable } from "../router.js";
 import { RunnableConfig } from "../config.js";
 import { JsonOutputParser } from "../../output_parsers/json.js";
+import { test, expect, describe } from "@jest/globals";
 
 test("Test batch", async () => {
   const llm = new FakeLLM({});

langchain-core/src/runnables/tests/runnable_binding.test.ts

@@ -1,6 +1,6 @@
 /* eslint-disable no-promise-executor-return */
 /* eslint-disable @typescript-eslint/no-explicit-any */
-import { test } from "@jest/globals";
+import { test, expect } from "@jest/globals";
 import { StringOutputParser } from "../../output_parsers/string.js";
 import { FakeChatModel, FakeStreamingLLM } from "../../utils/testing/index.js";
 

langchain-core/src/runnables/tests/runnable_branch.test.ts

@@ -1,7 +1,7 @@
 /* eslint-disable no-promise-executor-return */
 /* eslint-disable @typescript-eslint/no-explicit-any */
 
-import { test } from "@jest/globals";
+import { test, expect } from "@jest/globals";
 import { RunnableBranch } from "../branch.js";
 import { ChatPromptTemplate } from "../../prompts/chat.js";
 import { FakeStreamingLLM } from "../../utils/testing/index.js";

langchain-core/src/runnables/tests/runnable_graph.test.ts

@@ -4,6 +4,7 @@ import { StringOutputParser } from "../../output_parsers/string.js";
 import { FakeLLM } from "../../utils/testing/index.js";
 import { PromptTemplate } from "../../prompts/prompt.js";
 import { CommaSeparatedListOutputParser } from "../../output_parsers/list.js";
+import { test, expect } from "@jest/globals";
 
 test("Test graph single runnable", async () => {
   const jsonOutputParser = new StringOutputParser();

langchain-core/src/runnables/tests/runnable_history.test.ts

@@ -20,6 +20,7 @@ import {
 } from "../../utils/testing/index.js";
 import { ChatPromptTemplate, MessagesPlaceholder } from "../../prompts/chat.js";
 import { StringOutputParser } from "../../output_parsers/string.js";
+import { test, expect } from "@jest/globals";
 
 // For `BaseChatMessageHistory`
 async function getGetSessionHistory(): Promise<

langchain-core/src/runnables/tests/runnable_interface.test.ts

@@ -4,6 +4,7 @@
 import { StringOutputParser } from "../../output_parsers/string.js";
 import { PromptTemplate } from "../../prompts/prompt.js";
 import { RunnableSequence } from "../base.js";
+import { test, expect } from "@jest/globals";
 
 type RunnableBatchOptionsV0 = {
   maxConcurrency?: number;

langchain-core/src/runnables/tests/runnable_map.test.ts

@@ -16,6 +16,7 @@ import {
 } from "../../utils/testing/index.js";
 import { RunnableSequence, RunnableMap } from "../base.js";
 import { RunnablePassthrough } from "../passthrough.js";
+import { test, expect } from "@jest/globals";
 
 test("Create a runnable sequence with a runnable map", async () => {
   const promptTemplate = ChatPromptTemplate.fromMessages<{

langchain-core/src/runnables/tests/runnable_passthrough.test.ts

@@ -4,6 +4,7 @@ import { RunnablePassthrough } from "../passthrough.js";
 import { JsonOutputParser } from "../../output_parsers/json.js";
 import { RunnableSequence } from "../base.js";
 import { RunnableConfig } from "../config.js";
+import { test, expect } from "@jest/globals";
 
 test("RunnablePassthrough can call .assign and pass prev result through", async () => {
   const promptTemplate = PromptTemplate.fromTemplate("{input}");

langchain-core/src/runnables/tests/runnable_remote.int.test.ts

@@ -1,6 +1,7 @@
 import { HumanMessage } from "../../messages/index.js";
 import { applyPatch } from "../../utils/json_patch.js";
 import { RemoteRunnable } from "../remote.js";
+import { test, expect } from "@jest/globals";
 
 test("invoke hosted langserve", async () => {
   const remote = new RemoteRunnable({

langchain-core/src/runnables/tests/runnable_remote.test.ts

@@ -1,6 +1,6 @@
 /* eslint-disable no-promise-executor-return */
 /* eslint-disable @typescript-eslint/no-explicit-any */
-import { jest, test } from "@jest/globals";
+import { jest, test, expect, describe } from "@jest/globals";
 import {
   AIMessage,
   AIMessageChunk,

langchain-core/src/runnables/tests/runnable_retry.test.ts

@@ -1,7 +1,7 @@
 /* eslint-disable no-promise-executor-return */
 /* eslint-disable @typescript-eslint/no-explicit-any */
 
-import { test } from "@jest/globals";
+import { test, expect } from "@jest/globals";
 import { RunnableLambda } from "../base.js";
 
 test("RunnableRetry invoke", async () => {

langchain-core/src/runnables/tests/runnable_stream_events.test.ts

@@ -2,7 +2,7 @@
 /* eslint-disable no-process-env */
 /* eslint-disable @typescript-eslint/no-explicit-any */
 
-import { test } from "@jest/globals";
+import { test, expect, afterEach } from "@jest/globals";
 import { z } from "zod";
 import {
   RunnableLambda,

langchain-core/src/runnables/tests/runnable_stream_events_v2.test.ts

@@ -2,7 +2,7 @@
 /* eslint-disable no-process-env */
 /* eslint-disable @typescript-eslint/no-explicit-any */
 
-import { test } from "@jest/globals";
+import { test, expect, afterEach } from "@jest/globals";
 import { z } from "zod";
 import {
   RunnableLambda,

langchain-core/src/runnables/tests/runnable_stream_log.test.ts

@@ -1,7 +1,7 @@
 /* eslint-disable no-promise-executor-return */
 /* eslint-disable @typescript-eslint/no-explicit-any */
 
-import { test } from "@jest/globals";
+import { test, expect } from "@jest/globals";
 import {
   ChatPromptTemplate,
   HumanMessagePromptTemplate,

langchain-core/src/runnables/tests/runnable_tools.test.ts

@@ -2,6 +2,7 @@ import { z } from "zod";
 import { RunnableLambda, RunnableToolLike } from "../base.js";
 import { FakeRetriever } from "../../utils/testing/index.js";
 import { Document } from "../../documents/document.js";
+import { test, expect } from "@jest/globals";
 
 test("Runnable asTool works", async () => {
   const schema = z.object({

langchain-core/src/runnables/tests/runnable_tracing.int.test.ts

@@ -6,6 +6,7 @@ import {
 } from "../../output_parsers/base.js";
 import { FakeChatModel } from "../../utils/testing/index.js";
 import { getEnvironmentVariable } from "../../utils/env.js";
+import { test } from "@jest/globals";
 
 class FakeDateOutputParser extends BaseOutputParser<Date> {
   lc_namespace = ["langchain_core", "output_parsers", "testing"];

langchain-core/src/runnables/tests/runnable_with_fallbacks.test.ts

@@ -1,6 +1,6 @@
 /* eslint-disable no-promise-executor-return */
 /* eslint-disable @typescript-eslint/no-explicit-any */
-import { test } from "@jest/globals";
+import { test, expect } from "@jest/globals";
 import { FakeLLM } from "../../utils/testing/index.js";
 
 test("RunnableWithFallbacks", async () => {

langchain-core/src/structured_query/tests/utils.test.ts

@@ -1,5 +1,5 @@
 /* eslint-disable no-process-env */
-import { test } from "@jest/globals";
+import { test, expect } from "@jest/globals";
 import { castValue, isFloat, isInt, isString } from "../utils.js";
 
 test("Casting values correctly", () => {

langchain-core/src/tracers/tests/langchain_tracer.int.test.ts

@@ -1,6 +1,6 @@
 /* eslint-disable no-process-env */
 import * as uuid from "uuid";
-import { test } from "@jest/globals";
+import { test, expect } from "@jest/globals";
 
 import { LangChainTracer, Run } from "../tracer_langchain.js";
 import { Serialized } from "../../load/serializable.js";

langchain-core/src/tracers/tests/langsmith_interop.test.ts

@@ -2,7 +2,7 @@
 /* eslint-disable no-process-env */
 /* eslint-disable no-promise-executor-return */
 
-import { jest } from "@jest/globals";
+import { jest, test, beforeEach, afterEach, afterAll, expect } from "@jest/globals";
 import { traceable } from "langsmith/traceable";
 
 import { RunnableLambda } from "../../runnables/base.js";

langchain-core/src/utils/tests/enviroment.test.ts

@@ -1,4 +1,5 @@
 import { getRuntimeEnvironment } from "../env.js";
+import { test, expect } from "@jest/globals";
 
 test("test getRuntimeEnvironment", async () => {
   const runtimeEnvironment = await getRuntimeEnvironment();