From 6c23c711fb234083e427ad2c2def8501897c9a6d Mon Sep 17 00:00:00 2001 From: Mason Daugherty Date: Mon, 7 Jul 2025 13:11:06 -0400 Subject: [PATCH] fix: lint/format (#31894) --- .../langchain/memory/vectorstore_token_buffer_memory.py | 3 ++- libs/langchain/langchain/output_parsers/fix.py | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/libs/langchain/langchain/memory/vectorstore_token_buffer_memory.py b/libs/langchain/langchain/memory/vectorstore_token_buffer_memory.py index b839bb1d514..ddefc55f7b7 100644 --- a/libs/langchain/langchain/memory/vectorstore_token_buffer_memory.py +++ b/libs/langchain/langchain/memory/vectorstore_token_buffer_memory.py @@ -167,6 +167,7 @@ class ConversationVectorStoreTokenBufferMemory(ConversationTokenBufferMemory): self._pop_and_store_interaction(buffer) def _pop_and_store_interaction(self, buffer: list[BaseMessage]) -> None: + input_ = buffer.pop(0) output = buffer.pop(0) timestamp = self._timestamps.pop(0).strftime(TIMESTAMP_FORMAT) # Split AI output into smaller chunks to avoid creating documents @@ -174,7 +175,7 @@ class ConversationVectorStoreTokenBufferMemory(ConversationTokenBufferMemory): ai_chunks = self._split_long_ai_text(str(output.content)) for index, chunk in enumerate(ai_chunks): self.memory_retriever.save_context( - {"Human": f"<{timestamp}/00> {input.content!s}"}, + {"Human": f"<{timestamp}/00> {input_.content!s}"}, {"AI": f"<{timestamp}/{index:02}> {chunk}"}, ) diff --git a/libs/langchain/langchain/output_parsers/fix.py b/libs/langchain/langchain/output_parsers/fix.py index e8f26cb6220..34d214be51a 100644 --- a/libs/langchain/langchain/output_parsers/fix.py +++ b/libs/langchain/langchain/output_parsers/fix.py @@ -82,7 +82,7 @@ class OutputFixingParser(BaseOutputParser[T]): try: completion = self.retry_chain.invoke( { - "instructions": self.parser.get_format_instructions(), # noqa: E501 + "instructions": self.parser.get_format_instructions(), "completion": completion, "error": repr(e), } @@ -119,7 +119,7 @@ class OutputFixingParser(BaseOutputParser[T]): try: completion = await self.retry_chain.ainvoke( { - "instructions": self.parser.get_format_instructions(), # noqa: E501 + "instructions": self.parser.get_format_instructions(), "completion": completion, "error": repr(e), }