naman1102 commited on
Commit
477bf79
·
1 Parent(s): 266fff4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -32,7 +32,7 @@ def plan_node(state: AgentState) -> AgentState:
32
  If confident: return {"final_answer": "<answer>"}
33
  Otherwise: return exactly one of
34
  {"wiki_query": "..."},
35
- {"web_search_query": "..."},
36
  {"ocr_path": "..."},
37
  {"excel_path": "...", "excel_sheet_name": "..."},
38
  {"audio_path": "..."}
@@ -54,7 +54,7 @@ def plan_node(state: AgentState) -> AgentState:
54
  " and nothing else.\n"
55
  " • Otherwise, return exactly one of:\n"
56
  " {\"wiki_query\":\"<Wikipedia search>\"}\n"
57
- " {\"web_search_query\":\"<search terms>\"}\n"
58
  " {\"ocr_path\":\"<image path or task_id>\"}\n"
59
  " {\"excel_path\":\"<xlsx path>\", \"excel_sheet_name\":\"<sheet name>\"}\n"
60
  " {\"audio_path\":\"<audio path or task_id>\"}\n"
@@ -185,7 +185,7 @@ def inspect_node(state: AgentState) -> AgentState:
185
  "and nothing else.\n"
186
  "Otherwise, return exactly one of these JSON literals to fetch another tool:\n"
187
  " {\"wiki_query\":\"<query for Wikipedia>\"}\n"
188
- " {\"web_search_query\":\"<search terms>\"}\n"
189
  " {\"ocr_path\":\"<image path or task_id>\"}\n"
190
  " {\"excel_path\":\"<xls path>\", \"excel_sheet_name\":\"<sheet name>\"}\n"
191
  " {\"audio_path\":\"<audio path or task_id>\"}\n"
@@ -299,7 +299,7 @@ def respond_to_input(user_input: str, task_id) -> str:
299
  "You are an agent orchestrator. Decide whether to use a tool or answer directly.\n"
300
  "Tools available:\n"
301
  " • Wikipedia: set {\"wiki_query\":\"<search terms>\"}\n"
302
- " • Web search: set {\"web_search_query\":\"<search terms>\"}\n"
303
  " • OCR: set {\"ocr_path\":\"<image path or task_id>\"}\n"
304
  " • Excel: set {\"excel_path\":\"<xlsx path>\", \"excel_sheet_name\":\"<sheet>\"}\n"
305
  " • Audio transcription: set {\"audio_path\":\"<audio path or task_id>\"}\n"
 
32
  If confident: return {"final_answer": "<answer>"}
33
  Otherwise: return exactly one of
34
  {"wiki_query": "..."},
35
+
36
  {"ocr_path": "..."},
37
  {"excel_path": "...", "excel_sheet_name": "..."},
38
  {"audio_path": "..."}
 
54
  " and nothing else.\n"
55
  " • Otherwise, return exactly one of:\n"
56
  " {\"wiki_query\":\"<Wikipedia search>\"}\n"
57
+ # " {\"web_search_query\":\"<search terms>\"}\n"
58
  " {\"ocr_path\":\"<image path or task_id>\"}\n"
59
  " {\"excel_path\":\"<xlsx path>\", \"excel_sheet_name\":\"<sheet name>\"}\n"
60
  " {\"audio_path\":\"<audio path or task_id>\"}\n"
 
185
  "and nothing else.\n"
186
  "Otherwise, return exactly one of these JSON literals to fetch another tool:\n"
187
  " {\"wiki_query\":\"<query for Wikipedia>\"}\n"
188
+ # " {\"web_search_query\":\"<search terms>\"}\n"
189
  " {\"ocr_path\":\"<image path or task_id>\"}\n"
190
  " {\"excel_path\":\"<xls path>\", \"excel_sheet_name\":\"<sheet name>\"}\n"
191
  " {\"audio_path\":\"<audio path or task_id>\"}\n"
 
299
  "You are an agent orchestrator. Decide whether to use a tool or answer directly.\n"
300
  "Tools available:\n"
301
  " • Wikipedia: set {\"wiki_query\":\"<search terms>\"}\n"
302
+ # " • Web search: set {\"web_search_query\":\"<search terms>\"}\n"
303
  " • OCR: set {\"ocr_path\":\"<image path or task_id>\"}\n"
304
  " • Excel: set {\"excel_path\":\"<xlsx path>\", \"excel_sheet_name\":\"<sheet>\"}\n"
305
  " • Audio transcription: set {\"audio_path\":\"<audio path or task_id>\"}\n"