mcp

type AuthConfig struct {
	Type       string // "none", "basic", "bearer", "api-key"
	Username   string // For basic auth
	Password   string // For basic auth
	Token      string // For bearer auth
	ApiKey     string // For API key auth
	HeaderName string // Custom header name for API key
}

type Client struct {
	// Name is a friendly name for this MCP server connection
	Name string
	// contains filtered or unexported fields
}

type ClientConfig struct {
	// Common fields
	Name string

	// For stdio-based clients
	Command string
	Args    []string
	Env     []string

	// For HTTP-based clients
	URL          string
	Auth         *AuthConfig
	OAuthConfig  *OAuthConfig
	Timeout      int
	UseStreaming bool              // Whether to use streaming HTTP for better performance
	SkipVerify   bool              // Whether to skip TLS certificate verification for HTTPS connections
	Headers      map[string]string // Custom headers to include in HTTP requests

}

type Config struct {
	// Servers is a list of MCP server configurations
	Servers []ServerConfig `yaml:"servers,omitempty"`
}

type FunctionDefinition interface {
	// Schema returns a representation of the function schema
	Schema() any
}

type MCPClient interface {
	// Name returns the name of this client
	Name() string

	// Connect establishes a connection to the MCP server
	Connect(ctx context.Context) error

	// Close closes the connection to the MCP server
	Close() error

	// ListTools lists all available tools from the MCP server
	ListTools(ctx context.Context) ([]Tool, error)

	// CallTool calls a tool on the MCP server and returns the result as a string
	CallTool(ctx context.Context, toolName string, arguments map[string]interface{}) (string, error)
	// contains filtered or unexported methods
}

type MCPStatus struct {
	ServerInfoList []ServerConnectionInfo
	TotalServers   int
	ConnectedCount int
	FailedCount    int
	TotalTools     int
	ClientEnabled  bool
}

type Manager struct {
	// contains filtered or unexported fields
}

type OAuthConfig struct {
	ClientID     string
	ClientSecret string
	TokenURL     string
	AuthURL      string
	Scopes       []string
	RedirectURL  string
}

type RetryConfig struct {
	MaxRetries  int
	BaseDelay   time.Duration
	MaxDelay    time.Duration
	Multiplier  float64
	Description string
}

type SchemaBuilder func(tool *Tool) (FunctionDefinition, error)

type SchemaProperty interface {
	// Property returns a representation of the schema property
	Property() any
}

type ServerConfig struct {
	// Name is a friendly name for this MCP server
	Name string `yaml:"name"`
	// Command is the command to execute for stdio-based MCP servers
	Command string `yaml:"command"`
	// Args are the arguments to pass to the command
	Args []string `yaml:"args,omitempty"`
	// Env are the environment variables to set for the command
	Env map[string]string `yaml:"env,omitempty"`
	// URL is the URL for HTTP-based MCP servers
	URL string `yaml:"url,omitempty"`
	// Auth is the authentication configuration for HTTP-based MCP servers
	Auth *AuthConfig `yaml:"auth,omitempty"`
	// OAuthConfig is the OAuth configuration for HTTP-based MCP servers
	OAuthConfig *OAuthConfig `yaml:"oauth,omitempty"`
	// Timeout is the timeout in seconds for HTTP requests
	Timeout int `yaml:"timeout,omitempty"`
	// UseStreaming enables streaming HTTP for better performance
	UseStreaming bool `yaml:"use_streaming,omitempty"`
	// SkipVerify skips TLS certificate verification for HTTPS connections
	SkipVerify bool `yaml:"skip_verify,omitempty"`
}

type ServerConnectionInfo struct {
	Name           string
	Command        string
	IsLegacy       bool
	IsConnected    bool
	AvailableTools []Tool
}

type Tool struct {
	Name        string `json:"name"`
	Description string `json:"description,omitempty"`
	Server      string `json:"server,omitempty"`

	InputSchema *gollm.Schema `json:"inputSchema,omitempty"`
}