postgres_tools.py
"""
Postgres Tools
=============================
Demonstrates postgres tools.
"""
from agno.agent import Agent
from agno.tools.postgres import PostgresTools
# ---------------------------------------------------------------------------
# Create Agent
# ---------------------------------------------------------------------------
# Example 1: Include specific Postgres functions (default behavior - all functions included)
agent = Agent(
tools=[
PostgresTools(
host="localhost",
port=5532,
db_name="ai",
user="ai",
password="ai",
table_schema="ai",
)
]
)
# Example 2: Include only read-only operations
agent_readonly = Agent(
tools=[
PostgresTools(
host="localhost",
port=5532,
db_name="ai",
user="ai",
password="ai",
table_schema="ai",
include_tools=[
"show_tables",
"describe_table",
"summarize_table",
"inspect_query",
],
)
]
)
# Example 3: Exclude dangerous operations
agent_safe = Agent(
tools=[
PostgresTools(
host="localhost",
port=5532,
db_name="ai",
user="ai",
password="ai",
table_schema="ai",
exclude_tools=["run_query"], # Exclude direct query execution
)
]
)
# ---------------------------------------------------------------------------
# Run Agent
# ---------------------------------------------------------------------------
if __name__ == "__main__":
agent.print_response(
"List the tables in the database and summarize one of the tables", markdown=True
)
agent.print_response("""
Please run a SQL query to get all sessions in `agno_sessions` created in the last 24 hours and summarize the table.
""")
Run the Example
1
Set up your virtual environment
uv venv --python 3.12
source .venv/bin/activate
uv venv --python 3.12
.venv\Scripts\activate
2
Install dependencies
uv pip install -U agno openai psycopg psycopg-binary
3
Export your OpenAI API key
export OPENAI_API_KEY="your_openai_api_key_here"
$Env:OPENAI_API_KEY="your_openai_api_key_here"
4
Run the example
Save the code above as
postgres_tools.py, then run:python postgres_tools.py