vrr/agent-zero
2
0
Fork 0
mirror of https://github.com/agent0ai/agent-zero.git synced 2026-05-19 07:59:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Merge pull request #1258 from Deimos-AI/pr/extensible-models-get-api-key

Browse source
This commit is contained in:
Jan Tomášek 2026-03-26 12:23:49 +01:00 • committed by GitHub
commit 2e0ec63e75
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models.py
View file

@ -26,6 +26,7 @@ from helpers.providers import ModelType as ProviderModelType, get_provider_confi
from helpers.rate_limiter import RateLimiter
from helpers.tokens import approximate_tokens
from helpers import dirty_json
from helpers.extension import extensible # extensible: allows plugins to intercept get_api_key()
from langchain_core.language_models.chat_models import SimpleChatModel
from langchain_core.outputs.chat_generation import ChatGenerationChunk
@ -198,6 +199,7 @@ rate_limiters: dict[str, RateLimiter] = {}
api_keys_round_robin: dict[str, int] = {}
@extensible
def get_api_key(service: str) -> str:
# get api key for the service
key = (