Umer797 commited on
Commit
68550d1
·
verified ·
1 Parent(s): 91cb62a

Update llm_node.py

Browse files
Files changed (1) hide show
  1. llm_node.py +8 -1
llm_node.py CHANGED
@@ -2,6 +2,13 @@ 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"Answer this carefully and exactly:\n{question}"
 
 
 
 
 
 
 
6
  response = llm.invoke(prompt)
7
  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 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()