@gram-ai/elements 1.16.4 → 1.16.5

package/dist/types/index.d.ts

@@ -2,6 +2,11 @@ import { MODELS } from '../lib/models';
 import { AssistantTool, ImageMessagePartComponent, ReasoningGroupComponent, ReasoningMessagePartComponent, TextMessagePartComponent, ToolCallMessagePartComponent } from '@assistant-ui/react';
 import { ComponentType, Dispatch, PropsWithChildren, SetStateAction, ReactNode } from 'react';
 import { Plugin } from './plugins';
+/**
+ * Function to retrieve the session token from the backend endpoint.
+ * Override this if you have mounted your session endpoint at a different path.
+ */
+export type GetSessionFn = () => Promise<string>;
 export interface ElementsProviderProps {
     /**
      * The children to render.
@@ -11,6 +16,19 @@ export interface ElementsProviderProps {
      * Configuration object for the Elements library.
      */
     config: ElementsConfig;
+    /**
+     * Function to retrieve the session token from the backend endpoint.
+     *
+     * @example
+     * const config: ElementsConfig = {
+     *   getSession: async () => {
+     *     return fetch('/chat/session').then(res => res.json()).then(data => data.client_token)
+     *   },
+     * }
+     *
+     * @default Use this default if you are using the Elements server handlers, and have mounted the session handler at /chat/session.
+     */
+    getSession?: GetSessionFn;
 }
 type ServerUrl = string;
 export declare const VARIANTS: readonly ["widget", "sidecar", "standalone"];
@@ -85,17 +103,6 @@ export interface ElementsConfig {
      * }
      */
     mcp: ServerUrl;
-    /**
-     * The path of your backend's chat endpoint.
-     *
-     * @default '/chat/completions'
-     *
-     * @example
-     * const config: ElementsConfig = {
-     *   chatEndpoint: '/my-custom-chat-endpoint',
-     * }
-     */
-    chatEndpoint?: string;
     /**
      * Custom environment variable overrides for the Elements library.
      * Will be used to override the environment variables for the MCP server.
@@ -571,5 +578,10 @@ export type ElementsContextType = {
     isOpen: boolean;
     setIsOpen: (isOpen: boolean) => void;
     plugins: Plugin[];
+    /**
+     * Indicates if the process of discovering MCP tools is still ongoing.
+     * TODO: failure state
+     */
+    isLoadingMCPTools: boolean;
 };
 export {};

package/package.json

@@ -2,7 +2,7 @@
   "name": "@gram-ai/elements",
   "description": "Gram Elements is a library of UI primitives for building chat-like experiences for MCP Servers.",
   "type": "module",
-  "version": "1.16.4",
+  "version": "1.16.5",
   "main": "dist/index.js",
   "exports": {
     ".": {
@@ -85,6 +85,7 @@
   },
   "dependencies": {
     "@ai-sdk/mcp": "^0.0.11",
+    "@ai-sdk/react": "^2.0.118",
     "@openrouter/ai-sdk-provider": "^1.4.1",
     "@radix-ui/react-avatar": "^1.1.10",
     "@radix-ui/react-collapsible": "^1.1.12",
@@ -92,6 +93,7 @@
     "@radix-ui/react-popover": "^1.1.15",
     "@radix-ui/react-slot": "^1.2.3",
     "@radix-ui/react-tooltip": "^1.2.8",
+    "@tanstack/react-query": "5.90.10",
     "ai": "5.0.90",
     "assistant-stream": "^0.2.42",
     "class-variance-authority": "^0.7.1",