format n lint

langchain-ai · May 31, 2024 · 6b1c916 · 6b1c916
1 parent 968a047
commit 6b1c916
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 14 deletions.
diff --git a/libs/langchain-standard-tests/package.json b/libs/langchain-standard-tests/package.json
@@ -32,11 +32,11 @@
   "author": "LangChain",
   "license": "MIT",
   "dependencies": {
+    "@jest/globals": "^29.5.0",
     "@langchain/core": "workspace:*",
     "zod": "^3.22.4"
   },
   "devDependencies": {
-    "@jest/globals": "^29.5.0",
     "@langchain/scripts": "workspace:*",
     "@swc/core": "^1.3.90",
     "@swc/jest": "^0.2.29",

diff --git a/libs/langchain-standard-tests/src/base.ts b/libs/langchain-standard-tests/src/base.ts
@@ -7,25 +7,26 @@ import { BaseMessageChunk } from "@langchain/core/messages";
 export type BaseChatModelConstructor<
   CallOptions extends BaseChatModelCallOptions = BaseChatModelCallOptions,
   OutputMessageType extends BaseMessageChunk = BaseMessageChunk
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
 > = new (...args: any[]) => BaseChatModel<CallOptions, OutputMessageType>;
 
 export type BaseChatModelsTestsFields<
   CallOptions extends BaseChatModelCallOptions = BaseChatModelCallOptions,
   OutputMessageType extends BaseMessageChunk = BaseMessageChunk
 > = {
-  cls: BaseChatModelConstructor<CallOptions, OutputMessageType>;
+  Cls: BaseChatModelConstructor<CallOptions, OutputMessageType>;
 };
 
 export class BaseChatModelsTests<
   CallOptions extends BaseChatModelCallOptions = BaseChatModelCallOptions,
   OutputMessageType extends BaseMessageChunk = BaseMessageChunk
 > implements BaseChatModelsTestsFields<CallOptions, OutputMessageType>
 {
-  cls: BaseChatModelConstructor<CallOptions, OutputMessageType>;
+  Cls: BaseChatModelConstructor<CallOptions, OutputMessageType>;
 
   constructor(
     fields: BaseChatModelsTestsFields<CallOptions, OutputMessageType>
   ) {
-    this.cls = fields.cls;
+    this.Cls = fields.Cls;
   }
 }
diff --git a/libs/langchain-standard-tests/src/integration_tests/chat_models.ts b/libs/langchain-standard-tests/src/integration_tests/chat_models.ts
@@ -7,15 +7,16 @@ import { BaseMessageChunk } from "@langchain/core/messages";
 type BaseChatModelConstructor<
   CallOptions extends BaseChatModelCallOptions = BaseChatModelCallOptions,
   OutputMessageType extends BaseMessageChunk = BaseMessageChunk
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
 > = new (...args: any[]) => BaseChatModel<CallOptions, OutputMessageType>;
 
 export class ChatModelsIntegrationTests<
   CallOptions extends BaseChatModelCallOptions = BaseChatModelCallOptions,
   OutputMessageType extends BaseMessageChunk = BaseMessageChunk
 > {
-  cls: BaseChatModel<CallOptions, OutputMessageType>;
+  Cls: BaseChatModel<CallOptions, OutputMessageType>;
 
-  constructor(cls: BaseChatModelConstructor<CallOptions, OutputMessageType>) {
-    this.cls = new cls();
+  constructor(Cls: BaseChatModelConstructor<CallOptions, OutputMessageType>) {
+    this.Cls = new Cls();
   }
 }
diff --git a/libs/langchain-standard-tests/src/unit_tests/chat_models.ts b/libs/langchain-standard-tests/src/unit_tests/chat_models.ts
@@ -1,5 +1,4 @@
 import { expect } from "@jest/globals";
-import { BaseChatModelsTests, BaseChatModelsTestsFields } from "../base.js";
 import {
   BaseChatModelCallOptions,
   LangSmithParams,
@@ -9,6 +8,7 @@ import { z } from "zod";
 import { StructuredTool } from "@langchain/core/tools";
 import { ChatResult } from "@langchain/core/outputs";
 import { CallbackManagerForLLMRun } from "@langchain/core/callbacks/manager";
+import { BaseChatModelsTests, BaseChatModelsTestsFields } from "../base.js";
 
 const person = z
   .object({
@@ -43,19 +43,19 @@ export class ChatModelUnitTests<
   }
 
   testChatModelInit(constructorArgs: RecordStringAny) {
-    const chatModel = new this.cls(constructorArgs);
+    const chatModel = new this.Cls(constructorArgs);
     expect(chatModel).toBeDefined();
   }
 
   testChatModelInitApiKey(constructorArgs: RecordStringAny) {
     const params = { ...constructorArgs, apiKey: "test" };
-    const chatModel = new this.cls(params);
+    const chatModel = new this.Cls(params);
     expect(chatModel).toBeDefined();
   }
 
   testChatModelInitStreaming(constructorArgs: RecordStringAny) {
     const params = { ...constructorArgs, streaming: true };
-    const chatModel = new this.cls(params);
+    const chatModel = new this.Cls(params);
     expect(chatModel).toBeDefined();
   }
 
@@ -66,7 +66,7 @@ export class ChatModelUnitTests<
     if (!chatModelHasToolCalling) {
       return;
     }
-    const chatModel = new this.cls(constructorArgs);
+    const chatModel = new this.Cls(constructorArgs);
     expect(chatModel.bindTools?.([new PersonTool()])).toBeDefined();
   }
 
@@ -77,7 +77,8 @@ export class ChatModelUnitTests<
     if (!chatModelHasStructuredOutput) {
       return;
     }
-    const chatModel = new this.cls(constructorArgs);
+    const chatModel = new this.Cls(constructorArgs);
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
     expect((chatModel as any).withStructuredOutput?.(person)).toBeDefined();
   }
 
@@ -90,7 +91,7 @@ export class ChatModelUnitTests<
       ls_max_tokens: 0,
       ls_stop: ["Array<string>"],
     };
-    class ModelExtendsChatModel extends this.cls {
+    class ModelExtendsChatModel extends this.Cls {
       constructor(fields: RecordStringAny) {
         super(fields);
       }