mirror of
https://github.com/hwchase17/langchain.git
synced 2025-06-19 13:23:35 +00:00
mend
This commit is contained in:
parent
77196f40ee
commit
6fe3360d17
@ -1,9 +1,8 @@
|
|||||||
import re
|
import re
|
||||||
from typing import Any, Callable, Optional
|
from typing import Any, Callable, Optional
|
||||||
|
|
||||||
from pydantic import BaseModel, Field
|
|
||||||
|
|
||||||
from langchain_core.memory import BaseMemory
|
from langchain_core.memory import BaseMemory
|
||||||
|
from pydantic import BaseModel, Field
|
||||||
|
|
||||||
|
|
||||||
class EnrichedMessage(BaseModel):
|
class EnrichedMessage(BaseModel):
|
||||||
|
@ -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 json
|
||||||
|
import sys
|
||||||
|
import unittest
|
||||||
|
from contextlib import contextmanager
|
||||||
|
from io import StringIO
|
||||||
from typing import List, Optional
|
from typing import List, Optional
|
||||||
|
|
||||||
|
from langchain.memory.persona import EnrichedMessage, PersonaMemory
|
||||||
|
from langchain.schema import AIMessage, BaseMessage, HumanMessage
|
||||||
|
|
||||||
|
|
||||||
class MockChatModel:
|
class MockChatModel:
|
||||||
"""A simple mock chat model that can simulate both success and failure cases."""
|
"""A simple mock chat model that can simulate both success and failure cases."""
|
||||||
|
Loading…
Reference in New Issue
Block a user