✨ Add metadatas to embeddings

polyfact · Jul 19, 2024 · 04f450a · 04f450a
1 parent d5019a5
commit 04f450a
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/embeddings.ts b/lib/embeddings.ts
@@ -92,7 +92,7 @@ async function searchEmbeddings(
     id: string,
     input: string,
     clientOptions: InputClientOptions = {},
-): Promise<{ id: string; content: string; similarity: number }[]> {
+): Promise<{ id: string; content: string; similarity: number; metadatas: Record<string, any> }[]> {
     const { token, endpoint } = await defaultOptions(clientOptions);
 
     try {
@@ -166,7 +166,11 @@ class Embeddings {
         return this.memoryId;
     }
 
-    async search(input: string): Promise<{ id: string; content: string; similarity: number }[]> {
+    async search(
+        input: string,
+    ): Promise<
+        { id: string; content: string; similarity: number; metadatas: Record<string, any> }[]
+    > {
         const id = await this.memoryId;
         return searchEmbeddings(id, input, await this.clientOptions);
     }