Update base_graph.py

This commit is contained in:
Marco Vinciguerra 2024-08-06 16:29:59 +02:00
parent 48f07e7974
commit 6e4d04450f

View File

@ -157,9 +157,9 @@ class BaseGraph:
embedder_model = embedder_model.model
if hasattr(current_node, "node_config"):
if type(current_node.node_config) is dict:
if isinstance(current_node.node_config,dict):
if current_node.node_config.get("schema", None) and schema is None:
if type(current_node.node_config["schema"]) is not dict:
if not isinstance(current_node.node_config["schema"], dict):
# convert to dict
try:
schema = current_node.node_config["schema"].schema()