diff --git a/models.py b/models.py
index 7b8905205c..e3891d81fa 100644
--- a/models.py
+++ b/models.py
@@ -26,6 +26,7 @@
 from helpers.rate_limiter import RateLimiter
 from helpers.tokens import approximate_tokens
 from helpers import dirty_json, browser_use_monkeypatch
+from helpers.extension import extensible
 
 from langchain_core.language_models.chat_models import SimpleChatModel
 from langchain_core.outputs.chat_generation import ChatGenerationChunk
@@ -198,6 +199,7 @@ def output(self) -> ChatChunk:
 api_keys_round_robin: dict[str, int] = {}
 
 
+@extensible
 def get_api_key(service: str) -> str:
     # get api key for the service
     key = (