Accept LLM via constructor (#7794)

This commit is contained in:
William FH 2023-07-16 08:46:36 -07:00 committed by GitHub
parent c4ece52dac
commit ed97af423c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -88,6 +88,13 @@ def _wrap_in_chain_factory(
)
return lambda: chain
elif isinstance(llm_or_chain_factory, BaseLanguageModel):
return llm_or_chain_factory
elif callable(llm_or_chain_factory):
_model = llm_or_chain_factory()
if isinstance(_model, BaseLanguageModel):
return _model
return llm_or_chain_factory
return llm_or_chain_factory