@gram-ai/elements 1.16.4 → 1.17.0

package/dist/types/index.d.ts

@@ -2,6 +2,12 @@ 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';
+import { LanguageModel } from 'ai';
+/**
+ * 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 +17,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 +104,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.
@@ -165,6 +173,27 @@ export interface ElementsConfig {
      * }
      */
     composer?: ComposerConfig;
+    /**
+     * Optional property to override the LLM provider. If you override the model,
+     * then logs & usage metrics will not be tracked directly via Gram.
+     *
+     * Please ensure that you are using an AI SDK v2 compatible model (e.g a
+     * Vercel AI sdk provider in the v2 semver range), as this is the only variant
+     * compatible with AI SDK V5
+     *
+     * Example with Google Gemini:
+     * ```ts
+     * import { google } from '@ai-sdk/google';
+     *
+     * const googleGemini = google('gemini-3-pro-preview');
+     *
+     * const config: ElementsConfig = {
+     *   {other options}
+     *   languageModel: googleGemini,
+     * }
+     * ```
+     */
+    languageModel?: LanguageModel;
     /**
      * The configuration for the modal window.
      * Does not apply if variant is 'standalone'.
@@ -571,5 +600,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.17.0",
   "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",
@@ -102,6 +104,7 @@
     "zod": "^4.1.13"
   },
   "devDependencies": {
+    "@ai-sdk/google": "^2",
     "@ai-sdk/openai": "^2.0.0-beta.5",
     "@assistant-ui/react": "^0.11.53",
     "@assistant-ui/react-ai-sdk": "^1.1.16",