From 6fe3360d17fd33ef1b4cd668c471fdb66ee6977c Mon Sep 17 00:00:00 2001 From: Mikolaj Mikuliszyn Date: Sun, 27 Apr 2025 18:13:21 +0100 Subject: [PATCH] mend --- libs/langchain/langchain/memory/persona.py | 3 +-- .../tests/unit_tests/memory/test_persona.py | 13 +++++++------ 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/libs/langchain/langchain/memory/persona.py b/libs/langchain/langchain/memory/persona.py index 4717bdddfc8..369126da276 100644 --- a/libs/langchain/langchain/memory/persona.py +++ b/libs/langchain/langchain/memory/persona.py @@ -1,9 +1,8 @@ import re from typing import Any, Callable, Optional -from pydantic import BaseModel, Field - from langchain_core.memory import BaseMemory +from pydantic import BaseModel, Field class EnrichedMessage(BaseModel): diff --git a/libs/langchain/tests/unit_tests/memory/test_persona.py b/libs/langchain/tests/unit_tests/memory/test_persona.py index c06703ffd11..d6879b258f6 100644 --- a/libs/langchain/tests/unit_tests/memory/test_persona.py +++ b/libs/langchain/tests/unit_tests/memory/test_persona.py @@ -1,12 +1,13 @@ -import unittest -import sys -from io import StringIO -from contextlib import contextmanager -from langchain.memory.persona import PersonaMemory, EnrichedMessage -from langchain.schema import AIMessage, HumanMessage, BaseMessage import json +import sys +import unittest +from contextlib import contextmanager +from io import StringIO from typing import List, Optional +from langchain.memory.persona import EnrichedMessage, PersonaMemory +from langchain.schema import AIMessage, BaseMessage, HumanMessage + class MockChatModel: """A simple mock chat model that can simulate both success and failure cases."""