github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-28 01:19:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Accept LLM via constructor (#7794)

Browse Source
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
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
langchain/smith/evaluation/runner_utils.py
View File

@ -62,7 +62,7 @@ def _wrap_in_chain_factory(
if llm_or_chain_factory.memory is not None: if llm_or_chain_factory.memory is not None:
memory_class = chain.memory.__class__.__name__ memory_class = chain.memory.__class__.__name__
raise ValueError( raise ValueError(
"Cannot directly evaluate a chain with statefulmemory." "Cannot directly evaluate a chain with stateful memory."
" To evaluate this chain, pass in a chain constructor" " To evaluate this chain, pass in a chain constructor"
" that initializes fresh memory each time it is called." " that initializes fresh memory each time it is called."
" This will safegaurd against information" " This will safegaurd against information"
@ -88,6 +88,13 @@ def _wrap_in_chain_factory(
) )
return lambda: chain 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 return llm_or_chain_factory