Skip to content

Commit a3a1c81

Browse files
committed
fix(sdk): correct Together AI env var name lookup in vault middleware
1 parent 987ada0 commit a3a1c81

File tree

1 file changed

+7
-1
lines changed
  • sdk/agenta/sdk/middlewares/running

1 file changed

+7
-1
lines changed

sdk/agenta/sdk/middlewares/running/vault.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,10 @@
3838
if "mistral" not in _PROVIDER_KINDS:
3939
_PROVIDER_KINDS.append("mistral")
4040

41+
_PROVIDER_ENV_VAR_OVERRIDES = {
42+
"together_ai": "TOGETHERAI_API_KEY",
43+
}
44+
4145
_AUTH_ENABLED = (
4246
getenv("AGENTA_SERVICE_MIDDLEWARE_AUTH_ENABLED", "true").lower() in TRUTHY
4347
)
@@ -255,7 +259,9 @@ async def get_secrets(
255259
try:
256260
for provider_kind in _PROVIDER_KINDS:
257261
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+
)
259265
key = getenv(key_name)
260266

261267
if not key:

0 commit comments

Comments
 (0)