@memorylayerai/vercel-ai 0.3.2 → 0.4.0

package/dist/index.cjs

@@ -154,19 +154,38 @@ function memoryTool(config) {
 }
 function searchTool(config) {
   return {
-    description: "Search memories in the MemoryLayer system to retrieve relevant information",
+    description: "Search memories in the MemoryLayer system to retrieve relevant information with advanced retrieval options",
     parameters: import_zod.z.object({
       query: import_zod.z.string().describe("The search query to find relevant memories"),
       limit: import_zod.z.number().optional().describe("Maximum number of results to return (default: 10)"),
-      threshold: import_zod.z.number().optional().describe("Minimum relevance score threshold (0-1)")
+      threshold: import_zod.z.number().optional().describe("Minimum relevance score threshold (0-1)"),
+      enableQueryRewriting: import_zod.z.boolean().optional().describe("Enable query rewriting for better results (default: true)"),
+      enableEntityExpansion: import_zod.z.boolean().optional().describe("Enable entity expansion search (default: false)"),
+      enableGraphConnectivity: import_zod.z.boolean().optional().describe("Enable graph connectivity search (default: false)"),
+      enableSemanticDedup: import_zod.z.boolean().optional().describe("Enable semantic deduplication (default: false)"),
+      rerankingStrategy: import_zod.z.enum(["none", "cross-encoder", "llm"]).optional().describe("Reranking strategy (default: none)")
     }),
-    execute: async ({ query, limit, threshold }) => {
+    execute: async ({
+      query,
+      limit,
+      threshold,
+      enableQueryRewriting,
+      enableEntityExpansion,
+      enableGraphConnectivity,
+      enableSemanticDedup,
+      rerankingStrategy
+    }) => {
       try {
         const response = await config.client.search.search({
           query,
           projectId: config.projectId,
           limit,
-          threshold
+          threshold,
+          enableQueryRewriting,
+          enableEntityExpansion,
+          enableGraphConnectivity,
+          enableSemanticDedup,
+          rerankingStrategy
         });
         return {
           success: true,
@@ -174,7 +193,9 @@ function searchTool(config) {
             content: r.memory.content,
             score: r.score,
             metadata: r.memory.metadata,
-            id: r.memory.id
+            id: r.memory.id,
+            scoreBreakdown: r.scoreBreakdown,
+            connections: r.connections
           })),
           total: response.total
         };

package/dist/index.d.cts

@@ -199,19 +199,39 @@ declare function searchTool(config: MemoryToolConfig): {
         query: z.ZodString;
         limit: z.ZodOptional<z.ZodNumber>;
         threshold: z.ZodOptional<z.ZodNumber>;
+        enableQueryRewriting: z.ZodOptional<z.ZodBoolean>;
+        enableEntityExpansion: z.ZodOptional<z.ZodBoolean>;
+        enableGraphConnectivity: z.ZodOptional<z.ZodBoolean>;
+        enableSemanticDedup: z.ZodOptional<z.ZodBoolean>;
+        rerankingStrategy: z.ZodOptional<z.ZodEnum<["none", "cross-encoder", "llm"]>>;
     }, "strip", z.ZodTypeAny, {
         query: string;
         limit?: number | undefined;
         threshold?: number | undefined;
+        enableQueryRewriting?: boolean | undefined;
+        enableEntityExpansion?: boolean | undefined;
+        enableGraphConnectivity?: boolean | undefined;
+        enableSemanticDedup?: boolean | undefined;
+        rerankingStrategy?: "none" | "cross-encoder" | "llm" | undefined;
     }, {
         query: string;
         limit?: number | undefined;
         threshold?: number | undefined;
+        enableQueryRewriting?: boolean | undefined;
+        enableEntityExpansion?: boolean | undefined;
+        enableGraphConnectivity?: boolean | undefined;
+        enableSemanticDedup?: boolean | undefined;
+        rerankingStrategy?: "none" | "cross-encoder" | "llm" | undefined;
     }>;
-    execute: ({ query, limit, threshold }: {
+    execute: ({ query, limit, threshold, enableQueryRewriting, enableEntityExpansion, enableGraphConnectivity, enableSemanticDedup, rerankingStrategy, }: {
         query: string;
         limit?: number;
         threshold?: number;
+        enableQueryRewriting?: boolean;
+        enableEntityExpansion?: boolean;
+        enableGraphConnectivity?: boolean;
+        enableSemanticDedup?: boolean;
+        rerankingStrategy?: "none" | "cross-encoder" | "llm";
     }) => Promise<{
         success: boolean;
         results: any;

package/dist/index.d.ts

@@ -199,19 +199,39 @@ declare function searchTool(config: MemoryToolConfig): {
         query: z.ZodString;
         limit: z.ZodOptional<z.ZodNumber>;
         threshold: z.ZodOptional<z.ZodNumber>;
+        enableQueryRewriting: z.ZodOptional<z.ZodBoolean>;
+        enableEntityExpansion: z.ZodOptional<z.ZodBoolean>;
+        enableGraphConnectivity: z.ZodOptional<z.ZodBoolean>;
+        enableSemanticDedup: z.ZodOptional<z.ZodBoolean>;
+        rerankingStrategy: z.ZodOptional<z.ZodEnum<["none", "cross-encoder", "llm"]>>;
     }, "strip", z.ZodTypeAny, {
         query: string;
         limit?: number | undefined;
         threshold?: number | undefined;
+        enableQueryRewriting?: boolean | undefined;
+        enableEntityExpansion?: boolean | undefined;
+        enableGraphConnectivity?: boolean | undefined;
+        enableSemanticDedup?: boolean | undefined;
+        rerankingStrategy?: "none" | "cross-encoder" | "llm" | undefined;
     }, {
         query: string;
         limit?: number | undefined;
         threshold?: number | undefined;
+        enableQueryRewriting?: boolean | undefined;
+        enableEntityExpansion?: boolean | undefined;
+        enableGraphConnectivity?: boolean | undefined;
+        enableSemanticDedup?: boolean | undefined;
+        rerankingStrategy?: "none" | "cross-encoder" | "llm" | undefined;
     }>;
-    execute: ({ query, limit, threshold }: {
+    execute: ({ query, limit, threshold, enableQueryRewriting, enableEntityExpansion, enableGraphConnectivity, enableSemanticDedup, rerankingStrategy, }: {
         query: string;
         limit?: number;
         threshold?: number;
+        enableQueryRewriting?: boolean;
+        enableEntityExpansion?: boolean;
+        enableGraphConnectivity?: boolean;
+        enableSemanticDedup?: boolean;
+        rerankingStrategy?: "none" | "cross-encoder" | "llm";
     }) => Promise<{
         success: boolean;
         results: any;

package/dist/index.js

@@ -122,19 +122,38 @@ function memoryTool(config) {
 }
 function searchTool(config) {
   return {
-    description: "Search memories in the MemoryLayer system to retrieve relevant information",
+    description: "Search memories in the MemoryLayer system to retrieve relevant information with advanced retrieval options",
     parameters: z.object({
       query: z.string().describe("The search query to find relevant memories"),
       limit: z.number().optional().describe("Maximum number of results to return (default: 10)"),
-      threshold: z.number().optional().describe("Minimum relevance score threshold (0-1)")
+      threshold: z.number().optional().describe("Minimum relevance score threshold (0-1)"),
+      enableQueryRewriting: z.boolean().optional().describe("Enable query rewriting for better results (default: true)"),
+      enableEntityExpansion: z.boolean().optional().describe("Enable entity expansion search (default: false)"),
+      enableGraphConnectivity: z.boolean().optional().describe("Enable graph connectivity search (default: false)"),
+      enableSemanticDedup: z.boolean().optional().describe("Enable semantic deduplication (default: false)"),
+      rerankingStrategy: z.enum(["none", "cross-encoder", "llm"]).optional().describe("Reranking strategy (default: none)")
     }),
-    execute: async ({ query, limit, threshold }) => {
+    execute: async ({
+      query,
+      limit,
+      threshold,
+      enableQueryRewriting,
+      enableEntityExpansion,
+      enableGraphConnectivity,
+      enableSemanticDedup,
+      rerankingStrategy
+    }) => {
       try {
         const response = await config.client.search.search({
           query,
           projectId: config.projectId,
           limit,
-          threshold
+          threshold,
+          enableQueryRewriting,
+          enableEntityExpansion,
+          enableGraphConnectivity,
+          enableSemanticDedup,
+          rerankingStrategy
         });
         return {
           success: true,
@@ -142,7 +161,9 @@ function searchTool(config) {
             content: r.memory.content,
             score: r.score,
             metadata: r.memory.metadata,
-            id: r.memory.id
+            id: r.memory.id,
+            scoreBreakdown: r.scoreBreakdown,
+            connections: r.connections
           })),
           total: response.total
         };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@memorylayerai/vercel-ai",
-  "version": "0.3.2",
+  "version": "0.4.0",
   "description": "Vercel AI SDK integration for MemoryLayer",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
@@ -30,7 +30,7 @@
   "author": "MemoryLayer",
   "license": "MIT",
   "peerDependencies": {
-    "@memorylayerai/sdk": "^0.3.1",
+    "@memorylayerai/sdk": "^0.4.0",
     "ai": "^3.0.0",
     "zod": "^3.22.4"
   },

package/src/tools.ts

@@ -97,19 +97,47 @@ export function memoryTool(config: MemoryToolConfig) {
  */
 export function searchTool(config: MemoryToolConfig) {
   return {
-    description: 'Search memories in the MemoryLayer system to retrieve relevant information',
+    description: 'Search memories in the MemoryLayer system to retrieve relevant information with advanced retrieval options',
     parameters: z.object({
       query: z.string().describe('The search query to find relevant memories'),
       limit: z.number().optional().describe('Maximum number of results to return (default: 10)'),
       threshold: z.number().optional().describe('Minimum relevance score threshold (0-1)'),
+      enableQueryRewriting: z.boolean().optional().describe('Enable query rewriting for better results (default: true)'),
+      enableEntityExpansion: z.boolean().optional().describe('Enable entity expansion search (default: false)'),
+      enableGraphConnectivity: z.boolean().optional().describe('Enable graph connectivity search (default: false)'),
+      enableSemanticDedup: z.boolean().optional().describe('Enable semantic deduplication (default: false)'),
+      rerankingStrategy: z.enum(['none', 'cross-encoder', 'llm']).optional().describe('Reranking strategy (default: none)'),
     }),
-    execute: async ({ query, limit, threshold }: { query: string; limit?: number; threshold?: number }) => {
+    execute: async ({ 
+      query, 
+      limit, 
+      threshold,
+      enableQueryRewriting,
+      enableEntityExpansion,
+      enableGraphConnectivity,
+      enableSemanticDedup,
+      rerankingStrategy,
+    }: { 
+      query: string; 
+      limit?: number; 
+      threshold?: number;
+      enableQueryRewriting?: boolean;
+      enableEntityExpansion?: boolean;
+      enableGraphConnectivity?: boolean;
+      enableSemanticDedup?: boolean;
+      rerankingStrategy?: 'none' | 'cross-encoder' | 'llm';
+    }) => {
       try {
         const response = await config.client.search.search({
           query,
           projectId: config.projectId,
           limit,
           threshold,
+          enableQueryRewriting,
+          enableEntityExpansion,
+          enableGraphConnectivity,
+          enableSemanticDedup,
+          rerankingStrategy,
         });
 
         return {
@@ -119,6 +147,8 @@ export function searchTool(config: MemoryToolConfig) {
             score: r.score,
             metadata: r.memory.metadata,
             id: r.memory.id,
+            scoreBreakdown: r.scoreBreakdown,
+            connections: r.connections,
           })),
           total: response.total,
         };