@xsai/embed 0.0.11 → 0.0.13

package/dist/index.d.ts

@@ -1,10 +1,8 @@
 import { CommonRequestOptions } from '@xsai/shared';
 
-type EmbedModel = 'all-minilm' | 'mxbai-embed-large' | 'nomic-embed-text' | ({} & string);
-interface EmbedOptions extends CommonRequestOptions<'embeddings'> {
+interface EmbedOptions extends CommonRequestOptions {
     [key: string]: unknown;
     input: string;
-    model: EmbedModel;
 }
 interface EmbedResponseUsage {
     prompt_tokens: number;
@@ -17,10 +15,9 @@ interface EmbedResult {
 }
 declare const embed: (options: EmbedOptions) => Promise<EmbedResult>;
 
-interface EmbedManyOptions extends CommonRequestOptions<'embeddings'> {
+interface EmbedManyOptions extends CommonRequestOptions {
     [key: string]: unknown;
     input: string[];
-    model: EmbedModel;
 }
 interface EmbedManyResult {
     embeddings: number[][];
@@ -29,4 +26,4 @@ interface EmbedManyResult {
 }
 declare const embedMany: (options: EmbedManyOptions) => Promise<EmbedManyResult>;
 
-export { type EmbedManyOptions, type EmbedManyResult, type EmbedModel, type EmbedOptions, type EmbedResponseUsage, type EmbedResult, embed as default, embed, embedMany };
+export { type EmbedManyOptions, type EmbedManyResult, type EmbedOptions, type EmbedResponseUsage, type EmbedResult, embed as default, embed, embedMany };

package/dist/index.js

@@ -1,17 +1,11 @@
-import { requestUrl, clean } from '@xsai/shared';
+import { requestBody, requestHeaders } from '@xsai/shared';
 
-const embed = async (options) => await fetch(requestUrl(options.path ?? "embeddings", options.base), {
-  body: JSON.stringify(clean({
-    ...options,
-    abortSignal: void 0,
-    base: void 0,
-    headers: void 0,
-    path: void 0
-  })),
-  headers: {
+const embed = async (options) => await fetch(options.url, {
+  body: requestBody(options),
+  headers: requestHeaders({
     "Content-Type": "application/json",
     ...options.headers
-  },
+  }, options.apiKey),
   method: "POST",
   signal: options.abortSignal
 }).then((res) => res.json()).then(({ data, usage }) => ({
@@ -20,18 +14,12 @@ const embed = async (options) => await fetch(requestUrl(options.path ?? "embeddi
   usage
 }));
 
-const embedMany = async (options) => await fetch(requestUrl(options.path ?? "embeddings", options.base), {
-  body: JSON.stringify(clean({
-    ...options,
-    abortSignal: void 0,
-    base: void 0,
-    headers: void 0,
-    path: void 0
-  })),
-  headers: {
+const embedMany = async (options) => await fetch(options.url, {
+  body: requestBody(options),
+  headers: requestHeaders({
     "Content-Type": "application/json",
     ...options.headers
-  },
+  }, options.apiKey),
   method: "POST",
   signal: options.abortSignal
 }).then((res) => res.json()).then(({ data, usage }) => ({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@xsai/embed",
-  "version": "0.0.11",
+  "version": "0.0.13",
   "type": "module",
   "author": "Moeru AI",
   "license": "MIT",
@@ -25,6 +25,9 @@
   "dependencies": {
     "@xsai/shared": ""
   },
+  "devDependencies": {
+    "@xsai/providers": ""
+  },
   "scripts": {
     "build": "pkgroll",
     "build:watch": "pkgroll --watch",