@worktango/ai-assistant 0.0.25 → 0.0.28

package/dist/express.js

@@ -50,17 +50,18 @@ function setupAiAssistantRoutes(args) {
       if (!bearerToken) {
         return;
       }
-      const parsedUrl = new URL(req.url);
+      const targetPath = args.getTargetPath(req);
       const options = {
         hostname: args.targetHost,
         port: args.targetPort,
-        path: parsedUrl.pathname,
+        path: targetPath,
         method: req.method,
         headers: { ...req.headers }
       };
       options.headers["Authorization"] = `Bearer ${bearerToken}`;
       delete options.headers.host;
       logger.debug("Proxying request:", {
+        targetPath,
         method: req.method,
         headers: options.headers,
         proxyOptions: options

package/package.json

@@ -1,15 +1,11 @@
 {
   "name": "@worktango/ai-assistant",
-  "version": "0.0.25",
-  "types": "./types.d.ts",
+  "version": "0.0.28",
+  "types": "types.d.ts",
   "files": [
     "dist",
-    "src",
-    "main.js",
     "types.d.ts",
     "express.js",
-    "express.d.ts",
-    "react.ts",
     "react.js"
   ],
   "scripts": {
@@ -18,7 +14,7 @@
     "build:backend": "esbuild src/backend/express.ts --bundle --platform=node --outfile=dist/express.js \"--external:*\"",
     "build:frontend": "vite build",
     "build:all": "rm -rf dist && yarn build:backend && yarn build:frontend && yarn ts-node -T scripts/injectCssIntoBuild.ts",
-    "prepareBuildForPublish": "yarn ts-node -T scripts/prepareBuildForPublish.ts && mv react.public.js react.js && mv express.public.js express.js && mv react.ts react.d.ts",
+    "prepareBuildForPublish": "yarn ts-node -T scripts/prepareBuildForPublish.ts",
     "publish": "yarn build:all && yarn prepareBuildForPublish && yarn actuallyPublishPackage",
     "actuallyPublishPackage": "yarn npm publish --access public --tolerate-republish",
     "gql:generate": "gql-generate"

package/types.d.ts

@@ -32,8 +32,24 @@ declare module "@worktango/ai-assistant/react" {
     tools?: ToolDeclaration[];
   }): JSX.Element;
 
-  export * as tools from "./src/frontend/tools/llm-tools/index";
-  export * as config from "./src/frontend/config";
+  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" {
@@ -68,5 +84,11 @@ declare module "@worktango/ai-assistant/express" {
      * 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/express.d.ts

@@ -1 +0,0 @@
-export { setupAiAssistantRoutes } from "./src/backend/express";

package/src/backend/express.test.ts

@@ -1,208 +0,0 @@
-import http from "http";
-
-import { Express } from "express";
-
-import { setupAiAssistantRoutes } from "./express";
-
-jest.mock("http", () => ({
-  request: jest.fn().mockImplementation((options, callback) => {
-    const mockProxyRes = {
-      statusCode: 200,
-      headers: { "content-type": "application/json" },
-      pipe: jest.fn(),
-    };
-    callback(mockProxyRes);
-    return {
-      on: jest.fn(),
-      end: jest.fn(),
-    };
-  }),
-}));
-
-describe("setupAiAssistantRoutes", () => {
-  let mockApp: jest.Mocked<Express>;
-  let mockPost: jest.Mock;
-  let mockGetBearerToken: jest.Mock;
-  let mockReq: any;
-  let mockRes: any;
-  let handlerPromise: Promise<void>;
-
-  const mockLogger = {
-    log: jest.fn(),
-    error: jest.fn(),
-    debug: jest.fn(),
-  };
-
-  beforeEach(() => {
-    mockReq = {
-      url: "http://localhost/ai-assistant",
-      method: "POST",
-      headers: { "content-type": "application/json" },
-      readable: false,
-    };
-    mockRes = {
-      statusCode: 200,
-      setHeader: jest.fn(),
-      status: jest.fn().mockReturnThis(),
-      send: jest.fn(),
-    };
-    mockPost = jest.fn().mockImplementation((path, handler) => {
-      handlerPromise = handler(mockReq, mockRes);
-    });
-    mockGetBearerToken = jest.fn().mockResolvedValue("mock-token");
-
-    mockApp = {
-      post: mockPost,
-      all: mockPost,
-    } as unknown as jest.Mocked<Express>;
-
-    (http.request as jest.Mock).mockClear();
-  });
-
-  it("sets up the default route correctly", async () => {
-    setupAiAssistantRoutes({
-      app: mockApp,
-      getBearerToken: mockGetBearerToken,
-      targetHost: "localhost",
-      logger: mockLogger,
-    });
-
-    expect(mockPost).toHaveBeenCalledWith(
-      "/ai-assistant",
-      expect.any(Function)
-    );
-
-    await handlerPromise;
-
-    expect(http.request).toHaveBeenCalledWith(
-      expect.objectContaining({
-        hostname: "localhost",
-        path: "/ai-assistant",
-        headers: expect.objectContaining({
-          Authorization: "Bearer mock-token",
-        }),
-      }),
-      expect.any(Function)
-    );
-  });
-
-  it("sets up a custom route when provided", async () => {
-    mockReq.url = "http://localhost/custom-route";
-
-    setupAiAssistantRoutes({
-      app: mockApp,
-      getBearerToken: mockGetBearerToken,
-      targetHost: "localhost",
-      route: "/custom-route",
-      logger: mockLogger,
-    });
-
-    expect(mockPost).toHaveBeenCalledWith(
-      "/custom-route",
-      expect.any(Function)
-    );
-
-    await handlerPromise;
-
-    expect(http.request).toHaveBeenCalledWith(
-      expect.objectContaining({
-        hostname: "localhost",
-        path: "/custom-route",
-        headers: expect.objectContaining({
-          Authorization: "Bearer mock-token",
-        }),
-      }),
-      expect.any(Function)
-    );
-  });
-
-  it("handles errors when getting the bearer token", async () => {
-    mockGetBearerToken.mockRejectedValueOnce(
-      new Error("Error getting bearer token")
-    );
-
-    setupAiAssistantRoutes({
-      app: mockApp,
-      getBearerToken: mockGetBearerToken,
-      targetHost: "localhost",
-      logger: mockLogger,
-    });
-
-    await handlerPromise;
-
-    expect(mockRes.status).toHaveBeenCalledWith(500);
-    expect(mockRes.send).toHaveBeenCalledWith("Error getting bearer token");
-  });
-
-  it("handles proxy errors correctly", async () => {
-    const mockError = new Error("Proxy error");
-    (http.request as jest.Mock).mockImplementationOnce(() => ({
-      on: jest.fn().mockImplementation((event, handler) => {
-        if (event === "error") {
-          handler(mockError);
-        }
-      }),
-      end: jest.fn(),
-    }));
-
-    setupAiAssistantRoutes({
-      app: mockApp,
-      getBearerToken: mockGetBearerToken,
-      targetHost: "localhost",
-      logger: mockLogger,
-    });
-
-    await handlerPromise;
-
-    expect(mockRes.status).toHaveBeenCalledWith(500);
-    expect(mockRes.send).toHaveBeenCalledWith("Proxy Error: Proxy error");
-  });
-
-  it("handles readable request bodies", async () => {
-    mockReq.readable = true;
-    mockReq.pipe = jest.fn();
-
-    setupAiAssistantRoutes({
-      app: mockApp,
-      getBearerToken: mockGetBearerToken,
-      targetHost: "localhost",
-      logger: mockLogger,
-    });
-
-    await handlerPromise;
-
-    expect(mockReq.pipe).toHaveBeenCalled();
-  });
-
-  it("handles response headers and status code correctly", async () => {
-    const mockProxyRes = {
-      statusCode: undefined,
-      headers: {
-        "content-type": undefined,
-        "x-custom-header": "value",
-      },
-      pipe: jest.fn(),
-    };
-
-    (http.request as jest.Mock).mockImplementationOnce((options, callback) => {
-      callback(mockProxyRes);
-      return {
-        on: jest.fn(),
-        end: jest.fn(),
-      };
-    });
-
-    setupAiAssistantRoutes({
-      app: mockApp,
-      getBearerToken: mockGetBearerToken,
-      targetHost: "localhost",
-      logger: mockLogger,
-    });
-
-    await handlerPromise;
-
-    expect(mockRes.statusCode).toBe(500);
-    expect(mockRes.setHeader).toHaveBeenCalledWith("x-custom-header", "value");
-    expect(mockRes.setHeader).toHaveBeenCalledWith("content-type", "");
-  });
-});

package/src/backend/express.ts

@@ -1,98 +0,0 @@
-import http from "http";
-
-import { Express, RequestHandler } from "express";
-
-import { coalesce } from "@kazoohr/helpers";
-
-/**
- * Sets up the AI assistant routes on the given Express app.
- */
-export function setupAiAssistantRoutes(args: {
-  app: Express;
-  /**
-   * 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[];
-}) {
-  const logger = coalesce(args.logger, console);
-  const routePath = args.route ?? "/ai-assistant";
-
-  args.app.all(
-    routePath,
-    ...coalesce(args.preflightMiddlewares, []),
-    async (req, res) => {
-      const bearerToken = await args.getBearerToken().catch((error) => {
-        logger.error("Error getting bearer token:", error);
-        res.status(500).send("Error getting bearer token");
-        return null;
-      });
-
-      if (!bearerToken) {
-        return;
-      }
-
-      const parsedUrl = new URL(req.url);
-
-      const options = {
-        hostname: args.targetHost,
-        port: args.targetPort,
-        path: parsedUrl.pathname,
-        method: req.method,
-        headers: { ...req.headers },
-      };
-
-      options.headers["Authorization"] = `Bearer ${bearerToken}`;
-      delete options.headers.host;
-
-      logger.debug("Proxying request:", {
-        method: req.method,
-        headers: options.headers,
-        proxyOptions: options,
-      });
-
-      const proxyReq = http.request(options, (proxyRes) => {
-        res.statusCode = proxyRes.statusCode || 500;
-
-        Object.keys(proxyRes.headers).forEach((key) => {
-          res.setHeader(key, proxyRes.headers[key] || "");
-        });
-
-        proxyRes.pipe(res);
-      });
-
-      proxyReq.on("error", (error) => {
-        logger.error("Proxy request error:", error);
-        res.status(500).send("Proxy Error: " + error.message);
-      });
-
-      if (req.readable) {
-        req.pipe(proxyReq);
-      } else {
-        proxyReq.end();
-      }
-    }
-  );
-}

package/src/frontend/AiAssistant.test.tsx

@@ -1,19 +0,0 @@
-import React from "react";
-
-import { render, screen, waitFor } from "@kazoohr/test";
-
-import { AiAssistant } from "./AiAssistant";
-
-jest.mock("./components/AiAssistantComponent", () => ({
-  AiAssistantComponent: () => <div>AiAssistantComponent</div>,
-}));
-
-describe("AiAssistant", () => {
-  it("should render", async () => {
-    const { rerender } = render(<AiAssistant tools={[]} />);
-    await waitFor(() => {
-      rerender(<AiAssistant />);
-      expect(screen.queryByText("AiAssistantComponent")).toBeTruthy();
-    });
-  });
-});

package/src/frontend/AiAssistant.tsx

@@ -1,26 +0,0 @@
-import React from "react";
-
-import { ToastProvider } from "@kazoohr/confetti";
-import { ToolDeclaration } from "@kazoohr/llm/types";
-
-import { AiAssistantComponent } from "./components/AiAssistantComponent";
-import { useStyles } from "./useStyles";
-
-/**
- * This function loads the CSS styles for Confetti (and anything else we styled
- * custom in this app) and injects them into the page. While it's doing that, we
- * show a very minimal loader.
- */
-export function AiAssistant(props: { tools?: ToolDeclaration[] }) {
-  const { isLoaded } = useStyles();
-
-  if (!isLoaded) {
-    return null;
-  }
-
-  return (
-    <ToastProvider>
-      <AiAssistantComponent tools={props.tools} />
-    </ToastProvider>
-  );
-}

package/src/frontend/components/AiAssistantComponent.scss

@@ -1,24 +0,0 @@
-.ai-assistant-textarea {
-  border-radius: 0;
-  background-color: $color--system-gray-6;
-  border-radius: 8px;
-}
-
-.ai-assistant-textarea * {
-  border: none !important;
-  background-color: transparent !important;
-  border-radius: 8px !important;
-}
-
-.ai-assistant-textarea [class*="TextArea_textarea__highlighter"] {
-  min-height: 200px;
-}
-
-.ai-assistant-debug-textarea {
-  min-width: 100%;
-  height: auto;
-  max-width: 100%;
-  aspect-ratio: 3/1;
-  background-color: $color--system-gray-6;
-  padding: 8px;
-}