Update llm_node.py

llm_node.py

@@ -1,11 +1,28 @@
+import os
+from langchain_community.chat_models import ChatOpenAI
+
 def llm_node(question, search_result):
-    # Check if search_result is a list of dicts
-    if isinstance(search_result, list):
-        combined_result = " ".join(
-            item.get("content", "") for item in search_result if isinstance(item, dict)
-        )
-    else:
-        combined_result = str(search_result)
-
-    combined_answer = combined_result.strip()
-    return combined_answer
+    # Initialize GPT-4
+    llm = ChatOpenAI(
+        model="gpt-4",
+        temperature=0,
+        openai_api_key=os.getenv("OPENAI_API_KEY")
+    )
+
+    prompt = f"""You are solving a GAIA benchmark evaluation question.
+
+Here’s the question:
+{question}
+
+Here’s retrieved information:
+{search_result}
+
+⚠️ VERY IMPORTANT:
+- ONLY return the final answer, exactly as required.
+- Do NOT include explanations, prefixes, or notes.
+- If the question asks for a list, give only the list, in the requested format.
+
+Your answer:"""
+
+    response = llm.invoke(prompt)
+    return response.content.strip()