Refactor memory components to use updated Record schema

src/backend/base/langflow/components/memories/AstraDBMessageReader.py

@@ -1,8 +1,7 @@
 from typing import Optional, cast
 
-
 from langflow.base.memory.memory import BaseMemoryComponent
-from langflow.schema.schema import Record
+from langflow.schema.record import Record
 
 
 class AstraDBMessageReaderComponent(BaseMemoryComponent):

src/backend/base/langflow/components/memories/AstraDBMessageWriter.py

@@ -1,12 +1,9 @@
 from typing import Optional
 
-from langchain_astradb import AstraDBChatMessageHistory
 from langchain_core.messages import BaseMessage
 
 from langflow.base.memory.memory import BaseMemoryComponent
-from langflow.schema.schema import Record
-
-from langchain_core.messages import BaseMessage
+from langflow.schema.record import Record
 
 
 class AstraDBMessageWriterComponent(BaseMemoryComponent):

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

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

src/backend/base/langflow/components/prompts/Prompt.py

@@ -19,6 +19,6 @@ class PromptComponent(CustomComponent):
         template: Prompt,
         **kwargs,
     ) -> Prompt:
-        prompt = await Prompt.from_template_and_variables(template, kwargs)
+        prompt = await Prompt.from_template_and_variables(template, kwargs)  # type: ignore
         self.status = prompt.format_text()
         return prompt

src/backend/base/langflow/components/tools/PythonCodeStructuredTool.py

@@ -91,4 +91,4 @@ class PythonCodeStructuredTool(CustomComponent):
         tool = StructuredTool.from_function(
             func=func, args_schema=_class, name=name, description=description, return_direct=return_direct
         )
-        return tool
+        return tool  # type: ignore