@@ -6,7 +6,7 @@ import { BaseChatModelCallOptions } from "@langchain/core/language_models/chat_m
66import {
77 AIMessage ,
88 AIMessageChunk ,
9- BaseMessageChunk ,
9+ BaseMessage ,
1010 HumanMessage ,
1111 SystemMessage ,
1212 ToolMessage ,
@@ -29,9 +29,9 @@ import {
2929 BaseChatModelsTests ,
3030 BaseChatModelsTestsFields ,
3131 RecordStringAny ,
32- } from "../base.ts " ;
33- import { TestCallbackHandler } from "../utils.ts " ;
34- import { isContentBlock } from "../utils/types.ts " ;
32+ } from "../base.js " ;
33+ import { TestCallbackHandler } from "../utils.js " ;
34+ import { isContentBlock } from "../utils/types.js " ;
3535
3636// Placeholder data for content block tests
3737const TEST_IMAGE_URL =
@@ -89,7 +89,7 @@ const MATH_ADDITION_PROMPT = /* #__PURE__ */ ChatPromptTemplate.fromMessages([
8989
9090export interface ChatModelIntegrationTestsFields <
9191 CallOptions extends BaseChatModelCallOptions = BaseChatModelCallOptions ,
92- OutputMessageType extends BaseMessageChunk = BaseMessageChunk ,
92+ OutputMessageType extends AIMessageChunk = AIMessageChunk ,
9393 ConstructorArgs extends RecordStringAny = RecordStringAny
9494> extends BaseChatModelsTestsFields <
9595 CallOptions ,
@@ -117,7 +117,7 @@ export interface ChatModelIntegrationTestsFields<
117117
118118export abstract class ChatModelIntegrationTests <
119119 CallOptions extends BaseChatModelCallOptions = BaseChatModelCallOptions ,
120- OutputMessageType extends BaseMessageChunk = BaseMessageChunk ,
120+ OutputMessageType extends AIMessageChunk = AIMessageChunk ,
121121 ConstructorArgs extends RecordStringAny = RecordStringAny
122122> extends BaseChatModelsTests < CallOptions , OutputMessageType , ConstructorArgs > {
123123 functionId = "abc123" ;
@@ -1510,7 +1510,7 @@ export abstract class ChatModelIntegrationTests<
15101510 const modelWithTools = model . bindTools ( [ weatherTool ] ) ;
15111511
15121512 // Initialize the conversation with a weather query
1513- const messages = [
1513+ const messages : BaseMessage [ ] = [
15141514 new HumanMessage (
15151515 "What's the weather like in San Francisco right now? Use the 'get_current_weather' tool to find the answer."
15161516 ) ,
@@ -1602,7 +1602,7 @@ export abstract class ChatModelIntegrationTests<
16021602 const modelWithTools = model . bindTools ( [ weatherTool ] ) ;
16031603
16041604 // Initialize the conversation with a weather query
1605- const messages = [
1605+ const messages : BaseMessage [ ] = [
16061606 new HumanMessage (
16071607 "What's the weather like in San Francisco right now? Use the 'get_current_weather' tool to find the answer."
16081608 ) ,
0 commit comments