From 69f5f8280468994d10f0e84a0bb3770a58de579e Mon Sep 17 00:00:00 2001
From: Lukas Wolf <71704466+lu-wo@users.noreply.github.com>
Date: Tue, 24 Oct 2023 17:25:14 +0200
Subject: [PATCH] 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
---
 libs/langchain/langchain/chains/openai_functions/extraction.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/libs/langchain/langchain/chains/openai_functions/extraction.py b/libs/langchain/langchain/chains/openai_functions/extraction.py
index 7f52dd7baf6..c13d3ecca30 100644
--- a/libs/langchain/langchain/chains/openai_functions/extraction.py
+++ b/libs/langchain/langchain/chains/openai_functions/extraction.py
@@ -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