diff --git a/backend/modules/llm/litellm_caller.py b/backend/modules/llm/litellm_caller.py
index b81ef58..931f05f 100644
--- a/backend/modules/llm/litellm_caller.py
+++ b/backend/modules/llm/litellm_caller.py
@@ -99,6 +99,8 @@ def _get_model_kwargs(self, model_name: str, temperature: Optional[float] = None
                 os.environ["GOOGLE_API_KEY"] = api_key
             elif "cerebras" in model_config.model_url:
                 os.environ["CEREBRAS_API_KEY"] = api_key
+            else:
+                os.environ["OPENAI_API_KEY"] = api_key
         
         # Set custom API base for non-standard endpoints
         if hasattr(model_config, 'model_url') and model_config.model_url: