jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

refactor: Update Vertex class to use type annotations for display_name, description, frozen, is_input, and is_output

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-19 19:32:16 -03:00
commit b97ec2436c
1 changed files with 5 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/backend/base/langflow/graph/vertex/base.py
View file

@ -2,7 +2,6 @@ import ast
import asyncio import asyncio
import inspect import inspect
import os import os
import traceback
import types import types
from enum import Enum from enum import Enum
from typing import TYPE_CHECKING, Any, AsyncIterator, Callable, Dict, Iterator, List, Mapping, Optional from typing import TYPE_CHECKING, Any, AsyncIterator, Callable, Dict, Iterator, List, Mapping, Optional
@ -192,13 +191,13 @@ class Vertex:
self.outputs = self.data["node"].get("outputs", []) self.outputs = self.data["node"].get("outputs", [])
self.output = self.data["node"]["base_classes"] self.output = self.data["node"]["base_classes"]
self.display_name = self.data["node"].get("display_name", self.id.split("-")[0]) self.display_name: str = self.data["node"].get("display_name", self.id.split("-")[0])
self.description = self.data["node"].get("description", "") self.description: str = self.data["node"].get("description", "")
self.frozen = self.data["node"].get("frozen", False) self.frozen: bool = self.data["node"].get("frozen", False)
self.is_input = self.data["node"].get("is_input") or self.is_input self.is_input: bool = self.data["node"].get("is_input") or self.is_input
self.is_output = self.data["node"].get("is_output") or self.is_output self.is_output: bool = self.data["node"].get("is_output") or self.is_output
template_dicts = {key: value for key, value in self.data["node"]["template"].items() if isinstance(value, dict)} template_dicts = {key: value for key, value in self.data["node"]["template"].items() if isinstance(value, dict)}
self.has_session_id = "session_id" in template_dicts self.has_session_id = "session_id" in template_dicts