type.test.ts

import type { ChatMessage, MessageContent, MessageType } from "llamaindex";
import { expectTypeOf, test } from "vitest";
import type { ChatResponse } from "../src/index.js";

test("chat message type", () => {
  // if generic is not provided, `options` is not required
  expectTypeOf<ChatMessage>().toMatchTypeOf<{
    content: MessageContent;
    role: MessageType;
  }>();
  expectTypeOf<ChatMessage>().toMatchTypeOf<{
    content: MessageContent;
    role: MessageType;
    options?: object | undefined;
  }>();
  expectTypeOf<ChatMessage>().not.toMatchTypeOf<{
    content: MessageContent;
    role: MessageType;
    options: Record<string, unknown>;
  }>();
  type Options = {
    a: string;
    b: number;
  };
  expectTypeOf<ChatMessage<Options>>().toMatchTypeOf<{
    content: MessageContent;
    role: MessageType;
    options?: Options | undefined;
  }>();
});

test("chat response type", () => {
  // if generic is not provided, `options` is not required
  expectTypeOf<ChatResponse>().toMatchTypeOf<{
    message: ChatMessage;
    raw: object | null;
  }>();
  expectTypeOf<ChatResponse>().toMatchTypeOf<{
    message: ChatMessage;
    raw: object | null;
    options?: Record<string, unknown>;
  }>();
  expectTypeOf<ChatResponse>().not.toMatchTypeOf<{
    message: ChatMessage;
    raw: object | null;
    options: Record<string, unknown>;
  }>();
  type Options = {
    a: string;
    b: number;
  };
  expectTypeOf<ChatResponse<Options>>().toMatchTypeOf<{
    message: ChatMessage<Options>;
    raw: object | null;
  }>();
});