global connection and docs

Author: bboynton97

agentstack/_tools/py_sql/__init__.py agentstack/_tools/sql/__init__.pyagentstack/_tools/py_sql/__init__.py renamed to agentstack/_tools/sql/__init__.py

@@ -2,15 +2,22 @@
 import psycopg2
 from typing import Dict, Any
 
+connection = None
+
 def _get_connection():
     """Get PostgreSQL database connection"""
-    return psycopg2.connect(
-        dbname=os.getenv('POSTGRES_DB'),
-        user=os.getenv('POSTGRES_USER'),
-        password=os.getenv('POSTGRES_PASSWORD'),
-        host=os.getenv('POSTGRES_HOST', 'localhost'),
-        port=os.getenv('POSTGRES_PORT', '5432')
-    )
+
+    global connection
+    if connection is None:
+        connection = psycopg2.connect(
+            dbname=os.getenv('POSTGRES_DB'),
+            user=os.getenv('POSTGRES_USER'),
+            password=os.getenv('POSTGRES_PASSWORD'),
+            host=os.getenv('POSTGRES_HOST', 'localhost'),
+            port=os.getenv('POSTGRES_PORT', '5432')
+        )
+
+    return connection
 
 def get_schema() -> Dict[str, Any]:
     """
@@ -47,7 +54,7 @@ def get_schema() -> Dict[str, Any]:
             schema[table_name] = columns
 
         cursor.close()
-        conn.close()
+        # conn.close()
         return schema
 
     except Exception as e:
@@ -71,7 +78,7 @@ def execute_query(query: str) -> list:
         results = cursor.fetchall()
 
         cursor.close()
-        conn.close()
+        # conn.close()
         return results
 
     except Exception as e:

agentstack/_tools/py_sql/config.json agentstack/_tools/sql/config.jsonagentstack/_tools/py_sql/config.json renamed to agentstack/_tools/sql/config.json

@@ -1,13 +1,13 @@
 {
-    "name": "py_sql",
+    "name": "sql",
     "url": "https://pypi.org/project/psycopg2/",
     "category": "database",
     "env": {
-        "POSTGRES_DB": "...",
-        "POSTGRES_USER": "...",
-        "POSTGRES_PASSWORD": "...",
-        "POSTGRES_HOST": "...",
-        "POSTGRES_PORT": "..."
+        "POSTGRES_DB": null,
+        "POSTGRES_USER": null,
+        "POSTGRES_PASSWORD": null,
+        "POSTGRES_HOST": null,
+        "POSTGRES_PORT": null
     },
     "dependencies": [
         "psycopg2-binary>=2.9.9"

docs/tools/core.mdx

@@ -13,9 +13,12 @@ description: 'AgentStack tools that are not third-party integrations'
 
 - [Code Interpreter](/tools/tool/code-interpreter)
 
-## Data Input
+## Input
 - [Vision](/tools/tool/vision)
 
+## Data
+- [SQL](/tools/tool/sql)
+
 <CardGroup cols={1}>
   <Card
     title="Community Tools"

docs/tools/tool/sql.mdx

@@ -0,0 +1,26 @@
+---
+title: 'Perplexity'
+description: 'Agentic Search'
+---
+
+## Description
+Enable your agent to perform queries directly on a database
+
+<Note>
+    There is no built-in sandboxing using this tool. Agents may perform destructive queries that may be irreversible.
+</Note>
+
+## Installation
+
+```bash
+agentstack tools add sql
+```
+
+Set the API keys
+```env
+POSTGRES_DB=...
+POSTGRES_USER=...
+POSTGRES_PASSWORD=...
+POSTGRES_HOST=...
+POSTGRES_PORT=...
+```