diff --git a/api/core/llm_generator/llm_generator.py b/api/core/llm_generator/llm_generator.py
index 51b3a332c1..c98e4572e2 100644
--- a/api/core/llm_generator/llm_generator.py
+++ b/api/core/llm_generator/llm_generator.py
@@ -477,7 +477,6 @@ class LLMGenerator:
                 prompt_messages=complete_messages,
                 output_model=CodeNodeStructuredOutput,
                 model_parameters=model_parameters,
-                tenant_id=tenant_id,
             )
 
             return {
@@ -558,10 +557,14 @@ class LLMGenerator:
 
         completion_params = model_config.get("completion_params", {}) if model_config else {}
         try:
-            response = invoke_llm_with_pydantic_model(provider=model_instance.provider, model_schema=model_schema,
-                                                      model_instance=model_instance, prompt_messages=prompt_messages,
-                                                      output_model=SuggestedQuestionsOutput,
-                                                      model_parameters=completion_params, tenant_id=tenant_id)
+            response = invoke_llm_with_pydantic_model(
+                provider=model_instance.provider,
+                model_schema=model_schema,
+                model_instance=model_instance,
+                prompt_messages=prompt_messages,
+                output_model=SuggestedQuestionsOutput,
+                model_parameters=completion_params
+            )
 
             return {"questions": response.questions, "error": ""}