anywidget-vector 0.2.4__tar.gz → 0.2.5__tar.gz

{anywidget_vector-0.2.4 → anywidget_vector-0.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: anywidget-vector
-Version: 0.2.4
+Version: 0.2.5
 Summary: Interactive 3D vector visualization with query UI for vector databases
 Project-URL: Homepage, https://grafeo.dev/
 Project-URL: Repository, https://github.com/GrafeoDB/anywidget-vector

{anywidget_vector-0.2.4 → anywidget_vector-0.2.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "anywidget-vector"
-version = "0.2.4"
+version = "0.2.5"
 description = "Interactive 3D vector visualization with query UI for vector databases"
 readme = "README.md"
 license = "Apache-2.0"

{anywidget_vector-0.2.4 → anywidget_vector-0.2.5}/src/anywidget_vector/__init__.py

@@ -3,4 +3,4 @@
 from anywidget_vector.widget import VectorSpace
 
 __all__ = ["VectorSpace"]
-__version__ = "0.2.3"
+__version__ = "0.2.5"

{anywidget_vector-0.2.4 → anywidget_vector-0.2.5}/src/anywidget_vector/ui/canvas.js

@@ -100,14 +100,32 @@ export function createCanvas(model, container, callbacks) {
     setupRaycaster();
     setupTooltip();
     setupZoomControls();
-    setupModeControls();
     setupBoxSelection();
+    setupKeyboardShortcuts();
     createPoints();
     createConnections();
     fitToView();
     bindModelEvents();
   }
 
+  function setupKeyboardShortcuts() {
+    container.tabIndex = 0;
+    container.style.outline = "none";
+    container.addEventListener("keydown", (e) => {
+      const tag = e.target.tagName;
+      if (tag === "INPUT" || tag === "SELECT" || tag === "TEXTAREA") return;
+
+      if (e.key === "f" || e.key === "F") {
+        e.preventDefault();
+        fitToView();
+      } else if (e.key === "Escape") {
+        model.set("selected_points", []);
+        model.set("hovered_point", null);
+        model.save_changes();
+      }
+    });
+  }
+
   function setupAxesAndGrid() {
     while (axesGroup.children.length > 0) axesGroup.remove(axesGroup.children[0]);
     if (gridHelper) { scene.remove(gridHelper); gridHelper = null; }
@@ -471,27 +489,19 @@ export function createCanvas(model, container, callbacks) {
     zoomFitBtn.title = "Fit to view";
     zoomFitBtn.addEventListener("click", fitToView);
 
-    zoomControls.appendChild(zoomInBtn);
-    zoomControls.appendChild(zoomOutBtn);
-    zoomControls.appendChild(zoomFitBtn);
-    container.appendChild(zoomControls);
-  }
-
-  function setupModeControls() {
+    // Mode switcher (above zoom buttons, inside same column)
     const modeGroup = document.createElement("div");
     modeGroup.className = "avs-mode-group";
 
     const clickBtn = document.createElement("button");
-    clickBtn.className = "avs-mode-btn avs-mode-active";
+    clickBtn.className = "avs-zoom-btn avs-mode-btn avs-mode-active";
     clickBtn.innerHTML = ICONS.cursor;
     clickBtn.title = "Click / multi-select mode";
-    clickBtn.dataset.mode = "click";
 
     const boxBtn = document.createElement("button");
-    boxBtn.className = "avs-mode-btn";
+    boxBtn.className = "avs-zoom-btn avs-mode-btn";
     boxBtn.innerHTML = ICONS.boxSelect;
     boxBtn.title = "Box select mode";
-    boxBtn.dataset.mode = "box";
 
     function setActiveMode(mode) {
       currentMode = mode;
@@ -513,15 +523,20 @@ export function createCanvas(model, container, callbacks) {
     clickBtn.addEventListener("click", () => setActiveMode("click"));
     boxBtn.addEventListener("click", () => setActiveMode("box"));
 
-    modeGroup.appendChild(clickBtn);
-    modeGroup.appendChild(boxBtn);
-    container.appendChild(modeGroup);
-
     // Sync with model changes from Python side
     model.on("change:selection_mode", () => {
       const mode = model.get("selection_mode") || "click";
       if (mode !== currentMode) setActiveMode(mode);
     });
+
+    modeGroup.appendChild(clickBtn);
+    modeGroup.appendChild(boxBtn);
+
+    zoomControls.appendChild(modeGroup);
+    zoomControls.appendChild(zoomInBtn);
+    zoomControls.appendChild(zoomOutBtn);
+    zoomControls.appendChild(zoomFitBtn);
+    container.appendChild(zoomControls);
   }
 
   function setupBoxSelection() {

{anywidget_vector-0.2.4 → anywidget_vector-0.2.5}/src/anywidget_vector/ui/styles.css

@@ -47,7 +47,7 @@
   display: flex;
   flex-direction: column;
   gap: 2px;
-  z-index: 10;
+  z-index: 25;
 }
 
 .avs-zoom-btn {
@@ -87,37 +87,11 @@
 
 /* === Mode Switcher === */
 .avs-mode-group {
-  position: absolute;
-  bottom: 12px;
-  left: 12px;
   display: flex;
   gap: 2px;
-  z-index: 10;
-}
-
-.avs-mode-btn {
-  width: 32px;
-  height: 32px;
-  border: 1px solid var(--avs-border);
-  background: var(--avs-bg);
-  color: var(--avs-text);
-  cursor: pointer;
-  display: flex;
-  align-items: center;
-  justify-content: center;
-  border-radius: 4px;
-  transition: all 0.15s;
-  padding: 0;
-}
-
-.avs-mode-btn:hover {
-  background: var(--avs-surface);
-  border-color: var(--avs-text-muted);
-}
-
-.avs-mode-btn svg {
-  width: 16px;
-  height: 16px;
+  margin-bottom: 4px;
+  border-bottom: 1px solid rgba(255,255,255,0.15);
+  padding-bottom: 6px;
 }
 
 .avs-mode-btn.avs-mode-active {
@@ -159,6 +133,14 @@
   color: var(--avs-text-muted);
 }
 
+.avs-count-badge {
+  font-size: 11px;
+  color: var(--avs-text-muted);
+  white-space: nowrap;
+  flex-shrink: 0;
+  font-variant-numeric: tabular-nums;
+}
+
 .avs-filter-count {
   font-size: 11px;
   font-weight: 600;

{anywidget_vector-0.2.4 → anywidget_vector-0.2.5}/src/anywidget_vector/ui/toolbar.js

@@ -39,6 +39,17 @@ export function createToolbar(model, callbacks) {
   filterCount.style.display = "none";
   toolbar.appendChild(filterCount);
 
+  // Point count badge
+  const countBadge = document.createElement("span");
+  countBadge.className = "avs-count-badge";
+  function updatePointCount() {
+    const pts = model.get("points") || [];
+    countBadge.textContent = `${pts.length} points`;
+  }
+  updatePointCount();
+  model.on("change:points", updatePointCount);
+  toolbar.appendChild(countBadge);
+
   // Run button
   const runBtn = document.createElement("button");
   runBtn.className = "avs-btn avs-btn-primary";

{anywidget_vector-0.2.4 → anywidget_vector-0.2.5}/uv.lock

@@ -50,7 +50,7 @@ wheels = [
 
 [[package]]
 name = "anywidget-vector"
-version = "0.2.3"
+version = "0.2.5"
 source = { editable = "." }
 dependencies = [
     { name = "anywidget" },
@@ -109,30 +109,30 @@ requires-dist = [
     { name = "anywidget", specifier = ">=0.9.21" },
     { name = "chromadb", marker = "extra == 'all'", specifier = ">=1.5" },
     { name = "chromadb", marker = "extra == 'chroma'", specifier = ">=0.4" },
-    { name = "grafeo", marker = "extra == 'grafeo'", specifier = ">=0.5.0" },
+    { name = "grafeo", marker = "extra == 'grafeo'", specifier = ">=0.5" },
     { name = "lancedb", marker = "extra == 'all'", specifier = ">=0.29.2" },
     { name = "lancedb", marker = "extra == 'lancedb'", specifier = ">=0.1" },
-    { name = "marimo", marker = "extra == 'dev'", specifier = ">=0.19.9" },
+    { name = "marimo", marker = "extra == 'dev'", specifier = ">=0.20" },
     { name = "numpy", marker = "extra == 'all'", specifier = ">=2.4" },
     { name = "numpy", marker = "extra == 'numpy'", specifier = ">=1.24" },
-    { name = "pandas", marker = "extra == 'all'", specifier = ">=3.0" },
+    { name = "pandas", marker = "extra == 'all'", specifier = ">=3" },
     { name = "pandas", marker = "extra == 'pandas'", specifier = ">=2.0" },
     { name = "pinecone-client", marker = "extra == 'all'", specifier = ">=6.0" },
     { name = "pinecone-client", marker = "extra == 'pinecone'", specifier = ">=3.0" },
     { name = "prek", marker = "extra == 'dev'", specifier = ">=0.3.2" },
     { name = "pytest", marker = "extra == 'dev'", specifier = ">=9.0.2" },
-    { name = "qdrant-client", marker = "extra == 'all'", specifier = ">=1.0" },
+    { name = "qdrant-client", marker = "extra == 'all'", specifier = ">=1" },
     { name = "qdrant-client", marker = "extra == 'qdrant'", specifier = ">=1.0" },
-    { name = "ruff", marker = "extra == 'dev'", specifier = ">=0.15.0" },
-    { name = "ty", marker = "extra == 'dev'", specifier = ">=0.0.16" },
-    { name = "weaviate-client", marker = "extra == 'all'", specifier = ">=4.0" },
+    { name = "ruff", marker = "extra == 'dev'", specifier = ">=0.15" },
+    { name = "ty", marker = "extra == 'dev'", specifier = ">=0.0.20" },
+    { name = "weaviate-client", marker = "extra == 'all'", specifier = ">=4.20" },
     { name = "weaviate-client", marker = "extra == 'weaviate'", specifier = ">=4.0" },
 ]
 
 [package.metadata.requires-dev]
 dev = [
     { name = "pytest", specifier = ">=9.0.2" },
-    { name = "ruff", specifier = ">=0.14.14" },
+    { name = "ruff", specifier = ">=0.15" },
 ]
 
 [[package]]
@@ -2827,7 +2827,7 @@ wheels = [
 
 [[package]]
 name = "weaviate-client"
-version = "4.19.2"
+version = "4.20.1"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "authlib" },
@@ -2838,9 +2838,9 @@ dependencies = [
     { name = "pydantic" },
     { name = "validators" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/0b/57/f3201e5047d91aa9e7a06aa004b995bc8b05b2ea95dd178ace17eaaec16e/weaviate_client-4.19.2.tar.gz", hash = "sha256:99e76e912c95762436089cd5feedbfeea31e892aa13b6ad94729a2a54b316c45", size = 786448 }
+sdist = { url = "https://files.pythonhosted.org/packages/6a/0e/450732a620dec30e5c13d0e5ba6ea81cd2b168ffb5a1e6cfa034da2fd988/weaviate_client-4.20.1.tar.gz", hash = "sha256:7d3c4835292b17d54757c3f62921e87975c01f26869db787250cebef2a17bd80", size = 807802 }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/d5/09/636abb5ecfae2316b6584eaae7001a03abfe131f72d747fbec3419075524/weaviate_client-4.19.2-py3-none-any.whl", hash = "sha256:e78306d47c574c4035c87223e480bb77bd6e54142a21c4c58522dd43019fe493", size = 603652 },
+    { url = "https://files.pythonhosted.org/packages/de/3a/a2c74bfbdc5f8acbab8025205520448dcece82942206a9861ec48ccad03f/weaviate_client-4.20.1-py3-none-any.whl", hash = "sha256:6ca36bb8752c39589bfc856c1232cbcb627f69141e7a741b592e2516888f59d8", size = 618735 },
 ]
 
 [[package]]

anywidget_vector-0.2.4/src/anywidget_vector/traitlets.py

@@ -1,84 +0,0 @@
-"""Traitlet definitions for VectorSpace widget."""
-
-from __future__ import annotations
-
-import traitlets
-
-
-class VectorSpaceTraits(traitlets.HasTraits):
-    """All traitlets for the VectorSpace widget, grouped logically."""
-
-    # === Data ===
-    points = traitlets.List(trait=traitlets.Dict()).tag(sync=True)
-
-    # === Display ===
-    width = traitlets.Int(default_value=800).tag(sync=True)
-    height = traitlets.Int(default_value=600).tag(sync=True)
-    background = traitlets.Unicode(default_value="#1a1a2e").tag(sync=True)
-
-    # === Axes and Grid ===
-    show_axes = traitlets.Bool(default_value=True).tag(sync=True)
-    show_grid = traitlets.Bool(default_value=True).tag(sync=True)
-    axis_labels = traitlets.Dict(default_value={"x": "X", "y": "Y", "z": "Z"}).tag(sync=True)
-    grid_divisions = traitlets.Int(default_value=10).tag(sync=True)
-
-    # === Color Mapping ===
-    color_field = traitlets.Unicode(default_value=None, allow_none=True).tag(sync=True)
-    color_scale = traitlets.Unicode(default_value="viridis").tag(sync=True)
-    color_domain = traitlets.List(default_value=None, allow_none=True).tag(sync=True)
-
-    # === Size Mapping ===
-    size_field = traitlets.Unicode(default_value=None, allow_none=True).tag(sync=True)
-    size_range = traitlets.List(default_value=[0.02, 0.1]).tag(sync=True)
-
-    # === Shape Mapping ===
-    shape_field = traitlets.Unicode(default_value=None, allow_none=True).tag(sync=True)
-    shape_map = traitlets.Dict(default_value={}).tag(sync=True)
-
-    # === Camera ===
-    camera_position = traitlets.List(default_value=[2, 2, 2]).tag(sync=True)
-    camera_target = traitlets.List(default_value=[0, 0, 0]).tag(sync=True)
-
-    # === Interaction ===
-    selected_points = traitlets.List(default_value=[]).tag(sync=True)
-    hovered_point = traitlets.Dict(default_value=None, allow_none=True).tag(sync=True)
-    selection_mode = traitlets.Unicode(default_value="click").tag(sync=True)
-
-    # === Tooltip ===
-    show_tooltip = traitlets.Bool(default_value=True).tag(sync=True)
-    tooltip_fields = traitlets.List(default_value=["label", "x", "y", "z"]).tag(sync=True)
-
-    # === Performance ===
-    use_instancing = traitlets.Bool(default_value=True).tag(sync=True)
-    point_budget = traitlets.Int(default_value=100000).tag(sync=True)
-
-    # === Distance Metrics and Connections ===
-    distance_metric = traitlets.Unicode(default_value="euclidean").tag(sync=True)
-    show_connections = traitlets.Bool(default_value=False).tag(sync=True)
-    k_neighbors = traitlets.Int(default_value=0).tag(sync=True)
-    distance_threshold = traitlets.Float(default_value=None, allow_none=True).tag(sync=True)
-    reference_point = traitlets.Unicode(default_value=None, allow_none=True).tag(sync=True)
-    connection_color = traitlets.Unicode(default_value="#ffffff").tag(sync=True)
-    connection_opacity = traitlets.Float(default_value=0.3).tag(sync=True)
-
-    # === UI Visibility ===
-    show_toolbar = traitlets.Bool(default_value=False).tag(sync=True)
-    show_settings = traitlets.Bool(default_value=False).tag(sync=True)
-    show_properties = traitlets.Bool(default_value=False).tag(sync=True)
-
-    # === Backend Configuration ===
-    backend = traitlets.Unicode(default_value="qdrant").tag(sync=True)
-    backend_config = traitlets.Dict(default_value={}).tag(sync=True)
-    connection_status = traitlets.Unicode(default_value="disconnected").tag(sync=True)
-
-    # === Query Parameters ===
-    query_type = traitlets.Unicode(default_value="text").tag(sync=True)
-    query_input = traitlets.Unicode(default_value="").tag(sync=True)
-    query_limit = traitlets.Int(default_value=100).tag(sync=True)
-    query_error = traitlets.Unicode(default_value=None, allow_none=True).tag(sync=True)
-
-    # === Embedding Configuration ===
-    embedding_config = traitlets.Dict(default_value={}).tag(sync=True)
-
-    # === Internal Triggers ===
-    _execute_query = traitlets.Int(default_value=0).tag(sync=True)