004577c9cd* studio: import MCP servers from a config file * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * import config' on the add-server form * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * fix: defensively handle MCP config imports * fix: address MCP import review follow-ups * fix: preserve apostrophes in Windows MCP commands * fix: preserve apostrophe-wrapped Windows MCP args * fix: align Windows MCP parsing with list2cmdline * fix: preserve explicit MCP remote transport intent * fix: trim MCP remote URLs before transport checks --------- Co-authored-by: Roland Tannous <rolandtannous@gravityq.ai> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Roland Tannous <115670425+rolandtannous@users.noreply.github.com> Co-authored-by: Lee Jackson <130007945+Imagineer99@users.noreply.github.com> Co-authored-by: imagineer99 <samleejackson0@gmail.com>
57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
# SPDX-License-Identifier: AGPL-3.0-only
|
|
# Copyright 2026-present the Unsloth AI Inc. team. All rights reserved. See /studio/LICENSE.AGPL-3.0
|
|
|
|
from typing import Optional
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class McpServerCreate(BaseModel):
|
|
display_name: str
|
|
url: str
|
|
headers: Optional[dict[str, str]] = None
|
|
is_enabled: bool = True
|
|
use_oauth: bool = False
|
|
|
|
|
|
class McpServerUpdate(BaseModel):
|
|
display_name: Optional[str] = None
|
|
url: Optional[str] = None
|
|
# Absent in request body = leave as-is; null = drop all headers; dict = set.
|
|
headers: Optional[dict[str, str]] = None
|
|
is_enabled: Optional[bool] = None
|
|
use_oauth: Optional[bool] = None
|
|
|
|
|
|
class McpServerResponse(BaseModel):
|
|
id: str
|
|
display_name: str
|
|
url: str
|
|
headers: dict[str, str] = Field(default_factory = dict)
|
|
is_enabled: bool = True
|
|
use_oauth: bool = False
|
|
created_at: str
|
|
updated_at: str
|
|
|
|
|
|
class McpServerTestRequest(BaseModel):
|
|
url: str
|
|
headers: Optional[dict[str, str]] = None
|
|
use_oauth: bool = False
|
|
|
|
|
|
class McpServerProbeResult(BaseModel):
|
|
ok: bool
|
|
tool_count: int = 0
|
|
error: Optional[str] = None
|
|
|
|
|
|
class McpServerImportRequest(BaseModel):
|
|
# A standard mcpServers JSON config (Claude Desktop / Cursor / Cline / VS Code).
|
|
config: dict
|
|
|
|
|
|
class McpServerImportResult(BaseModel):
|
|
created: list[McpServerResponse] = Field(default_factory = list)
|
|
skipped: list[str] = Field(default_factory = list) # display names skipped as duplicates
|
|
errors: list[str] = Field(default_factory = list)
|