File size: 898 Bytes
00ff2c1 |
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 |
from langchain.tools import Tool
import os
import requests
def google_search(input: str) -> str:
"""A tool that simulates a Google search and returns top results."""
try:
response = requests.get(
"https://www.googleapis.com/customsearch/v1",
params={
"q": input,
"key": os.getenv("GOOGLE_API_KEY"),
"cx": os.getenv("GOOGLE_SEARCH_ENGINE_ID"),
"num": 1
}
)
data = response.json()
# Extract and return the top search result summary
return data.get("items", [])[0].get("snippet", "No results found.")
except Exception as e:
return f"Google search error: {str(e)}"
google_search_tool = Tool(
name="google_search",
func=google_search,
description="Search the web using Google and return the top summary from the results."
)
|