Postgres Tools

"""
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.
    """)