From 9205919ad23c8229f5c3db033b55c6b33e5bb6b5 Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Sun, 23 Jul 2023 18:02:45 -0700 Subject: [PATCH] actually use input key (#8136) --- libs/langchain/langchain/chains/openai_functions/openapi.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/langchain/langchain/chains/openai_functions/openapi.py b/libs/langchain/langchain/chains/openai_functions/openapi.py index 6a349f3a339..5b991eff9d5 100644 --- a/libs/langchain/langchain/chains/openai_functions/openapi.py +++ b/libs/langchain/langchain/chains/openai_functions/openapi.py @@ -212,8 +212,8 @@ class SimpleRequestChain(Chain): ) -> Dict[str, Any]: """Run the logic of this chain and return the output.""" _run_manager = run_manager or CallbackManagerForChainRun.get_noop_manager() - name = inputs["function"].pop("name") - args = inputs["function"].pop("arguments") + name = inputs[self.input_key].pop("name") + args = inputs[self.input_key].pop("arguments") _pretty_name = get_colored_text(name, "green") _pretty_args = get_colored_text(json.dumps(args, indent=2), "green") _text = f"Calling endpoint {_pretty_name} with arguments:\n" + _pretty_args