/**
 * @license
 * Copyright 2023 Google LLC
 * SPDX-License-Identifier: Apache-2.0
 */
import type { AnyProxyRequestMessage, AnyProxyResponseMessage, InputValues, Kit, NodeDescriptor, NodeHandlerContext, OutputValues } from "@breadboard-ai/types";
import { NodeProxyConfig, ProxyServerConfig } from "./config.js";
import { ClientTransport, ServerTransport } from "./types.js";
type ProxyServerTransport = ServerTransport<AnyProxyRequestMessage, AnyProxyResponseMessage>;
export declare class ProxyServer {
    #private;
    constructor(transport: ProxyServerTransport);
    serve(config: ProxyServerConfig): Promise<void>;
}
type ProxyClientTransport = ClientTransport<AnyProxyRequestMessage, AnyProxyResponseMessage>;
export declare class ProxyClient {
    #private;
    constructor(transport: ProxyClientTransport);
    shutdownServer(): void;
    proxy(node: NodeDescriptor, inputs: InputValues, context: NodeHandlerContext): Promise<OutputValues>;
    createProxyKit(args?: NodeProxyConfig, fallback?: Kit[]): {
        url: string;
        handlers: {
            [k: string]: {
                invoke: (inputs: InputValues, context: NodeHandlerContext) => Promise<void | OutputValues>;
            };
        };
    };
}
export {};
//# sourceMappingURL=proxy.d.ts.map