adds build

packages/plugin-embeddings/README.md

@@ -16,6 +16,7 @@ npm i @orama/plugin-embeddings
 - `@tensorflow/tfjs-node`
 - `@tensorflow/tfjs-backend-webgl`
 - `@tensorflow/tfjs-backend-cpu`
+- `@tensorflow/tfjs-node-gpu`
 - `@tensorflow/tfjs-backend-wasm`
 
 For example, if you're running Orama on the browser, we highly recommend using `@tensorflow/tfjs-backend-webgl`:

packages/plugin-embeddings/package.json

@@ -41,12 +41,12 @@
  "access": "public"
  },
  "devDependencies": {
- "@orama/orama": "workspace:*",
  "@types/node": "^20.9.0",
  "tsup": "^7.2.0",
  "typescript": "^5.0.0"
  },
  "dependencies": {
+ "@orama/orama": "workspace:*",
  "@tensorflow-models/universal-sentence-encoder": "^1.3.3",
  "@tensorflow/tfjs": "^4.21.0",
  "@tensorflow/tfjs-backend-cpu": "^4.21.0",

packages/plugin-embeddings/src/index.ts

@@ -1,5 +1,5 @@
 import type { AnyOrama, SearchParams, TypedDocument, OramaPluginAsync, PartialSchemaDeep } from '@orama/orama'
-import use from '@tensorflow-models/universal-sentence-encoder'
+import { load as loadModel } from '@tensorflow-models/universal-sentence-encoder'
 
 export type PluginEmbeddingsParams = {
  embeddings: {
@@ -28,7 +28,7 @@ function getPropertiesValues(schema: object, properties: string[]) {
 export const embeddingsType = 'vector[512]'
 
 export async function pluginEmbeddings(pluginParams: PluginEmbeddingsParams): OramaPluginAsync {
- const model = await use.load()
+ const model = await loadModel()
 
  return {
  name: 'orama-plugin-embeddings',

packages/plugin-embeddings/tsconfig.json

@@ -0,0 +1,20 @@
+{
+ "compilerOptions": {
+ "allowJs": true,
+ "target": "ES5",
+ "module": "NodeNext",
+ "outDir": "dist",
+ "jsx": "react",
+ "noImplicitAny": false,
+ "lib": ["ESNext", "DOM", "DOM.Iterable"],
+ "esModuleInterop": true,
+ "declaration": true,
+ "forceConsistentCasingInFileNames": true,
+ "strict": true,
+ "skipLibCheck": true,
+ "resolveJsonModule": true,
+ "sourceMap": true,
+ "moduleResolution": "nodenext"
+ },
+ "include": ["src/*.ts", "src/**/*.ts", "src/*.tsx", "src/**/*.tsx"]
+}

packages/plugin-embeddings/tsup.lib.js

@@ -0,0 +1,17 @@
+import { defineConfig } from 'tsup'
+
+const entry = new URL('src/index.ts', import.meta.url).pathname
+const outDir = new URL('dist', import.meta.url).pathname
+
+export default defineConfig({
+ entry: [entry],
+ splitting: false,
+ sourcemap: true,
+ minify: true,
+ format: ['cjs', 'esm', 'iife'],
+ globalName: 'pluginEmbeddings',
+ dts: true,
+ clean: true,
+ bundle: true,
+ outDir
+})

turbo.json

@@ -46,6 +46,10 @@
  "dependsOn": ["@orama/orama#build"],
  "outputs": ["dist/**"]
  },
+ "@orama/plugin-embeddings#build": {
+ "dependsOn": ["@orama/orama#build"],
+ "outputs": ["dist/**"]
+ },
  "@orama/tokenizers#build": {
  "dependsOn": ["@orama/orama#build"]
  },