refactor: Import Data class in utils.py for better code organization
This commit is contained in:
parent
d060d5308a
commit
1dbeec38c0
2 changed files with 2 additions and 2 deletions
|
|
@ -5,6 +5,7 @@ from langchain_core.documents import Document
|
|||
from pydantic import BaseModel
|
||||
|
||||
from langflow.interface.utils import extract_input_variables_from_prompt
|
||||
from langflow.schema.data import Data
|
||||
from langflow.schema.message import Message
|
||||
|
||||
|
||||
|
|
@ -67,7 +68,7 @@ def serialize_field(value):
|
|||
def get_artifact_type(value, build_result) -> str:
|
||||
result = ArtifactType.UNKNOWN
|
||||
match value:
|
||||
case Record():
|
||||
case Data():
|
||||
result = ArtifactType.RECORD
|
||||
|
||||
case str():
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue