diff --git a/reme/core/llm/base_llm.py b/reme/core/llm/base_llm.py
index 2e63da5f..7e086aa5 100644
--- a/reme/core/llm/base_llm.py
+++ b/reme/core/llm/base_llm.py
@@ -461,7 +461,7 @@ class BaseLLM(ABC):
     async def simple_request(
         self,
         prompt: str,
-        model_name: str,
+        model_name: str | None = None,
         callback_fn: Callable[[Message], Any] | None = None,
         default_value: Any = None,
         **kwargs,
@@ -479,7 +479,7 @@ class BaseLLM(ABC):
     async def simple_request_for_json(
         self,
         prompt: str,
-        model_name: str,
+        model_name: str | None = None,
         **kwargs,
     ) -> dict:
         """Make a simple request using the LLM and extract JSON."""