github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-12 14:23:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update extraction.py (#12207)

Browse Source 
Description: Pass tags as argument to create_extraction_chain
Issue: create_extraction_chain does not pass tags to chain yet 

@baskaryan
This commit is contained in:
Lukas Wolf 2023-10-24 17:25:14 +02:00 committed by GitHub
parent 34ffb94770
commit 69f5f82804
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libs/langchain/langchain/chains/openai_functions/extraction.py
View File

@ -47,6 +47,7 @@ def create_extraction_chain(
schema: dict, schema: dict,
llm: BaseLanguageModel, llm: BaseLanguageModel,
prompt: Optional[BasePromptTemplate] = None, prompt: Optional[BasePromptTemplate] = None,
tags: Optional[List[str]] = None,
verbose: bool = False, verbose: bool = False,
) -> Chain: ) -> Chain:
"""Creates a chain that extracts information from a passage. """Creates a chain that extracts information from a passage.
@ -71,6 +72,7 @@ def create_extraction_chain(
prompt=extraction_prompt, prompt=extraction_prompt,
llm_kwargs=llm_kwargs, llm_kwargs=llm_kwargs,
output_parser=output_parser, output_parser=output_parser,
tags=tags,
verbose=verbose, verbose=verbose,
) )
return chain return chain