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), }