mirror of
https://github.com/hwchase17/langchain.git
synced 2025-06-27 00:48:45 +00:00
Accept LLM via constructor (#7794)
This commit is contained in:
parent
c4ece52dac
commit
ed97af423c
@ -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
|
||||
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user