/**
 * This file is part of the NocoBase (R) project.
 * Copyright (c) 2020-2024 NocoBase Co., Ltd.
 * Authors: NocoBase Team.
 *
 * This project is dual-licensed under AGPL-3.0 and NocoBase Commercial License.
 * For more information, please refer to: https://www.nocobase.com/agreement.
 */
import type { MultiServerMCPClient } from '@langchain/mcp-adapters';
import type { Context } from '@nocobase/actions';
import type { DynamicToolsProvider, Permission } from '../tools-manager/types';
export interface MCPManager extends MCPRegistration {
    init(): Promise<void>;
    getMCP(name: string): Promise<MCPEntry>;
    listMCP(filter: MCPFilter): Promise<MCPEntry[]>;
    testConnection(options: MCPOptions, ctx?: Context): Promise<MCPTestResult>;
    rebuildClient(): Promise<void>;
    getClient(): MultiServerMCPClient | null;
    getMCPToolsProvider(): DynamicToolsProvider;
    listMCPTools(ctx?: Context): Promise<Record<string, MCPToolEntry[]>>;
    updateMCPToolPermission(toolName: string, permission: Permission): Promise<void>;
    clearUserContextCache(): Promise<void>;
}
export interface MCPRegistration {
    registerMCP(registration: {
        [key: string | symbol]: MCPOptions;
    }): Promise<void>;
}
export type MCPOptions = {
    transport: MCPTransport;
    command?: string;
    args?: string[];
    env?: Record<string, string>;
    url?: string;
    headers?: Record<string, string>;
    restart?: Record<string, any>;
    useUserContext?: boolean;
};
export type MCPEntry = MCPOptions & {
    name: string;
    enabled: boolean;
};
export type MCPFilter = {
    name?: string;
    enabled?: boolean;
    transport?: MCPTransport;
    useUserContext?: boolean;
};
export type MCPTransport = 'stdio' | 'sse' | 'http';
export type MCPTestResult = {
    success: boolean;
    message?: string;
    error?: string;
    details?: string;
    toolsCount?: number;
    tools?: string[];
    toolsTruncated?: boolean;
};
export type MCPToolEntry = {
    name: string;
    title: string;
    description?: string;
    serverName: string;
    permission: Permission;
};