openapi chain nit (#7012)

This commit is contained in:
Bagatur 2023-07-03 02:22:53 -06:00 committed by GitHub
parent a081e419a0
commit 1cdb33a090
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -233,7 +233,7 @@ def get_openapi_chain(
llm: Optional[BaseLanguageModel] = None, llm: Optional[BaseLanguageModel] = None,
prompt: Optional[BasePromptTemplate] = None, prompt: Optional[BasePromptTemplate] = None,
request_chain: Optional[Chain] = None, request_chain: Optional[Chain] = None,
llm_kwargs: Optional[Dict] = None, llm_chain_kwargs: Optional[Dict] = None,
verbose: bool = False, verbose: bool = False,
headers: Optional[Dict] = None, headers: Optional[Dict] = None,
params: Optional[Dict] = None, params: Optional[Dict] = None,
@ -275,7 +275,7 @@ def get_openapi_chain(
output_parser=JsonOutputFunctionsParser(args_only=False), output_parser=JsonOutputFunctionsParser(args_only=False),
output_key="function", output_key="function",
verbose=verbose, verbose=verbose,
**(llm_kwargs or {}), **(llm_chain_kwargs or {}),
) )
request_chain = request_chain or SimpleRequestChain( request_chain = request_chain or SimpleRequestChain(
request_method=lambda name, args: call_api_fn( request_method=lambda name, args: call_api_fn(