🔧 chore(utils.py): add support for handle_keys in build_input_keys_response function

The build_input_keys_response function now includes support for handle_keys in the response. This allows the function to populate the handle_keys field in the response with the values from the artifacts dictionary, if available.
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-06-29 19:30:51 -03:00
commit d6bd8a1a1b

View file

@ -30,6 +30,7 @@ def build_input_keys_response(langchain_object, artifacts):
input_keys_response = {
"input_keys": {key: "" for key in langchain_object.input_keys},
"memory_keys": [],
"handle_keys": artifacts.get("handle_keys", []),
}
# Set the input keys values from artifacts