Expose kwargs in LLMChainExtractor.from_llm (#3748)

Re: #3747
langchain-ai · Apr 29, 2023 · 57e0285 · 57e0285
1 parent 512c24f
commit 57e0285
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/langchain/retrievers/document_compressors/chain_extract.py b/langchain/retrievers/document_compressors/chain_extract.py
@@ -69,9 +69,10 @@ def from_llm(
         llm: BaseLanguageModel,
         prompt: Optional[PromptTemplate] = None,
         get_input: Optional[Callable[[str, Document], str]] = None,
+        llm_chain_kwargs: Optional[dict] = None,
     ) -> "LLMChainExtractor":
         """Initialize from LLM."""
         _prompt = prompt if prompt is not None else _get_default_chain_prompt()
         _get_input = get_input if get_input is not None else default_get_input
-        llm_chain = LLMChain(llm=llm, prompt=_prompt)
+        llm_chain = LLMChain(llm=llm, prompt=_prompt, **(llm_chain_kwargs or {}))
         return cls(llm_chain=llm_chain, get_input=_get_input)