@xsai/embed 0.0.7 → 0.0.9

package/dist/index.d.ts

@@ -3,35 +3,30 @@ import { CommonRequestOptions } from '@xsai/shared';
 type EmbedModel = 'all-minilm' | 'mxbai-embed-large' | 'nomic-embed-text' | ({} & string);
 interface EmbedOptions extends CommonRequestOptions<'embeddings'> {
     [key: string]: unknown;
-    input: string | string[];
+    input: string;
     model: EmbedModel;
 }
-interface EmbedResponse {
-    data: {
-        embedding: number[];
-        index: number;
-        object: 'embedding';
-    }[];
-    model: string;
-    object: 'list' | ({} & string);
-    system_fingerprint?: string;
-    usage: EmbedResponseUsage;
-}
 interface EmbedResponseUsage {
     prompt_tokens: number;
     total_tokens: number;
 }
 interface EmbedResult {
     embedding: number[];
-    request: Request;
-    response: Response;
+    input: string;
     usage: EmbedResponseUsage;
 }
-declare const embed: (options: EmbedOptions) => Promise<{
-    embedding: number[];
-    request: Request;
-    response: Response;
+declare const embed: (options: EmbedOptions) => Promise<EmbedResult>;
+
+interface EmbedManyOptions extends CommonRequestOptions<'embeddings'> {
+    [key: string]: unknown;
+    input: string[];
+    model: EmbedModel;
+}
+interface EmbedManyResult {
+    embeddings: number[][];
+    input: string[];
     usage: EmbedResponseUsage;
-}>;
+}
+declare const embedMany: (options: EmbedManyOptions) => Promise<EmbedManyResult>;
 
-export { type EmbedModel, type EmbedOptions, type EmbedResponse, type EmbedResponseUsage, type EmbedResult, embed as default, embed };
+export { type EmbedManyOptions, type EmbedManyResult, type EmbedModel, type EmbedOptions, type EmbedResponseUsage, type EmbedResult, embed as default, embed, embedMany };

package/dist/index.js

@@ -1,29 +1,43 @@
 import { requestUrl, clean } from '@xsai/shared';
 
-const embed = async (options) => {
-  const request = new Request(requestUrl(options.path ?? "embeddings", options.base), {
-    body: JSON.stringify(clean({
-      ...options,
-      abortSignal: void 0,
-      base: void 0,
-      headers: void 0,
-      path: void 0
-    })),
-    headers: {
-      "Content-Type": "application/json",
-      ...options.headers
-    },
-    method: "POST",
-    signal: options.abortSingal
-  });
-  const response = await fetch(request);
-  const json = await response.json();
-  return {
-    embedding: json.data[0].embedding,
-    request,
-    response,
-    usage: json.usage
-  };
-};
+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: {
+    "Content-Type": "application/json",
+    ...options.headers
+  },
+  method: "POST",
+  signal: options.abortSignal
+}).then((res) => res.json()).then(({ data, usage }) => ({
+  embedding: data[0].embedding,
+  input: options.input,
+  usage
+}));
 
-export { embed as default, embed };
+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: {
+    "Content-Type": "application/json",
+    ...options.headers
+  },
+  method: "POST",
+  signal: options.abortSignal
+}).then((res) => res.json()).then(({ data, usage }) => ({
+  embeddings: data.map((data2) => data2.embedding),
+  input: options.input,
+  usage
+}));
+
+export { embed as default, embed, embedMany };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@xsai/embed",
-  "version": "0.0.7",
+  "version": "0.0.9",
   "type": "module",
   "author": "藍+85CD",
   "license": "MIT",
@@ -23,7 +23,7 @@
     "dist"
   ],
   "dependencies": {
-    "@xsai/shared": "^0.0.7"
+    "@xsai/shared": "^0.0.9"
   },
   "scripts": {
     "build": "pkgroll",