Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2026-01-10 15:49:29 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

refactor(mcp): use includes http to detect streamable http type mcp server (#9865)

Browse Source 
refactor(mcp): 简化 McpServerTypeSchema 的类型校验逻辑

将联合类型替换为字符串校验并优化 http 相关类型的转换
This commit is contained in:
Phantom 2025-09-04 11:41:26 +08:00 committed by GitHub
parent 0a36869b3c
commit 9a92372c3e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/renderer/src/types/mcp.ts
View File

@ -16,22 +16,18 @@ export type MCPConfigSample = z.infer<typeof MCPConfigSampleSchema>
* 允许 inMemory 作为合法字段，需要额外校验 name 是否 builtin * 允许 inMemory 作为合法字段，需要额外校验 name 是否 builtin
*/ */
export const McpServerTypeSchema = z export const McpServerTypeSchema = z
.union([ .string()
z.literal('stdio'),
z.literal('sse'),
z.literal('streamableHttp'),
z.literal('http'),
z.literal('streamable_http'),
z.literal('inMemory')
])
.transform((type) => { .transform((type) => {
if (type === 'streamable_http' || type === 'http') { if (type.includes('http')) {
return 'streamableHttp' return 'streamableHttp'
} else { } else {
return type return type
} }
}) })
.default('stdio') // 大多数情况下默认使用 stdio .pipe(
z.union([z.literal('stdio'), z.literal('sse'), z.literal('streamableHttp'), z.literal('inMemory')]).default('stdio') // 大多数情况下默认使用 stdio
)
/** /**
* 定义单个 MCP 服务器的配置。 * 定义单个 MCP 服务器的配置。
* FIXME: 为了兼容性，暂时允许用户编辑任意字段，这可能会导致问题。 * FIXME: 为了兼容性，暂时允许用户编辑任意字段，这可能会导致问题。