diff --git a/src/backend/langflow/processing/process.py b/src/backend/langflow/processing/process.py index 647e8cd4b..9432b7f96 100644 --- a/src/backend/langflow/processing/process.py +++ b/src/backend/langflow/processing/process.py @@ -7,11 +7,12 @@ from langchain.schema import AgentAction, Document from langchain.vectorstores.base import VectorStore from langchain_core.messages import AIMessage from langchain_core.runnables.base import Runnable +from loguru import logger +from pydantic import BaseModel + from langflow.interface.custom.custom_component import CustomComponent from langflow.interface.run import build_sorted_vertices, get_memory_key, update_memory_keys from langflow.services.deps import get_session_service -from loguru import logger -from pydantic import BaseModel def fix_memory_inputs(langchain_object): @@ -106,9 +107,13 @@ def get_build_result(data_graph, session_id): return build_sorted_vertices(data_graph) -def process_inputs(inputs: Union[dict, List[dict]], artifacts: Dict[str, Any]) -> dict: +def process_inputs( + inputs: Optional[Union[dict, List[dict]]] = None, artifacts: Optional[Dict[str, Any]] = None +) -> Union[dict, List[dict]]: if inputs is None: inputs = {} + if artifacts is None: + artifacts = {} if isinstance(inputs, dict): inputs = update_inputs_dict(inputs, artifacts)