sql_agent.py

from langchain_google_genai import ChatGoogleGenerativeAI
from langchain_community.utilities import SQLDatabase
from langchain_community.agent_toolkits import create_sql_agent

# Load database
db = SQLDatabase.from_uri("sqlite:///chinook.db")

# Gemini model
llm = ChatGoogleGenerativeAI(
    model="gemini-2.5-flash",
    temperature=0
)

# Create SQL Agent
agent_executor = create_sql_agent(
    llm=llm,
    db=db,
    verbose=True
)

# Chat loop
while True:
    question = input("\nAsk a database question (type 'exit'): ")

    if question.lower() == "exit":
        break

    response = agent_executor.invoke({"input": question})
    print("\nAnswer:", response["output"])



from langchain_google_genai import ChatGoogleGenerativeAI
from langchain_community.utilities import SQLDatabase
from langchain_community.agent_toolkits import create_sql_agent


"""
Sample Questions to Test the Agent

Basic:
- List all tables in the database
- How many albums are there?
- How many artists exist?

Intermediate:
- Which artist has the most albums?
- Which genre has the most tracks?
- Show the first 10 tracks
- Which country has the most customers?

Analytics:
- Top 5 customers by spending
- Which artist generated the most revenue?
- Which album has the most songs?
- What is the longest track in the database?
"""