bigquery_search.py

from google.cloud import bigquery

def search_bigquery_for_remedy(search_query: str) -> str:
    try:
        client = bigquery.Client(project="gem-creation")
        query = """
            SELECT remedy_description FROM `gem-creation.maize_remedies.remedies`
            WHERE SEARCH(remedy_description, @query)
        """
        job_config = bigquery.QueryJobConfig(
            query_parameters=[bigquery.ScalarQueryParameter("query", "STRING", search_query)]
        )
        query_job = client.query(query, job_config=job_config)
        results = list(query_job)
        return results[0].remedy_description if results else "No remedy found."
    except Exception as e:
        return f"Error querying BigQuery: {e}"