From 2dddfe4aea31e1d024ae5c2d07d9783f24451762 Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Tue, 11 Jun 2024 10:15:02 -0300 Subject: [PATCH] refactor: Update import statements in CassandraMessageReader and CassandraMessageWriter --- .../components/memories/CassandraMessageReader.py | 2 +- .../components/memories/CassandraMessageWriter.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/src/backend/base/langflow/components/memories/CassandraMessageReader.py b/src/backend/base/langflow/components/memories/CassandraMessageReader.py index 3545d444a..3fd11d772 100644 --- a/src/backend/base/langflow/components/memories/CassandraMessageReader.py +++ b/src/backend/base/langflow/components/memories/CassandraMessageReader.py @@ -3,7 +3,7 @@ from typing import Optional, cast from langchain_community.chat_message_histories import CassandraChatMessageHistory from langflow.base.memory.memory import BaseMemoryComponent -from langflow.schema.schema import Record +from langflow.schema.record import Record class CassandraMessageReaderComponent(BaseMemoryComponent): diff --git a/src/backend/base/langflow/components/memories/CassandraMessageWriter.py b/src/backend/base/langflow/components/memories/CassandraMessageWriter.py index 716364b64..c8e3831a5 100644 --- a/src/backend/base/langflow/components/memories/CassandraMessageWriter.py +++ b/src/backend/base/langflow/components/memories/CassandraMessageWriter.py @@ -1,10 +1,10 @@ from typing import Optional -from langflow.base.memory.memory import BaseMemoryComponent -from langflow.schema.schema import Record - -from langchain_core.messages import BaseMessage from langchain_community.chat_message_histories import CassandraChatMessageHistory +from langchain_core.messages import BaseMessage + +from langflow.base.memory.memory import BaseMemoryComponent +from langflow.schema.record import Record class CassandraMessageWriterComponent(BaseMemoryComponent):