45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
#!/usr/bin/env python3
|
|
"""Test script to debug API key decryption issue."""
|
|
|
|
from src.core.database import SessionLocal
|
|
from src.models.llm import LLMProvider
|
|
from cryptography.fernet import Fernet
|
|
import os
|
|
|
|
# Get the encryption key
|
|
key = os.getenv("ENCRYPTION_KEY", "ZcytYzi0iHIl4Ttr-GdAEk117aGRogkGvN3wiTxrPpE=").encode()
|
|
print(f"Encryption key (first 20 chars): {key[:20]}")
|
|
print(f"Encryption key length: {len(key)}")
|
|
|
|
# Create Fernet instance
|
|
fernet = Fernet(key)
|
|
|
|
# Get provider from database
|
|
db = SessionLocal()
|
|
provider = db.query(LLMProvider).filter(LLMProvider.id == '6c899741-4108-4196-aea4-f38ad2f0150e').first()
|
|
|
|
if provider:
|
|
print(f"\nProvider found:")
|
|
print(f" ID: {provider.id}")
|
|
print(f" Name: {provider.name}")
|
|
print(f" Encrypted API Key (first 50 chars): {provider.api_key[:50]}")
|
|
print(f" Encrypted API Key Length: {len(provider.api_key)}")
|
|
|
|
# Test decryption
|
|
print(f"\nAttempting decryption...")
|
|
try:
|
|
decrypted = fernet.decrypt(provider.api_key.encode()).decode()
|
|
print(f"Decryption successful!")
|
|
print(f" Decrypted key length: {len(decrypted)}")
|
|
print(f" Decrypted key (first 8 chars): {decrypted[:8]}")
|
|
print(f" Decrypted key is empty: {len(decrypted) == 0}")
|
|
except Exception as e:
|
|
print(f"Decryption failed with error: {e}")
|
|
print(f"Error type: {type(e).__name__}")
|
|
import traceback
|
|
traceback.print_exc()
|
|
else:
|
|
print("Provider not found")
|
|
|
|
db.close()
|