@contextstream/mcp-server 0.3.17 → 0.3.18

package/README.md

@@ -23,7 +23,7 @@ One integration. Every AI editor. Persistent memory that never forgets.
 <div align="center">
 
 <a href="https://contextstream.io">
-  <img src="https://customer-vtx4jsqwkbsjpv5b.cloudflarestream.com/f083cfa709a679bd72ef48aca6fe0af2/thumbnails/thumbnail.gif?time=2s&height=600" alt="ContextStream Demo - AI that remembers across sessions" width="600" />
+  <img src="https://raw.githubusercontent.com/contextstream/mcp-server/main/mcp.gif.gif" alt="ContextStream Demo - AI that remembers across sessions" width="600" />
 </a>
 
 <sub>Your AI remembers decisions, preferences, and context — across sessions and tools.</sub>
@@ -119,6 +119,8 @@ CONTEXTSTREAM_API_KEY = "your_api_key"
 
 > Codex expects snake_case `mcp_servers` keys. After editing, fully restart Codex.
 
+> For workspace-pooled rate limiting (Team/Enterprise), the MCP server sends `X-Workspace-Id` based on the active repo/session (or explicit `workspace_id` in tool calls). You can optionally set `CONTEXTSTREAM_WORKSPACE_ID` as a fallback default, but it’s not required and isn’t a good fit if you frequently switch workspaces.
+
 ### AI Rules Files (Recommended)
 
 Adding rules files ensures your AI automatically uses ContextStream for memory on every conversation.
@@ -173,6 +175,19 @@ AI: "You prefer functional React components."
 
 ✨ **That's it. Your AI remembers now.**
 
+### 4. (Optional) Connect GitHub + Slack for richer context
+
+MCP gives your AI memory. Integrations make that memory richer by pulling in PRs, issues, and team conversations.
+
+- GitHub App setup + connect flow: https://contextstream.io/docs/integrations/github
+- Slack app setup + connect flow: https://contextstream.io/docs/integrations/slack
+
+After you connect, try prompts like:
+```
+"Search our Slack messages for the decision about rate limiting."
+"What did we decide in GitHub issues about the auth flow?"
+```
+
 ---
 
 ## Beyond Memory: Intelligence That Compounds

package/dist/index.js

@@ -4149,6 +4149,8 @@ async function request(config, path3, options = {}) {
   };
   if (apiKey) headers["X-API-Key"] = apiKey;
   if (jwt) headers["Authorization"] = `Bearer ${jwt}`;
+  const workspaceId = options.workspaceId || inferWorkspaceIdFromBody(options.body) || inferWorkspaceIdFromPath(apiPath) || config.defaultWorkspaceId;
+  if (workspaceId) headers["X-Workspace-Id"] = workspaceId;
   const fetchOptions = {
     method: options.method || (options.body ? "POST" : "GET"),
     headers
@@ -4192,8 +4194,12 @@ async function request(config, path3, options = {}) {
       payload = await response.text().catch(() => null);
     }
     if (!response.ok) {
-      const message = payload?.message || payload?.error || response.statusText;
-      lastError = new HttpError(response.status, message, payload);
+      const rateLimit = parseRateLimitHeaders(response.headers);
+      const enrichedPayload = attachRateLimit(payload, rateLimit);
+      const message = extractErrorMessage(enrichedPayload, response.statusText);
+      lastError = new HttpError(response.status, message, enrichedPayload);
+      const apiCode = extractErrorCode(enrichedPayload);
+      if (apiCode) lastError.code = apiCode;
       if (RETRYABLE_STATUSES.has(response.status) && attempt < maxRetries) {
         const retryAfter = response.headers.get("retry-after");
         const delay = retryAfter ? parseInt(retryAfter, 10) * 1e3 : baseDelay * Math.pow(2, attempt);
@@ -4206,6 +4212,72 @@ async function request(config, path3, options = {}) {
   }
   throw lastError || new HttpError(0, "Request failed after retries");
 }
+var UUID_RE = /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i;
+function isUuid(value) {
+  return typeof value === "string" && UUID_RE.test(value);
+}
+function inferWorkspaceIdFromBody(body) {
+  if (!body || typeof body !== "object") return void 0;
+  const maybe = body.workspace_id;
+  return isUuid(maybe) ? maybe : void 0;
+}
+function inferWorkspaceIdFromPath(apiPath) {
+  const qIndex = apiPath.indexOf("?");
+  if (qIndex >= 0) {
+    try {
+      const query = apiPath.slice(qIndex + 1);
+      const params = new URLSearchParams(query);
+      const ws = params.get("workspace_id");
+      if (isUuid(ws)) return ws;
+    } catch {
+    }
+  }
+  const match = apiPath.match(
+    /\/(?:workspaces|workspace)\/([0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12})/i
+  );
+  return match?.[1];
+}
+function parseRateLimitHeaders(headers) {
+  const limit = headers.get("X-RateLimit-Limit");
+  if (!limit) return null;
+  const retryAfter = headers.get("Retry-After");
+  return {
+    limit: parseInt(limit, 10),
+    remaining: parseInt(headers.get("X-RateLimit-Remaining") || "0", 10),
+    reset: parseInt(headers.get("X-RateLimit-Reset") || "0", 10),
+    scope: headers.get("X-RateLimit-Scope") || "unknown",
+    plan: headers.get("X-RateLimit-Plan") || "unknown",
+    group: headers.get("X-RateLimit-Group") || "default",
+    retryAfter: retryAfter ? parseInt(retryAfter, 10) : void 0
+  };
+}
+function attachRateLimit(payload, rateLimit) {
+  if (!rateLimit) return payload;
+  if (payload && typeof payload === "object") {
+    return { ...payload, rate_limit: rateLimit };
+  }
+  return { error: payload, rate_limit: rateLimit };
+}
+function extractErrorMessage(payload, fallback) {
+  if (!payload) return fallback;
+  const nested = payload?.error;
+  if (nested && typeof nested === "object" && typeof nested.message === "string") {
+    return nested.message;
+  }
+  if (typeof payload.message === "string") return payload.message;
+  if (typeof payload.error === "string") return payload.error;
+  if (typeof payload.detail === "string") return payload.detail;
+  return fallback;
+}
+function extractErrorCode(payload) {
+  if (!payload) return null;
+  const nested = payload?.error;
+  if (nested && typeof nested === "object" && typeof nested.code === "string" && nested.code.trim()) {
+    return nested.code.trim();
+  }
+  if (typeof payload.code === "string" && payload.code.trim()) return payload.code.trim();
+  return null;
+}
 
 // src/files.ts
 import * as fs from "fs";
@@ -4558,6 +4630,30 @@ var ContextStreamClient = class {
   constructor(config) {
     this.config = config;
   }
+  /**
+   * Update the client's default workspace/project IDs at runtime.
+   *
+   * This is useful for multi-workspace users: once a session is initialized
+   * (via repo mapping or explicit session_init), the MCP server can treat that
+   * workspace as the default for subsequent calls that don't explicitly include
+   * `workspace_id` in the request payload/path/query.
+   */
+  setDefaults(input) {
+    if (input.workspace_id) {
+      try {
+        uuidSchema.parse(input.workspace_id);
+        this.config.defaultWorkspaceId = input.workspace_id;
+      } catch {
+      }
+    }
+    if (input.project_id) {
+      try {
+        uuidSchema.parse(input.project_id);
+        this.config.defaultProjectId = input.project_id;
+      } catch {
+      }
+    }
+  }
   withDefaults(input) {
     const { defaultWorkspaceId, defaultProjectId } = this.config;
     return {
@@ -8246,6 +8342,11 @@ var SessionManager = class {
   markInitialized(context) {
     this.initialized = true;
     this.context = context;
+    const workspaceId = typeof context.workspace_id === "string" ? context.workspace_id : void 0;
+    const projectId = typeof context.project_id === "string" ? context.project_id : void 0;
+    if (workspaceId || projectId) {
+      this.client.setDefaults({ workspace_id: workspaceId, project_id: projectId });
+    }
   }
   /**
    * Set the folder path hint (can be passed from tools that know the workspace path)
@@ -8359,6 +8460,10 @@ var SessionManager = class {
       );
       this.initialized = true;
       this.context = context;
+      this.client.setDefaults({
+        workspace_id: typeof context.workspace_id === "string" ? context.workspace_id : void 0,
+        project_id: typeof context.project_id === "string" ? context.project_id : void 0
+      });
       console.error("[ContextStream] Workspace resolved:", context.workspace_name, "(source:", context.workspace_source, ")");
       const summary = this.buildContextSummary(context);
       console.error("[ContextStream] Auto-initialization complete");

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@contextstream/mcp-server",
-  "version": "0.3.17",
+  "version": "0.3.18",
   "description": "MCP server exposing ContextStream public API - code context, memory, search, and AI tools for developers",
   "type": "module",
   "license": "MIT",