refac: fix key and nodes

This commit is contained in:
Ibis Prevedello 2023-03-09 14:07:32 -03:00
commit 16e6aefa10
2 changed files with 2 additions and 1 deletions

View file

@ -21,6 +21,7 @@ def replace_zero_shot_prompt_with_prompt_template(nodes):
]
node["data"] = build_prompt_template(prompt=node["data"], tools=tools)
break
return nodes
def process_data_graph(data_graph: Dict[str, Any]):

View file

@ -283,7 +283,7 @@ def format_dict(d, name: Optional[str] = None):
"temperature",
"model_name",
]
or any(text in key for text in ["password", "token", "api", "key"])
or "api_key" in key
)
# Add password field