update , add mcp server stdio and sse

This commit is contained in:
duanfuxiang
2025-06-02 20:38:40 +08:00
parent 8ca5216b71
commit b1315aa6b1
30 changed files with 2639 additions and 955 deletions

View File

@@ -0,0 +1,88 @@
import { ChevronDown, ChevronRight, CheckCheck } from 'lucide-react'
import React, { PropsWithChildren, useEffect, useRef, useState } from 'react'
import { useDarkModeContext } from "../../../contexts/DarkModeContext"
import { t } from '../../../lang/helpers'
import { MemoizedSyntaxHighlighterWrapper } from "./SyntaxHighlighterWrapper"
const processContent = (content: string): { serverName: string; processedContent: string } => {
const lines = content.split('\n');
const firstLine = lines[0];
// 提取 serverName
const serverNameMatch = firstLine.match(/\[use_mcp_tool for '([^']+)'\]/);
const serverName = serverNameMatch ? serverNameMatch[1] : '';
// 移除第一行并重新组合内容
const processedContent = lines.slice(1).join('\n');
return { serverName, processedContent };
};
export default function MarkdownToolResult({
content,
}: PropsWithChildren<{
content: string
}>) {
const { isDarkMode } = useDarkModeContext()
const containerRef = useRef<HTMLDivElement>(null)
const [isOpen, setIsOpen] = useState(true)
const { serverName, processedContent } = React.useMemo(() => processContent(content), [content]);
useEffect(() => {
if (containerRef.current) {
containerRef.current.scrollTop = containerRef.current.scrollHeight
}
}, [processedContent])
return (
processedContent && (
<div
className={`infio-chat-code-block has-filename infio-reasoning-block`}
>
<div className={'infio-chat-code-block-header'}>
<div className={'infio-chat-code-block-header-filename'}>
<CheckCheck size={10} className="infio-chat-code-block-header-icon" />
response from tool
<span className="infio-mcp-tool-server-name">{serverName}</span>
</div>
<button
className="clickable-icon infio-chat-list-dropdown"
onClick={() => setIsOpen(!isOpen)}
>
{isOpen ? <ChevronDown size={16} /> : <ChevronRight size={16} />}
</button>
</div>
<div
ref={containerRef}
className="infio-reasoning-content-wrapper"
>
<MemoizedSyntaxHighlighterWrapper
isDarkMode={isDarkMode}
language="markdown"
hasFilename={true}
wrapLines={true}
isOpen={isOpen}
>
{processedContent}
</MemoizedSyntaxHighlighterWrapper>
</div>
<style>
{`
.infio-mcp-tool-server-name {
color: var(--text-accent);
border-radius: 4px;
margin-left: 4px;
margin-right: 4px;
font-weight: bold;
font-size: 13px;
display: inline-block;
}
`}
</style>
</div>
)
)
}

View File

@@ -0,0 +1,109 @@
import { Server } from 'lucide-react'
import React from 'react'
import { useSettings } from "../../../contexts/SettingsContext"
import { t } from '../../../lang/helpers'
import { ApplyStatus, SearchWebToolArgs } from "../../../types/apply"
export default function UseMcpToolBlock({
applyStatus,
onApply,
serverName,
toolName,
parameters,
finish
}: {
applyStatus: ApplyStatus
onApply: (args: SearchWebToolArgs) => void
serverName: string,
toolName: string,
parameters: Record<string, unknown>,
finish: boolean
}) {
const { settings } = useSettings()
React.useEffect(() => {
if (finish && applyStatus === ApplyStatus.Idle) {
onApply({
type: 'use_mcp_tool',
server_name: serverName,
tool_name: toolName,
parameters: parameters,
})
}
}, [finish])
return (
<div
className={`infio-chat-code-block has-filename`
}
>
<div className={'infio-chat-code-block-header'}>
<div className={'infio-chat-code-block-header-filename'}>
<Server size={14} className="infio-chat-code-block-header-icon" />
use mcp tool from
<span className="infio-mcp-tool-server-name">{serverName}</span>
</div>
</div>
<div
className="infio-reasoning-content-wrapper"
>
<div className="infio-mcp-tool-row">
<div className="infio-mcp-tool-row-header">
<div className="infio-mcp-tool-name-section">
<span className="infio-mcp-tool-name">{toolName}</span>
</div>
</div>
: <div className="infio-mcp-tool-parameters">
<pre className="infio-json-pre"><code>{JSON.stringify(parameters, null, 2)}</code></pre>
</div>
</div>
</div>
<style>{`
.infio-mcp-tool-row {
padding: 12px;
border-bottom: 1px solid var(--background-modifier-border);
background-color: var(--background-primary);
border-radius: var(--radius-s);
}
.infio-mcp-tool-row-header {
display: flex;
align-items: center;
gap: 8px;
margin-bottom: 8px;
}
.infio-mcp-tool-name {
font-weight: 600;
color: var(--text-normal);
font-size: 14px;
}
.infio-mcp-tool-server-name {
color: var(--text-accent);
border-radius: 4px;
margin-left: 4px;
margin-right: 4px;
font-weight: bold;
font-size: 13px;
display: inline-block;
}
.infio-mcp-tool-parameters {
font-size: 14px;
color: var(--text-muted);
line-height: 1.4;
margin: 8px 0 0 0;
}
.infio-json-pre {
background: #282c34;
color: #d4d4d4;
border-radius: 4px;
padding: 8px;
font-size: 13px;
overflow-x: auto;
margin: 0;
}
`}</style>
</div>
)
}