@tambo-ai/react 0.67.1 → 0.68.0

package/dist/hooks/use-tambo-threads.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-tambo-threads.js","sourceRoot":"","sources":["../../src/hooks/use-tambo-threads.ts"],"names":[],"mappings":";;AAyBA,gDAiCC;AAxDD,4CAA8C;AAC9C,8EAAoE;AACpE,2DAAoD;AAUpD;;;;;;;;;;GAUG;AACH,SAAgB,kBAAkB,CAChC,EAAE,SAAS,KAA+B,EAAE,EAC5C,UAEI,EAAE;IAEN,MAAM,MAAM,GAAG,IAAA,sCAAc,GAAE,CAAC;IAChC,MAAM,EAAE,UAAU,EAAE,GAAG,IAAA,0BAAc,GAAE,CAAC;IACxC,MAAM,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,cAAc,EAAE,GAAG,IAAA,iCAAa,EAAC;QAClE,GAAI,OAA8C;QAClD,QAAQ,EAAE,CAAC,WAAW,CAAC;QACvB,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,OAAO,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,EAAE,CAAC;QACtD,CAAC;KACF,CAAC,CAAC;IACH,MAAM,gBAAgB,GAAG,SAAS,IAAI,gBAAgB,CAAC;IAEvD,MAAM,WAAW,GAAG,IAAA,iCAAa,EAAC;QAChC,GAAG,OAAO;QACV,OAAO,EAAE,CAAC,CAAC,gBAAgB;QAC3B,QAAQ,EAAE,CAAC,SAAS,EAAE,gBAAgB,EAAE,UAAU,CAAC;QACnD,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACtB,OAAO,IAAI,CAAC;YACd,CAAC;YACD,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,gBAAgB,EAAE;gBAClE,UAAU,EAAE,UAAU;aACvB,CAAC,CAAC;YACH,OAAO,UAAU,CAAC;QACpB,CAAC;KACF,CAAC,CAAC;IAEH,OAAO,gBAAgB,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,cAAc,EAAE,CAAC;AAC5E,CAAC","sourcesContent":["import type TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { UseQueryOptions } from \"@tanstack/react-query\";\nimport { useTamboThread } from \"../providers\";\nimport { useTamboClient } from \"../providers/tambo-client-provider\";\nimport { useTamboQuery } from \"./react-query-hooks\";\n\ninterface UseTamboThreadListConfig {\n  /**\n   * The projectId to get the threads for. If not provided, the current project\n   * will be used.\n   */\n  projectId?: string;\n}\n\n/**\n * Get all the threads for the specified project.\n *\n * If contextKey is empty, then all threads for the project will be returned.\n * If contextKey is not empty, then only the threads for the specified context\n * key will be returned. The contextKey is obtained from the TamboThreadProvider\n * via useTamboThread().\n * @param config - The config for the useTamboThreadList hook\n * @param config.projectId - The projectId to get the threads for\n * @returns The threads for the specified project and optional context key\n */\nexport function useTamboThreadList(\n  { projectId }: UseTamboThreadListConfig = {},\n  options: Partial<\n    UseQueryOptions<TamboAI.Beta.Threads.ThreadsOffsetAndLimit | null>\n  > = {},\n) {\n  const client = useTamboClient();\n  const { contextKey } = useTamboThread();\n  const { data: queriedProjectId, ...projectIdState } = useTamboQuery({\n    ...(options as unknown as UseQueryOptions<string>),\n    queryKey: [\"projectId\"],\n    queryFn: async () => {\n      return (await client.beta.projects.getCurrent()).id;\n    },\n  });\n  const currentProjectId = projectId ?? queriedProjectId;\n\n  const threadState = useTamboQuery({\n    ...options,\n    enabled: !!currentProjectId,\n    queryKey: [\"threads\", currentProjectId, contextKey],\n    queryFn: async () => {\n      if (!currentProjectId) {\n        return null;\n      }\n      const threadIter = await client.beta.threads.list(currentProjectId, {\n        contextKey: contextKey,\n      });\n      return threadIter;\n    },\n  });\n\n  return currentProjectId ? threadState : { data: null, ...projectIdState };\n}\n"]}
+{"version":3,"file":"use-tambo-threads.js","sourceRoot":"","sources":["../../src/hooks/use-tambo-threads.ts"],"names":[],"mappings":";;AAyBA,gDAiCC;AAxDD,4CAA8C;AAC9C,8EAAoE;AACpE,2DAAoD;AAUpD;;;;;;;;;;GAUG;AACH,SAAgB,kBAAkB,CAChC,EAAE,SAAS,KAA+B,EAAE,EAC5C,UAEI,EAAE;IAEN,MAAM,MAAM,GAAG,IAAA,sCAAc,GAAE,CAAC;IAChC,MAAM,EAAE,UAAU,EAAE,GAAG,IAAA,0BAAc,GAAE,CAAC;IACxC,MAAM,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,cAAc,EAAE,GAAG,IAAA,iCAAa,EAAC;QAClE,GAAI,OAA8C;QAClD,QAAQ,EAAE,CAAC,WAAW,CAAC;QACvB,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,OAAO,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,EAAE,CAAC;QACtD,CAAC;KACF,CAAC,CAAC;IACH,MAAM,gBAAgB,GAAG,SAAS,IAAI,gBAAgB,CAAC;IAEvD,MAAM,WAAW,GAAG,IAAA,iCAAa,EAAC;QAChC,GAAG,OAAO;QACV,OAAO,EAAE,CAAC,CAAC,gBAAgB;QAC3B,QAAQ,EAAE,CAAC,SAAS,EAAE,gBAAgB,EAAE,UAAU,CAAC;QACnD,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACtB,OAAO,IAAI,CAAC;YACd,CAAC;YACD,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,gBAAgB,EAAE;gBAClE,UAAU;aACX,CAAC,CAAC;YACH,OAAO,UAAU,CAAC;QACpB,CAAC;KACF,CAAC,CAAC;IAEH,OAAO,gBAAgB,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,cAAc,EAAE,CAAC;AAC5E,CAAC","sourcesContent":["import type TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { UseQueryOptions } from \"@tanstack/react-query\";\nimport { useTamboThread } from \"../providers\";\nimport { useTamboClient } from \"../providers/tambo-client-provider\";\nimport { useTamboQuery } from \"./react-query-hooks\";\n\ninterface UseTamboThreadListConfig {\n  /**\n   * The projectId to get the threads for. If not provided, the current project\n   * will be used.\n   */\n  projectId?: string;\n}\n\n/**\n * Get all the threads for the specified project.\n *\n * If contextKey is empty, then all threads for the project will be returned.\n * If contextKey is not empty, then only the threads for the specified context\n * key will be returned. The contextKey is obtained from the TamboThreadProvider\n * via useTamboThread().\n * @param config - The config for the useTamboThreadList hook\n * @param config.projectId - The projectId to get the threads for\n * @returns The threads for the specified project and optional context key\n */\nexport function useTamboThreadList(\n  { projectId }: UseTamboThreadListConfig = {},\n  options: Partial<\n    UseQueryOptions<TamboAI.Beta.Threads.ThreadsOffsetAndLimit | null>\n  > = {},\n) {\n  const client = useTamboClient();\n  const { contextKey } = useTamboThread();\n  const { data: queriedProjectId, ...projectIdState } = useTamboQuery({\n    ...(options as unknown as UseQueryOptions<string>),\n    queryKey: [\"projectId\"],\n    queryFn: async () => {\n      return (await client.beta.projects.getCurrent()).id;\n    },\n  });\n  const currentProjectId = projectId ?? queriedProjectId;\n\n  const threadState = useTamboQuery({\n    ...options,\n    enabled: !!currentProjectId,\n    queryKey: [\"threads\", currentProjectId, contextKey],\n    queryFn: async () => {\n      if (!currentProjectId) {\n        return null;\n      }\n      const threadIter = await client.beta.threads.list(currentProjectId, {\n        contextKey,\n      });\n      return threadIter;\n    },\n  });\n\n  return currentProjectId ? threadState : { data: null, ...projectIdState };\n}\n"]}

package/dist/mcp/mcp-constants.d.ts

@@ -0,0 +1,19 @@
+/**
+ * Server type enum for distinguishing different kinds of MCP servers.
+ * Used to determine how resources should be resolved.
+ */
+export declare const ServerType: {
+    /** Client-side MCP server passed via mcpServers prop */
+    readonly BROWSER_SIDE: "browser-side";
+    /** Internal Tambo MCP server at /mcp endpoint (server-side MCP) */
+    readonly TAMBO_INTERNAL: "tambo-internal";
+    /** Virtual server representing browser-only registry resources */
+    readonly TAMBO_REGISTRY: "tambo-registry";
+};
+export type ServerType = (typeof ServerType)[keyof typeof ServerType];
+/**
+ * Synthetic server key for local registry resources.
+ * Used to give registry resources the same `@serverKey:uri` format as MCP resources.
+ */
+export declare const REGISTRY_SERVER_KEY = "registry";
+//# sourceMappingURL=mcp-constants.d.ts.map

package/dist/mcp/mcp-constants.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-constants.d.ts","sourceRoot":"","sources":["../../src/mcp/mcp-constants.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,eAAO,MAAM,UAAU;IACrB,wDAAwD;;IAExD,mEAAmE;;IAEnE,kEAAkE;;CAE1D,CAAC;AAEX,MAAM,MAAM,UAAU,GAAG,CAAC,OAAO,UAAU,CAAC,CAAC,MAAM,OAAO,UAAU,CAAC,CAAC;AAEtE;;;GAGG;AACH,eAAO,MAAM,mBAAmB,aAAa,CAAC"}

package/dist/mcp/mcp-constants.js

@@ -0,0 +1,21 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.REGISTRY_SERVER_KEY = exports.ServerType = void 0;
+/**
+ * Server type enum for distinguishing different kinds of MCP servers.
+ * Used to determine how resources should be resolved.
+ */
+exports.ServerType = {
+    /** Client-side MCP server passed via mcpServers prop */
+    BROWSER_SIDE: "browser-side",
+    /** Internal Tambo MCP server at /mcp endpoint (server-side MCP) */
+    TAMBO_INTERNAL: "tambo-internal",
+    /** Virtual server representing browser-only registry resources */
+    TAMBO_REGISTRY: "tambo-registry",
+};
+/**
+ * Synthetic server key for local registry resources.
+ * Used to give registry resources the same `@serverKey:uri` format as MCP resources.
+ */
+exports.REGISTRY_SERVER_KEY = "registry";
+//# sourceMappingURL=mcp-constants.js.map

package/dist/mcp/mcp-constants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-constants.js","sourceRoot":"","sources":["../../src/mcp/mcp-constants.ts"],"names":[],"mappings":";;;AAAA;;;GAGG;AACU,QAAA,UAAU,GAAG;IACxB,wDAAwD;IACxD,YAAY,EAAE,cAAc;IAC5B,mEAAmE;IACnE,cAAc,EAAE,gBAAgB;IAChC,kEAAkE;IAClE,cAAc,EAAE,gBAAgB;CACxB,CAAC;AAIX;;;GAGG;AACU,QAAA,mBAAmB,GAAG,UAAU,CAAC","sourcesContent":["/**\n * Server type enum for distinguishing different kinds of MCP servers.\n * Used to determine how resources should be resolved.\n */\nexport const ServerType = {\n  /** Client-side MCP server passed via mcpServers prop */\n  BROWSER_SIDE: \"browser-side\",\n  /** Internal Tambo MCP server at /mcp endpoint (server-side MCP) */\n  TAMBO_INTERNAL: \"tambo-internal\",\n  /** Virtual server representing browser-only registry resources */\n  TAMBO_REGISTRY: \"tambo-registry\",\n} as const;\n\nexport type ServerType = (typeof ServerType)[keyof typeof ServerType];\n\n/**\n * Synthetic server key for local registry resources.\n * Used to give registry resources the same `@serverKey:uri` format as MCP resources.\n */\nexport const REGISTRY_SERVER_KEY = \"registry\";\n"]}

package/dist/mcp/mcp-hooks.d.ts

@@ -1,6 +1,6 @@
 import { type ListPromptsResult, type ListResourcesResult } from "@modelcontextprotocol/sdk/types.js";
 import { UseQueryResult } from "@tanstack/react-query";
-import { type ConnectedMcpServer } from "./tambo-mcp-provider";
+import { type ConnectedMcpServer, type McpServer } from "./tambo-mcp-provider";
 export type ListPromptItem = ListPromptsResult["prompts"][number];
 export interface ListPromptEntry {
     server: ConnectedMcpServer;
@@ -77,6 +77,11 @@ export declare function useTamboMcpPromptList(): {
     isLoading: boolean;
     refetch: () => Promise<void>;
 };
+/**
+ * Type guard for narrowing McpServer to ConnectedMcpServer.
+ * A connected server has a non-null client.
+ */
+export declare function isConnectedMcpServer(server: McpServer): server is ConnectedMcpServer;
 /**
  * Hook to get the prompt for the specified name.
  * @param promptName - The name of the prompt to get. Can be prefixed with serverKey (e.g., "linear:issue") or unprefixed.
@@ -183,7 +188,29 @@ export declare function useTamboMcpPrompt(promptName: string | undefined, args?:
  * @returns The resources from MCP servers and the local registry, including the server that the resource was found on (null for registry resources).
  */
 export declare function useTamboMcpResourceList(): {
-    data: (RegistryResourceEntry | {
+    data: ({
+        resource: {
+            uri: string;
+            name: string;
+            description?: string | undefined;
+            mimeType?: string | undefined;
+            annotations?: {
+                audience?: ("user" | "assistant")[] | undefined;
+                priority?: number | undefined;
+                lastModified?: string | undefined;
+            } | undefined;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+            icons?: {
+                src: string;
+                mimeType?: string | undefined;
+                sizes?: string[] | undefined;
+            }[] | undefined;
+            title?: string | undefined;
+        };
+        server: null;
+    } | {
         resource: {
             uri: string;
             name: string;
@@ -219,7 +246,9 @@ export declare function useTamboMcpResourceList(): {
 };
 /**
  * Hook to get the resource for the specified URI.
- * @param resourceUri - The URI of the resource to get. Can be prefixed with serverKey (e.g., "linear:file://foo") for MCP resources, or unprefixed for registry resources.
+ * @param resourceUri - The URI of the resource to get. Must be prefixed:
+ *   - MCP resources: prefixed with serverKey (e.g., "linear:file://foo")
+ *   - Registry resources: prefixed with "registry:" (e.g., "registry:file://bar")
  * @returns The resource for the specified URI.
  */
 export declare function useTamboMcpResource(resourceUri: string | undefined): UseQueryResult<{

package/dist/mcp/mcp-hooks.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"mcp-hooks.d.ts","sourceRoot":"","sources":["../../src/mcp/mcp-hooks.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,iBAAiB,EACtB,KAAK,mBAAmB,EAEzB,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAGvD,OAAO,EACL,KAAK,kBAAkB,EAGxB,MAAM,sBAAsB,CAAC;AAE9B,MAAM,MAAM,cAAc,GAAG,iBAAiB,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC;AAClE,MAAM,WAAW,eAAe;IAE9B,MAAM,EAAE,kBAAkB,CAAC;IAC3B,MAAM,EAAE,cAAc,CAAC;CACxB;AAED,MAAM,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC,CAAC;AAExE;;;;;;GAMG;AACH,MAAM,WAAW,qBAAqB;IACpC,MAAM,EAAE,IAAI,CAAC;IACb,QAAQ,EAAE,gBAAgB,CAAC;CAC5B;AAED;;;;;GAKG;AACH,MAAM,WAAW,gBAAgB;IAC/B,MAAM,EAAE,kBAAkB,CAAC;IAC3B,QAAQ,EAAE,gBAAgB,CAAC;CAC5B;AAED;;;;;GAKG;AACH,MAAM,MAAM,iBAAiB,GAAG,qBAAqB,GAAG,gBAAgB,CAAC;AAEzE;;;GAGG;AACH,wBAAgB,kBAAkB,CAChC,KAAK,EAAE,iBAAiB,GACvB,KAAK,IAAI,gBAAgB,CAE3B;AAED;;;GAGG;AACH,wBAAgB,qBAAqB;;;;;;;;;;;;;;;;;;;;gBAnD3B,kBAAkB;;WAgGnB,KAAK,GAAG,IAAI;YACX,KAAK,EAAE;eACJ,OAAO;eACP,OAAO;aACT,OAAO;cACN,OAAO;kBACH,OAAO;gBACT,OAAO;eACR,OAAO;aACT,MAAM,OAAO,CAAC,IAAI,CAAC;EAb7B;AAwDD;;;;;GAKG;AACH,wBAAgB,iBAAiB,CAC/B,UAAU,EAAE,MAAM,GAAG,SAAS,EAC9B,IAAI,GAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBA0ClC;AAED;;;GAGG;AACH,wBAAgB,uBAAuB;;;;;;;;;;;;;;;;;;;;;;gBAnL7B,kBAAkB;;WAuEnB,KAAK,GAAG,IAAI;YACX,KAAK,EAAE;eACJ,OAAO;eACP,OAAO;aACT,OAAO;cACN,OAAO;kBACH,OAAO;gBACT,OAAO;eACR,OAAO;aACT,MAAM,OAAO,CAAC,IAAI,CAAC;EA6L7B;AAED;;;;GAIG;AACH,wBAAgB,mBAAmB,CAAC,WAAW,EAAE,MAAM,GAAG,SAAS;;;;;;;;;;;;;;;;;;;;;;;;iBAmElE"}
+{"version":3,"file":"mcp-hooks.d.ts","sourceRoot":"","sources":["../../src/mcp/mcp-hooks.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,iBAAiB,EACtB,KAAK,mBAAmB,EAEzB,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAIvD,OAAO,EACL,KAAK,kBAAkB,EACvB,KAAK,SAAS,EAEf,MAAM,sBAAsB,CAAC;AAE9B,MAAM,MAAM,cAAc,GAAG,iBAAiB,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC;AAClE,MAAM,WAAW,eAAe;IAE9B,MAAM,EAAE,kBAAkB,CAAC;IAC3B,MAAM,EAAE,cAAc,CAAC;CACxB;AAED,MAAM,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC,CAAC;AAExE;;;;;;GAMG;AACH,MAAM,WAAW,qBAAqB;IACpC,MAAM,EAAE,IAAI,CAAC;IACb,QAAQ,EAAE,gBAAgB,CAAC;CAC5B;AAED;;;;;GAKG;AACH,MAAM,WAAW,gBAAgB;IAC/B,MAAM,EAAE,kBAAkB,CAAC;IAC3B,QAAQ,EAAE,gBAAgB,CAAC;CAC5B;AAED;;;;;GAKG;AACH,MAAM,MAAM,iBAAiB,GAAG,qBAAqB,GAAG,gBAAgB,CAAC;AAEzE;;;GAGG;AACH,wBAAgB,kBAAkB,CAChC,KAAK,EAAE,iBAAiB,GACvB,KAAK,IAAI,gBAAgB,CAE3B;AAED;;;GAGG;AACH,wBAAgB,qBAAqB;;;;;;;;;;;;;;;;;;;;gBAnD3B,kBAAkB;;WAgGnB,KAAK,GAAG,IAAI;YACX,KAAK,EAAE;eACJ,OAAO;eACP,OAAO;aACT,OAAO;cACN,OAAO;kBACH,OAAO;gBACT,OAAO;eACR,OAAO;aACT,MAAM,OAAO,CAAC,IAAI,CAAC;EAb7B;AAmDD;;;GAGG;AACH,wBAAgB,oBAAoB,CAClC,MAAM,EAAE,SAAS,GAChB,MAAM,IAAI,kBAAkB,CAE9B;AAED;;;;;GAKG;AACH,wBAAgB,iBAAiB,CAC/B,UAAU,EAAE,MAAM,GAAG,SAAS,EAC9B,IAAI,GAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBA0ClC;AAED;;;GAGG;AACH,wBAAgB,uBAAuB;;;;;;;;;;;;;;;;;;;;;;gBAnM7B,IAAI;;;;;;;;;;;;;;;;;;;;;;gBAWJ,kBAAkB;;WAuEnB,KAAK,GAAG,IAAI;YACX,KAAK,EAAE;eACJ,OAAO;eACP,OAAO;aACT,OAAO;cACN,OAAO;kBACH,OAAO;gBACT,OAAO;eACR,OAAO;aACT,MAAM,OAAO,CAAC,IAAI,CAAC;EAwM7B;AAED;;;;;;GAMG;AACH,wBAAgB,mBAAmB,CAAC,WAAW,EAAE,MAAM,GAAG,SAAS;;;;;;;;;;;;;;;;;;;;;;;;iBAoElE"}

package/dist/mcp/mcp-hooks.js

@@ -2,11 +2,13 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.isMcpResourceEntry = isMcpResourceEntry;
 exports.useTamboMcpPromptList = useTamboMcpPromptList;
+exports.isConnectedMcpServer = isConnectedMcpServer;
 exports.useTamboMcpPrompt = useTamboMcpPrompt;
 exports.useTamboMcpResourceList = useTamboMcpResourceList;
 exports.useTamboMcpResource = useTamboMcpResource;
 const hooks_1 = require("../hooks");
 const tambo_registry_provider_1 = require("../providers/tambo-registry-provider");
+const mcp_constants_1 = require("./mcp-constants");
 const tambo_mcp_provider_1 = require("./tambo-mcp-provider");
 /**
  * Type guard for narrowing a `ListResourceEntry` to an MCP-backed resource
@@ -86,7 +88,10 @@ function combineArrayResults(results) {
         },
     };
 }
-// Type guard for narrowing to connected servers
+/**
+ * Type guard for narrowing McpServer to ConnectedMcpServer.
+ * A connected server has a non-null client.
+ */
 function isConnectedMcpServer(server) {
     return "client" in server && server.client != null;
 }
@@ -189,12 +194,18 @@ function useTamboMcpResourceList() {
             }));
             // Merge static resources with query results (registry resources first)
             const allData = [...staticEntries, ...combined.data];
-            // Apply serverKey prefix to ALL MCP resources (breaking change)
-            // Registry resources (server: null) are never prefixed
+            // Apply serverKey prefix to ALL resources for unified @serverKey:uri format
+            // Registry resources get REGISTRY_SERVER_KEY prefix, MCP resources get their serverKey
             const prefixedData = allData.map((entry) => {
                 if (entry.server === null) {
-                    // Registry resource - no prefix
-                    return entry;
+                    // Registry resource - prefix with REGISTRY_SERVER_KEY
+                    return {
+                        ...entry,
+                        resource: {
+                            ...entry.resource,
+                            uri: `${mcp_constants_1.REGISTRY_SERVER_KEY}:${entry.resource.uri}`,
+                        },
+                    };
                 }
                 // MCP resource - always prefix with serverKey
                 return {
@@ -215,7 +226,9 @@ function useTamboMcpResourceList() {
 }
 /**
  * Hook to get the resource for the specified URI.
- * @param resourceUri - The URI of the resource to get. Can be prefixed with serverKey (e.g., "linear:file://foo") for MCP resources, or unprefixed for registry resources.
+ * @param resourceUri - The URI of the resource to get. Must be prefixed:
+ *   - MCP resources: prefixed with serverKey (e.g., "linear:file://foo")
+ *   - Registry resources: prefixed with "registry:" (e.g., "registry:file://bar")
  * @returns The resource for the specified URI.
  */
 function useTamboMcpResource(resourceUri) {
@@ -223,30 +236,30 @@ function useTamboMcpResource(resourceUri) {
     const { data: resourceEntries } = useTamboMcpResourceList();
     // Find which server/source has the resource
     const resourceEntry = resourceEntries?.find((entry) => entry.resource.uri === resourceUri);
-    // Determine if this is a known registry resource or MCP resource
-    const isKnownEntry = resourceEntry != null;
-    const isRegistryResource = isKnownEntry && resourceEntry.server === null;
-    const mcpServer = isKnownEntry && resourceEntry.server != null ? resourceEntry.server : null;
-    // Strip the prefix to get the original resource URI for the MCP server call
-    // Only strip if we found a matching resource entry with an MCP server
-    const originalResourceUri = isKnownEntry && mcpServer
-        ? resourceUri?.replace(`${mcpServer.serverKey}:`, "")
-        : resourceUri;
-    // Check if we can fetch this resource. Registry resources can always be
-    // fetched when a resourceSource exists, even if they haven't appeared in a
-    // previous listResources result.
-    const hasRegistrySource = resourceSource != null;
-    const hasConnectedMcpServer = isKnownEntry && mcpServer != null && isConnectedMcpServer(mcpServer);
-    const canFetchResource = Boolean(resourceUri && (hasRegistrySource || hasConnectedMcpServer));
-    let locationKey;
-    if (isRegistryResource || (!isKnownEntry && hasRegistrySource)) {
-        locationKey = "registry";
+    // Determine if this is a registry resource or MCP resource
+    const isRegistryResource = resourceEntry?.server === null;
+    const mcpServer = resourceEntry?.server ?? null;
+    // Check if the URI has the registry prefix
+    const hasRegistryPrefix = Boolean(resourceUri?.startsWith(`${mcp_constants_1.REGISTRY_SERVER_KEY}:`));
+    // Strip the prefix to get the original resource URI for fetching
+    let originalResourceUri;
+    if (isRegistryResource || hasRegistryPrefix) {
+        const prefixLen = mcp_constants_1.REGISTRY_SERVER_KEY.length + 1; // +1 for the colon
+        originalResourceUri = resourceUri?.slice(prefixLen);
     }
     else if (mcpServer) {
-        locationKey = mcpServer.key;
+        const prefixLen = mcpServer.serverKey.length + 1; // +1 for the colon
+        originalResourceUri = resourceUri?.slice(prefixLen);
     }
+    // Check if we can fetch this resource
+    const hasRegistrySource = resourceSource != null;
+    const hasConnectedMcpServer = mcpServer != null && isConnectedMcpServer(mcpServer);
+    const canFetchFromRegistry = hasRegistrySource && (isRegistryResource || hasRegistryPrefix);
+    const canFetchResource = Boolean(resourceUri && (canFetchFromRegistry || hasConnectedMcpServer));
+    const locationKey = isRegistryResource || hasRegistryPrefix
+        ? mcp_constants_1.REGISTRY_SERVER_KEY
+        : mcpServer?.key;
     return (0, hooks_1.useTamboQuery)({
-        // Include server identity or "registry" to prevent stale cache hits
         queryKey: ["resource", resourceUri, locationKey],
         enabled: canFetchResource,
         queryFn: async () => {
@@ -254,9 +267,7 @@ function useTamboMcpResource(resourceUri) {
                 return null;
             }
             // Registry resource: use resourceSource.getResource
-            // If this is not a known entry but we have a resourceSource, treat it as
-            // a registry resource by default.
-            if (resourceSource && (!isKnownEntry || isRegistryResource)) {
+            if (resourceSource && (isRegistryResource || hasRegistryPrefix)) {
                 const result = await resourceSource.getResource(originalResourceUri);
                 return result ?? null;
             }

package/dist/mcp/mcp-hooks.js.map

@@ -1 +1 @@
-{"version":3,"file":"mcp-hooks.js","sourceRoot":"","sources":["../../src/mcp/mcp-hooks.ts"],"names":[],"mappings":";;AA2DA,gDAIC;AAMD,sDAyCC;AA8DD,8CA4CC;AAMD,0DA0FC;AAOD,kDAmEC;AA3XD,oCAA0D;AAC1D,kFAAwE;AACxE,6DAI8B;AA0C9B;;;GAGG;AACH,SAAgB,kBAAkB,CAChC,KAAwB;IAExB,OAAO,KAAK,CAAC,MAAM,KAAK,IAAI,IAAI,oBAAoB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;AACrE,CAAC;AAED;;;GAGG;AACH,SAAgB,qBAAqB;IACnC,MAAM,UAAU,GAAG,IAAA,uCAAkB,GAAE,CAAC;IAExC,MAAM,OAAO,GAAG,IAAA,uBAAe,EAAC;QAC9B,OAAO,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;YACtC,QAAQ,EAAE,CAAC,aAAa,EAAE,SAAS,CAAC,GAAG,CAAC;YACxC,oDAAoD;YACpD,OAAO,EAAE,oBAAoB,CAAC,SAAS,CAAC;YACxC,OAAO,EAAE,KAAK,IAAgC,EAAE;gBAC9C,6DAA6D;gBAC7D,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC;oBAAE,OAAO,EAAE,CAAC;gBAEhD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;gBAC3D,MAAM,OAAO,GAAqB,MAAM,EAAE,OAAO,IAAI,EAAE,CAAC;gBACxD,qEAAqE;gBACrE,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;oBAC9C,MAAM,EAAE,SAAS;oBACjB,MAAM;iBACP,CAAC,CAAC,CAAC;gBACJ,OAAO,cAAc,CAAC;YACxB,CAAC;SACF,CAAC,CAAC;QACH,OAAO,EAAE,CAAC,OAAO,EAAE,EAAE;YACnB,MAAM,QAAQ,GAAG,mBAAmB,CAAC,OAAO,CAAC,CAAC;YAE9C,gGAAgG;YAChG,iEAAiE;YACjE,OAAO;gBACL,GAAG,QAAQ;gBACX,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;oBAClC,GAAG,KAAK;oBACR,MAAM,EAAE;wBACN,GAAG,KAAK,CAAC,MAAM;wBACf,IAAI,EAAE,GAAG,KAAK,CAAC,MAAM,CAAC,SAAS,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE;qBACvD;iBACF,CAAC,CAAC;aACJ,CAAC;QACJ,CAAC;KACF,CAAC,CAAC;IAEH,OAAO,OAAO,CAAC;AACjB,CAAC;AACD,2CAA2C;AAC3C,SAAS,mBAAmB,CAAI,OAA8B;IAa5D,MAAM,MAAM,GAAG,OAAO;SACnB,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC;SAClC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAEjC,8EAA8E;IAC9E,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAC/B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,KAAK,MAAM,IAAI,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,OAAO,CAC5D,CAAC;IAEF,OAAO;QACL,oDAAoD;QACpD,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,CAC/B,MAAM,CAAC,SAAS,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAClE;QACD,qFAAqF;QACrF,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,IAAI;QACxB,MAAM;QACN,SAAS,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC;QACxD,SAAS,EACP,UAAU,CAAC,MAAM,GAAG,CAAC,IAAI,UAAU,CAAC,KAAK,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC;QACzE,OAAO,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC;QAC1B,QAAQ,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC;QACtD,YAAY,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC;QAC9D,UAAU,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC;QAC1D,SAAS,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC;QACxD,sDAAsD;QACtD,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,MAAM,OAAO,CAAC,GAAG,CACf,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE;gBACtB,MAAM,CAAC,CAAC,OAAO,EAAE,CAAC;YACpB,CAAC,CAAC,CACH,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED,gDAAgD;AAChD,SAAS,oBAAoB,CAAC,MAAiB;IAC7C,OAAO,QAAQ,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC;AACrD,CAAC;AAED;;;;;GAKG;AACH,SAAgB,iBAAiB,CAC/B,UAA8B,EAC9B,OAA+B,EAAE;IAEjC,yCAAyC;IACzC,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,qBAAqB,EAAE,CAAC;IACxD,MAAM,WAAW,GAAG,aAAa,EAAE,IAAI,CACrC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,KAAK,UAAU,CAC9C,CAAC;IACF,gFAAgF;IAChF,+BAA+B;IAC/B,MAAM,SAAS,GAAG,WAAW,EAAE,MAAM,CAAC;IAEtC,2EAA2E;IAC3E,MAAM,kBAAkB,GAAG,UAAU,EAAE,QAAQ,CAAC,GAAG,CAAC;QAClD,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;QAC1C,CAAC,CAAC,UAAU,CAAC;IAEf,4EAA4E;IAC5E,MAAM,iBAAiB,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC;SACxC,IAAI,EAAE;SACN,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,CAAU,CAAC,CAAC;IACrC,OAAO,IAAA,qBAAa,EAAC;QACnB,sEAAsE;QACtE,QAAQ,EAAE,CAAC,YAAY,EAAE,UAAU,EAAE,SAAS,EAAE,GAAG,EAAE,iBAAiB,CAAC;QACvE,6DAA6D;QAC7D,OAAO,EAAE,OAAO,CACd,UAAU,IAAI,SAAS,IAAI,oBAAoB,CAAC,SAAS,CAAC,CAC3D;QACD,OAAO,EAAE,KAAK,IAAqC,EAAE;YACnD,IACE,CAAC,kBAAkB;gBACnB,CAAC,SAAS;gBACV,CAAC,oBAAoB,CAAC,SAAS,CAAC,EAChC,CAAC;gBACD,OAAO,IAAI,CAAC;YACd,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC;gBACrD,IAAI,EAAE,kBAAkB;gBACxB,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;YACH,OAAO,MAAM,IAAI,IAAI,CAAC,CAAC,iEAAiE;QAC1F,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED;;;GAGG;AACH,SAAgB,uBAAuB;IACrC,MAAM,UAAU,GAAG,IAAA,uCAAkB,GAAE,CAAC;IACxC,MAAM,EAAE,SAAS,EAAE,eAAe,EAAE,cAAc,EAAE,GAAG,IAAA,0CAAgB,GAAE,CAAC;IAE1E,wEAAwE;IACxE,MAAM,YAAY,GAAG;QACnB,qBAAqB;QACrB,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;YAChC,QAAQ,EAAE,CAAC,eAAe,EAAE,SAAS,CAAC,GAAG,CAAC;YAC1C,oDAAoD;YACpD,OAAO,EAAE,oBAAoB,CAAC,SAAS,CAAC;YACxC,OAAO,EAAE,KAAK,IAAiC,EAAE;gBAC/C,6DAA6D;gBAC7D,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC;oBAAE,OAAO,EAAE,CAAC;gBAEhD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;gBAC7D,MAAM,SAAS,GAAuB,MAAM,EAAE,SAAS,IAAI,EAAE,CAAC;gBAC9D,uEAAuE;gBACvE,MAAM,eAAe,GAAuB,SAAS,CAAC,GAAG,CACvD,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;oBACb,MAAM,EAAE,SAAS;oBACjB,QAAQ;iBACT,CAAC,CACH,CAAC;gBACF,OAAO,eAAe,CAAC;YACzB,CAAC;SACF,CAAC,CAAC;QACH,4CAA4C;QAC5C,GAAG,CAAC,cAAc;YAChB,CAAC,CAAC;gBACE;oBACE,QAAQ,EAAE,CAAC,oBAAoB,EAAE,SAAS,CAAC;oBAC3C,OAAO,EAAE,IAAI;oBACb,OAAO,EAAE,KAAK,IAAsC,EAAE;wBACpD,IAAI,CAAC,cAAc;4BAAE,OAAO,EAAE,CAAC;wBAC/B,MAAM,SAAS,GAAG,MAAM,cAAc,CAAC,aAAa,EAAE,CAAC;wBACvD,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;4BAClC,MAAM,EAAE,IAAI;4BACZ,QAAQ;yBACT,CAAC,CAAC,CAAC;oBACN,CAAC;iBACF;aACF;YACH,CAAC,CAAC,EAAE,CAAC;KACR,CAAC;IAEF,MAAM,OAAO,GAAG,IAAA,uBAAe,EAAC;QAC9B,OAAO,EAAE,YAAY;QACrB,OAAO,EAAE,CAAC,OAAO,EAAE,EAAE;YACnB,yEAAyE;YACzE,MAAM,QAAQ,GAAG,mBAAmB,CAClC,OAAgD,CACjD,CAAC;YAEF,kDAAkD;YAClD,MAAM,aAAa,GAA4B,eAAe,CAAC,GAAG,CAChE,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;gBACb,MAAM,EAAE,IAAI;gBACZ,QAAQ;aACT,CAAC,CACH,CAAC;YAEF,uEAAuE;YACvE,MAAM,OAAO,GAAG,CAAC,GAAG,aAAa,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;YAErD,gEAAgE;YAChE,uDAAuD;YACvD,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;gBACzC,IAAI,KAAK,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC;oBAC1B,gCAAgC;oBAChC,OAAO,KAAK,CAAC;gBACf,CAAC;gBACD,8CAA8C;gBAC9C,OAAO;oBACL,GAAG,KAAK;oBACR,QAAQ,EAAE;wBACR,GAAG,KAAK,CAAC,QAAQ;wBACjB,GAAG,EAAE,GAAG,KAAK,CAAC,MAAM,CAAC,SAAS,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,EAAE;qBACvD;iBACF,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,OAAO;gBACL,GAAG,QAAQ;gBACX,IAAI,EAAE,YAAY;aACnB,CAAC;QACJ,CAAC;KACF,CAAC,CAAC;IAEH,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;;;GAIG;AACH,SAAgB,mBAAmB,CAAC,WAA+B;IACjE,MAAM,EAAE,cAAc,EAAE,GAAG,IAAA,0CAAgB,GAAE,CAAC;IAC9C,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,GAAG,uBAAuB,EAAE,CAAC;IAE5D,4CAA4C;IAC5C,MAAM,aAAa,GAAG,eAAe,EAAE,IAAI,CACzC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,KAAK,WAAW,CAC9C,CAAC;IAEF,iEAAiE;IACjE,MAAM,YAAY,GAAG,aAAa,IAAI,IAAI,CAAC;IAC3C,MAAM,kBAAkB,GAAG,YAAY,IAAI,aAAa,CAAC,MAAM,KAAK,IAAI,CAAC;IACzE,MAAM,SAAS,GACb,YAAY,IAAI,aAAa,CAAC,MAAM,IAAI,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC;IAE7E,4EAA4E;IAC5E,sEAAsE;IACtE,MAAM,mBAAmB,GACvB,YAAY,IAAI,SAAS;QACvB,CAAC,CAAC,WAAW,EAAE,OAAO,CAAC,GAAG,SAAS,CAAC,SAAS,GAAG,EAAE,EAAE,CAAC;QACrD,CAAC,CAAC,WAAW,CAAC;IAElB,wEAAwE;IACxE,2EAA2E;IAC3E,iCAAiC;IACjC,MAAM,iBAAiB,GAAG,cAAc,IAAI,IAAI,CAAC;IACjD,MAAM,qBAAqB,GACzB,YAAY,IAAI,SAAS,IAAI,IAAI,IAAI,oBAAoB,CAAC,SAAS,CAAC,CAAC;IACvE,MAAM,gBAAgB,GAAG,OAAO,CAC9B,WAAW,IAAI,CAAC,iBAAiB,IAAI,qBAAqB,CAAC,CAC5D,CAAC;IAEF,IAAI,WAA+B,CAAC;IACpC,IAAI,kBAAkB,IAAI,CAAC,CAAC,YAAY,IAAI,iBAAiB,CAAC,EAAE,CAAC;QAC/D,WAAW,GAAG,UAAU,CAAC;IAC3B,CAAC;SAAM,IAAI,SAAS,EAAE,CAAC;QACrB,WAAW,GAAG,SAAS,CAAC,GAAG,CAAC;IAC9B,CAAC;IAED,OAAO,IAAA,qBAAa,EAAC;QACnB,oEAAoE;QACpE,QAAQ,EAAE,CAAC,UAAU,EAAE,WAAW,EAAE,WAAW,CAAC;QAChD,OAAO,EAAE,gBAAgB;QACzB,OAAO,EAAE,KAAK,IAAwC,EAAE;YACtD,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBACzB,OAAO,IAAI,CAAC;YACd,CAAC;YAED,oDAAoD;YACpD,yEAAyE;YACzE,kCAAkC;YAClC,IAAI,cAAc,IAAI,CAAC,CAAC,YAAY,IAAI,kBAAkB,CAAC,EAAE,CAAC;gBAC5D,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,WAAW,CAAC,mBAAmB,CAAC,CAAC;gBACrE,OAAO,MAAM,IAAI,IAAI,CAAC;YACxB,CAAC;YAED,+BAA+B;YAC/B,IAAI,SAAS,IAAI,oBAAoB,CAAC,SAAS,CAAC,EAAE,CAAC;gBACjD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC;oBACxD,GAAG,EAAE,mBAAmB;iBACzB,CAAC,CAAC;gBACH,OAAO,MAAM,IAAI,IAAI,CAAC;YACxB,CAAC;YAED,OAAO,IAAI,CAAC;QACd,CAAC;KACF,CAAC,CAAC;AACL,CAAC","sourcesContent":["import {\n  GetPromptResult,\n  type ListPromptsResult,\n  type ListResourcesResult,\n  type ReadResourceResult,\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport { UseQueryResult } from \"@tanstack/react-query\";\nimport { useTamboQueries, useTamboQuery } from \"../hooks\";\nimport { useTamboRegistry } from \"../providers/tambo-registry-provider\";\nimport {\n  type ConnectedMcpServer,\n  type McpServer,\n  useTamboMcpServers,\n} from \"./tambo-mcp-provider\";\n\nexport type ListPromptItem = ListPromptsResult[\"prompts\"][number];\nexport interface ListPromptEntry {\n  // Only connected servers produce prompt entries, so expose the connected type\n  server: ConnectedMcpServer;\n  prompt: ListPromptItem;\n}\n\nexport type ListResourceItem = ListResourcesResult[\"resources\"][number];\n\n/**\n * Registry resource entry - resources from the local registry (not MCP servers).\n *\n * These entries always have `server === null` and represent resources that\n * are registered locally via `TamboRegistryProvider` (static `resources`\n * props or dynamic `ResourceSource` functions).\n */\nexport interface RegistryResourceEntry {\n  server: null;\n  resource: ListResourceItem;\n}\n\n/**\n * MCP server resource entry - resources from connected MCP servers.\n *\n * These entries always have a non-null `server` with connection metadata and\n * are produced by the active MCP clients managed by `TamboMcpProvider`.\n */\nexport interface McpResourceEntry {\n  server: ConnectedMcpServer;\n  resource: ListResourceItem;\n}\n\n/**\n * Union type for all resource entries returned by `useTamboMcpResourceList`.\n *\n * - Registry resources have `server === null`.\n * - MCP-backed resources have a non-null `server` with connection metadata.\n */\nexport type ListResourceEntry = RegistryResourceEntry | McpResourceEntry;\n\n/**\n * Type guard for narrowing a `ListResourceEntry` to an MCP-backed resource\n * (entries where `server` is non-null).\n */\nexport function isMcpResourceEntry(\n  entry: ListResourceEntry,\n): entry is McpResourceEntry {\n  return entry.server !== null && isConnectedMcpServer(entry.server);\n}\n\n/**\n * Hook to get the prompts for all the registered MCP servers.\n * @returns The prompts for the MCP servers, including the server that the prompt was found on.\n */\nexport function useTamboMcpPromptList() {\n  const mcpServers = useTamboMcpServers();\n\n  const queries = useTamboQueries({\n    queries: mcpServers.map((mcpServer) => ({\n      queryKey: [\"mcp-prompts\", mcpServer.key],\n      // Only run for connected servers that have a client\n      enabled: isConnectedMcpServer(mcpServer),\n      queryFn: async (): Promise<ListPromptEntry[]> => {\n        // Fast path: if this server doesn't have a client, skip work\n        if (!isConnectedMcpServer(mcpServer)) return [];\n\n        const result = await mcpServer.client.client.listPrompts();\n        const prompts: ListPromptItem[] = result?.prompts ?? [];\n        // Return prompts without prefixes - we'll apply prefixing in combine\n        const promptsEntries = prompts.map((prompt) => ({\n          server: mcpServer,\n          prompt,\n        }));\n        return promptsEntries;\n      },\n    })),\n    combine: (results) => {\n      const combined = combineArrayResults(results);\n\n      // Always apply serverKey prefix to MCP prompts (breaking change for consistency with resources)\n      // This ensures clear separation between local and remote prompts\n      return {\n        ...combined,\n        data: combined.data.map((entry) => ({\n          ...entry,\n          prompt: {\n            ...entry.prompt,\n            name: `${entry.server.serverKey}:${entry.prompt.name}`,\n          },\n        })),\n      };\n    },\n  });\n\n  return queries;\n}\n// TODO: find a more general place for this\nfunction combineArrayResults<T>(results: UseQueryResult<T[]>[]): {\n  data: T[];\n  error: Error | null;\n  errors: Error[];\n  isPending: boolean;\n  isSuccess: boolean;\n  isError: boolean;\n  isPaused: boolean;\n  isRefetching: boolean;\n  isFetching: boolean;\n  isLoading: boolean;\n  refetch: () => Promise<void>;\n} {\n  const errors = results\n    .filter((result) => result.isError)\n    .map((result) => result.error);\n\n  // Treat queries that are idle (disabled) as non-blocking for aggregate status\n  const enabledish = results.filter(\n    (r) => r.fetchStatus !== \"idle\" || r.isSuccess || r.isError,\n  );\n\n  return {\n    // Prefer flatMap to avoid extra intermediate arrays\n    data: results.flatMap((result) =>\n      result.isSuccess && Array.isArray(result.data) ? result.data : [],\n    ),\n    // Preserve a single error for compatibility and expose the full list for diagnostics\n    error: errors[0] ?? null,\n    errors,\n    isPending: enabledish.some((result) => result.isPending),\n    isSuccess:\n      enabledish.length > 0 && enabledish.every((result) => result.isSuccess),\n    isError: errors.length > 0,\n    isPaused: enabledish.some((result) => result.isPaused),\n    isRefetching: enabledish.some((result) => result.isRefetching),\n    isFetching: enabledish.some((result) => result.isFetching),\n    isLoading: enabledish.some((result) => result.isLoading),\n    // Aggregate refetch to trigger all underlying queries\n    refetch: async () => {\n      await Promise.all(\n        results.map(async (r) => {\n          await r.refetch();\n        }),\n      );\n    },\n  };\n}\n\n// Type guard for narrowing to connected servers\nfunction isConnectedMcpServer(server: McpServer): server is ConnectedMcpServer {\n  return \"client\" in server && server.client != null;\n}\n\n/**\n * Hook to get the prompt for the specified name.\n * @param promptName - The name of the prompt to get. Can be prefixed with serverKey (e.g., \"linear:issue\") or unprefixed.\n * @param args - The arguments to pass to the prompt.\n * @returns The prompt for the specified name.\n */\nexport function useTamboMcpPrompt(\n  promptName: string | undefined,\n  args: Record<string, string> = {},\n) {\n  // figure out which server has the prompt\n  const { data: promptEntries } = useTamboMcpPromptList();\n  const promptEntry = promptEntries?.find(\n    (prompt) => prompt.prompt.name === promptName,\n  );\n  // Use the stable server key (and the server instance itself) instead of brittle\n  // name/url/transport matching.\n  const mcpServer = promptEntry?.server;\n\n  // Strip the prefix to get the original prompt name for the MCP server call\n  const originalPromptName = promptName?.includes(\":\")\n    ? promptName.split(\":\").slice(1).join(\":\")\n    : promptName;\n\n  // Canonicalize args to avoid unstable cache keys from object identity/order\n  const sortedArgsEntries = Object.keys(args)\n    .sort()\n    .map((k) => [k, args[k]] as const);\n  return useTamboQuery({\n    // Include server identity and sorted args to prevent stale cache hits\n    queryKey: [\"mcp-prompt\", promptName, mcpServer?.key, sortedArgsEntries],\n    // Only run when we have a prompt name and a connected server\n    enabled: Boolean(\n      promptName && mcpServer && isConnectedMcpServer(mcpServer),\n    ),\n    queryFn: async (): Promise<GetPromptResult | null> => {\n      if (\n        !originalPromptName ||\n        !mcpServer ||\n        !isConnectedMcpServer(mcpServer)\n      ) {\n        return null;\n      }\n      const result = await mcpServer.client.client.getPrompt({\n        name: originalPromptName,\n        arguments: args,\n      });\n      return result ?? null; // return null because react-query doesn't like undefined results\n    },\n  });\n}\n\n/**\n * Hook to get the resources for all the registered MCP servers and registry.\n * @returns The resources from MCP servers and the local registry, including the server that the resource was found on (null for registry resources).\n */\nexport function useTamboMcpResourceList() {\n  const mcpServers = useTamboMcpServers();\n  const { resources: staticResources, resourceSource } = useTamboRegistry();\n\n  // Build list of queries: MCP servers + optional dynamic resource source\n  const queriesToRun = [\n    // MCP server queries\n    ...mcpServers.map((mcpServer) => ({\n      queryKey: [\"mcp-resources\", mcpServer.key],\n      // Only run for connected servers that have a client\n      enabled: isConnectedMcpServer(mcpServer),\n      queryFn: async (): Promise<McpResourceEntry[]> => {\n        // Fast path: if this server doesn't have a client, skip work\n        if (!isConnectedMcpServer(mcpServer)) return [];\n\n        const result = await mcpServer.client.client.listResources();\n        const resources: ListResourceItem[] = result?.resources ?? [];\n        // Return resources without prefixes - we'll apply prefixing in combine\n        const resourceEntries: McpResourceEntry[] = resources.map(\n          (resource) => ({\n            server: mcpServer,\n            resource,\n          }),\n        );\n        return resourceEntries;\n      },\n    })),\n    // Dynamic resource source query (if exists)\n    ...(resourceSource\n      ? [\n          {\n            queryKey: [\"registry-resources\", \"dynamic\"],\n            enabled: true,\n            queryFn: async (): Promise<RegistryResourceEntry[]> => {\n              if (!resourceSource) return [];\n              const resources = await resourceSource.listResources();\n              return resources.map((resource) => ({\n                server: null,\n                resource,\n              }));\n            },\n          },\n        ]\n      : []),\n  ];\n\n  const queries = useTamboQueries({\n    queries: queriesToRun,\n    combine: (results) => {\n      // Type assertion needed because queries can return different entry types\n      const combined = combineArrayResults(\n        results as UseQueryResult<ListResourceEntry[]>[],\n      );\n\n      // Add static registry resources (no query needed)\n      const staticEntries: RegistryResourceEntry[] = staticResources.map(\n        (resource) => ({\n          server: null,\n          resource,\n        }),\n      );\n\n      // Merge static resources with query results (registry resources first)\n      const allData = [...staticEntries, ...combined.data];\n\n      // Apply serverKey prefix to ALL MCP resources (breaking change)\n      // Registry resources (server: null) are never prefixed\n      const prefixedData = allData.map((entry) => {\n        if (entry.server === null) {\n          // Registry resource - no prefix\n          return entry;\n        }\n        // MCP resource - always prefix with serverKey\n        return {\n          ...entry,\n          resource: {\n            ...entry.resource,\n            uri: `${entry.server.serverKey}:${entry.resource.uri}`,\n          },\n        };\n      });\n\n      return {\n        ...combined,\n        data: prefixedData,\n      };\n    },\n  });\n\n  return queries;\n}\n\n/**\n * Hook to get the resource for the specified URI.\n * @param resourceUri - The URI of the resource to get. Can be prefixed with serverKey (e.g., \"linear:file://foo\") for MCP resources, or unprefixed for registry resources.\n * @returns The resource for the specified URI.\n */\nexport function useTamboMcpResource(resourceUri: string | undefined) {\n  const { resourceSource } = useTamboRegistry();\n  const { data: resourceEntries } = useTamboMcpResourceList();\n\n  // Find which server/source has the resource\n  const resourceEntry = resourceEntries?.find(\n    (entry) => entry.resource.uri === resourceUri,\n  );\n\n  // Determine if this is a known registry resource or MCP resource\n  const isKnownEntry = resourceEntry != null;\n  const isRegistryResource = isKnownEntry && resourceEntry.server === null;\n  const mcpServer =\n    isKnownEntry && resourceEntry.server != null ? resourceEntry.server : null;\n\n  // Strip the prefix to get the original resource URI for the MCP server call\n  // Only strip if we found a matching resource entry with an MCP server\n  const originalResourceUri =\n    isKnownEntry && mcpServer\n      ? resourceUri?.replace(`${mcpServer.serverKey}:`, \"\")\n      : resourceUri;\n\n  // Check if we can fetch this resource. Registry resources can always be\n  // fetched when a resourceSource exists, even if they haven't appeared in a\n  // previous listResources result.\n  const hasRegistrySource = resourceSource != null;\n  const hasConnectedMcpServer =\n    isKnownEntry && mcpServer != null && isConnectedMcpServer(mcpServer);\n  const canFetchResource = Boolean(\n    resourceUri && (hasRegistrySource || hasConnectedMcpServer),\n  );\n\n  let locationKey: string | undefined;\n  if (isRegistryResource || (!isKnownEntry && hasRegistrySource)) {\n    locationKey = \"registry\";\n  } else if (mcpServer) {\n    locationKey = mcpServer.key;\n  }\n\n  return useTamboQuery({\n    // Include server identity or \"registry\" to prevent stale cache hits\n    queryKey: [\"resource\", resourceUri, locationKey],\n    enabled: canFetchResource,\n    queryFn: async (): Promise<ReadResourceResult | null> => {\n      if (!originalResourceUri) {\n        return null;\n      }\n\n      // Registry resource: use resourceSource.getResource\n      // If this is not a known entry but we have a resourceSource, treat it as\n      // a registry resource by default.\n      if (resourceSource && (!isKnownEntry || isRegistryResource)) {\n        const result = await resourceSource.getResource(originalResourceUri);\n        return result ?? null;\n      }\n\n      // MCP resource: use MCP client\n      if (mcpServer && isConnectedMcpServer(mcpServer)) {\n        const result = await mcpServer.client.client.readResource({\n          uri: originalResourceUri,\n        });\n        return result ?? null;\n      }\n\n      return null;\n    },\n  });\n}\n"]}
+{"version":3,"file":"mcp-hooks.js","sourceRoot":"","sources":["../../src/mcp/mcp-hooks.ts"],"names":[],"mappings":";;AA4DA,gDAIC;AAMD,sDAyCC;AAuDD,oDAIC;AAQD,8CA4CC;AAMD,0DAgGC;AASD,kDAoEC;AA1YD,oCAA0D;AAC1D,kFAAwE;AACxE,mDAAsD;AACtD,6DAI8B;AA0C9B;;;GAGG;AACH,SAAgB,kBAAkB,CAChC,KAAwB;IAExB,OAAO,KAAK,CAAC,MAAM,KAAK,IAAI,IAAI,oBAAoB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;AACrE,CAAC;AAED;;;GAGG;AACH,SAAgB,qBAAqB;IACnC,MAAM,UAAU,GAAG,IAAA,uCAAkB,GAAE,CAAC;IAExC,MAAM,OAAO,GAAG,IAAA,uBAAe,EAAC;QAC9B,OAAO,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;YACtC,QAAQ,EAAE,CAAC,aAAa,EAAE,SAAS,CAAC,GAAG,CAAC;YACxC,oDAAoD;YACpD,OAAO,EAAE,oBAAoB,CAAC,SAAS,CAAC;YACxC,OAAO,EAAE,KAAK,IAAgC,EAAE;gBAC9C,6DAA6D;gBAC7D,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC;oBAAE,OAAO,EAAE,CAAC;gBAEhD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;gBAC3D,MAAM,OAAO,GAAqB,MAAM,EAAE,OAAO,IAAI,EAAE,CAAC;gBACxD,qEAAqE;gBACrE,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;oBAC9C,MAAM,EAAE,SAAS;oBACjB,MAAM;iBACP,CAAC,CAAC,CAAC;gBACJ,OAAO,cAAc,CAAC;YACxB,CAAC;SACF,CAAC,CAAC;QACH,OAAO,EAAE,CAAC,OAAO,EAAE,EAAE;YACnB,MAAM,QAAQ,GAAG,mBAAmB,CAAC,OAAO,CAAC,CAAC;YAE9C,gGAAgG;YAChG,iEAAiE;YACjE,OAAO;gBACL,GAAG,QAAQ;gBACX,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;oBAClC,GAAG,KAAK;oBACR,MAAM,EAAE;wBACN,GAAG,KAAK,CAAC,MAAM;wBACf,IAAI,EAAE,GAAG,KAAK,CAAC,MAAM,CAAC,SAAS,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE;qBACvD;iBACF,CAAC,CAAC;aACJ,CAAC;QACJ,CAAC;KACF,CAAC,CAAC;IAEH,OAAO,OAAO,CAAC;AACjB,CAAC;AACD,2CAA2C;AAC3C,SAAS,mBAAmB,CAAI,OAA8B;IAa5D,MAAM,MAAM,GAAG,OAAO;SACnB,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC;SAClC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAEjC,8EAA8E;IAC9E,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAC/B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,KAAK,MAAM,IAAI,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,OAAO,CAC5D,CAAC;IAEF,OAAO;QACL,oDAAoD;QACpD,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,CAC/B,MAAM,CAAC,SAAS,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAClE;QACD,qFAAqF;QACrF,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,IAAI;QACxB,MAAM;QACN,SAAS,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC;QACxD,SAAS,EACP,UAAU,CAAC,MAAM,GAAG,CAAC,IAAI,UAAU,CAAC,KAAK,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC;QACzE,OAAO,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC;QAC1B,QAAQ,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC;QACtD,YAAY,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC;QAC9D,UAAU,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC;QAC1D,SAAS,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC;QACxD,sDAAsD;QACtD,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,MAAM,OAAO,CAAC,GAAG,CACf,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE;gBACtB,MAAM,CAAC,CAAC,OAAO,EAAE,CAAC;YACpB,CAAC,CAAC,CACH,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,SAAgB,oBAAoB,CAClC,MAAiB;IAEjB,OAAO,QAAQ,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC;AACrD,CAAC;AAED;;;;;GAKG;AACH,SAAgB,iBAAiB,CAC/B,UAA8B,EAC9B,OAA+B,EAAE;IAEjC,yCAAyC;IACzC,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,qBAAqB,EAAE,CAAC;IACxD,MAAM,WAAW,GAAG,aAAa,EAAE,IAAI,CACrC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,KAAK,UAAU,CAC9C,CAAC;IACF,gFAAgF;IAChF,+BAA+B;IAC/B,MAAM,SAAS,GAAG,WAAW,EAAE,MAAM,CAAC;IAEtC,2EAA2E;IAC3E,MAAM,kBAAkB,GAAG,UAAU,EAAE,QAAQ,CAAC,GAAG,CAAC;QAClD,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;QAC1C,CAAC,CAAC,UAAU,CAAC;IAEf,4EAA4E;IAC5E,MAAM,iBAAiB,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC;SACxC,IAAI,EAAE;SACN,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,CAAU,CAAC,CAAC;IACrC,OAAO,IAAA,qBAAa,EAAC;QACnB,sEAAsE;QACtE,QAAQ,EAAE,CAAC,YAAY,EAAE,UAAU,EAAE,SAAS,EAAE,GAAG,EAAE,iBAAiB,CAAC;QACvE,6DAA6D;QAC7D,OAAO,EAAE,OAAO,CACd,UAAU,IAAI,SAAS,IAAI,oBAAoB,CAAC,SAAS,CAAC,CAC3D;QACD,OAAO,EAAE,KAAK,IAAqC,EAAE;YACnD,IACE,CAAC,kBAAkB;gBACnB,CAAC,SAAS;gBACV,CAAC,oBAAoB,CAAC,SAAS,CAAC,EAChC,CAAC;gBACD,OAAO,IAAI,CAAC;YACd,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC;gBACrD,IAAI,EAAE,kBAAkB;gBACxB,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;YACH,OAAO,MAAM,IAAI,IAAI,CAAC,CAAC,iEAAiE;QAC1F,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED;;;GAGG;AACH,SAAgB,uBAAuB;IACrC,MAAM,UAAU,GAAG,IAAA,uCAAkB,GAAE,CAAC;IACxC,MAAM,EAAE,SAAS,EAAE,eAAe,EAAE,cAAc,EAAE,GAAG,IAAA,0CAAgB,GAAE,CAAC;IAE1E,wEAAwE;IACxE,MAAM,YAAY,GAAG;QACnB,qBAAqB;QACrB,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;YAChC,QAAQ,EAAE,CAAC,eAAe,EAAE,SAAS,CAAC,GAAG,CAAC;YAC1C,oDAAoD;YACpD,OAAO,EAAE,oBAAoB,CAAC,SAAS,CAAC;YACxC,OAAO,EAAE,KAAK,IAAiC,EAAE;gBAC/C,6DAA6D;gBAC7D,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC;oBAAE,OAAO,EAAE,CAAC;gBAEhD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;gBAC7D,MAAM,SAAS,GAAuB,MAAM,EAAE,SAAS,IAAI,EAAE,CAAC;gBAC9D,uEAAuE;gBACvE,MAAM,eAAe,GAAuB,SAAS,CAAC,GAAG,CACvD,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;oBACb,MAAM,EAAE,SAAS;oBACjB,QAAQ;iBACT,CAAC,CACH,CAAC;gBACF,OAAO,eAAe,CAAC;YACzB,CAAC;SACF,CAAC,CAAC;QACH,4CAA4C;QAC5C,GAAG,CAAC,cAAc;YAChB,CAAC,CAAC;gBACE;oBACE,QAAQ,EAAE,CAAC,oBAAoB,EAAE,SAAS,CAAC;oBAC3C,OAAO,EAAE,IAAI;oBACb,OAAO,EAAE,KAAK,IAAsC,EAAE;wBACpD,IAAI,CAAC,cAAc;4BAAE,OAAO,EAAE,CAAC;wBAC/B,MAAM,SAAS,GAAG,MAAM,cAAc,CAAC,aAAa,EAAE,CAAC;wBACvD,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;4BAClC,MAAM,EAAE,IAAI;4BACZ,QAAQ;yBACT,CAAC,CAAC,CAAC;oBACN,CAAC;iBACF;aACF;YACH,CAAC,CAAC,EAAE,CAAC;KACR,CAAC;IAEF,MAAM,OAAO,GAAG,IAAA,uBAAe,EAAC;QAC9B,OAAO,EAAE,YAAY;QACrB,OAAO,EAAE,CAAC,OAAO,EAAE,EAAE;YACnB,yEAAyE;YACzE,MAAM,QAAQ,GAAG,mBAAmB,CAClC,OAAgD,CACjD,CAAC;YAEF,kDAAkD;YAClD,MAAM,aAAa,GAA4B,eAAe,CAAC,GAAG,CAChE,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;gBACb,MAAM,EAAE,IAAI;gBACZ,QAAQ;aACT,CAAC,CACH,CAAC;YAEF,uEAAuE;YACvE,MAAM,OAAO,GAAG,CAAC,GAAG,aAAa,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;YAErD,4EAA4E;YAC5E,uFAAuF;YACvF,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;gBACzC,IAAI,KAAK,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC;oBAC1B,sDAAsD;oBACtD,OAAO;wBACL,GAAG,KAAK;wBACR,QAAQ,EAAE;4BACR,GAAG,KAAK,CAAC,QAAQ;4BACjB,GAAG,EAAE,GAAG,mCAAmB,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,EAAE;yBACpD;qBACF,CAAC;gBACJ,CAAC;gBACD,8CAA8C;gBAC9C,OAAO;oBACL,GAAG,KAAK;oBACR,QAAQ,EAAE;wBACR,GAAG,KAAK,CAAC,QAAQ;wBACjB,GAAG,EAAE,GAAG,KAAK,CAAC,MAAM,CAAC,SAAS,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,EAAE;qBACvD;iBACF,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,OAAO;gBACL,GAAG,QAAQ;gBACX,IAAI,EAAE,YAAY;aACnB,CAAC;QACJ,CAAC;KACF,CAAC,CAAC;IAEH,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;;;;;GAMG;AACH,SAAgB,mBAAmB,CAAC,WAA+B;IACjE,MAAM,EAAE,cAAc,EAAE,GAAG,IAAA,0CAAgB,GAAE,CAAC;IAC9C,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,GAAG,uBAAuB,EAAE,CAAC;IAE5D,4CAA4C;IAC5C,MAAM,aAAa,GAAG,eAAe,EAAE,IAAI,CACzC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,KAAK,WAAW,CAC9C,CAAC;IAEF,2DAA2D;IAC3D,MAAM,kBAAkB,GAAG,aAAa,EAAE,MAAM,KAAK,IAAI,CAAC;IAC1D,MAAM,SAAS,GAAG,aAAa,EAAE,MAAM,IAAI,IAAI,CAAC;IAEhD,2CAA2C;IAC3C,MAAM,iBAAiB,GAAG,OAAO,CAC/B,WAAW,EAAE,UAAU,CAAC,GAAG,mCAAmB,GAAG,CAAC,CACnD,CAAC;IAEF,iEAAiE;IACjE,IAAI,mBAAuC,CAAC;IAC5C,IAAI,kBAAkB,IAAI,iBAAiB,EAAE,CAAC;QAC5C,MAAM,SAAS,GAAG,mCAAmB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,mBAAmB;QACrE,mBAAmB,GAAG,WAAW,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC;IACtD,CAAC;SAAM,IAAI,SAAS,EAAE,CAAC;QACrB,MAAM,SAAS,GAAG,SAAS,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,mBAAmB;QACrE,mBAAmB,GAAG,WAAW,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC;IACtD,CAAC;IAED,sCAAsC;IACtC,MAAM,iBAAiB,GAAG,cAAc,IAAI,IAAI,CAAC;IACjD,MAAM,qBAAqB,GACzB,SAAS,IAAI,IAAI,IAAI,oBAAoB,CAAC,SAAS,CAAC,CAAC;IACvD,MAAM,oBAAoB,GACxB,iBAAiB,IAAI,CAAC,kBAAkB,IAAI,iBAAiB,CAAC,CAAC;IACjE,MAAM,gBAAgB,GAAG,OAAO,CAC9B,WAAW,IAAI,CAAC,oBAAoB,IAAI,qBAAqB,CAAC,CAC/D,CAAC;IAEF,MAAM,WAAW,GACf,kBAAkB,IAAI,iBAAiB;QACrC,CAAC,CAAC,mCAAmB;QACrB,CAAC,CAAC,SAAS,EAAE,GAAG,CAAC;IAErB,OAAO,IAAA,qBAAa,EAAC;QACnB,QAAQ,EAAE,CAAC,UAAU,EAAE,WAAW,EAAE,WAAW,CAAC;QAChD,OAAO,EAAE,gBAAgB;QACzB,OAAO,EAAE,KAAK,IAAwC,EAAE;YACtD,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBACzB,OAAO,IAAI,CAAC;YACd,CAAC;YAED,oDAAoD;YACpD,IAAI,cAAc,IAAI,CAAC,kBAAkB,IAAI,iBAAiB,CAAC,EAAE,CAAC;gBAChE,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,WAAW,CAAC,mBAAmB,CAAC,CAAC;gBACrE,OAAO,MAAM,IAAI,IAAI,CAAC;YACxB,CAAC;YAED,+BAA+B;YAC/B,IAAI,SAAS,IAAI,oBAAoB,CAAC,SAAS,CAAC,EAAE,CAAC;gBACjD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC;oBACxD,GAAG,EAAE,mBAAmB;iBACzB,CAAC,CAAC;gBACH,OAAO,MAAM,IAAI,IAAI,CAAC;YACxB,CAAC;YAED,OAAO,IAAI,CAAC;QACd,CAAC;KACF,CAAC,CAAC;AACL,CAAC","sourcesContent":["import {\n  GetPromptResult,\n  type ListPromptsResult,\n  type ListResourcesResult,\n  type ReadResourceResult,\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport { UseQueryResult } from \"@tanstack/react-query\";\nimport { useTamboQueries, useTamboQuery } from \"../hooks\";\nimport { useTamboRegistry } from \"../providers/tambo-registry-provider\";\nimport { REGISTRY_SERVER_KEY } from \"./mcp-constants\";\nimport {\n  type ConnectedMcpServer,\n  type McpServer,\n  useTamboMcpServers,\n} from \"./tambo-mcp-provider\";\n\nexport type ListPromptItem = ListPromptsResult[\"prompts\"][number];\nexport interface ListPromptEntry {\n  // Only connected servers produce prompt entries, so expose the connected type\n  server: ConnectedMcpServer;\n  prompt: ListPromptItem;\n}\n\nexport type ListResourceItem = ListResourcesResult[\"resources\"][number];\n\n/**\n * Registry resource entry - resources from the local registry (not MCP servers).\n *\n * These entries always have `server === null` and represent resources that\n * are registered locally via `TamboRegistryProvider` (static `resources`\n * props or dynamic `ResourceSource` functions).\n */\nexport interface RegistryResourceEntry {\n  server: null;\n  resource: ListResourceItem;\n}\n\n/**\n * MCP server resource entry - resources from connected MCP servers.\n *\n * These entries always have a non-null `server` with connection metadata and\n * are produced by the active MCP clients managed by `TamboMcpProvider`.\n */\nexport interface McpResourceEntry {\n  server: ConnectedMcpServer;\n  resource: ListResourceItem;\n}\n\n/**\n * Union type for all resource entries returned by `useTamboMcpResourceList`.\n *\n * - Registry resources have `server === null`.\n * - MCP-backed resources have a non-null `server` with connection metadata.\n */\nexport type ListResourceEntry = RegistryResourceEntry | McpResourceEntry;\n\n/**\n * Type guard for narrowing a `ListResourceEntry` to an MCP-backed resource\n * (entries where `server` is non-null).\n */\nexport function isMcpResourceEntry(\n  entry: ListResourceEntry,\n): entry is McpResourceEntry {\n  return entry.server !== null && isConnectedMcpServer(entry.server);\n}\n\n/**\n * Hook to get the prompts for all the registered MCP servers.\n * @returns The prompts for the MCP servers, including the server that the prompt was found on.\n */\nexport function useTamboMcpPromptList() {\n  const mcpServers = useTamboMcpServers();\n\n  const queries = useTamboQueries({\n    queries: mcpServers.map((mcpServer) => ({\n      queryKey: [\"mcp-prompts\", mcpServer.key],\n      // Only run for connected servers that have a client\n      enabled: isConnectedMcpServer(mcpServer),\n      queryFn: async (): Promise<ListPromptEntry[]> => {\n        // Fast path: if this server doesn't have a client, skip work\n        if (!isConnectedMcpServer(mcpServer)) return [];\n\n        const result = await mcpServer.client.client.listPrompts();\n        const prompts: ListPromptItem[] = result?.prompts ?? [];\n        // Return prompts without prefixes - we'll apply prefixing in combine\n        const promptsEntries = prompts.map((prompt) => ({\n          server: mcpServer,\n          prompt,\n        }));\n        return promptsEntries;\n      },\n    })),\n    combine: (results) => {\n      const combined = combineArrayResults(results);\n\n      // Always apply serverKey prefix to MCP prompts (breaking change for consistency with resources)\n      // This ensures clear separation between local and remote prompts\n      return {\n        ...combined,\n        data: combined.data.map((entry) => ({\n          ...entry,\n          prompt: {\n            ...entry.prompt,\n            name: `${entry.server.serverKey}:${entry.prompt.name}`,\n          },\n        })),\n      };\n    },\n  });\n\n  return queries;\n}\n// TODO: find a more general place for this\nfunction combineArrayResults<T>(results: UseQueryResult<T[]>[]): {\n  data: T[];\n  error: Error | null;\n  errors: Error[];\n  isPending: boolean;\n  isSuccess: boolean;\n  isError: boolean;\n  isPaused: boolean;\n  isRefetching: boolean;\n  isFetching: boolean;\n  isLoading: boolean;\n  refetch: () => Promise<void>;\n} {\n  const errors = results\n    .filter((result) => result.isError)\n    .map((result) => result.error);\n\n  // Treat queries that are idle (disabled) as non-blocking for aggregate status\n  const enabledish = results.filter(\n    (r) => r.fetchStatus !== \"idle\" || r.isSuccess || r.isError,\n  );\n\n  return {\n    // Prefer flatMap to avoid extra intermediate arrays\n    data: results.flatMap((result) =>\n      result.isSuccess && Array.isArray(result.data) ? result.data : [],\n    ),\n    // Preserve a single error for compatibility and expose the full list for diagnostics\n    error: errors[0] ?? null,\n    errors,\n    isPending: enabledish.some((result) => result.isPending),\n    isSuccess:\n      enabledish.length > 0 && enabledish.every((result) => result.isSuccess),\n    isError: errors.length > 0,\n    isPaused: enabledish.some((result) => result.isPaused),\n    isRefetching: enabledish.some((result) => result.isRefetching),\n    isFetching: enabledish.some((result) => result.isFetching),\n    isLoading: enabledish.some((result) => result.isLoading),\n    // Aggregate refetch to trigger all underlying queries\n    refetch: async () => {\n      await Promise.all(\n        results.map(async (r) => {\n          await r.refetch();\n        }),\n      );\n    },\n  };\n}\n\n/**\n * Type guard for narrowing McpServer to ConnectedMcpServer.\n * A connected server has a non-null client.\n */\nexport function isConnectedMcpServer(\n  server: McpServer,\n): server is ConnectedMcpServer {\n  return \"client\" in server && server.client != null;\n}\n\n/**\n * Hook to get the prompt for the specified name.\n * @param promptName - The name of the prompt to get. Can be prefixed with serverKey (e.g., \"linear:issue\") or unprefixed.\n * @param args - The arguments to pass to the prompt.\n * @returns The prompt for the specified name.\n */\nexport function useTamboMcpPrompt(\n  promptName: string | undefined,\n  args: Record<string, string> = {},\n) {\n  // figure out which server has the prompt\n  const { data: promptEntries } = useTamboMcpPromptList();\n  const promptEntry = promptEntries?.find(\n    (prompt) => prompt.prompt.name === promptName,\n  );\n  // Use the stable server key (and the server instance itself) instead of brittle\n  // name/url/transport matching.\n  const mcpServer = promptEntry?.server;\n\n  // Strip the prefix to get the original prompt name for the MCP server call\n  const originalPromptName = promptName?.includes(\":\")\n    ? promptName.split(\":\").slice(1).join(\":\")\n    : promptName;\n\n  // Canonicalize args to avoid unstable cache keys from object identity/order\n  const sortedArgsEntries = Object.keys(args)\n    .sort()\n    .map((k) => [k, args[k]] as const);\n  return useTamboQuery({\n    // Include server identity and sorted args to prevent stale cache hits\n    queryKey: [\"mcp-prompt\", promptName, mcpServer?.key, sortedArgsEntries],\n    // Only run when we have a prompt name and a connected server\n    enabled: Boolean(\n      promptName && mcpServer && isConnectedMcpServer(mcpServer),\n    ),\n    queryFn: async (): Promise<GetPromptResult | null> => {\n      if (\n        !originalPromptName ||\n        !mcpServer ||\n        !isConnectedMcpServer(mcpServer)\n      ) {\n        return null;\n      }\n      const result = await mcpServer.client.client.getPrompt({\n        name: originalPromptName,\n        arguments: args,\n      });\n      return result ?? null; // return null because react-query doesn't like undefined results\n    },\n  });\n}\n\n/**\n * Hook to get the resources for all the registered MCP servers and registry.\n * @returns The resources from MCP servers and the local registry, including the server that the resource was found on (null for registry resources).\n */\nexport function useTamboMcpResourceList() {\n  const mcpServers = useTamboMcpServers();\n  const { resources: staticResources, resourceSource } = useTamboRegistry();\n\n  // Build list of queries: MCP servers + optional dynamic resource source\n  const queriesToRun = [\n    // MCP server queries\n    ...mcpServers.map((mcpServer) => ({\n      queryKey: [\"mcp-resources\", mcpServer.key],\n      // Only run for connected servers that have a client\n      enabled: isConnectedMcpServer(mcpServer),\n      queryFn: async (): Promise<McpResourceEntry[]> => {\n        // Fast path: if this server doesn't have a client, skip work\n        if (!isConnectedMcpServer(mcpServer)) return [];\n\n        const result = await mcpServer.client.client.listResources();\n        const resources: ListResourceItem[] = result?.resources ?? [];\n        // Return resources without prefixes - we'll apply prefixing in combine\n        const resourceEntries: McpResourceEntry[] = resources.map(\n          (resource) => ({\n            server: mcpServer,\n            resource,\n          }),\n        );\n        return resourceEntries;\n      },\n    })),\n    // Dynamic resource source query (if exists)\n    ...(resourceSource\n      ? [\n          {\n            queryKey: [\"registry-resources\", \"dynamic\"],\n            enabled: true,\n            queryFn: async (): Promise<RegistryResourceEntry[]> => {\n              if (!resourceSource) return [];\n              const resources = await resourceSource.listResources();\n              return resources.map((resource) => ({\n                server: null,\n                resource,\n              }));\n            },\n          },\n        ]\n      : []),\n  ];\n\n  const queries = useTamboQueries({\n    queries: queriesToRun,\n    combine: (results) => {\n      // Type assertion needed because queries can return different entry types\n      const combined = combineArrayResults(\n        results as UseQueryResult<ListResourceEntry[]>[],\n      );\n\n      // Add static registry resources (no query needed)\n      const staticEntries: RegistryResourceEntry[] = staticResources.map(\n        (resource) => ({\n          server: null,\n          resource,\n        }),\n      );\n\n      // Merge static resources with query results (registry resources first)\n      const allData = [...staticEntries, ...combined.data];\n\n      // Apply serverKey prefix to ALL resources for unified @serverKey:uri format\n      // Registry resources get REGISTRY_SERVER_KEY prefix, MCP resources get their serverKey\n      const prefixedData = allData.map((entry) => {\n        if (entry.server === null) {\n          // Registry resource - prefix with REGISTRY_SERVER_KEY\n          return {\n            ...entry,\n            resource: {\n              ...entry.resource,\n              uri: `${REGISTRY_SERVER_KEY}:${entry.resource.uri}`,\n            },\n          };\n        }\n        // MCP resource - always prefix with serverKey\n        return {\n          ...entry,\n          resource: {\n            ...entry.resource,\n            uri: `${entry.server.serverKey}:${entry.resource.uri}`,\n          },\n        };\n      });\n\n      return {\n        ...combined,\n        data: prefixedData,\n      };\n    },\n  });\n\n  return queries;\n}\n\n/**\n * Hook to get the resource for the specified URI.\n * @param resourceUri - The URI of the resource to get. Must be prefixed:\n *   - MCP resources: prefixed with serverKey (e.g., \"linear:file://foo\")\n *   - Registry resources: prefixed with \"registry:\" (e.g., \"registry:file://bar\")\n * @returns The resource for the specified URI.\n */\nexport function useTamboMcpResource(resourceUri: string | undefined) {\n  const { resourceSource } = useTamboRegistry();\n  const { data: resourceEntries } = useTamboMcpResourceList();\n\n  // Find which server/source has the resource\n  const resourceEntry = resourceEntries?.find(\n    (entry) => entry.resource.uri === resourceUri,\n  );\n\n  // Determine if this is a registry resource or MCP resource\n  const isRegistryResource = resourceEntry?.server === null;\n  const mcpServer = resourceEntry?.server ?? null;\n\n  // Check if the URI has the registry prefix\n  const hasRegistryPrefix = Boolean(\n    resourceUri?.startsWith(`${REGISTRY_SERVER_KEY}:`),\n  );\n\n  // Strip the prefix to get the original resource URI for fetching\n  let originalResourceUri: string | undefined;\n  if (isRegistryResource || hasRegistryPrefix) {\n    const prefixLen = REGISTRY_SERVER_KEY.length + 1; // +1 for the colon\n    originalResourceUri = resourceUri?.slice(prefixLen);\n  } else if (mcpServer) {\n    const prefixLen = mcpServer.serverKey.length + 1; // +1 for the colon\n    originalResourceUri = resourceUri?.slice(prefixLen);\n  }\n\n  // Check if we can fetch this resource\n  const hasRegistrySource = resourceSource != null;\n  const hasConnectedMcpServer =\n    mcpServer != null && isConnectedMcpServer(mcpServer);\n  const canFetchFromRegistry =\n    hasRegistrySource && (isRegistryResource || hasRegistryPrefix);\n  const canFetchResource = Boolean(\n    resourceUri && (canFetchFromRegistry || hasConnectedMcpServer),\n  );\n\n  const locationKey =\n    isRegistryResource || hasRegistryPrefix\n      ? REGISTRY_SERVER_KEY\n      : mcpServer?.key;\n\n  return useTamboQuery({\n    queryKey: [\"resource\", resourceUri, locationKey],\n    enabled: canFetchResource,\n    queryFn: async (): Promise<ReadResourceResult | null> => {\n      if (!originalResourceUri) {\n        return null;\n      }\n\n      // Registry resource: use resourceSource.getResource\n      if (resourceSource && (isRegistryResource || hasRegistryPrefix)) {\n        const result = await resourceSource.getResource(originalResourceUri);\n        return result ?? null;\n      }\n\n      // MCP resource: use MCP client\n      if (mcpServer && isConnectedMcpServer(mcpServer)) {\n        const result = await mcpServer.client.client.readResource({\n          uri: originalResourceUri,\n        });\n        return result ?? null;\n      }\n\n      return null;\n    },\n  });\n}\n"]}

package/dist/mcp/mcp-hooks.test.js

@@ -991,13 +991,14 @@ describe("useTamboMcpResource - read individual resource", () => {
             uri: "file:///home/user/doc.txt",
         });
     });
-    it("should read registry resources via resourceSource even when not listed", async () => {
-        const registryUri = "file:///local/registry-doc.txt";
+    it("should read registry resources via resourceSource with registry prefix", async () => {
+        const originalUri = "file:///local/registry-doc.txt";
+        const prefixedUri = `registry:${originalUri}`;
         const listResources = jest.fn().mockResolvedValue([]);
         const getResource = jest.fn().mockResolvedValue({
             contents: [
                 {
-                    uri: registryUri,
+                    uri: originalUri,
                     mimeType: "text/plain",
                     text: "Registry content",
                 },
@@ -1005,7 +1006,8 @@ describe("useTamboMcpResource - read individual resource", () => {
         });
         let capturedResourceData = null;
         const Capture = () => {
-            const { data: resourceData } = (0, mcp_hooks_1.useTamboMcpResource)(registryUri);
+            // Request with registry: prefix
+            const { data: resourceData } = (0, mcp_hooks_1.useTamboMcpResource)(prefixedUri);
             (0, react_2.useEffect)(() => {
                 if (resourceData) {
                     capturedResourceData = resourceData;
@@ -1025,7 +1027,8 @@ describe("useTamboMcpResource - read individual resource", () => {
         await (0, react_1.waitFor)(() => {
             expect(capturedResourceData).not.toBeNull();
         });
-        expect(getResource).toHaveBeenCalledWith(registryUri);
+        // getResource should be called with the original URI (prefix stripped)
+        expect(getResource).toHaveBeenCalledWith(originalUri);
         expect(capturedResourceData.contents[0].text).toBe("Registry content");
     });
 });