import { LanguageModelV3, ProviderV3 } from "@ai-sdk/provider";
import { FetchFunction } from "@ai-sdk/provider-utils";

//#region src/databricks-provider.d.ts
type DatabricksLanguageModelConfig = {
  provider: string;
  headers: () => Record<string, string | undefined>;
  url: (options: {
    path: string;
  }) => string;
  fetch?: FetchFunction;
  /**
   * When true, marks all tool calls as dynamic and provider-executed.
   * This indicates that tool execution is handled remotely by Databricks agents.
   * Defaults to true.
   */
  useRemoteToolCalling?: boolean;
};
interface DatabricksProvider extends ProviderV3 {
  /** Agents */
  chatAgent(modelId: string): LanguageModelV3;
  responses(modelId: string): LanguageModelV3;
  /** Foundation Models */
  chatCompletions(modelId: string): LanguageModelV3;
  /** Required overrides from ProviderV3 */
  textEmbeddingModel(modelId: string): never;
  languageModel(modelId: string): never;
}
interface DatabricksProviderSettings {
  /** Base URL for the Databricks API calls. */
  baseURL: string;
  /** Custom headers to include in the requests. */
  headers?: Record<string, string>;
  /** Provider name. Overrides the `databricks` default name for 3rd party providers. */
  provider?: string;
  /**
   * Custom fetch implementation. You can use it as a middleware to intercept requests,
   * or to provide a custom fetch implementation for e.g. testing.
   * */
  fetch?: FetchFunction;
  /**
   * Optional function to format the URL
   */
  formatUrl?: (options: {
    baseUrl?: string;
    path: string;
  }) => string;
  /**
   * When true, marks all tool calls as dynamic and provider-executed.
   * This indicates that tool execution is handled remotely by Databricks agents.
   * Defaults to true.
   */
  useRemoteToolCalling?: boolean;
}
declare const createDatabricksProvider: (settings: DatabricksProviderSettings) => DatabricksProvider; //#endregion

//# sourceMappingURL=databricks-provider.d.ts.map
export { DatabricksLanguageModelConfig, DatabricksProvider, DatabricksProviderSettings, createDatabricksProvider };
//# sourceMappingURL=index.d.cts.map