naman1102 commited on
Commit
61abebf
·
1 Parent(s): 3cdd63a
Files changed (2) hide show
  1. app.py +2 -2
  2. tools.py +1 -1
app.py CHANGED
@@ -99,7 +99,7 @@ def tool_selector(state: AgentState) -> AgentState:
99
  " {'action':'excel'}\n"
100
  " {'action':'final'}\n"
101
  "if the tool you want isnt listed above, return {'action':'final'}"
102
- "Use wiki if you need to search online for information"
103
  "Use ocr if the question is about an image"
104
  "Use audio if the question is about an audio file"
105
  "Use excel if the question is about an excel file"
@@ -149,7 +149,7 @@ def audio_tool(state: AgentState) -> AgentState:
149
  def excel_tool(state: AgentState) -> AgentState:
150
  result = parse_excel_tool({
151
  "task_id": state.task_id,
152
- "excel_sheet_name": state.sheet or ""
153
  })
154
  out = {"excel_result": result}
155
  state.tool_calls += 1
 
99
  " {'action':'excel'}\n"
100
  " {'action':'final'}\n"
101
  "if the tool you want isnt listed above, return {'action':'final'}"
102
+ "Use wiki if you need to search online for information. Keep the query short and concise and accurate."
103
  "Use ocr if the question is about an image"
104
  "Use audio if the question is about an audio file"
105
  "Use excel if the question is about an excel file"
 
149
  def excel_tool(state: AgentState) -> AgentState:
150
  result = parse_excel_tool({
151
  "task_id": state.task_id,
152
+ "excel_sheet_name": ""
153
  })
154
  out = {"excel_result": result}
155
  state.tool_calls += 1
tools.py CHANGED
@@ -253,7 +253,7 @@ def wikipedia_search_tool(args: dict) -> str:
253
  summary_text = summary_data.get("extract")
254
  if not summary_text:
255
  summary_text = summary_data.get("description", "No summary available.")
256
-
257
  return f"Title: {first_title}\n\n{summary_text}"
258
 
259
 
 
253
  summary_text = summary_data.get("extract")
254
  if not summary_text:
255
  summary_text = summary_data.get("description", "No summary available.")
256
+ print(f"Title: {first_title}\n\n{summary_text}")
257
  return f"Title: {first_title}\n\n{summary_text}"
258
 
259