@xsai/embed 0.0.7 → 0.0.8

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,7 +1,7 @@
 import { requestUrl, clean } from '@xsai/shared';
 
 const embed = async (options) => {
-  const request = new Request(requestUrl(options.path ?? "embeddings", options.base), {
+  const { data, usage } = await fetch(requestUrl(options.path ?? "embeddings", options.base), {
     body: JSON.stringify(clean({
       ...options,
       abortSignal: void 0,
@@ -15,15 +15,35 @@ const embed = async (options) => {
     },
     method: "POST",
     signal: options.abortSingal
-  });
-  const response = await fetch(request);
-  const json = await response.json();
+  }).then((res) => res.json());
   return {
-    embedding: json.data[0].embedding,
-    request,
-    response,
-    usage: json.usage
+    embedding: data[0].embedding,
+    input: options.input,
+    usage
   };
 };
 
-export { embed as default, embed };
+const embedMany = async (options) => {
+  const { data, usage } = 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.abortSingal
+  }).then((res) => res.json());
+  return {
+    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.8",
   "type": "module",
   "author": "藍+85CD",
   "license": "MIT",
@@ -23,7 +23,7 @@
     "dist"
   ],
   "dependencies": {
-    "@xsai/shared": "^0.0.7"
+    "@xsai/shared": "^0.0.8"
   },
   "scripts": {
     "build": "pkgroll",