Merge pull request #100 from UMLCloudComputing/environ

chore: specify dynamo_db table variables
UMLCloudComputing · Aug 23, 2024 · 893413b · 893413b
2 parents 0165ac7 + 23a7cdc
commit 893413b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/streamlit.py b/src/streamlit.py
@@ -6,6 +6,7 @@
 import app.llm as llm
 import streamlit as st
 import secrets
+import os
 from langchain_openai import ChatOpenAI
 from typing import List, Dict
 from pydantic import BaseModel
@@ -27,6 +28,8 @@
     }
 )
 
+os.environ["DYNAMO_TABLE"] = st.secrets["DYNAMO_TABLE"]
+
 # ------------------------------------------------------
 # Pydantic data model and helper function for Citations