Spaces:
Build error
Build error
File size: 3,026 Bytes
a258c1b 8d2f9d4 a258c1b 8d2f9d4 |
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 87 88 89 |
import requests
import logging
import httpx
from dotenv import load_dotenv
from typing import List, Dict, Optional
from app.utils.load_env import CX_CODE, CUSTOM_SEARCH_API_KEY
# Configure logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
# Load environment variables
load_dotenv()
async def google_search(query: str, num_results: str = "3") -> Optional[List[Dict[str, str]]]:
"""
Perform a Google search using the Custom Search JSON API.
Args:
query: The search query string.
num_results: The number of results to fetch as a string. Defaults to "10".
Returns:
Optional: A list of dictionaries containing the title and link of each search result.
Returns None if an error occurs or no results are found.
"""
if not query:
logger.error("Search query is empty.")
return None
try:
# Convert num_results to an integer
num_results_int = int(num_results)
if num_results_int > 10:
logger.warning("Google Custom Search API supports a maximum of 10 results per request.")
num_results_int = 10
elif num_results_int < 1:
logger.warning("num_results must be at least 1. Setting it to 1.")
num_results_int = 1
except ValueError:
logger.error("Invalid value for num_results. Must be a valid integer string.")
return None
# Define the URL and parameters
url = "https://www.googleapis.com/customsearch/v1"
params = {
"key": CUSTOM_SEARCH_API_KEY,
"cx": CX_CODE,
"q": query,
"num": num_results_int,
}
print(CUSTOM_SEARCH_API_KEY)
print(CX_CODE)
try:
# Make the HTTP request
async with httpx.AsyncClient() as client:
response = await client.get(url, params=params) # Properly encode query parameters
# Process the response
if response.status_code == 200:
results = response.json()
items = results.get("items", [])
return [{"title": item["title"], "link": item["link"], "snippet": item["snippet"]} for item in items]
else:
logger.error(f"Google Search API error: {response.status_code} - {response.text}")
return None
except httpx.RequestError as e:
logger.error("A network error occurred while performing the Google search.")
logger.error(f"Error details: {e}")
return None
def set_light_values(brightness: str, color_temp: str) -> Dict[str, str]:
"""Set the brightness and color temperature of a room light. (mock API).
Args:
brightness: Light level from 0 to 100. Zero is off and 100 is full brightness
color_temp: Color temperature of the light fixture, which can be `daylight`, `cool` or `warm`.
Returns:
A dictionary containing the set brightness and color temperature.
"""
return {
"brightness": brightness,
"colorTemperature": color_temp
}
|