Spaces:
Sleeping
Sleeping
File size: 2,925 Bytes
1397957 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 |
from typing import Dict, Any, List
from .tool import BaseTool, ToolContext, ToolResult
class WebSearchTool(BaseTool):
@property
def id(self) -> str:
return "websearch"
@property
def description(self) -> str:
return (
"Search the web using DuckDuckGo. Returns relevant search results "
"with titles, URLs, and snippets. Use this when you need current "
"information from the internet."
)
@property
def parameters(self) -> Dict[str, Any]:
return {
"type": "object",
"properties": {
"query": {
"type": "string",
"description": "The search query"
},
"max_results": {
"type": "integer",
"description": "Maximum number of results to return (default: 5)",
"default": 5
}
},
"required": ["query"]
}
async def execute(self, args: Dict[str, Any], ctx: ToolContext) -> ToolResult:
query = args["query"]
max_results = args.get("max_results", 5)
try:
from ddgs import DDGS
results = []
with DDGS() as ddgs:
# 한국 지역 기반 검색 결과
for r in ddgs.text(query, region="kr-kr", max_results=max_results):
results.append({
"title": r.get("title", ""),
"url": r.get("href", ""),
"snippet": r.get("body", "")
})
if not results:
return ToolResult(
title=f"Web search: {query}",
output="No results found.",
metadata={"query": query, "count": 0}
)
output_lines = []
for i, r in enumerate(results, 1):
output_lines.append(f"{i}. {r['title']}")
output_lines.append(f" URL: {r['url']}")
output_lines.append(f" {r['snippet']}")
output_lines.append("")
return ToolResult(
title=f"Web search: {query}",
output="\n".join(output_lines),
metadata={"query": query, "count": len(results)}
)
except ImportError:
return ToolResult(
title=f"Web search: {query}",
output="Error: ddgs package not installed. Run: pip install ddgs",
metadata={"error": "missing_dependency"}
)
except Exception as e:
return ToolResult(
title=f"Web search: {query}",
output=f"Error performing search: {str(e)}",
metadata={"error": str(e)}
)
|