@rcrsr/rill-ext-chroma 0.11.0 → 0.16.0

package/dist/index.d.ts

@@ -1,6 +1,6 @@
 // Generated by dts-bundle-generator v9.5.1
 
-import { ExtensionConfigSchema, ExtensionResult } from '@rcrsr/rill';
+import { ExtensionConfigSchema, ExtensionManifest, ExtensionResult } from '@rcrsr/rill';
 
 /**
  * Type definitions for ChromaDB extension.
@@ -88,7 +88,8 @@ export type ChromaExtensionConfig = ChromaConfig;
  * ```
  */
 export declare function createChromaExtension(config: ChromaConfig): ExtensionResult;
+export declare const VERSION: string;
 export declare const configSchema: ExtensionConfigSchema;
-export declare const CHROMA_EXTENSION_VERSION = "0.0.1";
+export declare const extensionManifest: ExtensionManifest;
 
 export {};

package/dist/index.js

@@ -1,9 +1,13 @@
+// src/index.ts
+import { createRequire } from "module";
+
 // src/factory.ts
 import { ChromaClient } from "chromadb";
 import {
   RuntimeError as RuntimeError6,
   emitExtensionEvent as emitExtensionEvent3,
-  createVector
+  createVector,
+  rillTypeToTypeValue
 } from "@rcrsr/rill";
 
 // ../../shared/ext-vector/dist/errors.js
@@ -178,13 +182,15 @@ var p = {
    * @param name - Parameter name (must be a valid identifier)
    * @param desc - Optional description
    * @param def - Optional default value
-   * @returns RillParam with type 'dict'
+   * @param fields - Optional structural field definitions (RillFieldDef with type and optional defaultValue)
+   * @returns RillParam with type 'dict' (with fields if provided)
    */
-  dict(name, desc, def) {
+  dict(name, desc, def, fields) {
     validateParamName2(name);
+    const type = fields !== void 0 ? { type: "dict", fields } : { type: "dict" };
     return {
       name,
-      type: { type: "dict" },
+      type,
       defaultValue: def,
       annotations: buildAnnotations(desc)
     };
@@ -247,9 +253,9 @@ function createChromaExtension(config) {
         const startTime = Date.now();
         try {
           checkDisposed(disposalState, "chroma");
-          const id = args[0];
-          const vector = args[1];
-          const metadata = args[2] ?? {};
+          const id = args["id"];
+          const vector = args["vector"];
+          const metadata = args["metadata"] ?? {};
           const collection = await client.getOrCreateCollection({
             name: factoryCollection
           });
@@ -282,8 +288,8 @@ function createChromaExtension(config) {
           throw rillError;
         }
       },
-      description: "Insert or update single vector with metadata",
-      returnType: { type: "dict" }
+      annotations: { description: "Insert or update single vector with metadata" },
+      returnType: rillTypeToTypeValue({ type: "dict", fields: { id: { type: { type: "string" } }, success: { type: { type: "bool" } } } })
     },
     // IR-2: chroma::upsert_batch
     upsert_batch: {
@@ -292,7 +298,7 @@ function createChromaExtension(config) {
         const startTime = Date.now();
         try {
           checkDisposed(disposalState, "chroma");
-          const items = args[0];
+          const items = args["items"];
           let succeeded = 0;
           const collection = await client.getOrCreateCollection({
             name: factoryCollection
@@ -367,21 +373,24 @@ function createChromaExtension(config) {
           throw rillError;
         }
       },
-      description: "Batch insert/update vectors",
-      returnType: { type: "dict" }
+      annotations: { description: "Batch insert/update vectors" },
+      returnType: rillTypeToTypeValue({ type: "dict", fields: { succeeded: { type: { type: "number" } }, failed: { type: { type: "string" } }, error: { type: { type: "string" } } } })
     },
     // IR-3: chroma::search
     search: {
       params: [
         vectorParam("vector"),
-        p.dict("options", void 0, {})
+        p.dict("options", void 0, {}, {
+          k: { type: { type: "number" }, defaultValue: 10 },
+          filter: { type: { type: "dict" }, defaultValue: {} }
+        })
       ],
       fn: async (args, ctx) => {
         const startTime = Date.now();
         try {
           checkDisposed(disposalState, "chroma");
-          const vector = args[0];
-          const options = args[1] ?? {};
+          const vector = args["vector"];
+          const options = args["options"] ?? {};
           const k = typeof options["k"] === "number" ? options["k"] : 10;
           const filter = options["filter"] ?? {};
           const collection = await client.getOrCreateCollection({
@@ -421,8 +430,8 @@ function createChromaExtension(config) {
           throw rillError;
         }
       },
-      description: "Search k nearest neighbors",
-      returnType: { type: "list" }
+      annotations: { description: "Search k nearest neighbors" },
+      returnType: rillTypeToTypeValue({ type: "list", element: { type: "dict", fields: { id: { type: { type: "string" } }, score: { type: { type: "number" } }, metadata: { type: { type: "dict" } } } } })
     },
     // IR-4: chroma::get
     get: {
@@ -431,7 +440,7 @@ function createChromaExtension(config) {
         const startTime = Date.now();
         try {
           checkDisposed(disposalState, "chroma");
-          const id = args[0];
+          const id = args["id"];
           const collection = await client.getOrCreateCollection({
             name: factoryCollection
           });
@@ -476,8 +485,8 @@ function createChromaExtension(config) {
           throw rillError;
         }
       },
-      description: "Fetch vector by ID",
-      returnType: { type: "dict" }
+      annotations: { description: "Fetch vector by ID" },
+      returnType: rillTypeToTypeValue({ type: "dict", fields: { id: { type: { type: "string" } }, vector: { type: { type: "vector" } }, metadata: { type: { type: "dict" } } } })
     },
     // IR-5: chroma::delete
     delete: {
@@ -486,7 +495,7 @@ function createChromaExtension(config) {
         const startTime = Date.now();
         try {
           checkDisposed(disposalState, "chroma");
-          const id = args[0];
+          const id = args["id"];
           const collection = await client.getOrCreateCollection({
             name: factoryCollection
           });
@@ -517,8 +526,8 @@ function createChromaExtension(config) {
           throw rillError;
         }
       },
-      description: "Delete vector by ID",
-      returnType: { type: "dict" }
+      annotations: { description: "Delete vector by ID" },
+      returnType: rillTypeToTypeValue({ type: "dict", fields: { id: { type: { type: "string" } }, deleted: { type: { type: "bool" } } } })
     },
     // IR-6: chroma::delete_batch
     delete_batch: {
@@ -527,7 +536,7 @@ function createChromaExtension(config) {
         const startTime = Date.now();
         try {
           checkDisposed(disposalState, "chroma");
-          const ids = args[0];
+          const ids = args["ids"];
           let succeeded = 0;
           const collection = await client.getOrCreateCollection({
             name: factoryCollection
@@ -579,8 +588,8 @@ function createChromaExtension(config) {
           throw rillError;
         }
       },
-      description: "Batch delete vectors",
-      returnType: { type: "dict" }
+      annotations: { description: "Batch delete vectors" },
+      returnType: rillTypeToTypeValue({ type: "dict", fields: { succeeded: { type: { type: "number" } }, failed: { type: { type: "string" } }, error: { type: { type: "string" } } } })
     },
     // IR-7: chroma::count
     count: {
@@ -613,21 +622,23 @@ function createChromaExtension(config) {
           throw rillError;
         }
       },
-      description: "Return total vector count in collection",
-      returnType: { type: "number" }
+      annotations: { description: "Return total vector count in collection" },
+      returnType: rillTypeToTypeValue({ type: "number" })
     },
     // IR-8: chroma::create_collection
     create_collection: {
       params: [
         p.str("name"),
-        p.dict("options", void 0, {})
+        p.dict("options", void 0, {}, {
+          metadata: { type: { type: "dict" }, defaultValue: {} }
+        })
       ],
       fn: async (args, ctx) => {
         const startTime = Date.now();
         try {
           checkDisposed(disposalState, "chroma");
-          const name = args[0];
-          const options = args[1] ?? {};
+          const name = args["name"];
+          const options = args["options"] ?? {};
           const metadata = options["metadata"] ?? {};
           await client.createCollection({
             name,
@@ -657,8 +668,8 @@ function createChromaExtension(config) {
           throw rillError;
         }
       },
-      description: "Create new vector collection",
-      returnType: { type: "dict" }
+      annotations: { description: "Create new vector collection" },
+      returnType: rillTypeToTypeValue({ type: "dict", fields: { name: { type: { type: "string" } }, created: { type: { type: "bool" } } } })
     },
     // IR-9: chroma::delete_collection
     delete_collection: {
@@ -667,7 +678,7 @@ function createChromaExtension(config) {
         const startTime = Date.now();
         try {
           checkDisposed(disposalState, "chroma");
-          const name = args[0];
+          const name = args["name"];
           await client.deleteCollection({ name });
           const result2 = {
             name,
@@ -693,8 +704,8 @@ function createChromaExtension(config) {
           throw rillError;
         }
       },
-      description: "Delete vector collection",
-      returnType: { type: "dict" }
+      annotations: { description: "Delete vector collection" },
+      returnType: rillTypeToTypeValue({ type: "dict", fields: { name: { type: { type: "string" } }, deleted: { type: { type: "bool" } } } })
     },
     // IR-10: chroma::list_collections
     list_collections: {
@@ -724,8 +735,8 @@ function createChromaExtension(config) {
           throw rillError;
         }
       },
-      description: "List all collection names",
-      returnType: { type: "list" }
+      annotations: { description: "List all collection names" },
+      returnType: rillTypeToTypeValue({ type: "list", element: { type: "string" } })
     },
     // IR-11: chroma::describe
     describe: {
@@ -762,8 +773,8 @@ function createChromaExtension(config) {
           throw rillError;
         }
       },
-      description: "Describe configured collection",
-      returnType: { type: "dict" }
+      annotations: { description: "Describe configured collection" },
+      returnType: rillTypeToTypeValue({ type: "dict", fields: { name: { type: { type: "string" } }, count: { type: { type: "number" } } } })
     }
   };
   result.dispose = async () => {
@@ -774,15 +785,23 @@ function createChromaExtension(config) {
 }
 
 // src/index.ts
+var _require = createRequire(import.meta.url);
+var _pkg = _require("../package.json");
+var VERSION = _pkg.version;
 var configSchema = {
   url: { type: "string" },
   collection: { type: "string", required: true },
   embeddingFunction: { type: "string" },
   timeout: { type: "number" }
 };
-var CHROMA_EXTENSION_VERSION = "0.0.1";
+var extensionManifest = {
+  factory: createChromaExtension,
+  configSchema,
+  version: VERSION
+};
 export {
-  CHROMA_EXTENSION_VERSION,
+  VERSION,
   configSchema,
-  createChromaExtension
+  createChromaExtension,
+  extensionManifest
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rcrsr/rill-ext-chroma",
-  "version": "0.11.0",
+  "version": "0.16.0",
   "description": "rill extension for ChromaDB vector database integration",
   "license": "MIT",
   "author": "Andre Bremer",
@@ -18,10 +18,10 @@
     "scripting"
   ],
   "peerDependencies": {
-    "@rcrsr/rill": "^0.11.0"
+    "@rcrsr/rill": "~0.16.0"
   },
   "devDependencies": {
-    "@rcrsr/rill": "^0.11.0",
+    "@rcrsr/rill": "~0.16.0",
     "@types/node": "^25.3.0",
     "dts-bundle-generator": "^9.5.1",
     "tsup": "^8.5.1",