Spaces:
Sleeping
Sleeping
Essi
commited on
Commit
·
9df38aa
1
Parent(s):
939bb6c
refactor: Rename verify_answer to normalize_answer and update reasoning steps format
Browse files
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("
|
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", "
|
169 |
-
workflow.add_edge("
|
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"
|
202 |
|
203 |
return state
|
204 |
|
@@ -234,7 +235,7 @@ class GAIAAgent:
|
|
234 |
return state
|
235 |
|
236 |
|
237 |
-
def
|
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
|