Update llm_node.py

llm_node.py

@@ -1,10 +1,7 @@
 from langchain_community.chat_models import ChatOpenAI
-from langchain.prompts import PromptTemplate
-from langchain.chains import LLMChain
-from prompts import PROMPT_TEMPLATE
 
-def llm_node():
+def llm_node(question):
     llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0)
-    prompt = PromptTemplate.from_template(PROMPT_TEMPLATE)
-    chain = LLMChain(llm=llm, prompt=prompt)
-    return chain
+    prompt = f"Answer this carefully and exactly:\n{question}"
+    response = llm.invoke(prompt)
+    return response.content.strip()