diff --git a/wren-ai-service/src/pipelines/ask/followup_generation.py b/wren-ai-service/src/pipelines/ask/followup_generation.py
index 6e80b19d4..1d6b8aff4 100644
--- a/wren-ai-service/src/pipelines/ask/followup_generation.py
+++ b/wren-ai-service/src/pipelines/ask/followup_generation.py
@@ -231,7 +231,7 @@ async def run(
 
     load_env_vars()
 
-    llm_provider, _, engine = init_providers()
+    llm_provider, _, _, engine = init_providers()
     pipeline = FollowUpGeneration(llm_provider=llm_provider, engine=engine)
 
     pipeline.visualize(
diff --git a/wren-ai-service/src/pipelines/ask/generation.py b/wren-ai-service/src/pipelines/ask/generation.py
index b5664bbb3..018af68a6 100644
--- a/wren-ai-service/src/pipelines/ask/generation.py
+++ b/wren-ai-service/src/pipelines/ask/generation.py
@@ -189,7 +189,7 @@ async def run(
 
     load_env_vars()
 
-    llm_provider, _, engine = init_providers()
+    llm_provider, _, _, engine = init_providers()
     pipeline = Generation(
         llm_provider=llm_provider,
         engine=engine,
diff --git a/wren-ai-service/src/pipelines/ask/sql_correction.py b/wren-ai-service/src/pipelines/ask/sql_correction.py
index 2aba540e1..b29f7c93f 100644
--- a/wren-ai-service/src/pipelines/ask/sql_correction.py
+++ b/wren-ai-service/src/pipelines/ask/sql_correction.py
@@ -155,7 +155,7 @@ async def run(
 
     load_env_vars()
 
-    llm_provider, _, engine = init_providers()
+    llm_provider, _, _, engine = init_providers()
     pipeline = SQLCorrection(
         llm_provider=llm_provider,
         engine=engine,
diff --git a/wren-ai-service/src/pipelines/ask_details/generation.py b/wren-ai-service/src/pipelines/ask_details/generation.py
index b815ea54a..6b1846fc7 100644
--- a/wren-ai-service/src/pipelines/ask_details/generation.py
+++ b/wren-ai-service/src/pipelines/ask_details/generation.py
@@ -194,7 +194,7 @@ async def run(self, sql: str):
 
     load_env_vars()
 
-    llm_provider, _, engine = init_providers()
+    llm_provider, _, _, engine = init_providers()
     pipeline = Generation(
         llm_provider=llm_provider,
         engine=engine,