update:1.更新mcp插件功能,目前只支持remote调用
This commit is contained in:
@@ -66,6 +66,7 @@ class ChapterGenerateRequest(BaseModel):
|
||||
ge=500, # 最小500字
|
||||
le=10000 # 最大10000字
|
||||
)
|
||||
enable_mcp: bool = Field(True, description="是否启用MCP工具增强(搜索参考资料)")
|
||||
|
||||
|
||||
class BatchGenerateRequest(BaseModel):
|
||||
@@ -80,6 +81,7 @@ class BatchGenerateRequest(BaseModel):
|
||||
le=10000
|
||||
)
|
||||
enable_analysis: bool = Field(False, description="是否启用同步分析")
|
||||
enable_mcp: bool = Field(True, description="是否启用MCP工具增强(搜索参考资料)")
|
||||
max_retries: int = Field(3, description="每个章节的最大重试次数", ge=0, le=5)
|
||||
|
||||
|
||||
|
||||
@@ -63,6 +63,7 @@ class CharacterGenerateRequest(BaseModel):
|
||||
role_type: Optional[str] = Field(None, description="角色类型")
|
||||
background: Optional[str] = Field(None, description="角色背景")
|
||||
requirements: Optional[str] = Field(None, description="特殊要求")
|
||||
enable_mcp: bool = Field(True, description="是否启用MCP工具增强(搜索人物原型参考)")
|
||||
|
||||
|
||||
class CharacterListResponse(BaseModel):
|
||||
|
||||
@@ -0,0 +1,105 @@
|
||||
"""MCP插件Pydantic模式"""
|
||||
from pydantic import BaseModel, Field
|
||||
from typing import Optional, Dict, Any, List
|
||||
from datetime import datetime
|
||||
|
||||
|
||||
class MCPToolSchema(BaseModel):
|
||||
"""MCP工具定义"""
|
||||
name: str
|
||||
description: Optional[str] = None
|
||||
inputSchema: Optional[Dict[str, Any]] = None
|
||||
category: Optional[str] = None
|
||||
|
||||
|
||||
class MCPPluginBase(BaseModel):
|
||||
"""插件基础模式"""
|
||||
plugin_name: str = Field(..., description="插件唯一标识")
|
||||
display_name: Optional[str] = Field(None, description="显示名称")
|
||||
description: Optional[str] = Field(None, description="插件描述")
|
||||
plugin_type: str = Field(default="http", description="插件类型:http/stdio")
|
||||
category: str = Field(default="general", description="分类")
|
||||
sort_order: int = Field(default=0, description="排序顺序")
|
||||
|
||||
|
||||
class MCPPluginCreate(MCPPluginBase):
|
||||
"""创建插件"""
|
||||
server_url: Optional[str] = Field(None, description="服务器URL(HTTP类型)")
|
||||
command: Optional[str] = Field(None, description="启动命令(stdio类型)")
|
||||
args: Optional[List[str]] = Field(None, description="命令参数")
|
||||
env: Optional[Dict[str, str]] = Field(None, description="环境变量")
|
||||
headers: Optional[Dict[str, str]] = Field(None, description="HTTP请求头")
|
||||
config: Optional[Dict[str, Any]] = Field(None, description="插件特定配置")
|
||||
enabled: bool = Field(default=True, description="是否启用")
|
||||
|
||||
|
||||
class MCPPluginSimpleCreate(BaseModel):
|
||||
"""简化的插件创建(通过标准MCP配置JSON)"""
|
||||
config_json: str = Field(..., description="标准MCP配置JSON字符串")
|
||||
enabled: bool = Field(default=True, description="是否启用")
|
||||
category: str = Field(default="general", description="插件分类")
|
||||
|
||||
|
||||
class MCPPluginUpdate(BaseModel):
|
||||
"""更新插件"""
|
||||
display_name: Optional[str] = None
|
||||
description: Optional[str] = None
|
||||
server_url: Optional[str] = None
|
||||
command: Optional[str] = None
|
||||
args: Optional[List[str]] = None
|
||||
env: Optional[Dict[str, str]] = None
|
||||
headers: Optional[Dict[str, str]] = None
|
||||
config: Optional[Dict[str, Any]] = None
|
||||
enabled: Optional[bool] = None
|
||||
category: Optional[str] = None
|
||||
sort_order: Optional[int] = None
|
||||
|
||||
|
||||
class MCPPluginResponse(BaseModel):
|
||||
"""插件响应 - 优化后只返回必要字段"""
|
||||
id: str
|
||||
plugin_name: str
|
||||
display_name: str
|
||||
description: Optional[str] = None
|
||||
plugin_type: str
|
||||
category: str
|
||||
|
||||
# HTTP类型字段
|
||||
server_url: Optional[str] = None
|
||||
headers: Optional[Dict[str, str]] = None
|
||||
|
||||
# Stdio类型字段
|
||||
command: Optional[str] = None
|
||||
args: Optional[List[str]] = None
|
||||
env: Optional[Dict[str, str]] = None
|
||||
|
||||
# 状态字段
|
||||
enabled: bool
|
||||
status: str
|
||||
last_error: Optional[str] = None
|
||||
last_test_at: Optional[datetime] = None
|
||||
|
||||
# 时间戳
|
||||
created_at: datetime
|
||||
|
||||
class Config:
|
||||
from_attributes = True
|
||||
|
||||
|
||||
class MCPToolCall(BaseModel):
|
||||
"""工具调用请求"""
|
||||
plugin_id: str = Field(..., description="插件ID")
|
||||
tool_name: str = Field(..., description="工具名称")
|
||||
arguments: Dict[str, Any] = Field(default_factory=dict, description="工具参数")
|
||||
|
||||
|
||||
class MCPTestResult(BaseModel):
|
||||
"""测试结果"""
|
||||
success: bool
|
||||
message: str
|
||||
response_time_ms: Optional[float] = None
|
||||
tools_count: Optional[int] = None
|
||||
tools: Optional[List[MCPToolSchema]] = None
|
||||
error: Optional[str] = None
|
||||
error_type: Optional[str] = None
|
||||
suggestions: Optional[List[str]] = None
|
||||
@@ -61,6 +61,7 @@ class OutlineGenerateRequest(BaseModel):
|
||||
story_direction: Optional[str] = Field(None, description="故事发展方向提示(续写时使用)")
|
||||
plot_stage: str = Field("development", description="情节阶段: development(发展), climax(高潮), ending(结局)")
|
||||
keep_existing: bool = Field(False, description="是否保留现有大纲(续写时)")
|
||||
enable_mcp: bool = Field(True, description="是否启用MCP工具增强(搜索情节设计参考)")
|
||||
|
||||
|
||||
class ChapterOutlineGenerateRequest(BaseModel):
|
||||
|
||||
Reference in New Issue
Block a user