From 8e7086681eff7260950f31e292ac9ca68fdd0822 Mon Sep 17 00:00:00 2001 From: Shroominic Date: Sat, 27 Jan 2024 11:31:27 +0000 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20Update=20type=20annotations?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/funcchain/utils/memory.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/funcchain/utils/memory.py b/src/funcchain/utils/memory.py index cfe5d8e..a0c97be 100644 --- a/src/funcchain/utils/memory.py +++ b/src/funcchain/utils/memory.py @@ -1,7 +1,5 @@ """langchain_community.chat_message_histories.in_memory.ChatMessageHistory""" -from typing import List - from langchain_core.chat_history import BaseChatMessageHistory from langchain_core.messages import BaseMessage from langchain_core.pydantic_v1 import BaseModel, Field @@ -13,7 +11,7 @@ class ChatMessageHistory(BaseChatMessageHistory, BaseModel): Stores messages in an in memory list. """ - messages: List[BaseMessage] = Field(default_factory=list) + messages: list[BaseMessage] = Field(default_factory=list) def add_message(self, message: BaseMessage) -> None: """Add a self-created message to the store"""