From 8116b653234930e298c17f13a1d6b783edcdd56e Mon Sep 17 00:00:00 2001 From: ehallmark Date: Sat, 30 Aug 2025 19:45:12 -0700 Subject: [PATCH 1/2] agent vault prefix --- src/agentstr/agents/agentstr.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/agentstr/agents/agentstr.py b/src/agentstr/agents/agentstr.py index 76a7678..151080f 100644 --- a/src/agentstr/agents/agentstr.py +++ b/src/agentstr/agents/agentstr.py @@ -111,15 +111,18 @@ def checkpointer(self): if key_manager: try: from agent_vault.langgraph import async_insecure_postgres_saver, async_secure_postgres_saver - from agent_vault.utils.key_manager import AWSSecretsManagerKeyManager, AzureKeyVaultKeyManager + from agent_vault.utils.key_manager import AWSParameterStoreKeyManager, AWSSecretsManagerKeyManager, AzureKeyVaultKeyManager except ImportError: raise ValueError("agent_vault is not installed") if key_manager == "none": checkpointer = async_insecure_postgres_saver(self.database.conn_str) elif key_manager == "aws": - checkpointer = async_secure_postgres_saver(self.database.conn_str, AWSSecretsManagerKeyManager(prefix=key_manager_prefix)) + checkpointer = async_secure_postgres_saver(self.database.conn_str, AWSParameterStoreKeyManager(prefix=key_manager_prefix)) elif key_manager == "azure": - checkpointer = async_secure_postgres_saver(self.database.conn_str, AzureKeyVaultKeyManager(prefix=key_manager_prefix)) + key_vault_url = os.getenv("AZURE_KEY_VAULT_URL") + if not key_vault_url: + raise ValueError("Agent Vault Azure Key Manager requested by AZURE_KEY_VAULT_URL environment variable is not set") + checkpointer = async_secure_postgres_saver(self.database.conn_str, AzureKeyVaultKeyManager(vault_url=key_vault_url, prefix=key_manager_prefix)) else: raise ValueError(f"Unsupported key manager: {key_manager}") else: From 41f7993b43751dcebf1e3688986db49a89f05681 Mon Sep 17 00:00:00 2001 From: ehallmark Date: Sat, 30 Aug 2025 19:46:47 -0700 Subject: [PATCH 2/2] agent vault prefix --- src/agentstr/agents/agentstr.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/agentstr/agents/agentstr.py b/src/agentstr/agents/agentstr.py index 151080f..6b35f6a 100644 --- a/src/agentstr/agents/agentstr.py +++ b/src/agentstr/agents/agentstr.py @@ -107,7 +107,7 @@ def checkpointer(self): checkpointer = None if self.database.conn_str.startswith("postgres"): key_manager = os.getenv("AGENT_VAULT_KEY_MANAGER") - key_manager_prefix = os.getenv("AGENT_VAULT_KEY_MANAGER_PREFIX") + key_manager_prefix = os.getenv("AGENT_VAULT_KEY_MANAGER_PREFIX", f"AGENTSTR-{self.name}-".upper().replace(' ', '-')) if key_manager: try: from agent_vault.langgraph import async_insecure_postgres_saver, async_secure_postgres_saver