@worktango/ai-assistant 0.0.104 → 0.0.106

package/express.d.ts

@@ -1,5 +1,7 @@
 import type { RequestHandler, Request, Application } from "express";
 
+export * from "./src/backend/types";
+
 export function setupAiAssistantRoutes(args: {
   app: Application;
   /**
@@ -52,4 +54,35 @@ export function setupAiAssistantRoutes(args: {
    * ```
    */
   getTargetPath: (fullPath: string) => string;
+  /**
+   * Optional callbacks to invoke after successful REST endpoint responses.
+   * Only invoked when the response status is 200.
+   *
+   * @example
+   * ```ts
+   *   onRestEndpointSuccess: {
+   *     '/api/api-ai-assistant/v1/chat': async ({ request, response }) => {
+   *       console.log('Chat complete:', response);
+   *     },
+   *   }
+   * ```
+   */
+  onRestEndpointSuccess?: import("./src/backend/types").RestEndpointCallbacks;
+  /**
+   * Optional callbacks to invoke after successful GraphQL operation responses.
+   * Only invoked when the response status is 200 and the operation name matches.
+   * Keys should match the GraphQL operation names (e.g., "AppendMessageToConversation").
+   *
+   * @example
+   * ```ts
+   *   onGraphQLOperationSuccess: {
+   *     AppendMessageToConversation: async ({ request, response }) => {
+   *       const conversationId = response.data?.appendMessage?.conversation?.id;
+   *       console.log('Message appended to conversation:', conversationId);
+   *       // Store conversation ID, update UI, etc.
+   *     },
+   *   }
+   * ```
+   */
+  onGraphQLOperationSuccess?: import("./src/backend/types").GraphQLOperationCallbacks;
 }): void;

package/package.json

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

package/sharedTypes.ts

@@ -55,6 +55,8 @@ export type AiAssistantSuggestedPromptsProps = {
 export type AiAssistantSuggestedPrompt = {
   title: string;
   fullPrompt?: string;
+  // Allow customization of the suggested action button in the UI
+  className?: string;
 };
 
 export type SupportedTextModel = "small" | "medium" | "large";
@@ -328,6 +330,10 @@ export type FollowUpSuggestion =
        * The function to call when the follow-up suggestion is clicked.
        */
       onClick: () => void;
+      /**
+       * Optional CSS class name for custom styling of the pill.
+       */
+      className?: string;
     };
 
 export interface Conversation {