neurocore-skill-qdrant 0.1.0__tar.gz

neurocore_skill_qdrant-0.1.0/.gitignore

@@ -0,0 +1,152 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.python-version
+
+# pipenv
+Pipfile.lock
+
+# poetry
+poetry.lock
+
+# pdm
+.pdm.toml
+.pdm-python
+.pdm-build/
+
+# PEP 582
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# Ruff
+.ruff_cache/
+
+# IDEs and editors
+.idea/
+.vscode/
+*.swp
+*.swo
+*~
+
+# OS files
+.DS_Store
+Thumbs.db

neurocore_skill_qdrant-0.1.0/PKG-INFO

@@ -0,0 +1,47 @@
+Metadata-Version: 2.4
+Name: neurocore-skill-qdrant
+Version: 0.1.0
+Summary: Qdrant vector-search skill for NeuroCore
+Author: NeuroCore Contributors
+License-Expression: Apache-2.0
+Keywords: ai,neurocore,qdrant,rag,skill,vector
+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: qdrant-client>=1.10
+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-qdrant
+
+Vector similarity search for NeuroCore over [Qdrant](https://qdrant.tech).
+
+```bash
+pip install neurocore-skill-qdrant
+docker run -p 6333:6333 qdrant/qdrant
+```
+
+```yaml
+components:
+  - name: retrieve
+    type: qdrant
+    config:
+      collection: documents
+      top_k: 5
+flow:
+  type: sequential
+  steps:
+    - component: retrieve
+```
+
+Reads `query_vector` (a list of floats from your embedder); writes
+`qdrant_results` (a list of `{id, score, payload}`).

neurocore_skill_qdrant-0.1.0/README.md

@@ -0,0 +1,24 @@
+# neurocore-skill-qdrant
+
+Vector similarity search for NeuroCore over [Qdrant](https://qdrant.tech).
+
+```bash
+pip install neurocore-skill-qdrant
+docker run -p 6333:6333 qdrant/qdrant
+```
+
+```yaml
+components:
+  - name: retrieve
+    type: qdrant
+    config:
+      collection: documents
+      top_k: 5
+flow:
+  type: sequential
+  steps:
+    - component: retrieve
+```
+
+Reads `query_vector` (a list of floats from your embedder); writes
+`qdrant_results` (a list of `{id, score, payload}`).

neurocore_skill_qdrant-0.1.0/pyproject.toml

@@ -0,0 +1,39 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "neurocore-skill-qdrant"
+version = "0.1.0"
+description = "Qdrant vector-search skill for NeuroCore"
+readme = "README.md"
+license = "Apache-2.0"
+requires-python = ">=3.13"
+authors = [{ name = "NeuroCore Contributors" }]
+keywords = ["neurocore", "skill", "qdrant", "vector", "rag", "ai"]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: Apache Software License",
+    "Programming Language :: Python :: 3.13",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    "Typing :: Typed",
+]
+dependencies = [
+    "neurocore-ai>=0.2.0",
+    "qdrant-client>=1.10",
+]
+
+[project.optional-dependencies]
+dev = ["pytest>=9.0.2", "pytest-asyncio>=0.24", "ruff>=0.8", "mypy>=1.8"]
+
+[project.entry-points."neurocore.skills"]
+qdrant = "neurocore_skill_qdrant:QdrantSkill"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/neurocore_skill_qdrant"]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+pythonpath = ["src"]
+asyncio_mode = "auto"

neurocore_skill_qdrant-0.1.0/src/neurocore_skill_qdrant/__init__.py

@@ -0,0 +1,5 @@
+"""neurocore-skill-qdrant — Qdrant vector-search skill for NeuroCore."""
+
+from neurocore_skill_qdrant.skill import QdrantSkill
+
+__all__ = ["QdrantSkill"]

neurocore_skill_qdrant-0.1.0/src/neurocore_skill_qdrant/skill.py

@@ -0,0 +1,80 @@
+"""QdrantSkill — vector similarity search over a Qdrant collection.
+
+Reads a query vector from ``query_vector`` (list[float]) and writes hits to
+``qdrant_results``.
+"""
+from __future__ import annotations
+
+import asyncio
+import logging
+import os
+from typing import Any
+
+from flowengine import FlowContext
+
+from neurocore import AsyncSkill, SkillMeta
+
+logger = logging.getLogger(__name__)
+
+
+class QdrantSkill(AsyncSkill):
+    """Async vector search against Qdrant."""
+
+    skill_meta = SkillMeta(
+        name="qdrant",
+        version="0.1.0",
+        description="Vector similarity search over a Qdrant collection",
+        author="NeuroCore Contributors",
+        requires=["qdrant-client>=1.10"],
+        provides=["qdrant_results"],
+        consumes=["query_vector"],
+        tags=["vector", "rag", "retrieval"],
+        max_retries=2,
+        config_schema={
+            "required": ["collection"],
+            "properties": {
+                "url": {"type": "string", "description": "Qdrant URL."},
+                "api_key": {"type": "string"},
+                "collection": {"type": "string"},
+                "top_k": {"type": "integer"},
+            },
+        },
+    )
+
+    def _build_client(self) -> Any:
+        from qdrant_client import QdrantClient  # type: ignore[import-untyped]
+
+        url = self.config.get("url") or os.environ.get(
+            "QDRANT_URL", "http://localhost:6333"
+        )
+        api_key = self.config.get("api_key") or os.environ.get("QDRANT_API_KEY")
+        return QdrantClient(url=url, api_key=api_key)
+
+    async def _search(self, vector: list[float]) -> list[dict[str, Any]]:
+        """Run a similarity query and return hit dicts."""
+        client = self._build_client()
+        collection = self.config["collection"]
+        top_k = int(self.config.get("top_k", 5))
+
+        def _run() -> list[dict[str, Any]]:
+            hits = client.query_points(
+                collection_name=collection, query=vector, limit=top_k
+            ).points
+            return [
+                {"id": h.id, "score": h.score, "payload": h.payload} for h in hits
+            ]
+
+        return await asyncio.to_thread(_run)
+
+    async def process(self, context: FlowContext) -> FlowContext:
+        vector = context.get("query_vector")
+        if not vector:
+            logger.warning("QdrantSkill: 'query_vector' is empty.")
+            context.set("qdrant_results", [])
+            return context
+        try:
+            context.set("qdrant_results", await self._search(list(vector)))
+        except Exception as exc:  # noqa: BLE001
+            logger.error("QdrantSkill search failed: %s", exc, exc_info=True)
+            context.set("qdrant_results", [])
+        return context

neurocore_skill_qdrant-0.1.0/tests/test_skill.py

@@ -0,0 +1,39 @@
+"""Tests for QdrantSkill (search mocked via the _search seam)."""
+from __future__ import annotations
+
+from flowengine import FlowContext
+from neurocore_skill_qdrant import QdrantSkill
+
+
+def test_skill_meta():
+    meta = QdrantSkill.skill_meta
+    assert meta.name == "qdrant"
+    assert "qdrant_results" in meta.provides
+
+
+def test_validate_config_requires_collection():
+    skill = QdrantSkill()
+    skill.init({})
+    assert any("collection" in e for e in skill.validate_config())
+
+
+async def test_process_sets_results(monkeypatch):
+    skill = QdrantSkill()
+    skill.init({"collection": "docs"})
+
+    async def fake_search(vector):
+        assert vector == [0.1, 0.2]
+        return [{"id": 1, "score": 0.9, "payload": {"text": "hi"}}]
+
+    monkeypatch.setattr(skill, "_search", fake_search)
+    ctx = FlowContext()
+    ctx.set("query_vector", [0.1, 0.2])
+    out = await skill.process(ctx)
+    assert out.get("qdrant_results")[0]["score"] == 0.9
+
+
+async def test_empty_vector_returns_empty():
+    skill = QdrantSkill()
+    skill.init({"collection": "docs"})
+    out = await skill.process(FlowContext())
+    assert out.get("qdrant_results") == []