added the state_storage_v1_responses flag, and use that to store state appropriately

This commit is contained in:
Salman Paracha 2025-12-16 11:18:19 -08:00
parent 142aaabc10
commit 1212b526b8
8 changed files with 134 additions and 10 deletions

View file

@ -148,6 +148,20 @@ def get_llm_provider_access_keys(arch_config_file):
if access_key is not None:
access_key_list.append(access_key)
# Extract environment variables from state_storage.connection_string
state_storage = arch_config_yaml.get("state_storage_v1_responses")
if state_storage:
connection_string = state_storage.get("connection_string")
if connection_string and isinstance(connection_string, str):
# Extract all $VAR and ${VAR} patterns from connection string
import re
# Match both $VAR and ${VAR} patterns
pattern = r"\$\{?([A-Z_][A-Z0-9_]*)\}?"
matches = re.findall(pattern, connection_string)
for var in matches:
access_key_list.append(f"${var}")
return access_key_list