update env variables

agent.py

@@ -6,18 +6,19 @@ from typing import List, TypedDict
 from langgraph.types import Command
 from typing import Literal, List
 from langgraph.graph import StateGraph, START, END
-import boto3
+# import boto3
+from dotenv import load_dotenv
 
-import streamlit as st
+load_dotenv()
 
-ssm = boto3.client(
-            'ssm',
-            region_name="us-east-2", 
-            # aws_access_key_id=os.environ['AWS_ACCESS_KEY'],
-            # aws_secret_access_key=os.environ['AWS_SECRET_ACCESS_KEY']
-        )
+# ssm = boto3.client(
+#             'ssm',
+#             region_name="us-east-2", 
+#             # aws_access_key_id=os.environ['AWS_ACCESS_KEY'],
+#             # aws_secret_access_key=os.environ['AWS_SECRET_ACCESS_KEY']
+#         )
 
-os.environ['TAVILY_API_KEY'] = ssm.get_parameter(Name="TAVILY_API_KEY", WithDecryption=True)['Parameter']['Value']
+# os.environ['TAVILY_API_KEY'] = ssm.get_parameter(Name="TAVILY_API_KEY", WithDecryption=True)['Parameter']['Value']
 
 def get_platform_tips(state) -> Command[Literal['web_search']]:
     """Conduct a web search to find up-to-date information on how to write an effective post for the provided platform."""
@@ -152,7 +153,7 @@ def fetch_image(state) -> Command[Literal[END]]:
     }
 
     headers = {
-        "Authorization": ssm.get_parameter(Name="PEXEL_API_KEY", WithDecryption=True)['Parameter']['Value']
+        "Authorization": os.getenv("PEXELS_API_KEY")#ssm.get_parameter(Name="PEXEL_API_KEY", WithDecryption=True)['Parameter']['Value']
     }
     
     response = requests.get(url, headers=headers, params=params)
@@ -182,7 +183,7 @@ model =  ChatGoogleGenerativeAI(
             max_tokens=None,
             timeout=None,
             max_retries=2,
-            google_api_key=ssm.get_parameter(Name="GOOGLE_API_KEY", WithDecryption=True)['Parameter']['Value']            
+            # google_api_key=ssm.get_parameter(Name="GOOGLE_API_KEY", WithDecryption=True)['Parameter']['Value']            
             )
 
 workflow = StateGraph(State)