diff --git a/run_agent.py b/run_agent.py
index fc7f72b731b..df49987fee1 100644
--- a/run_agent.py
+++ b/run_agent.py
@@ -947,6 +947,7 @@ class AIAgent:
                     client_kwargs["default_headers"] = headers
 
             self.api_key = client_kwargs.get("api_key", "")
+            self.base_url = client_kwargs.get("base_url", self.base_url)
             try:
                 self.client = self._create_openai_client(client_kwargs, reason="agent_init", shared=True)
                 if not self.quiet_mode: