Update retrieve langfuse api key through SSM

pagopa · Dec 2, 2024 · 9ba8e27 · 9ba8e27
1 parent dcd2e00
commit 9ba8e27
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/apps/chatbot/src/modules/chatbot.py b/apps/chatbot/src/modules/chatbot.py
@@ -21,6 +21,7 @@
 from src.modules.engine import get_automerging_engine
 from src.modules.handlers import EventHandler
 from src.modules.presidio import PresidioPII
+from src.modules.utils import get_ssm_parameter
 
 from dotenv import load_dotenv
 
@@ -42,8 +43,8 @@
     "Your role is to provide accurate, professional, and helpful responses to users' queries regarding "
     "the PagoPA DevPortal documentation available at: https://dev.developer.pagopa.it"
 )
-LANGFUSE_PUBLIC_KEY = os.getenv("LANGFUSE_INIT_PROJECT_PUBLIC_KEY")
-LANGFUSE_SECRET_KEY = os.getenv("LANGFUSE_INIT_PROJECT_SECRET_KEY")
+LANGFUSE_PUBLIC_KEY = get_ssm_parameter(os.getenv("LANGFUSE_INIT_PROJECT_PUBLIC_KEY"), os.getenv("LANGFUSE_INIT_PROJECT_PUBLIC_KEY"))
+LANGFUSE_SECRET_KEY = get_ssm_parameter(os.getenv("LANGFUSE_INIT_PROJECT_SECRET_KEY"), os.getenv("LANGFUSE_INIT_PROJECT_SECRET_KEY"))
 LANGFUSE_HOST = os.getenv("CHB_LANGFUSE_HOST")
 LANGFUSE_TAG = os.getenv("LANGFUSE_TAG", "development")
 LANGFUSE = Langfuse(