Essi commited on
Commit
9df38aa
·
1 Parent(s): 939bb6c

refactor: Rename verify_answer to normalize_answer and update reasoning steps format

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -4,6 +4,7 @@ import json
4
  import operator
5
  import os
6
  import re
 
7
  from typing import Annotated, TypedDict
8
  import gradio as gr
9
  from langchain_openai import ChatOpenAI
@@ -158,15 +159,15 @@ class GAIAAgent:
158
  workflow.add_node("search_or_calc", self._search_or_calc)
159
  workflow.add_node("process_info", self._process_info)
160
  workflow.add_node("generate_answer", self._generate_answer)
161
- workflow.add_node("verify_answer", self._verify_answer)
162
 
163
  # Add edges
164
  workflow.set_entry_point("analyze_question")
165
  workflow.add_edge("analyze_question", "search_or_calc")
166
  workflow.add_edge("search_or_calc", "process_info")
167
  workflow.add_edge("process_info", "generate_answer")
168
- workflow.add_edge("generate_answer", "verify_answer")
169
- workflow.add_edge("verify_answer", END)
170
 
171
  return workflow.compile()
172
 
@@ -198,7 +199,7 @@ class GAIAAgent:
198
 
199
  state["search_results"] = results_json
200
  state["tools_used"].append("web_search")
201
- state["reasoning_steps"].append(f"search:{query}")
202
 
203
  return state
204
 
@@ -234,7 +235,7 @@ class GAIAAgent:
234
  return state
235
 
236
 
237
- def _verify_answer(self, state: AgentState) -> AgentState:
238
  ans = state["answer"].strip()
239
 
240
  # Canonicalize numbers (remove commas) / lowercase yes|no
 
4
  import operator
5
  import os
6
  import re
7
+ import pandas as pd
8
  from typing import Annotated, TypedDict
9
  import gradio as gr
10
  from langchain_openai import ChatOpenAI
 
159
  workflow.add_node("search_or_calc", self._search_or_calc)
160
  workflow.add_node("process_info", self._process_info)
161
  workflow.add_node("generate_answer", self._generate_answer)
162
+ workflow.add_node("normalize_answer", self._normalize_answer)
163
 
164
  # Add edges
165
  workflow.set_entry_point("analyze_question")
166
  workflow.add_edge("analyze_question", "search_or_calc")
167
  workflow.add_edge("search_or_calc", "process_info")
168
  workflow.add_edge("process_info", "generate_answer")
169
+ workflow.add_edge("generate_answer", "normalize_answer")
170
+ workflow.add_edge("normalize_answer", END)
171
 
172
  return workflow.compile()
173
 
 
199
 
200
  state["search_results"] = results_json
201
  state["tools_used"].append("web_search")
202
+ state["reasoning_steps"].append(f"Search: {query}")
203
 
204
  return state
205
 
 
235
  return state
236
 
237
 
238
+ def _normalize_answer(self, state: AgentState) -> AgentState:
239
  ans = state["answer"].strip()
240
 
241
  # Canonicalize numbers (remove commas) / lowercase yes|no