@worktango/ai-assistant 0.0.90 → 0.0.91

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@worktango/ai-assistant",
-  "version": "0.0.90",
+  "version": "0.0.91",
   "files": [
     "constants.js",
     "dist",
@@ -31,8 +31,10 @@
     "html2canvas": "^1.4.1",
     "http-proxy-middleware": "^3",
     "json-nd": "^1",
+    "lodash.clonedeep": "^4",
     "micromark": "^4",
-    "react-apexcharts": "^1.7.0"
+    "react-apexcharts": "^1.7.0",
+    "uuid": "^9"
   },
   "devDependencies": {
     "@vitejs/plugin-react": "^4.2.1",
@@ -42,7 +44,13 @@
     "vite": "~6.2.4",
     "vite-plugin-css-modules": "0.0.1"
   },
-  "graphqlCodeGeneration": {},
+  "graphqlCodeGeneration": {
+    "apiHooks": {
+      "config": {
+        "enumsAsTypes": true
+      }
+    }
+  },
   "$schema": "../../packages/build-tools/schema.json",
   "peerDependencies": {
     "express": ">=4",

package/react.d.ts

@@ -9,9 +9,13 @@ import {
   AiAssistantSuggestedPromptsProps,
   GetFollowUpSuggestions,
   ChatDataFile,
+  AiAssistantControls,
+  FollowUpSuggestion,
 } from "./sharedTypes";
 
-export const defaultGetFollowUpSuggestions: GetFollowUpSuggestions;
+export const getDefaultGetFollowUpSuggestions: (args: {
+  promptOverride?: string;
+}) => GetFollowUpSuggestions;
 
 export function AiAssistant(args: AiAssistantProps): JSX.Element;
 

package/sharedTypes.ts

@@ -1,5 +1,7 @@
 import type { Static, TSchema, Type } from "@sinclair/typebox";
 
+import { Message } from "./generated/apiHooks";
+
 export type AiAssistantSuggestedPromptsProps = {
   suggestedPrompts: Record<string, AiAssistantSuggestedPrompt[]>;
   onSelect: (prompt: AiAssistantSuggestedPrompt) => void | Promise<void>;
@@ -29,6 +31,59 @@ export type ToolDeclaration<TParams extends TSchema = TSchema> = {
   fulfill: (result: Static<TParams>) => Promise<string | ChatDataFile>;
 };
 
+/**
+ * Debug info added during the execution of the AI assistant.
+ */
+export type DebugInfo = {
+  time: Date;
+  message: string;
+  data?: any;
+  level: "info" | "error" | "warn";
+}[];
+
+export type PropRefs = {
+  onAfterLlmCall?: (messages: Omit<Message, "__typename">[]) => void;
+  onBeforeLlmCall?: () => void;
+  onError?: (error: Error) => void;
+  onChunkAdded?: () => void;
+  tools?: ToolDeclaration[];
+  model?: SupportedTextModel;
+  temperature?: number;
+  isMounted: boolean;
+  systemInstruction?: () => string | Promise<string>;
+  getConversationById: (
+    conversationId: string
+  ) => ChatStateTree[string] | undefined;
+  getFollowUpSuggestions?: GetFollowUpSuggestions;
+  setFollowUpSuggestionsByConversationId: React.Dispatch<
+    React.SetStateAction<Record<string, FollowUpSuggestion[]>>
+  >;
+  handleSendUserMessage: (args: {
+    conversationId?: string;
+    messageText: string;
+  }) => Promise<void>;
+  setPrompt: (
+    conversationId: string | undefined,
+    prompt: string | ((prev: string) => string)
+  ) => void;
+  onSelectedConversationChange?: (
+    conversationId: string | undefined,
+    messages: ChatStateTree
+  ) => void;
+  selectedConversationId: string | undefined;
+  onSelectedConversationLoaded?: (
+    /**
+     * The conversation that loaded.
+     */
+    conversationId: string | undefined,
+    /**
+     * The conversation that is currently selected
+     */
+    selectedConversationId: string | undefined,
+    messages: ChatStateTree
+  ) => void;
+};
+
 export type ChatData =
   | ChatDataText
   | ChatDataToolCall
@@ -39,8 +94,10 @@ export type ChatDataText = { type: "text"; text: string };
 
 export type ChatDataFile = {
   type: "file";
+  toolName: string;
   fileUri: string;
   mimeType: string;
+  args?: unknown;
 };
 
 export type ChatDataToolCall = {
@@ -52,6 +109,7 @@ export type ChatDataToolCall = {
 export type ChatDataToolResult = {
   type: "toolResult";
   toolName: string;
+  args?: unknown;
   result: any;
 };
 
@@ -62,11 +120,8 @@ export type ChatHistoryItem = {
 };
 
 export type GetFollowUpSuggestions = (args: {
-  chatHistory: ChatItem[];
-  aiAssistantControls: React.MutableRefObject<AiAssistantControls>;
-  setFollowUpSuggestions: (suggestions: FollowUpSuggestion[]) => void;
-  followUpSuggestionsPromptOverride?: string;
-}) => Promise<FollowUpSuggestion[]>;
+  conversation: ChatStateTree[string];
+}) => Promise<AiAssistantSuggestedPrompt[]>;
 
 export type ChatHistory = Record<string, ChatHistoryItem[]>;
 
@@ -98,28 +153,6 @@ export type AiAssistantProps = {
    * The function to call when the AI assistant is closed.
    */
   onClose: () => void;
-  /**
-   * The function to call when the AI assistant receives a text reply from the LLM.
-   * Is invoked with the entire chat history, including the new message.
-   */
-  onLlmTextReply?: (chatHistory: ChatItem[]) => void;
-  /**
-   * The function to call when the selected conversation changes. Is invoked with
-   * the ID of the selected conversation, or undefined if there is no selected
-   * conversation.
-   */
-  onSelectedConversationChange?: (conversationId?: string) => void;
-  /**
-   * This is a callback event that is invoked when the selected conversation has
-   * loaded. We do not pass the conversation itself because this is to be used as
-   * an event to trigger other actions, like loading suggested follow ups.
-   *
-   * If the conversation is new, the `conversation` argument will be undefined.
-   */
-  onSelectedConversationLoaded?: (conversation?: {
-    conversationId: string;
-    messages: ChatItem[];
-  }) => void;
   /**
    * This is a callback event that is invoked when the user sends a message.
    */
@@ -173,9 +206,37 @@ export type AiAssistantProps = {
    */
   minimumCharacterCount?: number;
   /**
-   * Follow up suggestions to display in the chat window, just above the text area.
+   * The ID of the conversation to start with. If not provided, a new conversation will be created.
+   */
+  initialConversationId?: string;
+  /**
+   * Returns a set of follow-up suggestions for the given conversation.
    */
-  followUpSuggestions?: FollowUpSuggestion[];
+  getFollowUpSuggestions?: GetFollowUpSuggestions;
+  /**
+   * Invoked with a raw string of the chat (in plain text) to get a possible title
+   */
+  getChatTitle?: (chat: string) => Promise<string>;
+};
+
+export type ChatStateTree = Record<
+  string,
+  {
+    id: string;
+    title: string;
+    isUpdatingTitle?: boolean;
+    updatedAt: Date;
+    isLoading: boolean;
+    messages: Omit<Message, "__typename">[];
+  }
+>;
+
+// ChatItem augmented with some UI stuff
+export type AugmentedChatItem = ChatItem & {
+  id?: string;
+  timestamp?: Date;
+  isPending?: boolean;
+  filterFromApiCalls?: boolean;
 };
 
 /**
@@ -203,7 +264,7 @@ export type FollowUpSuggestion =
 export interface Conversation {
   id: string;
   title: string;
-  updatedAt: Date;
+  updatedAt?: Date;
   messages: {
     conversationId: string;
     id: string;

package/dist/DatePicker-l0sNRNKZ.mjs

@@ -1 +0,0 @@
-