@contextstream/mcp-server 0.4.34 → 0.4.36

package/dist/index.js

@@ -4297,7 +4297,7 @@ var RETRYABLE_STATUSES = /* @__PURE__ */ new Set([408, 429, 500, 502, 503, 504])
 var MAX_RETRIES = 3;
 var BASE_DELAY = 1e3;
 async function sleep(ms) {
-  return new Promise((resolve3) => setTimeout(resolve3, ms));
+  return new Promise((resolve4) => setTimeout(resolve4, ms));
 }
 async function request(config, path8, options = {}) {
   const { apiUrl, userAgent } = config;
@@ -4571,10 +4571,16 @@ var IGNORE_FILES = /* @__PURE__ */ new Set([
   "composer.lock"
 ]);
 var MAX_FILE_SIZE = 1024 * 1024;
+var MAX_BATCH_BYTES = 10 * 1024 * 1024;
+var LARGE_FILE_THRESHOLD = 2 * 1024 * 1024;
+var MAX_FILES_PER_BATCH = 200;
 async function* readAllFilesInBatches(rootPath, options = {}) {
-  const batchSize = options.batchSize ?? 50;
+  const maxBatchBytes = options.maxBatchBytes ?? MAX_BATCH_BYTES;
+  const largeFileThreshold = options.largeFileThreshold ?? LARGE_FILE_THRESHOLD;
+  const maxFilesPerBatch = options.maxFilesPerBatch ?? MAX_FILES_PER_BATCH;
   const maxFileSize = options.maxFileSize ?? MAX_FILE_SIZE;
   let batch = [];
+  let currentBatchBytes = 0;
   async function* walkDir(dir, relativePath = "") {
     let entries;
     try {
@@ -4596,28 +4602,46 @@ async function* readAllFilesInBatches(rootPath, options = {}) {
           const stat2 = await fs.promises.stat(fullPath);
           if (stat2.size > maxFileSize) continue;
           const content = await fs.promises.readFile(fullPath, "utf-8");
-          yield { path: relPath, content };
+          yield { path: relPath, content, sizeBytes: stat2.size };
         } catch {
         }
       }
     }
   }
   for await (const file of walkDir(rootPath)) {
-    batch.push(file);
-    if (batch.length >= batchSize) {
-      yield batch;
-      batch = [];
+    if (file.sizeBytes > largeFileThreshold) {
+      if (batch.length > 0) {
+        yield batch.map(({ sizeBytes, ...rest }) => rest);
+        batch = [];
+        currentBatchBytes = 0;
+      }
+      yield [{ path: file.path, content: file.content }];
+      continue;
     }
+    const wouldExceedBytes = currentBatchBytes + file.sizeBytes > maxBatchBytes;
+    const wouldExceedFiles = batch.length >= maxFilesPerBatch;
+    if (wouldExceedBytes || wouldExceedFiles) {
+      if (batch.length > 0) {
+        yield batch.map(({ sizeBytes, ...rest }) => rest);
+        batch = [];
+        currentBatchBytes = 0;
+      }
+    }
+    batch.push(file);
+    currentBatchBytes += file.sizeBytes;
   }
   if (batch.length > 0) {
-    yield batch;
+    yield batch.map(({ sizeBytes, ...rest }) => rest);
   }
 }
 async function* readChangedFilesInBatches(rootPath, sinceTimestamp, options = {}) {
-  const batchSize = options.batchSize ?? 50;
+  const maxBatchBytes = options.maxBatchBytes ?? MAX_BATCH_BYTES;
+  const largeFileThreshold = options.largeFileThreshold ?? LARGE_FILE_THRESHOLD;
+  const maxFilesPerBatch = options.maxFilesPerBatch ?? MAX_FILES_PER_BATCH;
   const maxFileSize = options.maxFileSize ?? MAX_FILE_SIZE;
   const sinceMs = sinceTimestamp.getTime();
   let batch = [];
+  let currentBatchBytes = 0;
   let filesScanned = 0;
   let filesChanged = 0;
   async function* walkDir(dir, relativePath = "") {
@@ -4644,21 +4668,36 @@ async function* readChangedFilesInBatches(rootPath, sinceTimestamp, options = {}
           if (stat2.size > maxFileSize) continue;
           const content = await fs.promises.readFile(fullPath, "utf-8");
           filesChanged++;
-          yield { path: relPath, content };
+          yield { path: relPath, content, sizeBytes: stat2.size };
         } catch {
         }
       }
     }
   }
   for await (const file of walkDir(rootPath)) {
-    batch.push(file);
-    if (batch.length >= batchSize) {
-      yield batch;
-      batch = [];
+    if (file.sizeBytes > largeFileThreshold) {
+      if (batch.length > 0) {
+        yield batch.map(({ sizeBytes, ...rest }) => rest);
+        batch = [];
+        currentBatchBytes = 0;
+      }
+      yield [{ path: file.path, content: file.content }];
+      continue;
+    }
+    const wouldExceedBytes = currentBatchBytes + file.sizeBytes > maxBatchBytes;
+    const wouldExceedFiles = batch.length >= maxFilesPerBatch;
+    if (wouldExceedBytes || wouldExceedFiles) {
+      if (batch.length > 0) {
+        yield batch.map(({ sizeBytes, ...rest }) => rest);
+        batch = [];
+        currentBatchBytes = 0;
+      }
     }
+    batch.push(file);
+    currentBatchBytes += file.sizeBytes;
   }
   if (batch.length > 0) {
-    yield batch;
+    yield batch.map(({ sizeBytes, ...rest }) => rest);
   }
   console.error(
     `[ContextStream] Incremental scan: ${filesChanged} changed files out of ${filesScanned} scanned (since ${sinceTimestamp.toISOString()})`
@@ -5078,7 +5117,7 @@ var ContextStreamClient = class {
         return "full";
       }
       if (planName.includes("pro")) return "lite";
-      if (planName.includes("free")) return "none";
+      if (planName.includes("free")) return "lite";
       return "lite";
     } catch {
       return "none";
@@ -6610,19 +6649,6 @@ var ContextStreamClient = class {
       used += next.length;
     }
     const summary = finalLines.join("\n");
-    this.trackTokenSavings({
-      tool: "session_summary",
-      workspace_id: withDefaults.workspace_id,
-      project_id: withDefaults.project_id,
-      candidate_chars: candidateSummary.length,
-      context_chars: summary.length,
-      max_tokens: maxTokens,
-      metadata: {
-        decision_count: decisionCount,
-        memory_count: memoryCount
-      }
-    }).catch(() => {
-    });
     return {
       summary,
       workspace_name: workspaceName,
@@ -6864,21 +6890,6 @@ var ContextStreamClient = class {
     const context = parts.join("");
     const candidateContext = candidateParts.join("");
     const tokenEstimate = Math.ceil(context.length / charsPerToken);
-    this.trackTokenSavings({
-      tool: "ai_context_budget",
-      workspace_id: withDefaults.workspace_id,
-      project_id: withDefaults.project_id,
-      candidate_chars: candidateContext.length,
-      context_chars: context.length,
-      max_tokens: maxTokens,
-      metadata: {
-        include_decisions: params.include_decisions !== false,
-        include_memory: params.include_memory !== false,
-        include_code: !!params.include_code,
-        sources: sources.length
-      }
-    }).catch(() => {
-    });
     return {
       context,
       token_estimate: tokenEstimate,
@@ -7222,21 +7233,6 @@ W:${wsHint}
       versionNotice = await getUpdateNotice();
     } catch {
     }
-    this.trackTokenSavings({
-      tool: "context_smart",
-      workspace_id: withDefaults.workspace_id,
-      project_id: withDefaults.project_id,
-      candidate_chars: candidateContext.length,
-      context_chars: context.length,
-      max_tokens: maxTokens,
-      metadata: {
-        format,
-        items: items.length,
-        keywords: keywords.slice(0, 10),
-        errors: errors.length
-      }
-    }).catch(() => {
-    });
     return {
       context,
       token_estimate: Math.ceil(context.length / 4),
@@ -7885,6 +7881,29 @@ W:${wsHint}
       { method: "GET" }
     );
   }
+  /**
+   * Create a new Notion database
+   */
+  async notionCreateDatabase(params) {
+    const withDefaults = this.withDefaults(params || {});
+    if (!withDefaults.workspace_id) {
+      throw new Error("workspace_id is required for creating Notion database");
+    }
+    const query = new URLSearchParams();
+    query.set("workspace_id", withDefaults.workspace_id);
+    return request(
+      this.config,
+      `/integrations/notion/databases?${query.toString()}`,
+      {
+        method: "POST",
+        body: {
+          title: params.title,
+          parent_page_id: params.parent_page_id,
+          description: params.description
+        }
+      }
+    );
+  }
   /**
    * Search/list pages in Notion with smart type detection filtering
    */
@@ -8925,13 +8944,21 @@ var PRETOOLUSE_HOOK_SCRIPT = `#!/usr/bin/env python3
 """
 ContextStream PreToolUse Hook for Claude Code
 Blocks Grep/Glob/Search/Task(Explore)/EnterPlanMode and redirects to ContextStream.
+
+Only blocks if the current project is indexed in ContextStream.
+If not indexed, allows local tools through with a suggestion to index.
 """
 
 import json
 import sys
 import os
+from pathlib import Path
+from datetime import datetime, timedelta
 
 ENABLED = os.environ.get("CONTEXTSTREAM_HOOK_ENABLED", "true").lower() == "true"
+INDEX_STATUS_FILE = Path.home() / ".contextstream" / "indexed-projects.json"
+# Consider index stale after 7 days
+STALE_THRESHOLD_DAYS = 7
 
 DISCOVERY_PATTERNS = ["**/*", "**/", "src/**", "lib/**", "app/**", "components/**"]
 
@@ -8953,6 +8980,44 @@ def is_discovery_grep(file_path):
         return True
     return False
 
+def is_project_indexed(cwd: str) -> tuple[bool, bool]:
+    """
+    Check if the current directory is in an indexed project.
+    Returns (is_indexed, is_stale).
+    """
+    if not INDEX_STATUS_FILE.exists():
+        return False, False
+
+    try:
+        with open(INDEX_STATUS_FILE, "r") as f:
+            data = json.load(f)
+    except:
+        return False, False
+
+    projects = data.get("projects", {})
+    cwd_path = Path(cwd).resolve()
+
+    # Check if cwd is within any indexed project
+    for project_path, info in projects.items():
+        try:
+            indexed_path = Path(project_path).resolve()
+            # Check if cwd is the project or a subdirectory
+            if cwd_path == indexed_path or indexed_path in cwd_path.parents:
+                # Check if stale
+                indexed_at = info.get("indexed_at")
+                if indexed_at:
+                    try:
+                        indexed_time = datetime.fromisoformat(indexed_at.replace("Z", "+00:00"))
+                        if datetime.now(indexed_time.tzinfo) - indexed_time > timedelta(days=STALE_THRESHOLD_DAYS):
+                            return True, True  # Indexed but stale
+                    except:
+                        pass
+                return True, False  # Indexed and fresh
+        except:
+            continue
+
+    return False, False
+
 def main():
     if not ENABLED:
         sys.exit(0)
@@ -8964,6 +9029,20 @@ def main():
 
     tool = data.get("tool_name", "")
     inp = data.get("tool_input", {})
+    cwd = data.get("cwd", os.getcwd())
+
+    # Check if project is indexed
+    is_indexed, is_stale = is_project_indexed(cwd)
+
+    if not is_indexed:
+        # Project not indexed - allow local tools but suggest indexing
+        # Don't block, just exit successfully
+        sys.exit(0)
+
+    if is_stale:
+        # Index is stale - allow with warning (printed but not blocking)
+        # Still allow the tool but remind about re-indexing
+        pass  # Continue to blocking logic but could add warning
 
     if tool == "Glob":
         pattern = inp.get("pattern", "")
@@ -8971,18 +9050,16 @@ def main():
             print(f"STOP: Use mcp__contextstream__search(mode=\\"hybrid\\", query=\\"{pattern}\\") instead of Glob.", file=sys.stderr)
             sys.exit(2)
 
-    elif tool == "Grep":
+    elif tool == "Grep" or tool == "Search":
+        # Block ALL Grep/Search operations - use ContextStream search or Read for specific files
         pattern = inp.get("pattern", "")
         path = inp.get("path", "")
-        if is_discovery_grep(path):
-            print(f"STOP: Use mcp__contextstream__search(mode=\\"keyword\\", query=\\"{pattern}\\") instead of Grep.", file=sys.stderr)
-            sys.exit(2)
-
-    elif tool == "Search":
-        # Block the Search tool which is Claude Code's default search
-        query = inp.get("pattern", "") or inp.get("query", "")
-        if query:
-            print(f"STOP: Use mcp__contextstream__search(mode=\\"hybrid\\", query=\\"{query}\\") instead of Search.", file=sys.stderr)
+        if pattern:
+            if path and not is_discovery_grep(path):
+                # Specific file - suggest Read instead
+                print(f"STOP: Use Read(\\"{path}\\") to view file content, or mcp__contextstream__search(mode=\\"keyword\\", query=\\"{pattern}\\") for codebase search.", file=sys.stderr)
+            else:
+                print(f"STOP: Use mcp__contextstream__search(mode=\\"hybrid\\", query=\\"{pattern}\\") instead of {tool}.", file=sys.stderr)
             sys.exit(2)
 
     elif tool == "Task":
@@ -9148,6 +9225,97 @@ async function installClaudeCodeHooks(options) {
   }
   return result;
 }
+function getIndexStatusPath() {
+  return path4.join(homedir2(), ".contextstream", "indexed-projects.json");
+}
+async function readIndexStatus() {
+  const statusPath = getIndexStatusPath();
+  try {
+    const content = await fs3.readFile(statusPath, "utf-8");
+    return JSON.parse(content);
+  } catch {
+    return { version: 1, projects: {} };
+  }
+}
+async function writeIndexStatus(status) {
+  const statusPath = getIndexStatusPath();
+  const dir = path4.dirname(statusPath);
+  await fs3.mkdir(dir, { recursive: true });
+  await fs3.writeFile(statusPath, JSON.stringify(status, null, 2));
+}
+async function markProjectIndexed(projectPath, options) {
+  const status = await readIndexStatus();
+  const resolvedPath = path4.resolve(projectPath);
+  status.projects[resolvedPath] = {
+    indexed_at: (/* @__PURE__ */ new Date()).toISOString(),
+    project_id: options?.project_id,
+    project_name: options?.project_name
+  };
+  await writeIndexStatus(status);
+}
+
+// src/token-savings.ts
+var TOKEN_SAVINGS_FORMULA_VERSION = 1;
+var MAX_CHARS_PER_EVENT = 2e7;
+var BASE_OVERHEAD_CHARS = 500;
+var CANDIDATE_MULTIPLIERS = {
+  // context_smart: Replaces reading multiple files to gather context
+  context_smart: 5,
+  ai_context_budget: 5,
+  // search: Semantic search replaces iterative Glob/Grep/Read cycles
+  search_semantic: 4.5,
+  search_hybrid: 4,
+  search_keyword: 2.5,
+  search_pattern: 3,
+  search_exhaustive: 3.5,
+  search_refactor: 3,
+  // session: Recall/search replaces reading through history
+  session_recall: 5,
+  session_smart_search: 4,
+  session_user_context: 3,
+  session_summary: 4,
+  // graph: Would require extensive file traversal
+  graph_dependencies: 8,
+  graph_impact: 10,
+  graph_call_path: 8,
+  graph_related: 6,
+  // memory: Context retrieval
+  memory_search: 3.5,
+  memory_decisions: 3,
+  memory_timeline: 3,
+  memory_summary: 4
+};
+function clampCharCount(value) {
+  if (!Number.isFinite(value) || value <= 0) return 0;
+  return Math.min(MAX_CHARS_PER_EVENT, Math.floor(value));
+}
+function trackToolTokenSavings(client, tool, contextText, params, extraMetadata) {
+  try {
+    const contextChars = clampCharCount(contextText.length);
+    const multiplier = CANDIDATE_MULTIPLIERS[tool] ?? 3;
+    const baseOverhead = contextChars > 0 ? BASE_OVERHEAD_CHARS : 0;
+    const estimatedCandidate = Math.round(contextChars * multiplier + baseOverhead);
+    const candidateChars = Math.max(contextChars, clampCharCount(estimatedCandidate));
+    client.trackTokenSavings({
+      tool,
+      workspace_id: params?.workspace_id,
+      project_id: params?.project_id,
+      candidate_chars: candidateChars,
+      context_chars: contextChars,
+      max_tokens: params?.max_tokens,
+      metadata: {
+        method: "multiplier_estimate",
+        formula_version: TOKEN_SAVINGS_FORMULA_VERSION,
+        source: "mcp-server",
+        multiplier,
+        base_overhead_chars: baseOverhead,
+        ...extraMetadata ?? {}
+      }
+    }).catch(() => {
+    });
+  } catch {
+  }
+}
 
 // src/tools.ts
 var LESSON_DEDUP_WINDOW_MS = 2 * 60 * 1e3;
@@ -10501,52 +10669,6 @@ function toStructured(data) {
   }
   return void 0;
 }
-var CANDIDATE_MULTIPLIERS = {
-  // context_smart: Replaces reading multiple files to gather context
-  context_smart: 5,
-  // search: Semantic search replaces iterative Glob/Grep/Read cycles
-  search_semantic: 4.5,
-  search_hybrid: 4,
-  search_keyword: 2.5,
-  search_pattern: 3,
-  search_exhaustive: 3.5,
-  search_refactor: 3,
-  // session: Recall/search replaces reading through history
-  session_recall: 5,
-  session_smart_search: 4,
-  session_user_context: 3,
-  session_summary: 4,
-  // graph: Would require extensive file traversal
-  graph_dependencies: 8,
-  graph_impact: 10,
-  graph_call_path: 8,
-  graph_related: 6,
-  // memory: Context retrieval
-  memory_search: 3.5,
-  memory_decisions: 3,
-  memory_timeline: 3,
-  memory_summary: 4
-};
-function trackToolTokenSavings(client, tool, resultContent, params) {
-  try {
-    const contextStr = typeof resultContent === "string" ? resultContent : JSON.stringify(resultContent ?? {});
-    const contextChars = contextStr.length;
-    const multiplier = CANDIDATE_MULTIPLIERS[tool] ?? 3;
-    const baseOverhead = 500;
-    const candidateChars = Math.round(contextChars * multiplier + baseOverhead);
-    client.trackTokenSavings({
-      tool,
-      workspace_id: params?.workspace_id,
-      project_id: params?.project_id,
-      candidate_chars: candidateChars,
-      context_chars: contextChars,
-      max_tokens: params?.max_tokens,
-      metadata: { multiplier, source: "mcp-server" }
-    }).catch(() => {
-    });
-  } catch {
-  }
-}
 function readStatNumber(payload, key) {
   if (!payload || typeof payload !== "object") return void 0;
   const direct = payload[key];
@@ -10790,7 +10912,8 @@ function registerTools(server, client, sessionManager) {
     ["graph_call_path", "full"],
     ["graph_circular_dependencies", "full"],
     ["graph_unused_code", "full"],
-    ["graph_ingest", "full"],
+    ["graph_ingest", "lite"],
+    // Pro can ingest (builds module-level graph), Elite gets full graph
     ["graph_contradictions", "full"]
   ]);
   const graphLiteMaxDepth = 1;
@@ -11262,6 +11385,12 @@ Hint: Run session_init(folder_path="<your_project_path>") first to establish a s
         console.error(
           `[ContextStream] Completed background ingestion: ${totalIndexed} files in ${batchCount} batches`
         );
+        try {
+          await markProjectIndexed(resolvedPath, { project_id: projectId });
+          console.error(`[ContextStream] Marked project as indexed: ${resolvedPath}`);
+        } catch (markError) {
+          console.error(`[ContextStream] Failed to mark project as indexed:`, markError);
+        }
       } catch (error) {
         console.error(`[ContextStream] Ingestion failed:`, error);
       }
@@ -13007,6 +13136,12 @@ ${benefitsList}` : "",
         noticeLines.push(
           `[INGEST_STATUS] Background indexing started. Codebase will be searchable shortly.`
         );
+      } else if (folderPathForRules && !ingestRec?.recommended) {
+        const projectId = typeof result.project_id === "string" ? result.project_id : void 0;
+        const projectName = typeof result.project_name === "string" ? result.project_name : void 0;
+        markProjectIndexed(folderPathForRules, { project_id: projectId, project_name: projectName }).catch(
+          (err) => console.error("[ContextStream] Failed to mark project as indexed:", err)
+        );
       }
       if (noticeLines.length > 0) {
         text = `${text}
@@ -15274,12 +15409,13 @@ Output formats: full (default, includes content), paths (file paths only - 80% t
           default:
             toolType = "search_hybrid";
         }
-        trackToolTokenSavings(client, toolType, result, {
+        const outputText = formatContent(result);
+        trackToolTokenSavings(client, toolType, outputText, {
           workspace_id: params.workspace_id,
           project_id: params.project_id
         });
         return {
-          content: [{ type: "text", text: formatContent(result) }],
+          content: [{ type: "text", text: outputText }],
           structuredContent: toStructured(result)
         };
       }
@@ -15490,12 +15626,13 @@ Output formats: full (default, includes content), paths (file paths only - 80% t
               include_related: input.include_related,
               include_decisions: input.include_decisions
             });
-            trackToolTokenSavings(client, "session_recall", result, {
+            const outputText = formatContent(result);
+            trackToolTokenSavings(client, "session_recall", outputText, {
               workspace_id: workspaceId,
               project_id: projectId
             });
             return {
-              content: [{ type: "text", text: formatContent(result) }],
+              content: [{ type: "text", text: outputText }],
               structuredContent: toStructured(result)
             };
           }
@@ -15517,11 +15654,12 @@ Output formats: full (default, includes content), paths (file paths only - 80% t
           }
           case "user_context": {
             const result = await client.getUserContext({ workspace_id: workspaceId });
-            trackToolTokenSavings(client, "session_user_context", result, {
+            const outputText = formatContent(result);
+            trackToolTokenSavings(client, "session_user_context", outputText, {
               workspace_id: workspaceId
             });
             return {
-              content: [{ type: "text", text: formatContent(result) }],
+              content: [{ type: "text", text: outputText }],
               structuredContent: toStructured(result)
             };
           }
@@ -15531,13 +15669,14 @@ Output formats: full (default, includes content), paths (file paths only - 80% t
               project_id: projectId,
               max_tokens: input.max_tokens
             });
-            trackToolTokenSavings(client, "session_summary", result, {
+            const outputText = formatContent(result);
+            trackToolTokenSavings(client, "session_summary", outputText, {
               workspace_id: workspaceId,
               project_id: projectId,
               max_tokens: input.max_tokens
             });
             return {
-              content: [{ type: "text", text: formatContent(result) }],
+              content: [{ type: "text", text: outputText }],
               structuredContent: toStructured(result)
             };
           }
@@ -15581,12 +15720,13 @@ Output formats: full (default, includes content), paths (file paths only - 80% t
               include_decisions: input.include_decisions,
               include_related: input.include_related
             });
-            trackToolTokenSavings(client, "session_smart_search", result, {
+            const outputText = formatContent(result);
+            trackToolTokenSavings(client, "session_smart_search", outputText, {
               workspace_id: workspaceId,
               project_id: projectId
             });
             return {
-              content: [{ type: "text", text: formatContent(result) }],
+              content: [{ type: "text", text: outputText }],
               structuredContent: toStructured(result)
             };
           }
@@ -15933,12 +16073,13 @@ Output formats: full (default, includes content), paths (file paths only - 80% t
               query: input.query,
               limit: input.limit
             });
-            trackToolTokenSavings(client, "memory_search", result, {
+            const outputText = formatContent(result);
+            trackToolTokenSavings(client, "memory_search", outputText, {
               workspace_id: workspaceId,
               project_id: projectId
             });
             return {
-              content: [{ type: "text", text: formatContent(result) }],
+              content: [{ type: "text", text: outputText }],
               structuredContent: toStructured(result)
             };
           }
@@ -15949,12 +16090,13 @@ Output formats: full (default, includes content), paths (file paths only - 80% t
               category: input.category,
               limit: input.limit
             });
-            trackToolTokenSavings(client, "memory_decisions", result, {
+            const outputText = formatContent(result);
+            trackToolTokenSavings(client, "memory_decisions", outputText, {
               workspace_id: workspaceId,
               project_id: projectId
             });
             return {
-              content: [{ type: "text", text: formatContent(result) }],
+              content: [{ type: "text", text: outputText }],
               structuredContent: toStructured(result)
             };
           }
@@ -15963,11 +16105,12 @@ Output formats: full (default, includes content), paths (file paths only - 80% t
               return errorResult("timeline requires workspace_id. Call session_init first.");
             }
             const result = await client.memoryTimeline(workspaceId);
-            trackToolTokenSavings(client, "memory_timeline", result, {
+            const outputText = formatContent(result);
+            trackToolTokenSavings(client, "memory_timeline", outputText, {
               workspace_id: workspaceId
             });
             return {
-              content: [{ type: "text", text: formatContent(result) }],
+              content: [{ type: "text", text: outputText }],
               structuredContent: toStructured(result)
             };
           }
@@ -15976,11 +16119,12 @@ Output formats: full (default, includes content), paths (file paths only - 80% t
               return errorResult("summary requires workspace_id. Call session_init first.");
             }
             const result = await client.memorySummary(workspaceId);
-            trackToolTokenSavings(client, "memory_summary", result, {
+            const outputText = formatContent(result);
+            trackToolTokenSavings(client, "memory_summary", outputText, {
               workspace_id: workspaceId
             });
             return {
-              content: [{ type: "text", text: formatContent(result) }],
+              content: [{ type: "text", text: outputText }],
               structuredContent: toStructured(result)
             };
           }
@@ -16165,12 +16309,13 @@ Output formats: full (default, includes content), paths (file paths only - 80% t
               max_depth: input.max_depth,
               include_transitive: input.include_transitive
             });
-            trackToolTokenSavings(client, "graph_dependencies", result, {
+            const outputText = formatContent(result);
+            trackToolTokenSavings(client, "graph_dependencies", outputText, {
               workspace_id: workspaceId,
               project_id: projectId
             });
             return {
-              content: [{ type: "text", text: formatContent(result) }],
+              content: [{ type: "text", text: outputText }],
               structuredContent: toStructured(result)
             };
           }
@@ -16182,12 +16327,13 @@ Output formats: full (default, includes content), paths (file paths only - 80% t
               target: input.target,
               max_depth: input.max_depth
             });
-            trackToolTokenSavings(client, "graph_impact", result, {
+            const outputText = formatContent(result);
+            trackToolTokenSavings(client, "graph_impact", outputText, {
               workspace_id: workspaceId,
               project_id: projectId
             });
             return {
-              content: [{ type: "text", text: formatContent(result) }],
+              content: [{ type: "text", text: outputText }],
               structuredContent: toStructured(result)
             };
           }
@@ -16200,12 +16346,13 @@ Output formats: full (default, includes content), paths (file paths only - 80% t
               target: input.target,
               max_depth: input.max_depth
             });
-            trackToolTokenSavings(client, "graph_call_path", result, {
+            const outputText = formatContent(result);
+            trackToolTokenSavings(client, "graph_call_path", outputText, {
               workspace_id: workspaceId,
               project_id: projectId
             });
             return {
-              content: [{ type: "text", text: formatContent(result) }],
+              content: [{ type: "text", text: outputText }],
               structuredContent: toStructured(result)
             };
           }
@@ -16219,12 +16366,13 @@ Output formats: full (default, includes content), paths (file paths only - 80% t
               project_id: projectId,
               limit: input.limit
             });
-            trackToolTokenSavings(client, "graph_related", result, {
+            const outputText = formatContent(result);
+            trackToolTokenSavings(client, "graph_related", outputText, {
               workspace_id: workspaceId,
               project_id: projectId
             });
             return {
-              content: [{ type: "text", text: formatContent(result) }],
+              content: [{ type: "text", text: outputText }],
               structuredContent: toStructured(result)
             };
           }
@@ -16688,7 +16836,7 @@ Output formats: full (default, includes content), paths (file paths only - 80% t
       "integration",
       {
         title: "Integration",
-        description: `Integration operations for Slack, GitHub, and Notion. Provider: slack, github, notion, all. Actions: status, search, stats, activity, contributors, knowledge, summary, channels (slack), discussions (slack), repos (github), issues (github), create_page (notion), list_databases (notion), search_pages (notion with smart type detection - filter by event_type, status, priority, has_due_date, tags), get_page (notion), query_database (notion), update_page (notion).`,
+        description: `Integration operations for Slack, GitHub, and Notion. Provider: slack, github, notion, all. Actions: status, search, stats, activity, contributors, knowledge, summary, channels (slack), discussions (slack), repos (github), issues (github), create_page (notion), create_database (notion), list_databases (notion), search_pages (notion with smart type detection - filter by event_type, status, priority, has_due_date, tags), get_page (notion), query_database (notion), update_page (notion).`,
         inputSchema: external_exports.object({
           provider: external_exports.enum(["slack", "github", "notion", "all"]).describe("Integration provider"),
           action: external_exports.enum([
@@ -16706,6 +16854,7 @@ Output formats: full (default, includes content), paths (file paths only - 80% t
             "issues",
             // Notion-specific actions
             "create_page",
+            "create_database",
             "list_databases",
             "search_pages",
             "get_page",
@@ -16719,10 +16868,11 @@ Output formats: full (default, includes content), paths (file paths only - 80% t
           since: external_exports.string().optional(),
           until: external_exports.string().optional(),
           // Notion-specific parameters
-          title: external_exports.string().optional().describe("Page title (for Notion create_page/update_page)"),
+          title: external_exports.string().optional().describe("Page/database title (for Notion create_page/update_page/create_database)"),
           content: external_exports.string().optional().describe("Page content in Markdown (for Notion create_page/update_page)"),
+          description: external_exports.string().optional().describe("Database description (for Notion create_database)"),
           parent_database_id: external_exports.string().optional().describe("Parent database ID (for Notion create_page)"),
-          parent_page_id: external_exports.string().optional().describe("Parent page ID (for Notion create_page)"),
+          parent_page_id: external_exports.string().optional().describe("Parent page ID (for Notion create_page/create_database)"),
           page_id: external_exports.string().optional().describe("Page ID (for Notion get_page/update_page)"),
           database_id: external_exports.string().optional().describe("Database ID (for Notion query_database/search_pages/activity)"),
           days: external_exports.number().optional().describe("Number of days for stats/summary (default: 7)"),
@@ -17014,6 +17164,39 @@ Created: ${result.created_time}`
               structuredContent: toStructured(result)
             };
           }
+          case "create_database": {
+            if (input.provider !== "notion") {
+              return errorResult("create_database is only available for notion provider");
+            }
+            if (!input.title) {
+              return errorResult("title is required for create_database action");
+            }
+            if (!input.parent_page_id) {
+              return errorResult("parent_page_id is required for create_database action");
+            }
+            if (!workspaceId) {
+              return errorResult(
+                "Error: workspace_id is required. Please call session_init first or provide workspace_id explicitly."
+              );
+            }
+            const newDatabase = await client.notionCreateDatabase({
+              workspace_id: workspaceId,
+              title: input.title,
+              parent_page_id: input.parent_page_id,
+              description: input.description
+            });
+            return {
+              content: [
+                {
+                  type: "text",
+                  text: `Created database "${newDatabase.title}"
+ID: ${newDatabase.id}
+URL: ${newDatabase.url}`
+                }
+              ],
+              structuredContent: toStructured(newDatabase)
+            };
+          }
           case "list_databases": {
             if (input.provider !== "notion") {
               return errorResult("list_databases is only available for notion provider");
@@ -19569,10 +19752,10 @@ Code: ${device.user_code}`);
           if (poll && poll.status === "pending") {
             const intervalSeconds = typeof poll.interval === "number" ? poll.interval : 5;
             const waitMs = Math.max(1, intervalSeconds) * 1e3;
-            await new Promise((resolve3) => setTimeout(resolve3, waitMs));
+            await new Promise((resolve4) => setTimeout(resolve4, waitMs));
             continue;
           }
-          await new Promise((resolve3) => setTimeout(resolve3, 1e3));
+          await new Promise((resolve4) => setTimeout(resolve4, 1e3));
         }
         if (!accessToken) {
           throw new Error(

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@contextstream/mcp-server",
   "mcpName": "io.github.contextstreamio/mcp-server",
-  "version": "0.4.34",
+  "version": "0.4.36",
   "description": "ContextStream MCP server - v0.4.x with consolidated domain tools (~11 tools, ~75% token reduction). Code context, memory, search, and AI tools.",
   "type": "module",
   "license": "MIT",
@@ -20,8 +20,8 @@
     "test-server": "tsx src/test-server.ts",
     "test-server:start": "node dist/test-server.js",
     "typecheck": "tsc --noEmit",
-    "test": "vitest run",
-    "test:watch": "vitest",
+    "test": "vitest run --config vitest.config.cjs",
+    "test:watch": "vitest --config vitest.config.cjs",
     "lint": "eslint src/",
     "lint:fix": "eslint src/ --fix",
     "format": "prettier --write src/",