File size: 2,314 Bytes
63522a5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 | import { buildHttpBaseUrl } from "#/utils/websocket-url";
import { getAgentServerWorkingDir } from "./agent-server-config";
import { getEffectiveLocalBackend } from "./backend-registry/active-store";
import type { Backend } from "./backend-registry/types";
export interface AgentServerClientOverrides {
host?: string;
apiKey?: string | null;
sessionApiKey?: string | null;
workingDir?: string;
conversationUrl?: string | null;
timeout?: number;
}
export interface AgentServerClientOptions {
host: string;
apiKey?: string;
workingDir: string;
timeout?: number;
}
export class NoBackendAvailableError extends Error {
constructor() {
super("No backend is configured.");
this.name = "NoBackendAvailableError";
}
}
export const isNoBackendAvailableError = (
error: unknown,
): error is NoBackendAvailableError =>
error instanceof NoBackendAvailableError ||
(typeof error === "object" &&
error !== null &&
"name" in error &&
error.name === "NoBackendAvailableError");
function normalizeHost(host: string): string {
return host.replace(/\/+$/, "");
}
function resolveHost(
overrides: AgentServerClientOverrides,
backend: Backend | null,
): string {
if (overrides.host) return normalizeHost(overrides.host);
if (overrides.conversationUrl)
return normalizeHost(buildHttpBaseUrl(overrides.conversationUrl));
return normalizeHost(backend?.host ?? "");
}
export function getAgentServerClientOptions(
overrides: AgentServerClientOverrides = {},
): AgentServerClientOptions {
const backend = getEffectiveLocalBackend();
if (!backend && !overrides.host && !overrides.conversationUrl) {
throw new NoBackendAvailableError();
}
const apiKey =
overrides.sessionApiKey ?? overrides.apiKey ?? backend?.apiKey ?? undefined;
return {
host: resolveHost(overrides, backend),
...(apiKey ? { apiKey } : {}),
workingDir: overrides.workingDir ?? getAgentServerWorkingDir(),
...(overrides.timeout !== undefined ? { timeout: overrides.timeout } : {}),
};
}
export function getAgentServerHttpClientOptions(
overrides?: AgentServerClientOverrides,
) {
const { host, apiKey, timeout } = getAgentServerClientOptions(overrides);
return {
baseUrl: host,
...(apiKey ? { apiKey } : {}),
timeout: timeout ?? 60000,
};
}
|