openAgent

1import { describe, expect, test } from "bun:test"
2import { LLM } from "../src"
3import * as OpenAIChat from "../src/protocols/openai-chat"
4import { Endpoint } from "../src/route"
5import { Model } from "../src/schema"
6 
7const request = () =>
8  LLM.request({
9    model: Model.make({
10      id: "model-1",
11      provider: "test",
12      route: OpenAIChat.route,
13    }),
14    prompt: "hello",
15  })
16 
17describe("Endpoint", () => {
18  test("appends a static path to the model's baseURL", () => {
19    const url = Endpoint.render(Endpoint.path("/chat", { baseURL: "https://api.example.test/v1/" }), {
20      request: request(),
21      body: {},
22    })
23 
24    expect(url.toString()).toBe("https://api.example.test/v1/chat")
25  })
26 
27  test("endpoint query params are appended to the rendered URL", () => {
28    const url = Endpoint.render(
29      Endpoint.path("/chat?alt=sse", {
30        baseURL: "https://custom.example.test/root/",
31        query: { "api-version": "2026-01-01", alt: "json" },
32      }),
33      {
34        request: request(),
35        body: {},
36      },
37    )
38 
39    expect(url.toString()).toBe("https://custom.example.test/root/chat?alt=json&api-version=2026-01-01")
40  })
41 
42  test("path may be a function of the validated body", () => {
43    const url = Endpoint.render(
44      Endpoint.path<{ readonly modelId: string }>(
45        ({ body }) => `/model/${encodeURIComponent(body.modelId)}/converse-stream`,
46        { baseURL: "https://bedrock-runtime.us-east-1.amazonaws.com" },
47      ),
48      {
49        request: request(),
50        body: { modelId: "us.amazon.nova-micro-v1:0" },
51      },
52    )
53 
54    expect(url.toString()).toBe(
55      "https://bedrock-runtime.us-east-1.amazonaws.com/model/us.amazon.nova-micro-v1%3A0/converse-stream",
56    )
57  })
58})
59