@lanonasis/mem-intel-sdk 2.0.5 → 2.1.0

package/dist/server/index.js

@@ -241,11 +241,12 @@ var HttpClient = class {
       clearTimeout(timeoutId);
       const responseData = await response.json();
       if (!response.ok) {
+        const errorMessage = responseData?.message || responseData?.error?.message || (typeof responseData?.error === "string" ? responseData.error : void 0) || "Request failed";
         return {
           status: response.status,
           error: {
-            message: responseData?.message || responseData?.error || "Request failed",
-            code: responseData?.code
+            message: errorMessage,
+            code: responseData?.code || responseData?.error?.code
           }
         };
       }
@@ -354,6 +355,20 @@ var HttpClient = class {
 
 // src/core/client.ts
 var DEFAULT_API_URL = "https://api.lanonasis.com/api/v1";
+function appendQueryContext(params, context) {
+  if (context.organizationId) {
+    params.append("organization_id", context.organizationId);
+  }
+  if (context.topicId) {
+    params.append("topic_id", context.topicId);
+  }
+  if (context.queryScope) {
+    params.append("query_scope", context.queryScope);
+  }
+  for (const memoryType of context.memoryTypes || []) {
+    params.append("memory_types", memoryType);
+  }
+}
 var MemoryIntelligenceClient = class {
   httpClient;
   defaultResponseFormat;
@@ -403,7 +418,15 @@ var MemoryIntelligenceClient = class {
    * Query memories from the API
    */
   async queryMemories(userId, options = {}) {
-    const { type, limit = 100, offset = 0 } = options;
+    const {
+      type,
+      limit = 100,
+      offset = 0,
+      organizationId,
+      topicId,
+      memoryTypes,
+      queryScope
+    } = options;
     const params = new URLSearchParams({
       user_id: userId,
       limit: limit.toString(),
@@ -412,7 +435,13 @@ var MemoryIntelligenceClient = class {
     if (type) {
       params.append("type", type);
     }
-    const response = await this.httpClient.get(
+    appendQueryContext(params, {
+      organizationId,
+      topicId,
+      memoryTypes,
+      queryScope
+    });
+    const response = await this.httpClient.getEnhanced(
       `/intelligence/memories?${params.toString()}`
     );
     if (response.error) {
@@ -580,6 +609,9 @@ var MemoryIntelligenceClient = class {
       "/intelligence/predictive-recall",
       {
         userId: params.userId,
+        organizationId: params.organizationId,
+        topicId: params.topicId,
+        queryScope: params.queryScope,
         context: params.context,
         limit: params.limit || 5,
         minConfidence: params.minConfidence || 40,
@@ -636,11 +668,11 @@ var MemoryIntelligenceClient = class {
    *     active_files: ["src/components/DataTable.tsx"]
    *   },
    *   actions: [
-   *     { tool: "read_file", params: { path: "src/components/DataTable.tsx" } },
-   *     { tool: "analyze_code", params: { focus: "class_components" } },
-   *     { tool: "edit_file", params: { path: "src/components/DataTable.tsx" } }
+   *     { tool: "read_file", parameters: { path: "src/components/DataTable.tsx" }, outcome: "success" },
+   *     { tool: "analyze_code", parameters: { focus: "class_components" }, outcome: "success" },
+   *     { tool: "edit_file", parameters: { path: "src/components/DataTable.tsx" }, outcome: "success" }
    *   ],
-   *   final_outcome: "Successfully converted class component to functional with hooks"
+   *   final_outcome: "success"
    * });
    *
    * if (result.data.was_duplicate) {
@@ -649,13 +681,20 @@ var MemoryIntelligenceClient = class {
    * ```
    */
   async recordBehavior(params) {
+    const actions = params.actions.map((action) => ({
+      tool: action.tool,
+      ...action.parameters || action.params ? { parameters: action.parameters ?? action.params } : {},
+      ...action.outcome ? { outcome: action.outcome } : {},
+      ...action.timestamp ? { timestamp: action.timestamp } : {},
+      ...action.duration_ms !== void 0 ? { duration_ms: action.duration_ms } : {}
+    }));
     const response = await this.httpClient.postEnhanced(
       "/intelligence/behavior-record",
       {
         user_id: params.user_id,
         trigger: params.trigger,
         context: params.context || {},
-        actions: params.actions,
+        actions,
         final_outcome: params.final_outcome,
         confidence: params.confidence ?? 0.7
       }
@@ -732,8 +771,8 @@ var MemoryIntelligenceClient = class {
    *   current_state: {
    *     task_description: "Implementing user authentication with OAuth",
    *     completed_steps: [
-   *       { tool: "create_file", params: { path: "src/auth/oauth.ts" } },
-   *       { tool: "read_file", params: { path: "package.json" } }
+   *       "create src/auth/oauth.ts",
+   *       "inspect package.json"
    *     ]
    *   },
    *   max_suggestions: 3
@@ -798,6 +837,12 @@ var MemoryType = z.enum([
   "personal",
   "workflow"
 ]);
+var QueryScope = z.enum([
+  "personal",
+  "team",
+  "organization",
+  "hybrid"
+]);
 
 // src/utils/formatting.ts
 var CHARACTER_LIMIT = 5e4;
@@ -817,6 +862,15 @@ function truncateIfNeeded(text) {
 }
 
 // src/server/mcp-server.ts
+var optionalContextFields = {
+  organization_id: z.string().uuid().optional(),
+  topic_id: z.string().optional(),
+  query_scope: QueryScope.optional()
+};
+var optionalCollectionContextFields = {
+  ...optionalContextFields,
+  memory_types: z.array(MemoryType).optional()
+};
 function createMCPServer(config) {
   const client = new MemoryIntelligenceClient(config);
   const server = new McpServer({
@@ -831,6 +885,7 @@ function createMCPServer(config) {
       inputSchema: z.object({
         user_id: z.string().uuid(),
         time_range_days: z.number().int().min(1).max(365).default(30),
+        ...optionalCollectionContextFields,
         response_format: z.enum([ResponseFormat.JSON, ResponseFormat.MARKDOWN]).default(ResponseFormat.MARKDOWN)
       }).strict(),
       annotations: {
@@ -845,6 +900,10 @@ function createMCPServer(config) {
         const params = {
           userId: rawParams.user_id,
           timeRangeDays: rawParams.time_range_days,
+          organizationId: rawParams.organization_id,
+          topicId: rawParams.topic_id,
+          memoryTypes: rawParams.memory_types,
+          queryScope: rawParams.query_scope,
           responseFormat: rawParams.response_format
         };
         const response = await client.analyzePatterns(params);
@@ -990,6 +1049,7 @@ function createMCPServer(config) {
         user_id: z.string().uuid(),
         limit: z.number().int().min(1).max(50).default(10),
         similarity_threshold: z.number().min(0).max(1).default(0.7),
+        ...optionalCollectionContextFields,
         response_format: z.enum([ResponseFormat.JSON, ResponseFormat.MARKDOWN]).default(ResponseFormat.MARKDOWN)
       }).strict(),
       annotations: {
@@ -1006,6 +1066,10 @@ function createMCPServer(config) {
           userId: rawParams.user_id,
           limit: rawParams.limit,
           similarityThreshold: rawParams.similarity_threshold,
+          organizationId: rawParams.organization_id,
+          topicId: rawParams.topic_id,
+          memoryTypes: rawParams.memory_types,
+          queryScope: rawParams.query_scope,
           responseFormat: rawParams.response_format
         };
         const response = await client.findRelated(params);
@@ -1063,6 +1127,7 @@ function createMCPServer(config) {
         user_id: z.string().uuid(),
         similarity_threshold: z.number().min(0.8).max(0.99).default(0.9),
         max_pairs: z.number().int().min(1).max(100).default(20),
+        ...optionalCollectionContextFields,
         response_format: z.enum([ResponseFormat.JSON, ResponseFormat.MARKDOWN]).default(ResponseFormat.MARKDOWN)
       }).strict(),
       annotations: {
@@ -1078,6 +1143,10 @@ function createMCPServer(config) {
           userId: rawParams.user_id,
           similarityThreshold: rawParams.similarity_threshold,
           maxPairs: rawParams.max_pairs,
+          organizationId: rawParams.organization_id,
+          topicId: rawParams.topic_id,
+          memoryTypes: rawParams.memory_types,
+          queryScope: rawParams.query_scope,
           responseFormat: rawParams.response_format
         };
         const response = await client.detectDuplicates(params);
@@ -1141,6 +1210,7 @@ function createMCPServer(config) {
         topic: z.string().optional(),
         memory_type: z.string().optional(),
         max_memories: z.number().int().min(5).max(100).default(20),
+        ...optionalCollectionContextFields,
         response_format: z.enum([ResponseFormat.JSON, ResponseFormat.MARKDOWN]).default(ResponseFormat.MARKDOWN)
       }).strict(),
       annotations: {
@@ -1157,6 +1227,10 @@ function createMCPServer(config) {
           topic: rawParams.topic,
           memoryType: rawParams.memory_type,
           maxMemories: rawParams.max_memories,
+          organizationId: rawParams.organization_id,
+          topicId: rawParams.topic_id,
+          memoryTypes: rawParams.memory_types,
+          queryScope: rawParams.query_scope,
           responseFormat: rawParams.response_format
         };
         const response = await client.extractInsights(params);
@@ -1224,6 +1298,7 @@ ${data.overall_summary}
       description: `Analyze the health and organization quality of memories.`,
       inputSchema: z.object({
         user_id: z.string().uuid(),
+        ...optionalCollectionContextFields,
         response_format: z.enum([ResponseFormat.JSON, ResponseFormat.MARKDOWN]).default(ResponseFormat.MARKDOWN)
       }).strict(),
       annotations: {
@@ -1237,6 +1312,10 @@ ${data.overall_summary}
       try {
         const params = {
           userId: rawParams.user_id,
+          organizationId: rawParams.organization_id,
+          topicId: rawParams.topic_id,
+          memoryTypes: rawParams.memory_types,
+          queryScope: rawParams.query_scope,
           responseFormat: rawParams.response_format
         };
         const response = await client.healthCheck(params);
@@ -1322,6 +1401,7 @@ ${data.overall_summary}
       description: `AI-powered prediction of memories you'll need based on your current context. Uses semantic similarity, temporal relevance, usage patterns, and serendipity scoring to anticipate what you need before you realize it.`,
       inputSchema: z.object({
         user_id: z.string().uuid(),
+        ...optionalContextFields,
         context: z.object({
           current_project: z.string().optional().describe("Current project or task description"),
           recent_topics: z.array(z.string()).optional().describe("Recent topics you've been working on"),
@@ -1347,6 +1427,9 @@ ${data.overall_summary}
       try {
         const params = {
           userId: rawParams.user_id,
+          organizationId: rawParams.organization_id,
+          topicId: rawParams.topic_id,
+          queryScope: rawParams.query_scope,
           context: {
             currentProject: rawParams.context?.current_project,
             recentTopics: rawParams.context?.recent_topics,
@@ -1632,10 +1715,16 @@ Was duplicate: ${response.data.was_duplicate}`
         user_id: z.string().uuid(),
         current_state: z.object({
           task_description: z.string().min(1, "Task description is required"),
-          completed_steps: z.array(z.object({
-            tool: z.string(),
-            params: z.record(z.unknown()).optional()
-          })).optional()
+          completed_steps: z.array(
+            z.union([
+              z.string().min(1, "Completed step must be a non-empty string"),
+              z.object({
+                tool: z.string().min(1, "Legacy step objects must include tool"),
+                params: z.record(z.unknown()).optional(),
+                parameters: z.record(z.unknown()).optional()
+              }).transform((step) => step.tool)
+            ])
+          ).optional()
         }),
         max_suggestions: z.number().int().min(1).max(10).default(3)
       }).strict(),
@@ -1687,6 +1776,6 @@ Was duplicate: ${response.data.was_duplicate}`
   return server;
 }
 
-export { ConfigurationError, DatabaseError, EmbeddingError, MemoryIntelligenceError, MemoryNotFoundError, MemoryType, ResponseFormat, ValidationError, createMCPServer };
+export { ConfigurationError, DatabaseError, EmbeddingError, MemoryIntelligenceError, MemoryNotFoundError, MemoryType, QueryScope, ResponseFormat, ValidationError, createMCPServer };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map