| from smolagents import tool |
| from duckduckgo_search import DDGS |
| from PyPDF2 import PdfReader |
|
|
|
|
| @tool |
| def web_search(query: str) -> str: |
| """ |
| Search the web for information. |
| |
| Args: |
| query: The search query to look up. |
| |
| Returns: |
| Search results as text. |
| """ |
|
|
| results = DDGS().text(query, max_results=5) |
|
|
| output = [] |
|
|
| for r in results: |
| output.append( |
| f"{r['title']}\n{r['body']}\n{r['href']}" |
| ) |
|
|
| return "\n\n".join(output) |
|
|
|
|
| @tool |
| def read_pdf(path: str) -> str: |
| """ |
| Read text content from a PDF file. |
| |
| Args: |
| path: Path to the PDF file. |
| |
| Returns: |
| Extracted text from the PDF. |
| """ |
|
|
| reader = PdfReader(path) |
|
|
| text = "" |
|
|
| for page in reader.pages: |
| extracted = page.extract_text() |
|
|
| if extracted: |
| text += extracted |
|
|
| return text[:12000] |
