ruvector-attention 0.2.0__py3-none-any.whl

ruvector_attention/__init__.py

@@ -0,0 +1,37 @@
+"""Validated Python convenience APIs backed by RuVector Rust."""
+import numpy as np
+
+_MAX_DIM = 65536
+
+def _integer(value, name, upper=_MAX_DIM):
+    if isinstance(value, bool) or not isinstance(value, int) or not 1 <= value <= upper:
+        raise ValueError(f"{name} must be an integer in [1, {upper}]")
+    return value
+
+def _array(value, ndim, name):
+    data = np.ascontiguousarray(value, dtype=np.float32)
+    if data.ndim != ndim or not data.size or data.shape[-1] > _MAX_DIM:
+        raise ValueError(f"{name} has an invalid or empty shape")
+    if not np.isfinite(data).all():
+        raise ValueError(f"{name} must be finite")
+    return data
+
+from dataclasses import dataclass
+from ruvector import AttentionReranker
+__all__ = ["AttentionResult", "attend", "AttentionReranker"]
+
+@dataclass(frozen=True)
+class AttentionResult:
+    blended: np.ndarray
+    weights: np.ndarray
+
+def attend(query, candidates) -> AttentionResult:
+    """Native scaled dot product attention; keys and values are candidates."""
+    q = _array(query, 1, "query")
+    data = _array(candidates, 2, "candidates")
+    if data.shape[1] != len(q):
+        raise ValueError("query and candidate dimensions must match")
+    blended, weights = AttentionReranker(len(q)).rerank(q, data)
+    if not np.isfinite(blended).all() or not np.isfinite(weights).all():
+        raise ValueError("attention overflow; rescale input")
+    return AttentionResult(blended, weights)

ruvector_attention-0.2.0.dist-info/METADATA

@@ -0,0 +1,56 @@
+Metadata-Version: 2.4
+Name: ruvector-attention
+Version: 0.2.0
+Summary: Native scaled dot product attention for embedding aggregation.
+Project-URL: Homepage, https://github.com/ruvnet/RuVector
+Project-URL: Documentation, https://github.com/ruvnet/RuVector/tree/main/python/portfolio/ruvector-attention
+Project-URL: Repository, https://github.com/ruvnet/RuVector
+Project-URL: Issues, https://github.com/ruvnet/RuVector/issues
+Project-URL: Changelog, https://github.com/ruvnet/RuVector/blob/main/python/portfolio/CHANGELOG.md
+Author: RuVector Team
+License-Expression: MIT
+License-File: LICENSE
+Keywords: ai,attention,embeddings,rust,ruvector
+Classifier: Development Status :: 3 - Alpha
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Rust
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.10
+Requires-Dist: numpy>=1.21
+Requires-Dist: ruvector==0.1.3
+Description-Content-Type: text/markdown
+
+# ruvector-attention
+
+![RuVector attention animated overview](https://raw.githubusercontent.com/ruvnet/RuVector/main/docs/python-portfolio/attention.svg)
+
+Native scaled dot product attention for embedding aggregation.
+
+## Getting started
+
+Requires Python 3.10 or later and a platform supported by the `ruvector==0.1.3` native wheel.
+
+```sh
+python -m pip install ruvector-attention==0.2.0
+```
+
+```python
+from ruvector_attention import attend
+result = attend([1, 0], [[1, 0], [0, 1]])
+assert result.weights[0] > result.weights[1]
+assert result.blended.shape == (2,)
+```
+
+## How it works
+
+Python sequences → finite contiguous float32 validation → shared RuVector native Rust runtime → typed Python or NumPy results. No Node.js process is used.
+
+## Scope and limits
+
+This exposes scaled dot product attention with candidates as both keys and values. It does not expose every npm attention mechanism or a trained cross encoder. Candidate ordering is equivalent to dot product.
+
+This is a Python convenience API backed by the shared native runtime, not complete npm API parity. Invalid dimensions and nonfinite inputs are rejected.
+
+## Documentation
+
+[Portfolio and architecture](https://github.com/ruvnet/RuVector/tree/main/python/portfolio) · [Source](https://github.com/ruvnet/RuVector/tree/main/python/portfolio/ruvector-attention) · [Issues](https://github.com/ruvnet/RuVector/issues)

ruvector_attention-0.2.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+ruvector_attention/__init__.py,sha256=V-zcce3e2isn_f0C8z5JKohLsKazjG7AzNFZUaYQKIg,1442
+ruvector_attention/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ruvector_attention-0.2.0.dist-info/METADATA,sha256=Of7FEud6d8sDrTLETOqXG6YoU93QDF-8CdG4M3JGQJE,2358
+ruvector_attention-0.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+ruvector_attention-0.2.0.dist-info/licenses/LICENSE,sha256=UP77k1YK9EITacmck8whTNg1sxgat_FjI_nm8t7YcOM,1060
+ruvector_attention-0.2.0.dist-info/RECORD,,

ruvector_attention-0.2.0.dist-info/WHEEL

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

ruvector_attention-0.2.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 rUv
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.