@blocklet/search-kit-js 0.2.6 → 0.2.8

package/dist/index.cjs

@@ -2,6 +2,17 @@
 
 const meilisearch = require('meilisearch');
 const component = require('@blocklet/sdk/lib/component');
+const getWallet = require('@blocklet/sdk/lib/wallet');
+const env = require('@blocklet/sdk/lib/env');
+const ufo = require('ufo');
+const sdk = require('@blocklet/sdk');
+const crypto = require('crypto');
+
+function _interopDefaultCompat (e) { return e && typeof e === 'object' && 'default' in e ? e.default : e; }
+
+const getWallet__default = /*#__PURE__*/_interopDefaultCompat(getWallet);
+const env__default = /*#__PURE__*/_interopDefaultCompat(env);
+const crypto__default = /*#__PURE__*/_interopDefaultCompat(crypto);
 
 const SEARCH_KIT_DID = "z8iZorY6mvb5tZrxXTqhBmwu89xjEEazrgT3t";
 class SearchKitClient extends meilisearch.MeiliSearch {
@@ -13,7 +24,31 @@ class SearchKitClient extends meilisearch.MeiliSearch {
   }
 }
 
+const hash = (str, salt) => crypto__default.createHash("sha256").update(str + salt).digest("hex");
+
+const wallet = getWallet__default();
+const resolveRestEmbedders = ({ documentTemplate, ...rest }) => {
+  const mountPoint = sdk.getComponentMountPoint("ai-kit");
+  if (!wallet.secretKey || !mountPoint) {
+    throw new Error("wallet.secretKey or ai-kit blocklet is not found");
+  }
+  return {
+    default: {
+      source: "rest",
+      url: ufo.joinURL(env__default.appUrl, mountPoint, "/api/meilisearch/embeddings"),
+      dimensions: 1536,
+      apiKey: hash(wallet.secretKey, ":/ai-kit/api/meilisearch/embeddings"),
+      request: { data: { text: "{{text}}" } },
+      response: { data: { embedding: "{{embedding}}" } },
+      distribution: { mean: 0.7, sigma: 0.3 },
+      documentTemplate,
+      ...rest
+    }
+  };
+};
+
 exports.SearchKitClient = SearchKitClient;
+exports.resolveRestEmbedders = resolveRestEmbedders;
 Object.keys(meilisearch).forEach(function (k) {
   if (k !== 'default' && !Object.prototype.hasOwnProperty.call(exports, k)) exports[k] = meilisearch[k];
 });

package/dist/index.d.cts

@@ -1,8 +1,13 @@
-import { MeiliSearch, Config } from 'meilisearch';
+import { MeiliSearch, Config, Embedders, RestEmbedder } from 'meilisearch';
 export * from 'meilisearch';
 
 declare class SearchKitClient extends MeiliSearch {
-    constructor(config: Omit<Config, 'host' | 'apiKey'>);
+    constructor(config?: Omit<Config, 'host' | 'apiKey'>);
 }
 
-export { SearchKitClient };
+type PartialRestEmbedder = Partial<RestEmbedder> & {
+    documentTemplate: string;
+};
+declare const resolveRestEmbedders: ({ documentTemplate, ...rest }: PartialRestEmbedder) => Embedders;
+
+export { SearchKitClient, resolveRestEmbedders };

package/dist/index.d.mts

@@ -1,8 +1,13 @@
-import { MeiliSearch, Config } from 'meilisearch';
+import { MeiliSearch, Config, Embedders, RestEmbedder } from 'meilisearch';
 export * from 'meilisearch';
 
 declare class SearchKitClient extends MeiliSearch {
-    constructor(config: Omit<Config, 'host' | 'apiKey'>);
+    constructor(config?: Omit<Config, 'host' | 'apiKey'>);
 }
 
-export { SearchKitClient };
+type PartialRestEmbedder = Partial<RestEmbedder> & {
+    documentTemplate: string;
+};
+declare const resolveRestEmbedders: ({ documentTemplate, ...rest }: PartialRestEmbedder) => Embedders;
+
+export { SearchKitClient, resolveRestEmbedders };

package/dist/index.d.ts

@@ -1,8 +1,13 @@
-import { MeiliSearch, Config } from 'meilisearch';
+import { MeiliSearch, Config, Embedders, RestEmbedder } from 'meilisearch';
 export * from 'meilisearch';
 
 declare class SearchKitClient extends MeiliSearch {
-    constructor(config: Omit<Config, 'host' | 'apiKey'>);
+    constructor(config?: Omit<Config, 'host' | 'apiKey'>);
 }
 
-export { SearchKitClient };
+type PartialRestEmbedder = Partial<RestEmbedder> & {
+    documentTemplate: string;
+};
+declare const resolveRestEmbedders: ({ documentTemplate, ...rest }: PartialRestEmbedder) => Embedders;
+
+export { SearchKitClient, resolveRestEmbedders };

package/dist/index.mjs

@@ -1,6 +1,11 @@
 import { MeiliSearch } from 'meilisearch';
 export * from 'meilisearch';
 import { getComponentWebEndpoint } from '@blocklet/sdk/lib/component';
+import getWallet from '@blocklet/sdk/lib/wallet';
+import env from '@blocklet/sdk/lib/env';
+import { joinURL } from 'ufo';
+import { getComponentMountPoint } from '@blocklet/sdk';
+import crypto from 'crypto';
 
 const SEARCH_KIT_DID = "z8iZorY6mvb5tZrxXTqhBmwu89xjEEazrgT3t";
 class SearchKitClient extends MeiliSearch {
@@ -12,4 +17,27 @@ class SearchKitClient extends MeiliSearch {
   }
 }
 
-export { SearchKitClient };
+const hash = (str, salt) => crypto.createHash("sha256").update(str + salt).digest("hex");
+
+const wallet = getWallet();
+const resolveRestEmbedders = ({ documentTemplate, ...rest }) => {
+  const mountPoint = getComponentMountPoint("ai-kit");
+  if (!wallet.secretKey || !mountPoint) {
+    throw new Error("wallet.secretKey or ai-kit blocklet is not found");
+  }
+  return {
+    default: {
+      source: "rest",
+      url: joinURL(env.appUrl, mountPoint, "/api/meilisearch/embeddings"),
+      dimensions: 1536,
+      apiKey: hash(wallet.secretKey, ":/ai-kit/api/meilisearch/embeddings"),
+      request: { data: { text: "{{text}}" } },
+      response: { data: { embedding: "{{embedding}}" } },
+      distribution: { mean: 0.7, sigma: 0.3 },
+      documentTemplate,
+      ...rest
+    }
+  };
+};
+
+export { SearchKitClient, resolveRestEmbedders };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blocklet/search-kit-js",
-  "version": "0.2.6",
+  "version": "0.2.8",
   "author": "Arcblock <blocklet@arcblock.io> https://github.com/blocklet",
   "main": "./dist/index.cjs",
   "module": "./dist/index.mjs",
@@ -19,8 +19,9 @@
     "README.md"
   ],
   "dependencies": {
-    "@blocklet/sdk": "^1.16.28",
-    "meilisearch": "^0.41.0"
+    "@blocklet/sdk": "^1.16.31",
+    "meilisearch": "^0.42.0",
+    "ufo": "^1.5.4"
   },
   "scripts": {
     "build": "unbuild",