DH-4917 Encrypt llm_api_key

Dataherald · Oct 30, 2023 · f08225d · f08225d
1 parent d8439f3
commit f08225d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dataherald/sql_database/models/types.py b/dataherald/sql_database/models/types.py
@@ -68,7 +68,7 @@ def set_uri_without_ssh(cls, v, values):
             raise ValueError("When use_ssh is False set uri")
         return v
 
-    @validator("uri", pre=True, always=True)
+    @validator("uri", "llm_api_key", pre=True, always=True)
     def encrypt(cls, value: str):
         fernet_encrypt = FernetEncrypt()
         try: