This commit is contained in:
Mikolaj Mikuliszyn 2025-04-27 18:13:21 +01:00
parent 77196f40ee
commit 6fe3360d17
2 changed files with 8 additions and 8 deletions

View File

@ -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):

View File

@ -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."""