@worktango/ai-assistant 0.0.68 → 0.0.70

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@worktango/ai-assistant",
-  "version": "0.0.68",
+  "version": "0.0.70",
   "files": [
     "dist",
     "express.d.ts",
@@ -22,7 +22,9 @@
   "dependencies": {
     "@apollo/client": "^3.0.0",
     "@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",
@@ -31,7 +33,6 @@
     "react-apexcharts": "^1.7.0"
   },
   "devDependencies": {
-    "@sinclair/typebox": "^0.34",
     "@vitejs/plugin-react": "^4.2.1",
     "esbuild": "^0.17.11",
     "ts-node": "^10.9.1",

package/react.d.ts

@@ -1,104 +1,12 @@
 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,
+} from "./sharedTypes";
 
 export function AiAssistant(args: AiAssistantProps): JSX.Element;
 
@@ -112,13 +20,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,8 +72,36 @@ export const config: {
   getScreenshotMainContentSelector: () => string;
 };
 
-export interface Conversation {
-  id: string;
-  title: string;
-  updatedAt: Date;
-}
+/**
+ * Makes a structured response request to the LLM API.
+ */
+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 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";