| import os | |
| import sys | |
| from dotenv import load_dotenv | |
| # Add backend to path | |
| sys.path.append(os.getcwd()) | |
| from core.byok_endpoints import get_byok_manager | |
| def refresh_keys(): | |
| load_dotenv() | |
| manager = get_byok_manager() | |
| keys_to_refresh = { | |
| "openai": os.getenv("OPENAI_API_KEY"), | |
| "anthropic": os.getenv("ANTHROPIC_API_KEY"), | |
| "deepseek": os.getenv("DEEPSEEK_API_KEY") | |
| } | |
| print(f"Refreshing keys with BYOK_ENCRYPTION_KEY: {os.getenv('BYOK_ENCRYPTION_KEY')[:10]}...") | |
| for provider, key in keys_to_refresh.items(): | |
| try: | |
| manager.store_api_key(provider, key) | |
| print(f"Successfully refreshed {provider}") | |
| except Exception as e: | |
| print(f"Failed to refresh {provider}: {e}") | |
| if __name__ == "__main__": | |
| refresh_keys() | |