Spaces:
Paused
Paused
File size: 1,535 Bytes
12f2295 |
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 |
import os
import logging
from googleapiclient.discovery import build
from dotenv import load_dotenv
load_dotenv()
class GoogleSearch:
"""Performs Google Custom Search API queries."""
def __init__(self):
self.api_key = os.getenv("GOOGLE_API_KEY")
self.cse_id = os.getenv("GOOGLE_CSE_ID")
self.service = None
if self.api_key and self.cse_id:
try:
self.service = build("customsearch", "v1", developerKey=self.api_key)
logging.info("✅ Google Custom Search initialized.")
except Exception as e:
logging.error(f"❌ CSE init failed: {e}")
else:
logging.warning("⚠️ GOOGLE_API_KEY or GOOGLE_CSE_ID not set; search disabled.")
def search(self, queries: list, num_results: int = 1) -> list:
if not self.service:
return []
out = []
for q in queries:
try:
resp = self.service.cse().list(q=q, cx=self.cse_id, num=num_results).execute()
items = resp.get("items", [])
formatted = [
{"title": it.get("title"), "link": it.get("link"), "snippet": it.get("snippet")}
for it in items
]
out.append({"query": q, "results": formatted})
except Exception as e:
logging.error(f"❌ Search error for '{q}': {e}")
out.append({"query": q, "results": []})
return out
google_search = GoogleSearch()
|