@worktango/ai-assistant 0.0.28 → 0.0.30

package/express.d.ts

@@ -0,0 +1,38 @@
+import type { RequestHandler, Request, Application } from "express";
+
+export function setupAiAssistantRoutes(args: {
+  app: Application;
+  /**
+   * A function that returns the bearer token for the AI assistant to use for
+   * back-channel request proxying. This should be a JWT token that can be
+   * decoded on the `kazoo-web` application to authenticate the request.
+   */
+  getBearerToken: () => Promise<string>;
+  /**
+   * The host of the application to proxy requests to.
+   */
+  targetHost: string;
+  /**
+   * The port of the application to proxy requests to. Defaults to 80.
+   */
+  targetPort?: number;
+  /**
+   * The route path to the local service's AI assistant. Defaults to
+   * "/ai-assistant".
+   */
+  route?: string;
+  /**
+   * The logger to use for logging.
+   */
+  logger?: Pick<typeof console, "log" | "error" | "debug">;
+  /**
+   * Middleware functions to run before the proxy request.
+   */
+  preflightMiddlewares?: RequestHandler[];
+  /**
+   * A function that should return the target path for the proxy request, based
+   * on the incoming request object. The path returned will be appended to the
+   * target host and port to form the final URL.
+   */
+  getTargetPath: (req: Request) => string;
+}): void;

package/package.json

@@ -1,11 +1,11 @@
 {
   "name": "@worktango/ai-assistant",
-  "version": "0.0.28",
-  "types": "types.d.ts",
+  "version": "0.0.30",
   "files": [
     "dist",
-    "types.d.ts",
+    "express.d.ts",
     "express.js",
+    "react.d.ts",
     "react.js"
   ],
   "scripts": {

package/react.d.ts

@@ -0,0 +1,47 @@
+type ToolDeclaration = {
+  name: string;
+  description: string;
+  parameters?: any;
+  chatItemMatches?: (chatItem: {
+    role: "user" | "model" | "tool";
+    parts: ChatData[];
+  }) => boolean;
+  fulfill: (result: any) => Promise<string>;
+};
+
+type ChatData = ChatDataText | ChatDataToolCall | ChatDataToolResult;
+
+type ChatDataText = { type: "text"; text: string };
+
+type ChatDataToolCall = {
+  type: "toolCall";
+  toolName: string;
+  args: Record<string, unknown>;
+};
+
+type ChatDataToolResult = {
+  type: "toolResult";
+  toolName: string;
+  result: any;
+};
+
+export function AiAssistant({
+  tools,
+}: {
+  tools?: ToolDeclaration[];
+}): JSX.Element;
+
+export const tools: {
+  kazooTools: {
+    getRewardsAndRecognitionCurrentUserTool: ToolDeclaration;
+  };
+};
+
+export const config: {
+  getLlmCallUrl: () => string;
+  getKazooUrl: (urlPath: string) => string;
+  getPulseUrl: (urlPath: string) => string;
+  setKazooUrlStrategy: (strategy: (urlPath: string) => string) => void;
+  setLlmCallUrlStrategy: (strategy: () => string) => void;
+  setPulseUrlStrategy: (strategy: (urlPath: string) => string) => void;
+};

package/types.d.ts

@@ -1,94 +0,0 @@
-type ToolDeclaration = {
-  name: string;
-  description: string;
-  parameters?: JsonSchema;
-  chatItemMatches?: (chatItem: {
-    role: "user" | "model" | "tool";
-    parts: ChatData[];
-  }) => boolean;
-  fulfill: (result: any) => Promise<string>;
-};
-
-type ChatData = ChatDataText | ChatDataToolCall | ChatDataToolResult;
-
-type ChatDataText = { type: "text"; text: string };
-
-type ChatDataToolCall = {
-  type: "toolCall";
-  toolName: string;
-  args: Record<string, unknown>;
-};
-
-type ChatDataToolResult = {
-  type: "toolResult";
-  toolName: string;
-  result: any;
-};
-
-declare module "@worktango/ai-assistant/react" {
-  export function AiAssistant({
-    tools,
-  }: {
-    tools?: ToolDeclaration[];
-  }): JSX.Element;
-
-  export namespace tools {
-    export const kazooTools: {
-      getRewardsAndRecognitionCurrentUserTool: ToolDeclaration;
-    };
-  }
-
-  export namespace config {
-    export function getLlmCallUrl(): string;
-    export function getKazooUrl(urlPath: string): string;
-    export function getPulseUrl(urlPath: string): string;
-    export function setKazooUrlStrategy(
-      strategy: (urlPath: string) => string
-    ): void;
-    export function setLlmCallUrlStrategy(strategy: () => string): void;
-    export function setPulseUrlStrategy(
-      strategy: (urlPath: string) => string
-    ): void;
-  }
-}
-
-declare module "@worktango/ai-assistant/express" {
-  export function setupAiAssistantRoutes(args: {
-    app: {
-      all: (path: string, ...args: any[]) => void;
-    };
-    /**
-     * A function that returns the bearer token for the AI assistant to use for
-     * back-channel request proxying. This should be a JWT token that can be
-     * decoded on the `kazoo-web` application to authenticate the request.
-     */
-    getBearerToken: () => Promise<string>;
-    /**
-     * The host of the application to proxy requests to.
-     */
-    targetHost: string;
-    /**
-     * The port of the application to proxy requests to. Defaults to 80.
-     */
-    targetPort?: number;
-    /**
-     * The route path to the local service's AI assistant. Defaults to
-     * "/ai-assistant".
-     */
-    route?: string;
-    /**
-     * The logger to use for logging.
-     */
-    logger?: Pick<typeof console, "log" | "error" | "debug">;
-    /**
-     * Middleware functions to run before the proxy request.
-     */
-    preflightMiddlewares?: RequestHandler[];
-    /**
-     * A function that should return the target path for the proxy request, based
-     * on the incoming request object. The path returned will be appended to the
-     * target host and port to form the final URL.
-     */
-    getTargetPath: (req: Request) => string;
-  }): void;
-}