@worktango/ai-assistant 0.0.72-beta05 → 0.0.72-beta07

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@worktango/ai-assistant",
-  "version": "0.0.72-beta05",
+  "version": "0.0.72-beta07",
   "files": [
     "constants.js",
     "dist",

package/react.d.ts

@@ -7,8 +7,11 @@ import {
   GetStructuredResponseOptions,
   ToolDeclaration,
   AiAssistantSuggestedPromptsProps,
+  GetFollowUpSuggestions,
 } from "./sharedTypes";
 
+export const defaultGetFollowUpSuggestions: GetFollowUpSuggestions;
+
 export function AiAssistant(args: AiAssistantProps): JSX.Element;
 
 export function AiAssistantSuggestedPrompts(
@@ -102,25 +105,4 @@ export function getChatCompletion(
   args: GetChatCompletionOptions
 ): Promise<ChatData[]>;
 
-export {
-  AiAssistantControls,
-  ChatData,
-  ChatItem,
-  GetChatCompletionOptions,
-  GetStructuredResponseOptions,
-  ToolDeclaration,
-  AiAssistantProps,
-  ChatDataFile,
-  ChatDataText,
-  ChatDataToolCall,
-  ChatDataToolResult,
-  ChatHistory,
-  ChatHistoryItem,
-  ChatSizes,
-  Conversation,
-  SupportedTextModel,
-  AiAssistantSuggestedPrompt,
-  AiAssistantSuggestedPromptsProps,
-  AiAssistantControls,
-  FollowUpSuggestion,
-} from "./sharedTypes";
+export * from "./sharedTypes";

package/sharedTypes.ts

@@ -61,6 +61,12 @@ export type ChatHistoryItem = {
   [key: string]: any;
 };
 
+export type GetFollowUpSuggestions = (args: {
+  chatHistory: ChatItem[];
+  aiAssistantControls: React.MutableRefObject<AiAssistantControls>;
+  setFollowUpSuggestions: (suggestions: FollowUpSuggestion[]) => void;
+}) => Promise<FollowUpSuggestion[]>;
+
 export type ChatHistory = Record<string, ChatHistoryItem[]>;
 
 export type ChatSizes = "small" | "default" | "large" | "full";
@@ -167,20 +173,27 @@ export type AiAssistantProps = {
   followUpSuggestions?: FollowUpSuggestion[];
 };
 
-export type FollowUpSuggestion = {
-  /**
-   * A unique identifier for the follow-up suggestion. This will be used to keep user-level state for dismissed items.
-   */
-  id: string;
-  /**
-   * The title of the follow-up suggestion. This will be displayed in the UI as a pill button, so keep it short.
-   */
-  title: string;
-  /**
-   * The function to call when the follow-up suggestion is clicked.
-   */
-  onClick: () => void;
-};
+/**
+ * If { loading: true }, a small loader will be shown. You can use this show a
+ * loading state in any place of the UI.
+ */
+export type FollowUpSuggestion =
+  | { type: "loading"; id?: string }
+  | {
+      type?: "suggestion";
+      /**
+       * A unique identifier for the follow-up suggestion. This will be used to keep user-level state for dismissed items.
+       */
+      id: string;
+      /**
+       * The title of the follow-up suggestion. This will be displayed in the UI as a pill button, so keep it short.
+       */
+      title: string;
+      /**
+       * The function to call when the follow-up suggestion is clicked.
+       */
+      onClick: () => void;
+    };
 
 export interface Conversation {
   id: string;