basic google search custom engine

gistfile1.txt

import json
import sys

from article_agent.config import my_api_key, my_cse_id
from googleapiclient.discovery import build


def get_search_results(search_term: str, num_search_results: int = 10) -> json:
    """Perform a Google search using Custom Search API"""
    # Build request
    service = build("customsearch", "v1", developerKey=my_api_key)
    # Execute request
    query_result = service.cse().list(q=search_term, cx=my_cse_id, num=num_search_results).execute()

    print(query_result)
    #with open(f'{"-".join(search_term.split("-"))}.json', 'w') as output_file:
    with open(f'openai-alpha.json', 'w') as output_file:
        json.dump(query_result, output_file)

    return query_result

if __name__ == '__main__':
    # take first argument from cli
    search_term = sys.argv[1]
    num_search_results = 10
    get_search_results(search_term, num_search_results)