Update llm.py

llm.py

@@ -2,6 +2,7 @@
 # Description: 
 # ------------------------------------------------
 from langchain_openai import ChatOpenAI
+import os
 
 def invoke_llm(context, task):
     prompt = f"""You are an assistant of a newspaper. 
@@ -15,7 +16,7 @@ def invoke_llm(context, task):
 
     llm=ChatOpenAI(model_name="gpt-3.5-turbo",
                        temperature=0,
-                       openai_api_key=OPENAI_KEY,
+                       openai_api_key=os.getenv('OPENAI_KEY'),
                        max_tokens=1000)
     result = llm.invoke(prompt)
     return result.content