/**
 * @license
 * Copyright 2024 Google LLC
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
import { SingleRequestOptions } from '../types';
import { ApiSettings } from '../types/internal';
export declare enum Task {
    GENERATE_CONTENT = "generateContent",
    STREAM_GENERATE_CONTENT = "streamGenerateContent",
    COUNT_TOKENS = "countTokens",
    PREDICT = "predict"
}
export declare const enum ServerPromptTemplateTask {
    TEMPLATE_GENERATE_CONTENT = "templateGenerateContent",
    TEMPLATE_STREAM_GENERATE_CONTENT = "templateStreamGenerateContent",
    TEMPLATE_PREDICT = "templatePredict"
}
export declare class RequestUrl {
    model: string;
    task: Task;
    apiSettings: ApiSettings;
    stream: boolean;
    requestOptions?: SingleRequestOptions | undefined;
    constructor(model: string, task: Task, apiSettings: ApiSettings, stream: boolean, requestOptions?: SingleRequestOptions | undefined);
    toString(): string;
    private get baseUrl();
    private get queryParams();
}
export declare class TemplateRequestUrl {
    templateId: string;
    task: ServerPromptTemplateTask;
    apiSettings: ApiSettings;
    stream: boolean;
    requestOptions?: SingleRequestOptions | undefined;
    constructor(templateId: string, task: ServerPromptTemplateTask, apiSettings: ApiSettings, stream: boolean, requestOptions?: SingleRequestOptions | undefined);
    toString(): string;
    private get baseUrl();
    private get queryParams();
}
export declare function getHeaders(url: RequestUrl): Promise<Headers>;
export declare function getTemplateHeaders(url: TemplateRequestUrl): Promise<Headers>;
export declare function constructRequest(model: string, task: Task, apiSettings: ApiSettings, stream: boolean, body: string, requestOptions?: SingleRequestOptions): Promise<{
    url: string;
    fetchOptions: RequestInit;
}>;
export declare function constructTemplateRequest(templateId: string, task: ServerPromptTemplateTask, apiSettings: ApiSettings, stream: boolean, body: string, requestOptions?: SingleRequestOptions): Promise<{
    url: string;
    fetchOptions: RequestInit;
}>;
export declare function makeRequest(params: {
    model: string;
    task: Task;
    apiSettings: ApiSettings;
    stream: boolean;
    requestOptions?: SingleRequestOptions;
}, body: string): Promise<Response>;
export declare function makeRequest(params: {
    templateId: string;
    task: ServerPromptTemplateTask;
    apiSettings: ApiSettings;
    stream: boolean;
    requestOptions?: SingleRequestOptions;
}, body: string): Promise<Response>;
export declare class WebSocketUrl {
    apiSettings: ApiSettings;
    constructor(apiSettings: ApiSettings);
    toString(): string;
    private get pathname();
}
//# sourceMappingURL=request.d.ts.map