bigquery_agents 0.1.0

No project description provided

BigQuery Agents

This project implements a conversational agent powered by LangGraph and Google Vertex AI's Gemini model to troubleshoot BigQuery jobs. It utilizes custom tools to fetch job details and table schemas, enabling the agent to provide informed suggestions for fixing errors.

Features

• BigQuery Job Details Retrieval: Fetches details of a BigQuery job, including the query and any error messages.
• BigQuery Table Schema Retrieval: Retrieves the schema of a BigQuery table, including column names and data types.
• Error Analysis and Fix Suggestions: Analyzes job details and errors to suggest possible fixes, leveraging table schema information when necessary.
• Conversational Interface: Allows users to interact with the agent through a command-line interface.
• LangGraph State Management: Uses LangGraph to manage the conversation flow and state, including messages, job details, and table schemas.
• Mermaid Graph Visualization: Generates a Mermaid diagram of the LangGraph flow, saved as graph.jpg.

Prerequisites

• Python 3.12+
• Google Cloud Platform (GCP) account with BigQuery enabled
• PROJECT_ID environment variable set to your GCP project ID or hardcoded in the script.
• Poetry for dependency management

Installation

1. Clone the repository:

   git clone <repository_url>
   cd <repository_directory>
   

2. Install dependencies using Poetry:

   poetry install
   

3. Set your GCP project ID:

   export PROJECT_ID="your-gcp-project-id"
   

   Or change the PROJECT_ID variable inside the python script.

4. Authenticate with GCP:

   gcloud auth application-default login
   

Usage

Run the agent using the bq_fix script defined in poetry.toml:

poetry run bq_fix