Umer797 commited on
Commit
e3c0d0c
·
verified ·
1 Parent(s): 87167b1

Update llm_node.py

Browse files
Files changed (1) hide show
  1. llm_node.py +9 -4
llm_node.py CHANGED
@@ -2,13 +2,18 @@ from langchain_community.chat_models import ChatOpenAI
2
 
3
  def llm_node(question):
4
  llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0)
5
- prompt = f"""You are solving a GAIA benchmark question.
6
- ONLY return the direct, final answer.
7
- Do NOT include any explanation or prefix.
 
 
 
 
 
8
  Here’s the question:
9
 
10
  {question}
11
 
12
- Your answer:"""
13
  response = llm.invoke(prompt)
14
  return response.content.strip()
 
2
 
3
  def llm_node(question):
4
  llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0)
5
+ prompt = f"""You are solving a GAIA benchmark evaluation question.
6
+
7
+ ⚠️ VERY IMPORTANT:
8
+ - ONLY return the final answer, exactly as required.
9
+ - DO NOT include explanations, prefixes, or notes.
10
+ - Format the answer exactly as asked (e.g., comma-separated, plural, in requested order).
11
+ - If the question asks for a list, give only the list, no intro.
12
+
13
  Here’s the question:
14
 
15
  {question}
16
 
17
+ Your direct answer:"""
18
  response = llm.invoke(prompt)
19
  return response.content.strip()