@rcrsr/rill-ext-chroma 0.9.0 → 0.11.0

package/dist/index.js

@@ -1,7 +1,7 @@
 // src/factory.ts
 import { ChromaClient } from "chromadb";
 import {
-  RuntimeError as RuntimeError4,
+  RuntimeError as RuntimeError6,
   emitExtensionEvent as emitExtensionEvent3,
   createVector
 } from "@rcrsr/rill";
@@ -87,6 +87,144 @@ function assertRequired(value, fieldName) {
 // ../../shared/ext-vector/dist/wrapper.js
 import { emitExtensionEvent as emitExtensionEvent2 } from "@rcrsr/rill";
 
+// ../../shared/ext-vector/dist/param.js
+import { RuntimeError as RuntimeError4 } from "@rcrsr/rill";
+function validateParamName(name) {
+  if (name === "")
+    throw new RuntimeError4("RILL-R001", "param name must not be empty");
+  if (/\s/.test(name))
+    throw new RuntimeError4("RILL-R001", "param name must be a valid identifier");
+}
+function vectorParam(name) {
+  validateParamName(name);
+  return {
+    name,
+    type: { type: "vector" },
+    defaultValue: void 0,
+    annotations: {}
+  };
+}
+
+// ../../shared/ext-param/dist/param.js
+import { RuntimeError as RuntimeError5 } from "@rcrsr/rill";
+function validateParamName2(name) {
+  if (name === "") {
+    throw new RuntimeError5("RILL-R001", "param name must not be empty");
+  }
+  if (/\s/.test(name)) {
+    throw new RuntimeError5("RILL-R001", "param name must be a valid identifier");
+  }
+}
+function buildAnnotations(desc) {
+  if (desc !== void 0) {
+    return { description: desc };
+  }
+  return {};
+}
+var p = {
+  /**
+   * IR-1: Creates a string parameter descriptor.
+   *
+   * @param name - Parameter name (must be a valid identifier)
+   * @param desc - Optional description
+   * @returns RillParam with type 'string'
+   */
+  str(name, desc) {
+    validateParamName2(name);
+    return {
+      name,
+      type: { type: "string" },
+      defaultValue: void 0,
+      annotations: buildAnnotations(desc)
+    };
+  },
+  /**
+   * IR-2: Creates a number parameter descriptor.
+   *
+   * @param name - Parameter name (must be a valid identifier)
+   * @param desc - Optional description
+   * @param def - Optional default value
+   * @returns RillParam with type 'number'
+   */
+  num(name, desc, def) {
+    validateParamName2(name);
+    return {
+      name,
+      type: { type: "number" },
+      defaultValue: def,
+      annotations: buildAnnotations(desc)
+    };
+  },
+  /**
+   * IR-3: Creates a boolean parameter descriptor.
+   *
+   * @param name - Parameter name (must be a valid identifier)
+   * @param desc - Optional description
+   * @param def - Optional default value
+   * @returns RillParam with type 'bool'
+   */
+  bool(name, desc, def) {
+    validateParamName2(name);
+    return {
+      name,
+      type: { type: "bool" },
+      defaultValue: def,
+      annotations: buildAnnotations(desc)
+    };
+  },
+  /**
+   * IR-4: Creates a dict parameter descriptor.
+   *
+   * @param name - Parameter name (must be a valid identifier)
+   * @param desc - Optional description
+   * @param def - Optional default value
+   * @returns RillParam with type 'dict'
+   */
+  dict(name, desc, def) {
+    validateParamName2(name);
+    return {
+      name,
+      type: { type: "dict" },
+      defaultValue: def,
+      annotations: buildAnnotations(desc)
+    };
+  },
+  /**
+   * IR-5: Creates a list parameter descriptor.
+   *
+   * @param name - Parameter name (must be a valid identifier)
+   * @param itemType - Optional element type; omitted when not provided
+   * @param desc - Optional description
+   * @returns RillParam with type 'list' (with element if itemType provided)
+   */
+  list(name, itemType, desc) {
+    validateParamName2(name);
+    const type = itemType !== void 0 ? { type: "list", element: itemType } : { type: "list" };
+    return {
+      name,
+      type,
+      defaultValue: void 0,
+      annotations: buildAnnotations(desc)
+    };
+  },
+  /**
+   * IR-6: Creates a callable parameter descriptor.
+   *
+   * @param name - Parameter name (must be a valid identifier)
+   * @param desc - Optional description
+   * @returns RillParam with type 'closure'
+   */
+  callable(name, desc) {
+    validateParamName2(name);
+    return {
+      name,
+      type: { type: "closure" },
+      defaultValue: void 0,
+      annotations: buildAnnotations(desc)
+    };
+  }
+};
+
 // src/factory.ts
 function createChromaExtension(config) {
   assertRequired(config.collection, "collection");
@@ -101,9 +239,9 @@ function createChromaExtension(config) {
     // IR-1: chroma::upsert
     upsert: {
       params: [
-        { name: "id", type: "string" },
-        { name: "vector", type: "vector" },
-        { name: "metadata", type: "dict", defaultValue: {} }
+        p.str("id"),
+        vectorParam("vector"),
+        p.dict("metadata", void 0, {})
       ],
       fn: async (args, ctx) => {
         const startTime = Date.now();
@@ -145,11 +283,11 @@ function createChromaExtension(config) {
         }
       },
       description: "Insert or update single vector with metadata",
-      returnType: "dict"
+      returnType: { type: "dict" }
     },
     // IR-2: chroma::upsert_batch
     upsert_batch: {
-      params: [{ name: "items", type: "list" }],
+      params: [p.list("items")],
       fn: async (args, ctx) => {
         const startTime = Date.now();
         try {
@@ -230,13 +368,13 @@ function createChromaExtension(config) {
         }
       },
       description: "Batch insert/update vectors",
-      returnType: "dict"
+      returnType: { type: "dict" }
     },
     // IR-3: chroma::search
     search: {
       params: [
-        { name: "vector", type: "vector" },
-        { name: "options", type: "dict", defaultValue: {} }
+        vectorParam("vector"),
+        p.dict("options", void 0, {})
       ],
       fn: async (args, ctx) => {
         const startTime = Date.now();
@@ -284,11 +422,11 @@ function createChromaExtension(config) {
         }
       },
       description: "Search k nearest neighbors",
-      returnType: "list"
+      returnType: { type: "list" }
     },
     // IR-4: chroma::get
     get: {
-      params: [{ name: "id", type: "string" }],
+      params: [p.str("id")],
       fn: async (args, ctx) => {
         const startTime = Date.now();
         try {
@@ -301,12 +439,12 @@ function createChromaExtension(config) {
             ids: [id]
           });
           if (response.ids.length === 0) {
-            throw new RuntimeError4("RILL-R004", "chroma: id not found");
+            throw new RuntimeError6("RILL-R004", "chroma: id not found");
           }
           const embedding = response.embeddings?.[0];
           const metadata = response.metadatas?.[0];
           if (!embedding || !Array.isArray(embedding)) {
-            throw new RuntimeError4(
+            throw new RuntimeError6(
               "RILL-R004",
               "chroma: invalid vector format"
             );
@@ -339,11 +477,11 @@ function createChromaExtension(config) {
         }
       },
       description: "Fetch vector by ID",
-      returnType: "dict"
+      returnType: { type: "dict" }
     },
     // IR-5: chroma::delete
     delete: {
-      params: [{ name: "id", type: "string" }],
+      params: [p.str("id")],
       fn: async (args, ctx) => {
         const startTime = Date.now();
         try {
@@ -380,11 +518,11 @@ function createChromaExtension(config) {
         }
       },
       description: "Delete vector by ID",
-      returnType: "dict"
+      returnType: { type: "dict" }
     },
     // IR-6: chroma::delete_batch
     delete_batch: {
-      params: [{ name: "ids", type: "list" }],
+      params: [p.list("ids")],
       fn: async (args, ctx) => {
         const startTime = Date.now();
         try {
@@ -442,7 +580,7 @@ function createChromaExtension(config) {
         }
       },
       description: "Batch delete vectors",
-      returnType: "dict"
+      returnType: { type: "dict" }
     },
     // IR-7: chroma::count
     count: {
@@ -476,13 +614,13 @@ function createChromaExtension(config) {
         }
       },
       description: "Return total vector count in collection",
-      returnType: "number"
+      returnType: { type: "number" }
     },
     // IR-8: chroma::create_collection
     create_collection: {
       params: [
-        { name: "name", type: "string" },
-        { name: "options", type: "dict", defaultValue: {} }
+        p.str("name"),
+        p.dict("options", void 0, {})
       ],
       fn: async (args, ctx) => {
         const startTime = Date.now();
@@ -520,11 +658,11 @@ function createChromaExtension(config) {
         }
       },
       description: "Create new vector collection",
-      returnType: "dict"
+      returnType: { type: "dict" }
     },
     // IR-9: chroma::delete_collection
     delete_collection: {
-      params: [{ name: "name", type: "string" }],
+      params: [p.str("name")],
       fn: async (args, ctx) => {
         const startTime = Date.now();
         try {
@@ -556,7 +694,7 @@ function createChromaExtension(config) {
         }
       },
       description: "Delete vector collection",
-      returnType: "dict"
+      returnType: { type: "dict" }
     },
     // IR-10: chroma::list_collections
     list_collections: {
@@ -587,7 +725,7 @@ function createChromaExtension(config) {
         }
       },
       description: "List all collection names",
-      returnType: "list"
+      returnType: { type: "list" }
     },
     // IR-11: chroma::describe
     describe: {
@@ -625,7 +763,7 @@ function createChromaExtension(config) {
         }
       },
       description: "Describe configured collection",
-      returnType: "dict"
+      returnType: { type: "dict" }
     }
   };
   result.dispose = async () => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rcrsr/rill-ext-chroma",
-  "version": "0.9.0",
+  "version": "0.11.0",
   "description": "rill extension for ChromaDB vector database integration",
   "license": "MIT",
   "author": "Andre Bremer",
@@ -18,10 +18,10 @@
     "scripting"
   ],
   "peerDependencies": {
-    "@rcrsr/rill": "^0.9.0"
+    "@rcrsr/rill": "^0.11.0"
   },
   "devDependencies": {
-    "@rcrsr/rill": "^0.9.0",
+    "@rcrsr/rill": "^0.11.0",
     "@types/node": "^25.3.0",
     "dts-bundle-generator": "^9.5.1",
     "tsup": "^8.5.1",
@@ -44,7 +44,8 @@
     "access": "public"
   },
   "dependencies": {
-    "chromadb": "^3.3.1"
+    "chromadb": "^3.3.1",
+    "@rcrsr/rill-ext-param-shared": "^0.0.1"
   },
   "scripts": {
     "build": "tsup && dts-bundle-generator --config dts-bundle-generator.config.cjs",