17 lines
431 B
Python
17 lines
431 B
Python
|
from pydantic import BaseModel, Field
|
||
|
|
from langchain_core.tools import Tool, StructuredTool
|
||
|
|
|
||
|
|
from src.utils.web_search import WebSearcher
|
||
|
|
|
||
|
|
class SearchArgsSchema(BaseModel):
|
||
|
|
query: str = Field(..., description="The query to search the web for")
|
||
|
|
|
||
|
|
search_tool = Tool(
|
||
|
|
name="search",
|
||
|
|
description="Search the web for information",
|
||
|
|
func=WebSearcher().search,
|
||
|
|
args_schema=SearchArgsSchema,
|
||
|
|
return_direct=True,
|
||
|
|
)
|
||
|
|
|