From fef95e10b83a9bc94d8956e68d4ebfa208188091 Mon Sep 17 00:00:00 2001 From: Matt Rubens Date: Sun, 2 Feb 2025 01:34:45 -0500 Subject: [PATCH] Fix type --- src/core/sliding-window/index.ts | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/core/sliding-window/index.ts b/src/core/sliding-window/index.ts index 6fbd5b8a9c..3ecb785f5e 100644 --- a/src/core/sliding-window/index.ts +++ b/src/core/sliding-window/index.ts @@ -1,6 +1,5 @@ import { Anthropic } from "@anthropic-ai/sdk" import { ModelInfo } from "../../shared/api" -import { MessageParam } from "@anthropic-ai/sdk/resources/messages.mjs" export function truncateConversation( messages: Anthropic.Messages.MessageParam[], @@ -16,10 +15,10 @@ export function truncateConversation( } export function truncateConversationIfNeeded( - messages: MessageParam[], + messages: Anthropic.Messages.MessageParam[], totalTokens: number, modelInfo: ModelInfo, -): MessageParam[] { +): Anthropic.Messages.MessageParam[] { if (modelInfo.supportsPromptCache) { return totalTokens < getMaxTokensForPromptCachingModels(modelInfo) ? messages