naman1102 commited on
Commit
fadaecd
·
1 Parent(s): c40ba50

Update tools.py

Browse files
Files changed (1) hide show
  1. tools.py +6 -6
tools.py CHANGED
@@ -1,15 +1,15 @@
1
  # tools.py
2
 
3
  import pandas as pd
4
- from langchain_community.tools import DuckDuckGoSearchRun
5
  from pathlib import Path
6
- from PIL import Image
7
- import pytesseract
8
  from state import AgentState
9
  from langchain.schema import HumanMessage
10
  import regex as re
11
  import time
12
- from duckduckgo_search import DDGS
13
 
14
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
15
 
@@ -50,13 +50,13 @@ def web_search_tool(state: AgentState) -> AgentState:
50
  if not query:
51
  return {} # nothing to do
52
 
53
- ddg = DDGS()
54
  max_retries = 5
55
  result_text = ""
56
 
57
  for attempt in range(1, max_retries + 1):
58
  try:
59
- result_text = ddg(query, max_results=5)
60
  except Exception as e:
61
  # Network error or timeout—retry up to max_retries
62
  if attempt < max_retries:
 
1
  # tools.py
2
 
3
  import pandas as pd
4
+ # from langchain_community.tools import DuckDuckGoSearchRun
5
  from pathlib import Path
6
+ # from PIL import Image
7
+ # import pytesseract
8
  from state import AgentState
9
  from langchain.schema import HumanMessage
10
  import regex as re
11
  import time
12
+ from duckduckgo_search import ddg
13
 
14
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
15
 
 
50
  if not query:
51
  return {} # nothing to do
52
 
53
+ # ddg = DDGS()
54
  max_retries = 5
55
  result_text = ""
56
 
57
  for attempt in range(1, max_retries + 1):
58
  try:
59
+ result_text = str(ddg(query, max_results=5))
60
  except Exception as e:
61
  # Network error or timeout—retry up to max_retries
62
  if attempt < max_retries: