This commit is contained in:
duanfuxiang 2025-06-14 13:44:03 +08:00
parent 5c24993ab9
commit 93a3c0969d
5 changed files with 115 additions and 19 deletions

View File

@ -16,7 +16,7 @@ export default function MarkdownFetchUrlsContentBlock({
finish: boolean finish: boolean
}) { }) {
const containerRef = useRef<HTMLDivElement>(null) const containerRef = useRef<HTMLDivElement>(null)
const [isOpen, setIsOpen] = useState(true) const [isOpen, setIsOpen] = useState(false)
React.useEffect(() => { React.useEffect(() => {
if (finish && applyStatus === ApplyStatus.Idle) { if (finish && applyStatus === ApplyStatus.Idle) {

View File

@ -13,7 +13,7 @@ export default function MarkdownReasoningBlock({
}>) { }>) {
const { isDarkMode } = useDarkModeContext() const { isDarkMode } = useDarkModeContext()
const containerRef = useRef<HTMLDivElement>(null) const containerRef = useRef<HTMLDivElement>(null)
const [isOpen, setIsOpen] = useState(true) const [isOpen, setIsOpen] = useState(false)
useEffect(() => { useEffect(() => {
if (containerRef.current) { if (containerRef.current) {

View File

@ -1,4 +1,4 @@
import { ChevronDown, ChevronRight, CheckCheck } from 'lucide-react' import { CheckCheck, ChevronDown, ChevronRight } from 'lucide-react'
import React, { PropsWithChildren, useEffect, useRef, useState } from 'react' import React, { PropsWithChildren, useEffect, useRef, useState } from 'react'
import { useDarkModeContext } from "../../../contexts/DarkModeContext" import { useDarkModeContext } from "../../../contexts/DarkModeContext"
@ -11,7 +11,8 @@ const processContent = (content: string): { serverName: string; processedContent
const firstLine = lines[0]; const firstLine = lines[0];
// 提取 serverName // 提取 serverName
const serverNameMatch = firstLine.match(/\[use_mcp_tool for '([^']+)'\]/); const serverNameRegex = /\[use_mcp_tool for '([^']+)'\]/;
const serverNameMatch = serverNameRegex.exec(firstLine);
const serverName = serverNameMatch ? serverNameMatch[1] : ''; const serverName = serverNameMatch ? serverNameMatch[1] : '';
// 移除第一行并重新组合内容 // 移除第一行并重新组合内容
@ -27,7 +28,7 @@ export default function MarkdownToolResult({
}>) { }>) {
const { isDarkMode } = useDarkModeContext() const { isDarkMode } = useDarkModeContext()
const containerRef = useRef<HTMLDivElement>(null) const containerRef = useRef<HTMLDivElement>(null)
const [isOpen, setIsOpen] = useState(true) const [isOpen, setIsOpen] = useState(false)
const { serverName, processedContent } = React.useMemo(() => processContent(content), [content]); const { serverName, processedContent } = React.useMemo(() => processContent(content), [content]);
@ -40,12 +41,12 @@ export default function MarkdownToolResult({
return ( return (
processedContent && ( processedContent && (
<div <div
className={`infio-chat-code-block has-filename infio-reasoning-block`} className={`infio-chat-code-block-response has-filename infio-reasoning-block`}
> >
<div className={'infio-chat-code-block-header'}> <div className={'infio-chat-code-block-response-header'}>
<div className={'infio-chat-code-block-header-filename'}> <div className={'infio-chat-code-block-response-header-filename'}>
<CheckCheck size={10} className="infio-chat-code-block-header-icon" /> <CheckCheck size={10} className="infio-chat-code-block-response-header-icon" />
response from tool {t('response_from_tool')}
<span className="infio-mcp-tool-server-name">{serverName}</span> <span className="infio-mcp-tool-server-name">{serverName}</span>
</div> </div>
<button <button
@ -71,6 +72,110 @@ export default function MarkdownToolResult({
</div> </div>
<style> <style>
{` {`
.infio-chat-code-block-response {
position: relative;
border: 1px solid var(--background-modifier-border);
border-radius: var(--radius-s);
margin-top: -10px;
margin-bottom: 12px;
}
.infio-chat-code-block-response.infio-reasoning-block {
max-height: 200px;
overflow: hidden;
}
.infio-chat-code-block-response code {
padding: 0;
}
.infio-chat-code-block-response-header {
display: none;
justify-content: space-between;
align-items: center;
font-size: var(--font-smallest);
padding: 0 var(--size-4-1) 0 0;
}
.infio-chat-code-block-response:hover .infio-chat-code-block-response-header {
position: absolute;
top: calc(var(--size-4-3) * -1);
right: var(--size-4-1);
display: flex;
}
.infio-chat-code-block-response.has-filename .infio-chat-code-block-response-header {
display: flex;
border-bottom: 1px solid var(--background-modifier-border);
background-color: var(--background-secondary);
border-radius: var(--radius-s) var(--radius-s) 0 0;
height: calc(var(--size-4-8) - var(--size-4-1));
}
.infio-chat-code-block-response.has-filename:hover .infio-chat-code-block-response-header {
position: inherit;
top: 0;
left: 0;
}
.infio-chat-code-block-response-header-filename {
padding-left: var(--size-4-2);
font-size: var(--font-medium);
overflow: hidden;
text-overflow: ellipsis;
white-space: nowrap;
}
.infio-chat-code-block-response-header-icon {
margin-right: 6px;
}
.infio-chat-code-block-response-header-button {
display: flex;
gap: var(--size-4-1);
right: 0;
font-family: var(--font-interface);
padding: 0;
font-size: var(--font-small);
font-weight: var(--font-medium);
color: var(--text-muted);
}
.infio-chat-code-block-response.has-filename .infio-chat-code-block-response-header-button {
gap: 0;
overflow: hidden;
min-width: fit-content;
height: 100%;
}
.infio-chat-code-block-response.has-filename
.infio-chat-code-block-response-header-button
button {
box-shadow: none;
border: 0;
padding: 0 var(--size-4-2);
border-radius: 0;
background-color: var(--background-primary);
font-size: var(--font-medium);
height: 100%;
cursor: pointer;
&:hover {
background-color: var(--background-modifier-hover);
}
}
.infio-chat-code-block-response-header-button button {
display: flex;
gap: var(--size-4-1);
font-size: var(--font-ui-smaller);
}
.infio-chat-code-block-response-content {
margin: 0;
}
.infio-mcp-tool-server-name { .infio-mcp-tool-server-name {
color: var(--text-accent); color: var(--text-accent);
border-radius: 4px; border-radius: 4px;

View File

@ -53,7 +53,6 @@ export class RAGEngine {
throw new Error('Embedding model is not set') throw new Error('Embedding model is not set')
} }
await this.initializeDimension() await this.initializeDimension()
console.log("updateVaultIndex")
await this.vectorManager.updateVaultIndex( await this.vectorManager.updateVaultIndex(
this.embeddingModel, this.embeddingModel,
@ -70,7 +69,6 @@ export class RAGEngine {
}) })
}, },
) )
console.log("updateVaultIndex done")
this.initialized = true this.initialized = true
} }

View File

@ -85,7 +85,6 @@ export class VectorManager {
}, },
updateProgress?: (indexProgress: IndexProgress) => void, updateProgress?: (indexProgress: IndexProgress) => void,
): Promise<void> { ): Promise<void> {
console.log("updateVaultIndex start")
let filesToIndex: TFile[] let filesToIndex: TFile[]
if (options.reindexAll) { if (options.reindexAll) {
filesToIndex = await this.getFilesToIndex({ filesToIndex = await this.getFilesToIndex({
@ -96,7 +95,6 @@ export class VectorManager {
}) })
await this.repository.clearAllVectors(embeddingModel) await this.repository.clearAllVectors(embeddingModel)
} else { } else {
console.log("updateVaultIndex cleanVectorsForDeletedFiles")
await this.cleanVectorsForDeletedFiles(embeddingModel) await this.cleanVectorsForDeletedFiles(embeddingModel)
filesToIndex = await this.getFilesToIndex({ filesToIndex = await this.getFilesToIndex({
embeddingModel: embeddingModel, embeddingModel: embeddingModel,
@ -484,9 +482,7 @@ export class VectorManager {
private async cleanVectorsForDeletedFiles( private async cleanVectorsForDeletedFiles(
embeddingModel: EmbeddingModel, embeddingModel: EmbeddingModel,
) { ) {
console.log("cleanVectorsForDeletedFiles start")
const indexedFilePaths = await this.repository.getAllIndexedFilePaths(embeddingModel) const indexedFilePaths = await this.repository.getAllIndexedFilePaths(embeddingModel)
console.log("indexedFilePaths: ", indexedFilePaths)
const needToDelete = indexedFilePaths.filter(filePath => !this.app.vault.getAbstractFileByPath(filePath)) const needToDelete = indexedFilePaths.filter(filePath => !this.app.vault.getAbstractFileByPath(filePath))
if (needToDelete.length > 0) { if (needToDelete.length > 0) {
await this.repository.deleteVectorsForMultipleFiles( await this.repository.deleteVectorsForMultipleFiles(
@ -494,7 +490,6 @@ export class VectorManager {
embeddingModel, embeddingModel,
) )
} }
console.log("cleanVectorsForDeletedFiles done")
} }
private async getFilesToIndex({ private async getFilesToIndex({
@ -508,7 +503,6 @@ export class VectorManager {
includePatterns: string[] includePatterns: string[]
reindexAll?: boolean reindexAll?: boolean
}): Promise<TFile[]> { }): Promise<TFile[]> {
console.log("getFilesToIndex")
let filesToIndex = this.app.vault.getMarkdownFiles() let filesToIndex = this.app.vault.getMarkdownFiles()
filesToIndex = filesToIndex.filter((file) => { filesToIndex = filesToIndex.filter((file) => {
@ -524,7 +518,6 @@ export class VectorManager {
if (reindexAll) { if (reindexAll) {
return filesToIndex return filesToIndex
} }
console.log("filesToIndex: ", filesToIndex)
// Check for updated or new files // Check for updated or new files
filesToIndex = await Promise.all( filesToIndex = await Promise.all(
filesToIndex.map(async (file) => { filesToIndex.map(async (file) => {