From 4b414b18de419deb8ec0348be8ec18b84223100f Mon Sep 17 00:00:00 2001
From: "jinli.yl" <jinli.yl@alibaba-inc.com>
Date: Thu, 12 Feb 2026 16:20:54 +0800
Subject: [PATCH] feat(application): add parameters to update_api_envs method

---
 reme/core/application.py | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/reme/core/application.py b/reme/core/application.py
index c64a3bc0..389b96e9 100644
--- a/reme/core/application.py
+++ b/reme/core/application.py
@@ -60,9 +60,20 @@ class Application:
         self.prompt_handler = PromptHandler(language=self.service_context.language)
         self._started: bool = False
 
-    def update_api_envs(self):
+    def update_api_envs(
+        self,
+        llm_api_key: str | None = None,
+        llm_base_url: str | None = None,
+        embedding_api_key: str | None = None,
+        embedding_base_url: str | None = None,
+    ):
         """Update the API environment variables."""
-        self.service_context.update_api_envs()
+        self.service_context.update_api_envs(
+            llm_api_key=llm_api_key,
+            llm_base_url=llm_base_url,
+            embedding_api_key=embedding_api_key,
+            embedding_base_url=embedding_base_url,
+        )
 
     @classmethod
     async def create(