import { generationInputSchema, type GenerationResult as BaseGenerationResult, type GatewayModelId } from "@automate.ax/api-contract/runtime";
import { type Encodable, type ProducingSchema } from "@automate.ax/codec";
import { createAnthropic } from "@ai-sdk/anthropic";
import { createCerebras } from "@ai-sdk/cerebras";
import { createCohere } from "@ai-sdk/cohere";
import { createDeepInfra } from "@ai-sdk/deepinfra";
import { createDeepSeek } from "@ai-sdk/deepseek";
import { createFireworks } from "@ai-sdk/fireworks";
import { createGoogleGenerativeAI } from "@ai-sdk/google";
import { createGroq } from "@ai-sdk/groq";
import { createHuggingFace } from "@ai-sdk/huggingface";
import { createMistral } from "@ai-sdk/mistral";
import { createOpenAI } from "@ai-sdk/openai";
import { createPerplexity } from "@ai-sdk/perplexity";
import { createTogetherAI } from "@ai-sdk/togetherai";
import { createXai } from "@ai-sdk/xai";
import type { StandardSchemaV1 } from "@standard-schema/spec";
import type { DistributedOmit } from "type-fest";
import { type ActionObjectInput } from "../../automation/actions.js";
import type { IntegrationAccountReference } from "../../automation/integrations.js";
import type { Signal } from "../../automation/signal-protocol.js";
declare const AI_ACCOUNT_SERVICE_IDS: readonly ["anthropic", "cerebras", "cohere", "deepinfra", "deepseek", "fireworks", "google-genai", "groq", "huggingface", "mistral", "openai", "openrouter", "perplexity", "togetherai", "vercel-ai-gateway", "xai"];
type AiAccountServiceId = (typeof AI_ACCOUNT_SERVICE_IDS)[number];
type GenerateSchema = ProducingSchema<Encodable>;
interface AiModelIdsByService {
    anthropic: Parameters<ReturnType<typeof createAnthropic>>[0];
    cerebras: Parameters<ReturnType<typeof createCerebras>>[0];
    cohere: Parameters<ReturnType<typeof createCohere>>[0];
    deepinfra: Parameters<ReturnType<typeof createDeepInfra>>[0];
    deepseek: Parameters<ReturnType<typeof createDeepSeek>>[0];
    fireworks: Parameters<ReturnType<typeof createFireworks>>[0];
    "google-genai": Parameters<ReturnType<typeof createGoogleGenerativeAI>>[0];
    groq: Parameters<ReturnType<typeof createGroq>>[0];
    huggingface: Parameters<ReturnType<typeof createHuggingFace>>[0];
    mistral: Parameters<ReturnType<typeof createMistral>>[0];
    openai: Parameters<ReturnType<typeof createOpenAI>>[0];
    openrouter: never;
    perplexity: Parameters<ReturnType<typeof createPerplexity>>[0];
    togetherai: Parameters<ReturnType<typeof createTogetherAI>>[0];
    "vercel-ai-gateway": GatewayModelId;
    xai: Parameters<ReturnType<typeof createXai>>[0];
}
/** Model IDs suggested for one provider, with support for newly released IDs. */
export type GenerateModelId<TServiceId extends AiAccountServiceId = AiAccountServiceId> = AiModelIdsByService[TServiceId] | (string & {});
/** Serializable AI generation result with a schema-dependent output. */
export type GenerateResult<TOutput extends Encodable = string> = Omit<BaseGenerationResult, "output"> & {
    /** Generated text or structured value. */
    output: TOutput;
};
/** Inputs accepted by the built-in AI generation action. */
export type GenerateOptions<TSchema extends GenerateSchema | undefined = undefined, TServiceId extends AiAccountServiceId = AiAccountServiceId> = DistributedOmit<ActionObjectInput<typeof generationInputSchema>, "model"> & {
    /** User-managed provider account. Omit to use the platform AI Gateway. */
    account?: IntegrationAccountReference<TServiceId>;
    /** Provider model ID. Defaults according to the selected account. */
    model?: GenerateModelId<NoInfer<TServiceId>> | Exclude<ActionObjectInput<typeof generationInputSchema>["model"], GatewayModelId | undefined>;
    /** Standard Schema used to constrain and type structured output. */
    schema?: TSchema;
    /** Additional model guidance describing the structured output. */
    schemaDescription?: string;
    /** Provider-facing name for the structured output. */
    schemaName?: string;
};
type PlatformGenerateOptions<TSchema extends GenerateSchema | undefined = undefined> = DistributedOmit<GenerateOptions<TSchema, "vercel-ai-gateway">, "account"> & {
    account?: never;
};
/**
 * Generates text or schema-validated structured data.
 *
 * Omit `account` to use the platform AI Gateway. Explicit provider accounts use
 * their own credentials and accept provider-native model IDs.
 *
 * @param options - Prompt, model, generation settings, and optional schema.
 */
export declare function generate<TSchema extends GenerateSchema>(options: PlatformGenerateOptions<TSchema> & {
    schema: TSchema;
}): Signal<GenerateResult<StandardSchemaV1.InferOutput<TSchema>>>;
/**
 * Generates structured data through an explicit provider account.
 *
 * @param options - Provider account, prompt, model, and output schema.
 */
export declare function generate<TSchema extends GenerateSchema, const TServiceId extends AiAccountServiceId>(options: GenerateOptions<TSchema, TServiceId> & {
    account: IntegrationAccountReference<TServiceId>;
    schema: TSchema;
}): Signal<GenerateResult<StandardSchemaV1.InferOutput<TSchema>>>;
/**
 * Generates structured data when the account may vary at authoring time.
 *
 * @param options - Account, prompt, model, and output schema.
 */
export declare function generate<TSchema extends GenerateSchema>(options: GenerateOptions<TSchema> & {
    schema: TSchema;
}): Signal<GenerateResult<StandardSchemaV1.InferOutput<TSchema>>>;
/**
 * Generates plain text through the platform gateway or an explicit provider.
 *
 * @param options - Prompt, model, and generation settings.
 */
export declare function generate(options: PlatformGenerateOptions): Signal<GenerateResult>;
/**
 * Generates text through an explicit provider account.
 *
 * @param options - Provider account, prompt, and model settings.
 */
export declare function generate<const TServiceId extends AiAccountServiceId>(options: GenerateOptions<undefined, TServiceId> & {
    account: IntegrationAccountReference<TServiceId>;
}): Signal<GenerateResult>;
/**
 * Generates text when the account may vary at authoring time.
 *
 * @param options - Account, prompt, and model settings.
 */
export declare function generate(options: GenerateOptions): Signal<GenerateResult>;
export {};
//# sourceMappingURL=generate.d.ts.map