From d8d20d7aa5a3fee4152873f39fadf1da6bfecaf2 Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Mon, 17 Jun 2024 12:17:59 -0300 Subject: [PATCH] fix: Add type checking import --- src/backend/base/langflow/schema/message.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/src/backend/base/langflow/schema/message.py b/src/backend/base/langflow/schema/message.py index 865d684bf..30a1d6a68 100644 --- a/src/backend/base/langflow/schema/message.py +++ b/src/backend/base/langflow/schema/message.py @@ -1,13 +1,14 @@ from datetime import datetime, timezone -from typing import Annotated, Any, AsyncIterator, Iterator, Optional +from typing import TYPE_CHECKING, Annotated, Any, AsyncIterator, Iterator, Optional from langchain_core.messages import AIMessage, BaseMessage, HumanMessage from langchain_core.prompt_values import ImagePromptValue from langchain_core.prompts.image import ImagePromptTemplate +from langflow.schema.image import Image, get_file_paths, is_image_file from pydantic import BaseModel, BeforeValidator, ConfigDict, Field, field_serializer -from langflow.schema.image import Image, get_file_paths, is_image_file -from langflow.schema.record import Record +if TYPE_CHECKING: + from langflow.schema.record import Record def _timestamp_to_str(timestamp: datetime) -> str: @@ -68,7 +69,7 @@ class Message(BaseModel): return AIMessage(content=self.text) @classmethod - def from_record(cls, record: Record) -> "Message": + def from_record(cls, record: "Record") -> "Message": """ Converts a BaseMessage to a Record.