import { Client } from '@modelcontextprotocol/sdk/client/index.js';
import { Genkit, MultipartToolAction, ToolAction } from 'genkit';

/**
 * 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.
 */

/**
 * Lookup all tools available in the server and register each as a Genkit tool.
 */
declare function registerAllTools(ai: Genkit, client: Client, params: {
    name: string;
    serverName: string;
    rawToolResponses?: boolean;
    multipart?: boolean;
}): Promise<void>;
/**
 * Lookup all tools available in the server and fetches as a Genkit dynamic tool.
 */
declare function fetchDynamicTools<Multipart extends boolean = false>(ai: Genkit, client: Client, params: {
    name: string;
    serverName: string;
    rawToolResponses?: boolean;
    multipart?: Multipart;
}): Promise<(Multipart extends true ? MultipartToolAction : ToolAction)[]>;

export { fetchDynamicTools, registerAllTools };