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

🐛 fix(utils.py): add support for initializing class objects with a template parameter

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-13 20:01:33 -03:00
commit 94003cf39a
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/backend/langflow/interface/initialize/utils.py
View file

@ -5,6 +5,7 @@ from typing import Any, Dict, List
import orjson import orjson
from langchain.agents import ZeroShotAgent from langchain.agents import ZeroShotAgent
from langchain.schema import BaseOutputParser, Document from langchain.schema import BaseOutputParser, Document
from langflow.services.database.models.base import orjson_dumps from langflow.services.database.models.base import orjson_dumps
@ -16,6 +17,8 @@ def handle_node_type(node_type, class_object, params: Dict):
prompt = instantiate_from_template(class_object, params) prompt = instantiate_from_template(class_object, params)
elif node_type == "ChatPromptTemplate": elif node_type == "ChatPromptTemplate":
prompt = class_object.from_messages(**params) prompt = class_object.from_messages(**params)
elif hasattr(class_object, "from_template") and params.get("template"):
prompt = class_object.from_template(template=params.pop("template"))
else: else:
prompt = class_object(**params) prompt = class_object(**params)
return params, prompt return params, prompt