Update AgentGPT.py

AgentGPT.py

@@ -14,6 +14,11 @@ from langchain.llms import HuggingFaceHub
 from langchain.llms.fireworks import Fireworks
 from langchain.chat_models.fireworks import ChatFireworks
 
+GOOGLE_CSE_ID = os.getenv("GOOGLE_CSE_ID")
+GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
+FIREWORKS_API_KEY = os.getenv("FIREWORKS_API_KEY")
+HUGGINGFACEHUB_API_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
+
 client_ports = []
 
 # Define the websocket client class
@@ -24,10 +29,10 @@ class AgentsGPT:
         
     async def get_response(self, question):
           
-        GOOGLE_CSE_ID = os.environ["GOOGLE_CSE_ID"]
-        GOOGLE_API_KEY = os.environ["GOOGLE_API_KEY"]
-        FIREWORKS_API_KEY = os.environ["FIREWORKS_API_KEY"]
-        HUGGINGFACEHUB_API_TOKEN = os.environ["HUGGINGFACEHUB_API_TOKEN"]
+        os.environ["GOOGLE_CSE_ID"] = GOOGLE_CSE_ID
+        os.environ["GOOGLE_API_KEY"] = GOOGLE_API_KEY
+        os.environ["FIREWORKS_API_KEY"] = FIREWORKS_API_KEY
+        os.environ["HUGGINGFACEHUB_API_TOKEN"] = HUGGINGFACEHUB_API_TOKEN
 
         llm = Fireworks(model="accounts/fireworks/models/llama-v2-70b-chat", model_kwargs={"temperature":0, "max_tokens":1500, "top_p":1.0}, streaming=True)
         tools = load_tools(["google-search", "llm-math"], llm=llm)