From 5dfcda85750bb2084e4fed18e4714cd12d5e8b68 Mon Sep 17 00:00:00 2001 From: "jinli.yl" Date: Mon, 9 Feb 2026 16:13:51 +0800 Subject: [PATCH] feat(llm): make model_name optional in simple_request methods --- reme/core/llm/base_llm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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."""