github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-02-21 14:43:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

lint

Browse Source
This commit is contained in:
Sydney Runkle
2025-11-18 12:37:01 -05:00
parent 5d6c7e2433
commit 99ece77cd4
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
libs/langchain_v1/langchain/agents/middleware/context_editing.py
View File

@@ -17,7 +17,6 @@ from typing import Any, Literal
from langchain_core.messages import (
AIMessage,
AnyMessage,
BaseMessage,
SystemMessage,
ToolMessage,
)
@@ -442,15 +441,13 @@ class ContextEditingMiddleware(AgentMiddleware):
return await handler(request)
if self.token_count_method == "approximate": # noqa: S105
def count_tokens(messages: Sequence[BaseMessage]) -> int:
return count_tokens_approximately(messages)
count_tokens: TokenCounter = count_tokens_approximately
else:
system_msg = (
[SystemMessage(content=request.system_prompt)] if request.system_prompt else []
)
def count_tokens(messages: Sequence[BaseMessage]) -> int:
def count_tokens(messages: Iterable[Any]) -> int:
return request.model.get_num_tokens_from_messages(
system_msg + list(messages), request.tools
)