Skip to content

Commit 5dfcda8

Browse files
committed
feat(llm): make model_name optional in simple_request methods
1 parent e472562 commit 5dfcda8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

reme/core/llm/base_llm.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -461,7 +461,7 @@ def chat_sync(
461461
async def simple_request(
462462
self,
463463
prompt: str,
464-
model_name: str,
464+
model_name: str | None = None,
465465
callback_fn: Callable[[Message], Any] | None = None,
466466
default_value: Any = None,
467467
**kwargs,
@@ -479,7 +479,7 @@ async def simple_request(
479479
async def simple_request_for_json(
480480
self,
481481
prompt: str,
482-
model_name: str,
482+
model_name: str | None = None,
483483
**kwargs,
484484
) -> dict:
485485
"""Make a simple request using the LLM and extract JSON."""

0 commit comments

Comments
 (0)