github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-02-21 14:43:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

x

Browse Source
This commit is contained in:
Eugene Yurtsev
2024-04-23 16:34:50 -04:00
parent 9ac0b0026b
commit 017ae731d4
7 changed files with 10 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
libs/core/langchain_core/legacy/memory/__init__.py
View File

@@ -1,10 +1,10 @@
from langchain_core.legacy.memory.base import BaseMemory
from langchain_core.legacy.memory.chat_memory import BaseChatMemory
from langchain_core.legacy.memory.buffer import (
ConversationBufferMemory,
ConversationStringBufferMemory,
)
from langchain_core.legacy.memory.buffer_window import ConversationBufferWindowMemory
from langchain_core.legacy.memory.chat_memory import BaseChatMemory
from langchain_core.legacy.memory.combined import CombinedMemory
from langchain_core.legacy.memory.entity import (
ConversationEntityMemory,

2
libs/core/langchain_core/legacy/memory/base.py
View File

@@ -69,7 +69,7 @@ class BaseMemory(Serializable, ABC):
"""Save the context of this chain run to memory."""
async def asave_context(
self, inputs: Dict[str, Any], outputs: Dict[str, str]
self, inputs: Dict[str, Any], outputs: Dict[str, str]
) -> None:
"""Save the context of this chain run to memory."""
await run_in_executor(None, self.save_context, inputs, outputs)

3
libs/core/langchain_core/legacy/memory/buffer_window.py
View File

@@ -1,8 +1,7 @@
from typing import Any, Dict, List, Union
from langchain_core.messages import BaseMessage, get_buffer_string
from langchain_core.legacy.memory.chat_memory import BaseChatMemory
from langchain_core.messages import BaseMessage, get_buffer_string
class ConversationBufferWindowMemory(BaseChatMemory):

2
libs/core/langchain_core/legacy/memory/entity.py
View File

@@ -1,6 +1,6 @@
import logging
from abc import ABC, abstractmethod
from typing import Dict, Optional, List, Any
from typing import Any, Dict, List, Optional
from pydantic import Field

7
libs/core/langchain_core/legacy/memory/summary.py
View File

@@ -4,13 +4,12 @@ from typing import Any, Dict, List, Type
from langchain_core.chat_history import BaseChatMessageHistory
from langchain_core.language_models import BaseLanguageModel
from langchain_core.messages import BaseMessage, SystemMessage, get_buffer_string
from langchain_core.prompts import BasePromptTemplate
from langchain_core.pydantic_v1 import BaseModel, root_validator
from langchain_core.legacy.chains.llm import LLMChain
from langchain_core.legacy.memory.chat_memory import BaseChatMemory
from langchain_core.legacy.memory.prompt import SUMMARY_PROMPT
from langchain_core.messages import BaseMessage, SystemMessage, get_buffer_string
from langchain_core.prompts import BasePromptTemplate
from langchain_core.pydantic_v1 import BaseModel, root_validator
class SummarizerMixin(BaseModel):

5
libs/core/langchain_core/legacy/memory/summary_buffer.py
View File

@@ -1,10 +1,9 @@
from typing import Any, Dict, List
from langchain_core.messages import BaseMessage, get_buffer_string
from langchain_core.pydantic_v1 import root_validator
from langchain_core.legacy.memory.chat_memory import BaseChatMemory
from langchain_core.legacy.memory.summary import SummarizerMixin
from langchain_core.messages import BaseMessage, get_buffer_string
from langchain_core.pydantic_v1 import root_validator
class ConversationSummaryBufferMemory(BaseChatMemory, SummarizerMixin):

3
libs/core/langchain_core/legacy/memory/token_buffer.py
View File

@@ -1,9 +1,8 @@
from typing import Any, Dict, List
from langchain_core.language_models import BaseLanguageModel
from langchain_core.messages import BaseMessage, get_buffer_string
from langchain_core.legacy.memory.chat_memory import BaseChatMemory
from langchain_core.messages import BaseMessage, get_buffer_string
class ConversationTokenBufferMemory(BaseChatMemory):