import * as t from "io-ts";
import type { TypeOf } from "io-ts";
import type { ModelApi, ModelRequestOptions } from "@typeDefs";
import { FnTemplate } from "../../utils/Template";
/**
 * @category Requests
 * @category Cohere Generate
 */
export interface CohereGenerateOptions extends ModelRequestOptions {
    num_generations?: number;
    stream?: boolean;
    max_tokens?: number;
    truncate?: "NONE" | "START" | "END";
    temperature?: number;
    seed?: number;
    preset?: string;
    end_sequences?: string[];
    stop_sequences?: string[];
    k?: number;
    p?: number;
    frequency_penalty?: number;
    presence_penalty?: number;
    return_likelihoods?: "GENERATION" | "ALL" | "NONE";
    logit_bias?: {
        [token_id: number]: number;
    };
}
/**
 * @category Templates
 * @category Cohere Generate
 */
export declare const CohereGenerateTemplate: FnTemplate<CohereGenerateOptions>;
declare const CohereGenerateResponseCodec: t.IntersectionC<[t.TypeC<{
    id: t.StringC;
    prompt: t.StringC;
    generations: t.ArrayC<t.TypeC<{
        id: t.StringC;
        text: t.StringC;
        finish_reason: t.StringC;
    }>>;
}>, t.PartialC<{
    meta: t.TypeC<{
        api_version: t.TypeC<{
            version: t.StringC;
        }>;
        billed_units: t.TypeC<{
            input_tokens: t.NumberC;
            output_tokens: t.NumberC;
        }>;
    }>;
}>]>;
/**
 * @category Responses
 * @category Cohere Generate
 */
export interface CohereGenerateResponse extends TypeOf<typeof CohereGenerateResponseCodec> {
}
export declare function isCohereGenerateResponse(response: unknown): response is CohereGenerateResponse;
export interface CohereGenerateApi extends ModelApi<CohereGenerateOptions, CohereGenerateResponse> {
}
/**
 *
 * ## Reference
 * [Cohere Generate](https://docs.cohere.com/reference/generate)
 *
 * ## Providers using this API
 * - {@link createCohereModelProvider | Cohere}
 * - {@link createAwsBedrockModelProvider | AWS Bedrock}
 *
 * @category APIs
 * @category Cohere Generate
 * @category Provider: AWS Bedrock
 * @category Provider: Cohere
 *
 */
export declare const CohereGenerateApi: CohereGenerateApi;
export {};