From 69f51c7e8ac1dd933563405c2c1ca3fd980b850c Mon Sep 17 00:00:00 2001 From: jacoblee93 Date: Fri, 17 Jan 2025 23:36:41 -0800 Subject: [PATCH] Format, use BaseLanguageModel instead of BaseChatModel --- examples/src/tools/google_calendar.ts | 2 +- libs/langchain-community/src/tools/google_calendar/base.ts | 6 +++--- .../src/tools/google_calendar/commands/run-create-events.ts | 4 ++-- .../src/tools/google_calendar/commands/run-view-events.ts | 4 ++-- 4 files changed, 8 insertions(+), 8 deletions(-) diff --git a/examples/src/tools/google_calendar.ts b/examples/src/tools/google_calendar.ts index b6b167f05c30..948af910c161 100644 --- a/examples/src/tools/google_calendar.ts +++ b/examples/src/tools/google_calendar.ts @@ -34,7 +34,7 @@ export async function run() { const calendarAgent = createReactAgent({ llm: model, - tools + tools, }); const createInput = `Create a meeting with John Doe next Friday at 4pm - adding to the agenda of it the result of 99 + 99`; diff --git a/libs/langchain-community/src/tools/google_calendar/base.ts b/libs/langchain-community/src/tools/google_calendar/base.ts index 9da0c8f1b487..f1962091acf9 100644 --- a/libs/langchain-community/src/tools/google_calendar/base.ts +++ b/libs/langchain-community/src/tools/google_calendar/base.ts @@ -1,7 +1,7 @@ import { google } from "googleapis"; import { Tool } from "@langchain/core/tools"; import { getEnvironmentVariable } from "@langchain/core/utils/env"; -import { BaseChatModel } from "@langchain/core/language_models/chat_models"; +import { BaseLanguageModel } from "@langchain/core/language_models/base"; export interface GoogleCalendarAgentParams { credentials?: { @@ -10,7 +10,7 @@ export interface GoogleCalendarAgentParams { calendarId?: string; }; scopes?: string[]; - model?: BaseChatModel; + model?: BaseLanguageModel; } export class GoogleCalendarBase extends Tool { @@ -27,7 +27,7 @@ export class GoogleCalendarBase extends Tool { protected scopes: string[]; - protected llm: BaseChatModel; + protected llm: BaseLanguageModel; constructor( fields: GoogleCalendarAgentParams = { diff --git a/libs/langchain-community/src/tools/google_calendar/commands/run-create-events.ts b/libs/langchain-community/src/tools/google_calendar/commands/run-create-events.ts index 6abcd9a792b9..2eea293ab68c 100644 --- a/libs/langchain-community/src/tools/google_calendar/commands/run-create-events.ts +++ b/libs/langchain-community/src/tools/google_calendar/commands/run-create-events.ts @@ -2,7 +2,7 @@ import { google, calendar_v3 } from "googleapis"; import type { JWT, GaxiosResponse } from "googleapis-common"; import { PromptTemplate } from "@langchain/core/prompts"; import { CallbackManagerForToolRun } from "@langchain/core/callbacks/manager"; -import { BaseChatModel } from "@langchain/core/language_models/chat_models"; +import { BaseLanguageModel } from "@langchain/core/language_models/base"; import { StringOutputParser } from "@langchain/core/output_parsers"; import { CREATE_EVENT_PROMPT } from "../prompts/index.js"; import { getTimezoneOffsetInHours } from "../utils/get-timezone-offset-in-hours.js"; @@ -61,7 +61,7 @@ const createEvent = async ( type RunCreateEventParams = { calendarId: string; auth: JWT; - model: BaseChatModel; + model: BaseLanguageModel; }; const runCreateEvent = async ( diff --git a/libs/langchain-community/src/tools/google_calendar/commands/run-view-events.ts b/libs/langchain-community/src/tools/google_calendar/commands/run-view-events.ts index 731f9cad162c..52767d80bff9 100644 --- a/libs/langchain-community/src/tools/google_calendar/commands/run-view-events.ts +++ b/libs/langchain-community/src/tools/google_calendar/commands/run-view-events.ts @@ -1,7 +1,7 @@ import { calendar_v3 } from "googleapis"; import type { JWT } from "googleapis-common"; import { PromptTemplate } from "@langchain/core/prompts"; -import { BaseChatModel } from "@langchain/core/language_models/chat_models"; +import { BaseLanguageModel } from "@langchain/core/language_models/base"; import { CallbackManagerForToolRun } from "@langchain/core/callbacks/manager"; import { StringOutputParser } from "@langchain/core/output_parsers"; @@ -11,7 +11,7 @@ import { getTimezoneOffsetInHours } from "../utils/get-timezone-offset-in-hours. type RunViewEventParams = { calendarId: string; auth: JWT; - model: BaseChatModel; + model: BaseLanguageModel; }; const runViewEvents = async (