neurocore-skill-postgres 0.1.0__py3-none-any.whl

neurocore_skill_postgres/__init__.py

@@ -0,0 +1,5 @@
+"""neurocore-skill-postgres — PostgreSQL query skill for NeuroCore."""
+
+from neurocore_skill_postgres.skill import PostgresSkill
+
+__all__ = ["PostgresSkill"]

neurocore_skill_postgres/skill.py

@@ -0,0 +1,68 @@
+"""PostgresSkill — run a parameterized SQL query against PostgreSQL.
+
+Reads ``sql`` (and optional ``sql_params``) from context (config provides
+defaults) and writes returned rows to ``postgres_rows``.
+"""
+from __future__ import annotations
+
+import logging
+import os
+from typing import Any
+
+from flowengine import FlowContext
+
+from neurocore import AsyncSkill, SkillMeta
+
+logger = logging.getLogger(__name__)
+
+
+class PostgresSkill(AsyncSkill):
+    """Async parameterized SQL execution against PostgreSQL (psycopg 3)."""
+
+    skill_meta = SkillMeta(
+        name="postgres",
+        version="0.1.0",
+        description="Run parameterized SQL against PostgreSQL",
+        author="NeuroCore Contributors",
+        requires=["psycopg[binary]>=3.1"],
+        provides=["postgres_rows"],
+        consumes=["sql", "sql_params"],
+        tags=["database", "sql", "postgres"],
+        config_schema={
+            "properties": {
+                "dsn": {"type": "string", "description": "Connection string."},
+                "sql": {"type": "string", "description": "Default SQL (overridable)."},
+            }
+        },
+    )
+
+    def _resolve_dsn(self) -> str:
+        return self.config.get("dsn", "") or os.environ.get("DATABASE_URL", "")
+
+    async def _execute(self, sql: str, params: Any) -> list[dict[str, Any]]:
+        """Execute ``sql`` and return rows as dicts (empty for non-SELECT)."""
+        import psycopg
+        from psycopg.rows import dict_row
+
+        async with await psycopg.AsyncConnection.connect(
+            self._resolve_dsn(), row_factory=dict_row
+        ) as conn, conn.cursor() as cur:
+            await cur.execute(sql, params)
+            if cur.description is None:
+                await conn.commit()
+                return []
+            return [dict(row) for row in await cur.fetchall()]
+
+    async def process(self, context: FlowContext) -> FlowContext:
+        sql = str(context.get("sql") or self.config.get("sql", ""))
+        if not sql:
+            logger.warning("PostgresSkill: no 'sql' provided.")
+            context.set("postgres_rows", [])
+            return context
+        params = context.get("sql_params")
+        try:
+            context.set("postgres_rows", await self._execute(sql, params))
+        except Exception as exc:  # noqa: BLE001
+            logger.error("PostgresSkill query failed: %s", exc, exc_info=True)
+            context.set("postgres_rows", {"error": str(exc)})
+        return context

neurocore_skill_postgres-0.1.0.dist-info/METADATA

@@ -0,0 +1,50 @@
+Metadata-Version: 2.4
+Name: neurocore-skill-postgres
+Version: 0.1.0
+Summary: PostgreSQL query skill for NeuroCore
+Author: NeuroCore Contributors
+License-Expression: Apache-2.0
+Keywords: ai,database,neurocore,postgres,skill,sql
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Typing :: Typed
+Requires-Python: >=3.13
+Requires-Dist: neurocore-ai>=0.2.0
+Requires-Dist: psycopg[binary]>=3.1
+Provides-Extra: dev
+Requires-Dist: mypy>=1.8; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.24; extra == 'dev'
+Requires-Dist: pytest>=9.0.2; extra == 'dev'
+Requires-Dist: ruff>=0.8; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# neurocore-skill-postgres
+
+Run parameterized SQL against PostgreSQL from a NeuroCore flow (psycopg 3).
+
+```bash
+pip install neurocore-skill-postgres
+export DATABASE_URL=postgresql://user:pass@localhost/db
+```
+
+```yaml
+components:
+  - name: query
+    type: postgres
+    config:
+      sql: "SELECT id, name FROM users WHERE active = %s"
+flow:
+  type: sequential
+  steps:
+    - component: query
+```
+
+Provide `sql` (and optional `sql_params`) via config or context. Results are
+written to `postgres_rows` (a list of row dicts; empty list for non-SELECT
+statements, which are committed).
+
+> Use parameterized queries (`%s` placeholders + `sql_params`) — never string
+> interpolation — to avoid SQL injection.

neurocore_skill_postgres-0.1.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+neurocore_skill_postgres/__init__.py,sha256=LqqCjpEBYqFLlNmf5gIWEZ7_dfBsd3PfFp4vAApy61g,160
+neurocore_skill_postgres/skill.py,sha256=Fp9tF8Zd1-kZ2c_ugRDR5NDWGa2DwnkH99LOULA6MiY,2456
+neurocore_skill_postgres-0.1.0.dist-info/METADATA,sha256=uUEJAHWfSf9cm4DEy0kVAWEVvA564Oi2o2VCidpUXn0,1572
+neurocore_skill_postgres-0.1.0.dist-info/WHEEL,sha256=mffPy8wBnZQn2VnJUU5jE99KsxaSfiyMHV9Yt0aLVxs,87
+neurocore_skill_postgres-0.1.0.dist-info/entry_points.txt,sha256=umlQVsAYOny0WGAj6yYzC5lbhq4Km5VgLsRwJaPLKwc,69
+neurocore_skill_postgres-0.1.0.dist-info/RECORD,,

neurocore_skill_postgres-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.30.1
+Root-Is-Purelib: true
+Tag: py3-none-any

neurocore_skill_postgres-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[neurocore.skills]
+postgres = neurocore_skill_postgres:PostgresSkill