github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-03 21:54:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

The predict_and_parse is deprecated, instead pass an output parser directly to LLMChain. (#20130)

Browse Source 
The `predict_and_parse` method is deprecated, instead pass an output
parser directly to LLMChain.

- [x] **PR title**: "langchain: update chain_extract.py"


![image](https://github.com/langchain-ai/langchain/assets/40889019/e950d79f-5a0f-4086-86e9-89f627990fe5)

---------

Co-authored-by: Chester Curme <chester.curme@gmail.com>
This commit is contained in:
Silas Xu 2024-05-08 21:32:17 +08:00 committed by GitHub
parent 3c31bd0ed0
commit aafaf3e193
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
libs/langchain/langchain/retrievers/document_compressors/chain_extract.py
View File

@ -64,7 +64,10 @@ class LLMChainExtractor(BaseDocumentCompressor):
compressed_docs = [] compressed_docs = []
for doc in documents: for doc in documents:
_input = self.get_input(query, doc) _input = self.get_input(query, doc)
output = self.llm_chain.predict_and_parse(**_input, callbacks=callbacks) output_dict = self.llm_chain.invoke(_input, config={"callbacks": callbacks})
output = output_dict[self.llm_chain.output_key]
if self.llm_chain.prompt.output_parser is not None:
output = self.llm_chain.prompt.output_parser.parse(output)
if len(output) == 0: if len(output) == 0:
continue continue
compressed_docs.append( compressed_docs.append(