@worktango/ai-assistant 0.0.69 → 0.0.71-beta01

package/package.json

@@ -1,7 +1,8 @@
 {
   "name": "@worktango/ai-assistant",
-  "version": "0.0.69",
+  "version": "0.0.71-beta01",
   "files": [
+    "constants.js",
     "dist",
     "express.d.ts",
     "express.js",
@@ -24,6 +25,7 @@
     "@ricky0123/vad-react": "^0.0.28",
     "@sinclair/typebox": "^0.34",
     "apexcharts": "^4.7.0",
+    "classnames": "^2.2.6",
     "dom-to-semantic-markdown": "^1.3",
     "html2canvas": "^1.4.1",
     "http-proxy-middleware": "^3",

package/react.d.ts

@@ -1,107 +1,20 @@
 import type { TSchema, Type, Static, TObject } from "@sinclair/typebox";
-
-type ChatItem = {
-  role: "user" | "model" | "tool";
-  parts: ChatData[];
-};
-
-type ToolDeclaration = {
-  name: string;
-  description: string;
-  parameters?: any;
-  chatItemMatches?: (chatItem: {
-    role: "user" | "model" | "tool";
-    parts: ChatData[];
-  }) => boolean;
-  fulfill: (result: any) => Promise<string | ChatDataFile>;
-};
-
-type ChatData =
-  | ChatDataText
-  | ChatDataToolCall
-  | ChatDataToolResult
-  | ChatDataFile;
-
-type ChatDataText = { type: "text"; text: string };
-
-type ChatDataFile = {
-  type: "file";
-  fileUri: string;
-  mimeType: string;
-};
-
-type ChatDataToolCall = {
-  type: "toolCall";
-  toolName: string;
-  args: Record<string, unknown>;
-};
-
-type ChatDataToolResult = {
-  type: "toolResult";
-  toolName: string;
-  result: any;
-};
-
-export type ChatHistoryItem = {
-  title: string;
-  messages: any[];
-  [key: string]: any;
-};
-
-export type ChatHistory = Record<string, ChatHistoryItem[]>;
-
-export type ChatSizes = "small" | "default" | "large" | "full";
-
-export type AiAssistantControls = {
-  setPrompt?: (prompt: string) => void;
-};
-
-export type AiAssistantProps = {
-  /**
-   * The title to display in the window. Default: "WorkTango AI Assistant".
-   */
-  windowTitle?: React.ReactNode;
-  tools?: ToolDeclaration[];
-  enableDebug: boolean;
-  onClose: () => void;
-  hasCloseButton?: boolean;
-  chatWindowZeroState?: React.ReactNode;
-  hasToggleChatSizeButton?: boolean;
-  hasChatHistory?: boolean;
-  chatHistory?: ChatHistory;
-  startingWidth?: ChatSizes;
-  /**
-   * The system instruction to use for the AI assistant.
-   */
-  systemInstruction?: () => string | Promise<string>;
-  /**
-   * The initial prompt to use for the AI assistant.
-   */
-  initialPrompt?: string;
-  /**
-   * A ref to the controls object for the AI assistant. Used to allow some form
-   * of control over the AI assistant from outside the component, allowing things
-   * like recommended prompts to be set from outside the component.
-   */
-  controls?: React.MutableRefObject<AiAssistantControls>;
-  /**
-   * Whether to show the chat history by default. Default: true.
-   */
-  chatHistoryShownByDefault?: boolean;
-  /**
-   * Otherwise, if there's a chat history, we start with the latest history, so
-   * the user can continue the previous conversation.
-   */
-  startWithNewChat?: boolean;
-  /**
-   * The minimum number of characters that must be entered before the user can
-   * submit a message. Default: 5.
-   */
-  minimumCharacterCount?: number;
-};
+import {
+  AiAssistantProps,
+  ChatData,
+  ChatItem,
+  GetChatCompletionOptions,
+  GetStructuredResponseOptions,
+  ToolDeclaration,
+  AiAssistantSuggestedPromptsProps,
+} from "./sharedTypes";
 
 export function AiAssistant(args: AiAssistantProps): JSX.Element;
 
+export function AiAssistantSuggestedPrompts(
+  args: AiAssistantSuggestedPromptsProps
+): JSX.Element;
+
 export const tools: {
   kazooTools: {
     getRewardsAndRecognitionCurrentUserTool: ToolDeclaration;
@@ -112,13 +25,7 @@ export const tools: {
     askSingleShotQuestion: ToolDeclaration;
     analyzeCurrentPage: ToolDeclaration;
   };
-  createTool<TParams extends TSchema = TSchema>({
-    name,
-    description,
-    parameters: userParameters,
-    fulfill,
-    chatItemMatches,
-  }: {
+  createTool<TParams extends TSchema = TSchema>(args: {
     name: string;
     description: string;
     parameters?: (t: typeof Type) => TParams;
@@ -170,50 +77,36 @@ export const config: {
   getScreenshotMainContentSelector: () => string;
 };
 
-export interface Conversation {
-  id: string;
-  title: string;
-  updatedAt: Date;
-}
-
-import type { Static, TSchema, Type } from "@sinclair/typebox";
-
-export type GetStructuredResponseOptions<T extends TSchema> = {
-  requestBody: {
-    model: SupportedTextModel | undefined;
-    temperature: number | undefined;
-    prompt: string;
-    schema: (t: typeof Type) => TSchema;
-  };
-};
-
 /**
  * Makes a structured response request to the LLM API.
  */
-export async function getStructuredResponse<T extends TSchema>({
-  requestBody,
-}: GetStructuredResponseOptions<T>): Promise<Static<T>>;
-
-export type GetChatCompletionOptions = {
-  requestBody: {
-    model: SupportedTextModel | undefined;
-    temperature: number | undefined;
-    stream: boolean;
-    messages: ChatItem[];
-    tools?: ToolDeclaration[] | undefined;
-  };
-  onChunk?: (chunk: ChatData) => void;
-  onError?: (error: Error) => void;
-  onComplete?: () => void;
-};
+export function getStructuredResponse<T extends TSchema>(
+  args: GetStructuredResponseOptions<T>
+): Promise<Static<T>>;
 
 /**
  * Makes a chat completion request to the LLM API. Can be used both with streaming and non-streaming responses.
  * For streaming responses, provides callbacks for handling chunks and errors.
  */
-export async function getChatCompletion({
-  requestBody,
-  onChunk: userOnChunk,
-  onError,
-  onComplete,
-}: GetChatCompletionOptions): Promise<ChatData[]>;
+export function getChatCompletion(
+  args: GetChatCompletionOptions
+): Promise<ChatData[]>;
+
+export {
+  AiAssistantControls,
+  ChatData,
+  ChatItem,
+  GetChatCompletionOptions,
+  GetStructuredResponseOptions,
+  ToolDeclaration,
+  AiAssistantProps,
+  ChatDataFile,
+  ChatDataText,
+  ChatDataToolCall,
+  ChatDataToolResult,
+  ChatHistory,
+  ChatHistoryItem,
+  ChatSizes,
+  Conversation,
+  SupportedTextModel,
+} from "./sharedTypes";