We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 987ada0 commit a3a1c81Copy full SHA for a3a1c81
sdk/agenta/sdk/middlewares/running/vault.py
@@ -38,6 +38,10 @@
38
if "mistral" not in _PROVIDER_KINDS:
39
_PROVIDER_KINDS.append("mistral")
40
41
+_PROVIDER_ENV_VAR_OVERRIDES = {
42
+ "together_ai": "TOGETHERAI_API_KEY",
43
+}
44
+
45
_AUTH_ENABLED = (
46
getenv("AGENTA_SERVICE_MIDDLEWARE_AUTH_ENABLED", "true").lower() in TRUTHY
47
)
@@ -255,7 +259,9 @@ async def get_secrets(
255
259
try:
256
260
for provider_kind in _PROVIDER_KINDS:
257
261
provider = provider_kind
258
- key_name = f"{provider.upper()}_API_KEY"
262
+ key_name = _PROVIDER_ENV_VAR_OVERRIDES.get(
263
+ provider, f"{provider.upper()}_API_KEY"
264
+ )
265
key = getenv(key_name)
266
267
if not key:
0 commit comments