Refactor process_inputs function to handle optional inputs and artifacts
This commit is contained in:
parent
d07150ad17
commit
f7ae3ae852
1 changed files with 8 additions and 3 deletions
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue