mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-13 21:47:12 +00:00
Update extraction.py (#12207)
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:
@@ -47,6 +47,7 @@ def create_extraction_chain(
|
||||
schema: dict,
|
||||
llm: BaseLanguageModel,
|
||||
prompt: Optional[BasePromptTemplate] = None,
|
||||
tags: Optional[List[str]] = None,
|
||||
verbose: bool = False,
|
||||
) -> Chain:
|
||||
"""Creates a chain that extracts information from a passage.
|
||||
@@ -71,6 +72,7 @@ def create_extraction_chain(
|
||||
prompt=extraction_prompt,
|
||||
llm_kwargs=llm_kwargs,
|
||||
output_parser=output_parser,
|
||||
tags=tags,
|
||||
verbose=verbose,
|
||||
)
|
||||
return chain
|
||||
|
Reference in New Issue
Block a user