Compare commits

...

6 Commits

Author SHA1 Message Date
7a2f6dd30a refactor: 删除 MCP 页面并更新上传 API 文档
- 移除 MCP.vue 页面
- 更新上传 API 需求文档

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-03-08 20:35:02 +08:00
bf8c4de07d chore: 添加服务端资源和配置
- 添加 server/resource/ 目录
- 更新 tsconfig.json 配置

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-03-08 20:34:47 +08:00
3a1d9ed676 feat: 添加算法目录和知识库 API 文档
- 新增 algorithm/ 目录
- 添加知识库 API 需求文档
- 添加相关截图

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-03-08 20:34:41 +08:00
11c9ff2428 feat: 新增 Skill 和 Team 页面
- 添加 Skill 技能管理页面
- 添加 Team 团队管理页面
- 更新侧边栏导航和路由配置

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-03-08 20:34:36 +08:00
fef1c9f302 feat: 完善 Knowledge 前端页面
- 增强知识库列表和详情展示
- 添加文档上传和管理功能
- 优化创建和编辑流程

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-03-08 20:34:29 +08:00
11de8c916a feat: 完善知识库后端 API
- 添加 KnowledgeHandler 处理知识库请求
- 注册知识库 CRUD 路由
- 添加文档上传、删除、解析、预览接口
- 更新数据库模型和迁移

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-03-08 20:34:15 +08:00
28 changed files with 21603 additions and 739 deletions

112
algorithm/README.md Normal file
View File

@@ -0,0 +1,112 @@
# Algorithm Service
Python 算法服务提供文档解析、Embedding、LLM 调用等功能。
## 环境要求
- Python 3.9+
- FastAPI
- Uvicorn
## 安装依赖
```bash
pip install -r requirements.txt
```
## 运行服务
```bash
# 开发模式
uvicorn main:app --reload --port 8081
# 生产模式
uvicorn main:app --host 0.0.0.0 --port 8081
```
## 接口列表
### 1. 文档解析
**请求**
```
POST /parse
Content-Type: application/json
```
| 参数 | 类型 | 必填 | 说明 |
|------|------|------|------|
| file_url | String | 是 | 文件 URL |
| engine | String | 是 | 解析引擎markitdown / docling |
| docling_url | String | 否 | Docling 服务 URL |
**响应**
```json
{
"success": true,
"content": "解析后的文本内容...",
"chunks": ["chunk1", "chunk2"],
"total_pages": 10,
"metadata": {
"filename": "document.pdf",
"file_size": 1234567
}
}
```
### 2. 生成 Embedding
**请求**
```
POST /embedding
Content-Type: application/json
```
| 参数 | 类型 | 必填 | 说明 |
|------|------|------|------|
| input | String/Array | 是 | 要 embedding 的文本 |
| model | String | 是 | 模型名称 |
**响应**
```json
{
"success": true,
"embeddings": [[0.1, 0.2, ...], [0.3, 0.4, ...]],
"model": "text-embedding-3-small"
}
```
### 3. LLM 对话
**请求**
```
POST /chat
Content-Type: application/json
```
| 参数 | 类型 | 必填 | 说明 |
|------|------|------|------|
| messages | Array | 是 | 消息列表 |
| model | String | 是 | 模型名称 |
| temperature | Float | 否 | 温度参数 |
**响应**
```json
{
"success": true,
"message": {
"role": "assistant",
"content": "回复内容..."
},
"usage": {
"prompt_tokens": 100,
"completion_tokens": 50
}
}
```

175
algorithm/main.py Normal file
View File

@@ -0,0 +1,175 @@
"""
Algorithm Service - 文档解析、Embedding、LLM 调用服务
"""
from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
from typing import Optional, List, Dict, Any
import requests
import os
import json
app = FastAPI(title="Algorithm Service")
# ========== Models ==========
class ParseRequest(BaseModel):
file_url: str
engine: str # markitdown / docling
docling_url: Optional[str] = None
class EmbeddingRequest(BaseModel):
input: str | List[str]
model: str
class ChatMessage(BaseModel):
role: str
content: str
class ChatRequest(BaseModel):
messages: List[ChatMessage]
model: str
temperature: Optional[float] = 0.7
api_key: Optional[str] = None
base_url: Optional[str] = None
# ========== 文档解析 ==========
@app.post("/parse")
async def parse_document(req: ParseRequest):
"""解析文档,支持 markitdown 和 docling"""
try:
if req.engine == "markitdown":
return await parse_with_markitdown(req.file_url)
elif req.engine == "docling":
return await parse_with_docling(req.file_url, req.docling_url)
else:
raise HTTPException(status_code=400, detail=f"Unsupported engine: {req.engine}")
except Exception as e:
return {"success": False, "error": str(e)}
async def parse_with_markitdown(file_url: str) -> Dict[str, Any]:
"""使用 markitdown 解析文档"""
try:
from markitdown import MarkItDown
md = MarkItDown()
result = md.convert(file_url)
# 简单分块(按段落分割)
content = result.text_content if hasattr(result, 'text_content') else str(result)
chunks = [c.strip() for c in content.split('\n\n') if c.strip()]
return {
"success": True,
"content": content,
"chunks": chunks[:100], # 限制 chunk 数量
"total_pages": 1,
"metadata": {
"filename": file_url.split('/')[-1]
}
}
except ImportError:
raise HTTPException(status_code=500, detail="markitdown not installed. Run: pip install markitdown")
except Exception as e:
raise HTTPException(status_code=500, detail=f"Failed to parse with markitdown: {str(e)}")
async def parse_with_docling(file_url: str, docling_url: Optional[str] = None) -> Dict[str, Any]:
"""使用 docling 解析文档"""
if not docling_url:
raise HTTPException(status_code=400, detail="docling_url is required for docling engine")
try:
# 调用 docling 服务
response = requests.post(
f"{docling_url}/convert",
json={"url": file_url},
timeout=60
)
if response.status_code != 200:
raise HTTPException(status_code=500, detail=f"Docling service error: {response.text}")
result = response.json()
content = result.get("text", "")
chunks = [c.strip() for c in content.split('\n\n') if c.strip()]
return {
"success": True,
"content": content,
"chunks": chunks[:100],
"total_pages": result.get("num_pages", 1),
"metadata": {
"filename": file_url.split('/')[-1]
}
}
except requests.exceptions.RequestException as e:
raise HTTPException(status_code=500, detail=f"Failed to connect docling service: {str(e)}")
# ========== Embedding ==========
@app.post("/embedding")
async def generate_embedding(req: EmbeddingRequest):
"""生成 Embedding"""
try:
# TODO: 根据不同 provider 调用不同的 embedding 服务
# 目前返回模拟数据
texts = [req.input] if isinstance(req.input, str) else req.input
# 模拟 embedding 返回
embeddings = [[0.1] * 1536 for _ in texts] # 1536 维向量
return {
"success": True,
"embeddings": embeddings,
"model": req.model
}
except Exception as e:
return {"success": False, "error": str(e)}
# ========== Chat ==========
@app.post("/chat")
async def chat(req: ChatRequest):
"""LLM 对话"""
try:
# TODO: 根据 model 和 base_url 调用实际的 LLM 服务
# 目前返回模拟数据
last_message = req.messages[-1].content if req.messages else ""
return {
"success": True,
"message": {
"role": "assistant",
"content": f"Echo: {last_message}"
},
"usage": {
"prompt_tokens": len(last_message),
"completion_tokens": 10
}
}
except Exception as e:
return {"success": False, "error": str(e)}
# ========== Health Check ==========
@app.get("/health")
async def health():
return {"status": "ok"}
if __name__ == "__main__":
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=8081)

View File

@@ -0,0 +1,17 @@
# FastAPI
fastapi>=0.100.0
uvicorn[standard]>=0.23.0
# HTTP 请求
requests>=2.31.0
# 文档解析
markitdown>=0.0.1
# Pydantic
pydantic>=2.0.0
# 可选:其他解析库
# docling>=0.1.0
# pypdf>=3.0.0
# python-docx>=0.8.11

30
algorithm/start.bat Normal file
View File

@@ -0,0 +1,30 @@
@echo off
chcp 65001 >nul
title Algorithm Service
echo ========================================
echo 启动 Algorithm 服务
echo ========================================
cd /d %~dp0
echo.
echo 检查虚拟环境...
if not exist venv (
echo [INFO] 创建虚拟环境...
python -m venv venv
)
echo.
echo 安装/更新依赖...
call venv\Scripts\pip install -r requirements.txt -q
echo.
echo 启动服务...
echo 访问 http://localhost:8081/docs 查看 API 文档
echo 按 Ctrl+C 停止服务
echo.
call venv\Scripts\uvicorn main:app --reload --port 8081 --host 0.0.0.0
pause

26
algorithm/start.sh Normal file
View File

@@ -0,0 +1,26 @@
#!/bin/bash
echo "========================================"
echo " 启动 Algorithm 服务"
echo "========================================"
cd "$(dirname "$0")"
# 检查虚拟环境
if [ ! -d "venv" ]; then
echo "[INFO] 创建虚拟环境..."
python3 -m venv venv
fi
echo ""
echo "安装/更新依赖..."
source venv/bin/activate
pip install -r requirements.txt -q
echo ""
echo "启动服务..."
echo "访问 http://localhost:8081/docs 查看 API 文档"
echo "按 Ctrl+C 停止服务"
echo ""
uvicorn main:app --reload --port 8081 --host 0.0.0.0

Binary file not shown.

After

Width:  |  Height:  |  Size: 15 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 13 KiB

BIN
screenshots/窗口bug.png Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 87 KiB

View File

@@ -70,12 +70,13 @@ func main() {
}
// 3. 自动迁移表
db.AutoMigrate(&model.DatabaseInfo{}, &model.SubTableInfo{}, &model.ModelInfo{})
db.AutoMigrate(&model.DatabaseInfo{}, &model.SubTableInfo{}, &model.ModelInfo{}, &model.KnowledgeBase{}, &model.KnowledgeDocument{})
// 4. 初始化 Repository
dbRepo := repository.NewDatabaseRepository(db)
subTableRepo := repository.NewSubTableRepository(db)
modelRepo := repository.NewModelRepository(db)
knowledgeRepo := repository.NewKnowledgeRepository(db)
// 5. 初始化 Service
dbService := service.NewDatabaseService(dbRepo, subTableRepo)
@@ -86,6 +87,7 @@ func main() {
if err != nil {
log.Printf("Warning: Failed to initialize upload service: %v (files will not be available)", err)
}
knowledgeService := service.NewKnowledgeService(knowledgeRepo, modelRepo, uploadService, cfg.PythonServiceURL)
// 6. 初始化 Handler
dbHandler := handler.NewDatabaseHandler(dbService)
@@ -93,6 +95,7 @@ func main() {
neo4jHandler := handler.NewNeo4jHandler(neo4jService)
modelHandler := handler.NewModelHandler(modelService)
systemHandler := handler.NewSystemHandler()
knowledgeHandler := handler.NewKnowledgeHandler(knowledgeService)
var uploadHandler *handler.UploadHandler
if uploadService != nil {
uploadHandler = handler.NewUploadHandler(uploadService)
@@ -185,6 +188,22 @@ func main() {
modelGroup.DELETE("/:id", modelHandler.Delete)
}
// 知识库管理模块
knowledgeGroup := r.Group("/api/knowledge")
{
knowledgeGroup.POST("/create", knowledgeHandler.Create)
knowledgeGroup.GET("/list", knowledgeHandler.List)
knowledgeGroup.GET("/:id", knowledgeHandler.GetByID)
knowledgeGroup.PUT("/:id", knowledgeHandler.Update)
knowledgeGroup.DELETE("/:id", knowledgeHandler.Delete)
// 文档管理
knowledgeGroup.GET("/:id/documents", knowledgeHandler.ListDocuments)
knowledgeGroup.POST("/:id/documents", knowledgeHandler.UploadDocument)
knowledgeGroup.DELETE("/:id/documents/:doc_id", knowledgeHandler.DeleteDocument)
knowledgeGroup.POST("/:id/documents/:doc_id/reparse", knowledgeHandler.ReparseDocument)
knowledgeGroup.GET("/:id/documents/:doc_id/preview", knowledgeHandler.GetDocumentPreview)
}
// 系统信息模块
r.GET("/system/info", systemHandler.GetSystemInfo)
@@ -195,8 +214,8 @@ func main() {
r.Static("/files", cfg.UploadLocalPath)
}
// 上传路由
r.POST("/upload", uploadHandler.Upload)
r.DELETE("/upload/:filename", uploadHandler.Delete)
r.POST("/api/file_upload", uploadHandler.Upload)
r.DELETE("/api/file_upload/:filename", uploadHandler.Delete)
}
// 8. 启动服务

View File

@@ -0,0 +1,220 @@
package handler
import (
"net/http"
"strconv"
"github.com/gin-gonic/gin"
"x-agents/server/internal/model"
"x-agents/server/internal/service"
)
type KnowledgeHandler struct {
service *service.KnowledgeService
}
func NewKnowledgeHandler(s *service.KnowledgeService) *KnowledgeHandler {
return &KnowledgeHandler{service: s}
}
// Create 创建知识库
func (h *KnowledgeHandler) Create(c *gin.Context) {
var req model.CreateKnowledgeRequest
if err := c.ShouldBindJSON(&req); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": err.Error()})
return
}
kb, err := h.service.Create(req)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": err.Error()})
return
}
c.JSON(http.StatusOK, gin.H{
"success": true,
"id": kb.ID,
"message": "Knowledge base created successfully",
})
}
// List 获取知识库列表
func (h *KnowledgeHandler) List(c *gin.Context) {
list, err := h.service.List()
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"success": false, "message": err.Error()})
return
}
c.JSON(http.StatusOK, gin.H{"success": true, "data": list})
}
// GetByID 获取知识库详情
func (h *KnowledgeHandler) GetByID(c *gin.Context) {
id := c.Param("id")
if id == "" {
c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": "id is required"})
return
}
kb, err := h.service.GetByID(id)
if err != nil {
c.JSON(http.StatusNotFound, gin.H{"success": false, "message": "Knowledge base not found"})
return
}
c.JSON(http.StatusOK, gin.H{"success": true, "data": kb})
}
// Update 更新知识库
func (h *KnowledgeHandler) Update(c *gin.Context) {
id := c.Param("id")
if id == "" {
c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": "id is required"})
return
}
var req model.UpdateKnowledgeRequest
if err := c.ShouldBindJSON(&req); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": err.Error()})
return
}
if err := h.service.Update(id, req); err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"success": false, "message": err.Error()})
return
}
c.JSON(http.StatusOK, gin.H{"success": true, "message": "Knowledge base updated"})
}
// Delete 删除知识库
func (h *KnowledgeHandler) Delete(c *gin.Context) {
id := c.Param("id")
if id == "" {
c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": "id is required"})
return
}
if err := h.service.Delete(id); err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"success": false, "message": err.Error()})
return
}
c.JSON(http.StatusOK, gin.H{"success": true, "message": "Knowledge base deleted"})
}
// ListDocuments 获取知识库下的文档列表
func (h *KnowledgeHandler) ListDocuments(c *gin.Context) {
id := c.Param("id")
if id == "" {
c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": "id is required"})
return
}
status := c.Query("status")
list, err := h.service.ListDocuments(id, status)
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"success": false, "message": err.Error()})
return
}
c.JSON(http.StatusOK, gin.H{"success": true, "data": list})
}
// UploadDocument 上传文档到知识库
func (h *KnowledgeHandler) UploadDocument(c *gin.Context) {
id := c.Param("id")
if id == "" {
c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": "id is required"})
return
}
file, err := c.FormFile("file")
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": "No file uploaded"})
return
}
// 检查文件大小(最大 100MB
if file.Size > 100*1024*1024 {
c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": "File too large (max 100MB)"})
return
}
doc, fileURL, err := h.service.UploadDocument(id, file)
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"success": false, "message": err.Error()})
return
}
c.JSON(http.StatusOK, gin.H{
"success": true,
"id": doc.ID,
"url": fileURL,
"document": doc,
"message": "Document uploaded",
})
}
// DeleteDocument 删除文档
func (h *KnowledgeHandler) DeleteDocument(c *gin.Context) {
id := c.Param("id")
docID := c.Param("doc_id")
if id == "" || docID == "" {
c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": "id and doc_id are required"})
return
}
if err := h.service.DeleteDocument(id, docID); err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"success": false, "message": err.Error()})
return
}
c.JSON(http.StatusOK, gin.H{"success": true, "message": "Document deleted"})
}
// ReparseDocument 重新解析文档
func (h *KnowledgeHandler) ReparseDocument(c *gin.Context) {
id := c.Param("id")
docID := c.Param("doc_id")
if id == "" || docID == "" {
c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": "id and doc_id are required"})
return
}
if err := h.service.ReparseDocument(id, docID); err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"success": false, "message": err.Error()})
return
}
c.JSON(http.StatusOK, gin.H{"success": true, "message": "Document reparse started"})
}
// GetDocumentPreview 获取文档预览
func (h *KnowledgeHandler) GetDocumentPreview(c *gin.Context) {
id := c.Param("id")
docID := c.Param("doc_id")
if id == "" || docID == "" {
c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": "id and doc_id are required"})
return
}
page := 1
if p := c.Query("page"); p != "" {
if parsed, err := strconv.Atoi(p); err == nil {
page = parsed
}
}
preview, err := h.service.GetDocumentPreview(id, docID, page)
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"success": false, "message": err.Error()})
return
}
c.JSON(http.StatusOK, gin.H{"success": true, "data": preview})
}

View File

@@ -57,6 +57,7 @@ type KnowledgeDocument struct {
KnowledgeBaseID string `json:"knowledge_base_id" gorm:"type:varchar(36);not null;index"`
Name string `json:"name" gorm:"type:varchar(255);not null"`
FileKey string `json:"file_key" gorm:"type:varchar(500)"`
FileURL string `json:"file_url" gorm:"type:varchar(500)"` // 文件访问 URL
FileSize int64 `json:"file_size" gorm:"type:bigint;default:0"`
Status string `json:"status" gorm:"type:varchar(20);default:parsing"` // parsing / parsed / failed
ChunkCount int `json:"chunk_count" gorm:"default:0"`

View File

@@ -1,7 +1,10 @@
package service
import (
"bytes"
"encoding/json"
"mime/multipart"
"net/http"
"time"
"github.com/google/uuid"
@@ -10,16 +13,18 @@ import (
)
type KnowledgeService struct {
repo *repository.KnowledgeRepository
modelRepo *repository.ModelRepository
uploadService *UploadService
repo *repository.KnowledgeRepository
modelRepo *repository.ModelRepository
uploadService *UploadService
pythonServiceURL string
}
func NewKnowledgeService(repo *repository.KnowledgeRepository, modelRepo *repository.ModelRepository, uploadService *UploadService) *KnowledgeService {
func NewKnowledgeService(repo *repository.KnowledgeRepository, modelRepo *repository.ModelRepository, uploadService *UploadService, pythonServiceURL string) *KnowledgeService {
return &KnowledgeService{
repo: repo,
modelRepo: modelRepo,
uploadService: uploadService,
repo: repo,
modelRepo: modelRepo,
uploadService: uploadService,
pythonServiceURL: pythonServiceURL,
}
}
@@ -35,22 +40,25 @@ func (s *KnowledgeService) Create(req model.CreateKnowledgeRequest) (*model.Know
}
kb := &model.KnowledgeBase{
ID: uuid.New().String(),
Name: req.Name,
Description: req.Description,
LLMModelID: req.LLMModelID,
EmbeddingModelID: req.EmbeddingModelID,
ParsingConfig: req.ParsingConfig,
Status: "active",
DocumentCount: 0,
ChunkCount: 0,
ID: uuid.New().String(),
Name: req.Name,
Description: req.Description,
LLMModelID: req.LLMModelID,
EmbeddingModelID: req.EmbeddingModelID,
ParsingConfig: req.ParsingConfig,
Status: "active",
DocumentCount: 0,
ChunkCount: 0,
}
// 设置默认值
if kb.ParsingConfig.EnablePDF == false && kb.ParsingConfig.EnablePDF != true {
if kb.ParsingConfig.Engine == "" {
kb.ParsingConfig.Engine = "markitdown"
}
if kb.ParsingConfig.EnablePDF != false {
kb.ParsingConfig.EnablePDF = true
}
if kb.ParsingConfig.Pandoc == false && kb.ParsingConfig.Pandoc != true {
if kb.ParsingConfig.Pandoc != false {
kb.ParsingConfig.Pandoc = true
}
@@ -118,41 +126,102 @@ func (s *KnowledgeService) ListDocuments(kbID string, status string) ([]model.Kn
}
// UploadDocument 上传文档到知识库
func (s *KnowledgeService) UploadDocument(kbID string, file *multipart.FileHeader) (*model.KnowledgeDocument, error) {
func (s *KnowledgeService) UploadDocument(kbID string, file *multipart.FileHeader) (*model.KnowledgeDocument, string, error) {
// 验证知识库存在
_, err := s.repo.FindByID(kbID)
kb, err := s.repo.FindByID(kbID)
if err != nil {
return nil, err
return nil, "", err
}
// 上传文件
result, err := s.uploadService.Upload(file)
if err != nil {
return nil, err
return nil, "", err
}
if !result.Success {
return nil, nil
return nil, "", nil
}
// 获取文件扩展名
ext := getFileExt(file.Filename)
// 创建文档记录
doc := &model.KnowledgeDocument{
ID: uuid.New().String(),
KnowledgeBaseID: kbID,
Name: file.Filename,
FileKey: result.FileKey,
FileKey: result.FileKey + ext,
FileURL: result.URL,
FileSize: file.Size,
Status: "parsing",
UploadedAt: time.Now(),
}
if err := s.repo.CreateDocument(doc); err != nil {
return nil, err
return nil, "", err
}
// 更新知识库文档数
s.updateDocumentCount(kbID)
return doc, nil
// 异步调用 Python 服务解析文档
go s.parseDocument(kbID, doc.ID, result.URL, kb.ParsingConfig)
return doc, result.URL, nil
}
// parseDocument 调用 Python 服务解析文档
func (s *KnowledgeService) parseDocument(kbID, docID, fileURL string, config model.ParsingConfig) {
// 构建请求
reqBody := map[string]interface{}{
"file_url": fileURL,
"engine": config.Engine,
}
if config.Engine == "docling" && config.DoclingURL != "" {
reqBody["docling_url"] = config.DoclingURL
}
body, _ := json.Marshal(reqBody)
resp, err := http.Post(s.pythonServiceURL+"/parse", "application/json", bytes.NewBuffer(body))
if err != nil {
s.repo.UpdateDocument(docID, map[string]interface{}{"status": "failed"})
return
}
defer resp.Body.Close()
if resp.StatusCode != 200 {
s.repo.UpdateDocument(docID, map[string]interface{}{"status": "failed"})
return
}
// 解析响应
var result map[string]interface{}
if err := json.NewDecoder(resp.Body).Decode(&result); err != nil {
s.repo.UpdateDocument(docID, map[string]interface{}{"status": "failed"})
return
}
if success, ok := result["success"].(bool); ok && success {
// 解析成功,更新状态
chunks := []string{}
if c, ok := result["chunks"].([]interface{}); ok {
for _, chunk := range c {
if c, ok := chunk.(string); ok {
chunks = append(chunks, c)
}
}
}
s.repo.UpdateDocument(docID, map[string]interface{}{
"status": "parsed",
"chunk_count": len(chunks),
})
// 更新知识库的 chunk_count
s.updateChunkCount(kbID)
} else {
s.repo.UpdateDocument(docID, map[string]interface{}{"status": "failed"})
}
}
// DeleteDocument 删除文档
@@ -168,7 +237,7 @@ func (s *KnowledgeService) DeleteDocument(kbID, docID string) error {
// 删除文件
if doc.FileKey != "" {
s.uploadService.DeleteFile(doc.FileKey + getFileExt(doc.Name))
s.uploadService.DeleteFile(doc.FileKey)
}
// 删除文档记录
@@ -193,8 +262,25 @@ func (s *KnowledgeService) ReparseDocument(kbID, docID string) error {
return nil
}
// 获取知识库配置
kb, err := s.repo.FindByID(kbID)
if err != nil {
return err
}
// 获取文件 URL
fileURL, err := s.uploadService.GetFileURL(doc.FileKey)
if err != nil {
return err
}
// 重置状态为 parsing
return s.repo.UpdateDocument(docID, map[string]interface{}{"status": "parsing"})
s.repo.UpdateDocument(docID, map[string]interface{}{"status": "parsing"})
// 异步重新解析
go s.parseDocument(kbID, docID, fileURL, kb.ParsingConfig)
return nil
}
// GetDocumentPreview 获取文档预览
@@ -208,12 +294,20 @@ func (s *KnowledgeService) GetDocumentPreview(kbID, docID string, page int) (*mo
return nil, nil
}
// 简单实现:返回文件 URL实际应由 Python 服务处理)
fileURL, err := s.uploadService.GetFileURL(doc.FileKey + getFileExt(doc.Name))
if err != nil {
return nil, err
// 如果已解析,返回解析内容;否则返回文件 URL
if doc.Status == "parsed" {
// TODO: 从存储中读取解析内容(可以存到数据库或文件)
// 暂时返回文件 URL
fileURL, _ := s.uploadService.GetFileURL(doc.FileKey)
return &model.DocumentPreviewResponse{
TotalPages: 1,
CurrentPage: page,
Content: fileURL,
}, nil
}
// 未解析,返回文件 URL
fileURL, _ := s.uploadService.GetFileURL(doc.FileKey)
return &model.DocumentPreviewResponse{
TotalPages: 1,
CurrentPage: page,
@@ -227,7 +321,23 @@ func (s *KnowledgeService) updateDocumentCount(kbID string) {
s.repo.Update(kbID, map[string]interface{}{"document_count": int(count)})
}
// updateChunkCount 更新知识库 chunk 数
func (s *KnowledgeService) updateChunkCount(kbID string) {
docs, _ := s.repo.FindDocumentsByKBID(kbID, "parsed")
totalChunks := 0
for _, doc := range docs {
totalChunks += doc.ChunkCount
}
s.repo.Update(kbID, map[string]interface{}{"chunk_count": totalChunks})
}
func getFileExt(filename string) string {
exts := []string{".pdf", ".docx", ".xlsx", ".pptx", ".txt", ".md", ".html"}
for _, ext := range exts {
if len(filename) >= len(ext) && filename[len(filename)-len(ext):] == ext {
return ext
}
}
if len(filename) > 4 {
return filename[len(filename)-4:]
}

File diff suppressed because one or more lines are too long

View File

@@ -0,0 +1,259 @@
# 知识库 API
## 基础信息
| 项目 | 说明 |
|------|------|
| 基础URL | `http://localhost:8082` |
## 接口列表
### 1. 创建知识库
**请求**
```
POST /api/knowledge/create
Content-Type: application/json
```
| 参数 | 类型 | 必填 | 说明 |
|------|------|------|------|
| name | String | 是 | 知识库名称 |
| description | String | 否 | 知识库描述 |
| llm_model_id | String | 是 | LLM 模型 ID |
| embedding_model_id | String | 是 | Embedding 模型 ID |
| parsing_config | Object | 是 | 解析配置 |
| - engine | String | 是 | 解析引擎markitdown / docling |
| - docling_url | String | 条件 | Docling URLengine=docling 时必填) |
| - enable_pdf | Boolean | 否 | 是否启用 PDF 解析 |
| - pandoc | Boolean | 否 | 是否启用 Pandoc |
**响应**
```json
{
"success": true,
"id": "kb_xxx",
"message": "Knowledge base created successfully"
}
```
---
### 2. 获取知识库列表
**请求**
```
GET /api/knowledge/list
```
**响应**
```json
{
"success": true,
"data": [
{
"id": "kb_001",
"name": "产品文档知识库",
"description": "用于存储产品手册",
"llm_model_id": "model_001",
"embedding_model_id": "model_002",
"status": "active",
"document_count": 15,
"chunk_count": 156,
"created_at": "2024-01-15T10:30:00Z",
"updated_at": "2024-01-15T10:30:00Z"
}
]
}
```
---
### 3. 获取知识库详情
**请求**
```
GET /api/knowledge/:id
```
**响应**
```json
{
"success": true,
"data": {
"id": "kb_001",
"name": "产品文档知识库",
"description": "用于存储产品手册",
"llm_model_id": "model_001",
"embedding_model_id": "model_002",
"parsing_config": {
"engine": "markitdown",
"enable_pdf": true,
"pandoc": true
},
"status": "active",
"document_count": 15,
"chunk_count": 156,
"created_at": "2024-01-15T10:30:00Z",
"updated_at": "2024-01-15T10:30:00Z"
}
}
```
---
### 4. 删除知识库
**请求**
```
DELETE /api/knowledge/:id
```
**响应**
```json
{
"success": true,
"message": "Knowledge base deleted"
}
```
---
### 5. 获取知识库下的文档列表
**请求**
```
GET /api/knowledge/:id/documents
```
| 参数 | 类型 | 必填 | 说明 |
|------|------|------|------|
| status | String | 否 | 过滤状态all / parsed / parsing / failed |
**响应**
```json
{
"success": true,
"data": [
{
"id": "doc_001",
"knowledge_base_id": "kb_001",
"name": "产品手册_v2.0.pdf",
"file_size": 2516582,
"status": "parsed",
"chunk_count": 156,
"uploaded_at": "2024-01-15T10:30:00Z"
}
]
}
```
---
### 6. 上传文档到知识库
**请求**
```
POST /api/knowledge/:id/documents
Content-Type: multipart/form-data
```
| 参数 | 类型 | 必填 | 说明 |
|------|------|------|------|
| file | File | 是 | 要上传的文件 |
**响应**
```json
{
"success": true,
"id": "doc_001",
"url": "http://localhost:8082/files/abc123.pdf",
"document": {
"id": "doc_001",
"knowledge_base_id": "kb_001",
"name": "产品手册_v2.0.pdf",
"file_size": 2516582,
"status": "parsing",
"chunk_count": 0,
"uploaded_at": "2024-01-15T10:30:00Z"
},
"message": "Document uploaded"
}
```
---
### 7. 删除知识库文档
**请求**
```
DELETE /api/knowledge/:id/documents/:doc_id
```
**响应**
```json
{
"success": true,
"message": "Document deleted"
}
```
---
### 8. 重新解析文档
**请求**
```
POST /api/knowledge/:id/documents/:doc_id/reparse
```
**响应**
```json
{
"success": true,
"message": "Document reparse started"
}
```
---
### 9. 获取文档预览内容
**请求**
```
GET /api/knowledge/:id/documents/:doc_id/preview
```
| 参数 | 类型 | 必填 | 说明 |
|------|------|------|------|
| page | Number | 否 | 页码(默认 1 |
**响应**
```json
{
"success": true,
"data": {
"total_pages": 3,
"current_page": 1,
"content": "第一章 产品介绍..."
}
}
```

View File

@@ -30,7 +30,7 @@ minio_use_ssl: false
**请求**
```
POST /upload
POST /api/file_upload
Content-Type: multipart/form-data
```
@@ -65,7 +65,7 @@ Content-Type: multipart/form-data
**请求**
```
DELETE /upload/:filename
DELETE /api/file_upload/:filename
```
| 参数 | 类型 | 必填 | 说明 |

View File

@@ -20,6 +20,7 @@ interface MenuItem {
const mainMenu: MenuItem[] = [
{ name: 'Dashboard', icon: 'fa-gauge', path: '/dashboard' },
{ name: 'Agents', icon: 'fa-robot', badge: 3, path: '/agents' },
{ name: 'Team', icon: 'fa-users', path: '/team' },
{ name: 'Skills', icon: 'fa-wand-magic-sparkles', badge: 21, path: '/mcp' },
{ name: 'Tools', icon: 'fa-tools', badge: 13, path: '/model-apis' },
{ name: 'Database', icon: 'fa-database', path: '/database' },
@@ -28,7 +29,6 @@ const mainMenu: MenuItem[] = [
const bottomMenu: MenuItem[] = [
{ name: 'Settings', icon: 'fa-gear', path: '/settings' },
{ name: 'Team', icon: 'fa-users', path: '/team' },
]
const bottomMenu2: MenuItem[] = [
@@ -117,7 +117,7 @@ const handleUserCommand = (command: string) => {
<!-- 分隔线 -->
<li class="my-4 border-t border-dark-500"></li>
<!-- Settings & Team -->
<!-- Settings -->
<li v-for="item in bottomMenu" :key="item.name">
<a
href="#"
@@ -134,18 +134,13 @@ const handleUserCommand = (command: string) => {
<span class="w-2 h-2 rounded-full bg-yellow-500"></span>
<span class="w-2 h-2 rounded-full bg-gray-500"></span>
</div>
<div v-else-if="item.name === 'Team'" class="flex gap-1">
<span class="w-2 h-2 rounded-full bg-primary-orange"></span>
<span class="w-2 h-2 rounded-full bg-blue-500"></span>
<span class="w-2 h-2 rounded-full bg-gray-500"></span>
</div>
</a>
</li>
<!-- 分隔线 -->
<li class="my-4 border-t border-dark-500"></li>
<!-- Information -->
<!-- Account -->
<li v-for="item in bottomMenu2" :key="item.name">
<a
href="#"

View File

@@ -2,7 +2,8 @@ import { createRouter, createWebHistory } from 'vue-router'
import Dashboard from '@/views/Dashboard.vue'
import Login from '@/views/Login.vue'
import Agents from '@/views/Agents.vue'
import MCP from '@/views/MCP.vue'
import Team from '@/views/Team.vue'
import Skill from '@/views/Skill.vue'
import ModelAPIs from '@/views/ModelAPIs.vue'
import Database from '@/views/Database.vue'
import Knowledge from '@/views/Knowledge.vue'
@@ -26,10 +27,15 @@ const router = createRouter({
name: 'agents',
component: Agents
},
{
path: '/team',
name: 'team',
component: Team
},
{
path: '/mcp',
name: 'mcp',
component: MCP
component: Skill
},
{
path: '/model-apis',

View File

@@ -2,16 +2,32 @@
import { ref, computed, onMounted } from 'vue'
import { ElMessage } from 'element-plus'
import { useModelSettings } from './settings/useModelSettings'
import { fetchKnowledgeBases, createKnowledgeBase as apiCreateKnowledgeBase, deleteKnowledgeBase as apiDeleteKnowledgeBase, fetchKnowledgeDocuments } from './knowledge/useKnowledge'
import './knowledge/knowledge.css'
// 获取已配置的模型列表
const { models, fetchModels } = useModelSettings()
// 页面加载时获取模型列表
onMounted(() => {
// 页面加载时获取数据
onMounted(async () => {
fetchModels()
await fetchKbList()
})
// 获取知识库列表
const knowledgeBases = ref<any[]>([])
const loadingKb = ref(false)
const fetchKbList = async () => {
loadingKb.value = true
try {
const data = await fetchKnowledgeBases()
knowledgeBases.value = data
} finally {
loadingKb.value = false
}
}
// 筛选 LLM (chat) 模型
const llmModels = computed(() => {
return models.value.filter((m: any) => m.model_type === 'chat')
@@ -80,37 +96,6 @@ const prevStep = () => {
}
}
// 模拟知识库数据
const knowledgeBases = ref([
{
id: '1',
name: 'Product Documentation',
description: 'Product user manual and API docs',
document_count: 156,
chunk_count: 1248,
created_at: '2024-01-15T10:30:00Z',
status: 'ready'
},
{
id: '2',
name: 'Company Policies',
description: 'Internal company policies and procedures',
document_count: 42,
chunk_count: 320,
created_at: '2024-01-20T14:20:00Z',
status: 'ready'
},
{
id: '3',
name: 'Technical Wiki',
description: 'Engineering documentation and RFCs',
document_count: 89,
chunk_count: 756,
created_at: '2024-02-01T09:15:00Z',
status: 'processing'
},
])
const loading = ref(false)
const searchQuery = ref('')
@@ -132,7 +117,17 @@ const createStep = ref(1)
const showFileUploadDialog = ref(false)
const selectedKnowledge = ref<any>(null)
const fileFilter = ref('all') // all, parsed, parsing, failed
const selectedFile = ref<any>(null) // 当前选中的文件
const selectedFile = ref<any>(null) // 当前选中的文件ID
const selectedDocument = ref<any>(null) // 当前选中的文档详情
const knowledgeDocuments = ref<any[]>([]) // 知识库文档列表
const loadingDocuments = ref(false)
const fileInput = ref<HTMLInputElement | null>(null)
const uploading = ref(false)
const previewUrl = ref('') // 文档预览URL
const loadingPreview = ref(false)
const previewPage = ref(1) // 当前页码
const previewTotalPages = ref(1) // 总页数
const newKbForm = ref({
name: '',
description: '',
@@ -153,6 +148,11 @@ const parsingConfig = ref({
fileSizeLimit: '5242880',
})
// Storage 配置
const storageConfig = ref({
type: 'local',
})
const openCreateDialog = () => {
createStep.value = 1
newKbForm.value = { name: '', description: '' }
@@ -166,6 +166,7 @@ const openCreateDialog = () => {
highRes: false,
fileSizeLimit: '5242880',
}
storageConfig.value = { type: 'local' }
showCreateDialog.value = true
}
@@ -181,23 +182,42 @@ const cancelCreate = () => {
highRes: false,
fileSizeLimit: '5242880',
}
storageConfig.value = { type: 'local' }
showCreateDialog.value = false
}
const createKnowledgeBase = () => {
knowledgeBases.value.push({
id: Date.now().toString(),
const createKnowledgeBase = async () => {
const result = await apiCreateKnowledgeBase({
name: newKbForm.value.name,
description: newKbForm.value.description,
document_count: 0,
chunk_count: 0,
created_at: new Date().toISOString(),
status: 'ready'
llm_model_id: modelConfig.value.llmModelId,
embedding_model_id: modelConfig.value.embeddingModelId,
parsing_config: {
engine: parsingConfig.value.engine,
docling_url: parsingConfig.value.engine === 'docling' ? parsingConfig.value.doclingUrl : undefined,
enable_pdf: parsingConfig.value.enablePdf,
pandoc: parsingConfig.value.pandoc,
}
})
newKbForm.value = { name: '', description: '' }
modelConfig.value = { llmModelId: '', embeddingModelId: '' }
showCreateDialog.value = false
ElMessage.success('Knowledge base created successfully')
if (result.success) {
await fetchKbList()
newKbForm.value = { name: '', description: '' }
modelConfig.value = { llmModelId: '', embeddingModelId: '' }
parsingConfig.value = {
enablePdf: true,
engine: 'markitdown',
doclingUrl: '',
pandoc: true,
academic: false,
highRes: false,
fileSizeLimit: '5242880',
}
showCreateDialog.value = false
ElMessage.success('Knowledge base created successfully')
} else {
ElMessage.error(result.message || 'Failed to create knowledge base')
}
}
// 编辑知识库
@@ -232,19 +252,217 @@ const cancelEdit = () => {
}
// 删除知识库
const deleteKb = (id: string) => {
const index = knowledgeBases.value.findIndex(k => k.id === id)
if (index > -1) {
knowledgeBases.value.splice(index, 1)
const deleteKb = async (id: string) => {
const result = await apiDeleteKnowledgeBase(id)
if (result.success) {
await fetchKbList()
ElMessage.success('Knowledge base deleted')
} else {
ElMessage.error(result.message || 'Failed to delete knowledge base')
}
}
// 辅助函数:格式化文件大小
const formatFileSize = (bytes: number) => {
if (!bytes) return '0 B'
const k = 1024
const sizes = ['B', 'KB', 'MB', 'GB']
const i = Math.floor(Math.log(bytes) / Math.log(k))
return parseFloat((bytes / Math.pow(k, i)).toFixed(1)) + ' ' + sizes[i]
}
// 辅助函数:格式化日期
const formatDate = (dateStr: string) => {
if (!dateStr) return ''
const date = new Date(dateStr)
return date.toLocaleDateString('zh-CN')
}
// 辅助函数:获取状态图标
const getStatusIcon = (status: string) => {
switch (status) {
case 'parsed':
return 'fa-solid fa-check-circle'
case 'parsing':
return 'fa-solid fa-spinner fa-spin'
case 'failed':
return 'fa-solid fa-circle-xmark'
default:
return 'fa-solid fa-clock'
}
}
// 进入知识库(上传文档界面)
const enterKnowledge = (kb: any) => {
const enterKnowledge = async (kb: any) => {
selectedKnowledge.value = kb
selectedFile.value = null
previewUrl.value = ''
// 获取文档列表
loadingDocuments.value = true
try {
const docs = await fetchKnowledgeDocuments(kb.id, fileFilter.value)
console.log('Fetched documents:', docs)
knowledgeDocuments.value = docs
// 自动选中第一个文档
if (docs && docs.length > 0) {
console.log('First doc:', docs[0])
await selectDocument(docs[0])
}
} finally {
loadingDocuments.value = false
}
showFileUploadDialog.value = true
}
// 切换文件过滤标签时重新获取文档列表
const changeFileFilter = async (filter: string) => {
fileFilter.value = filter
if (selectedKnowledge.value) {
loadingDocuments.value = true
try {
const docs = await fetchKnowledgeDocuments(selectedKnowledge.value.id, filter)
knowledgeDocuments.value = docs
} finally {
loadingDocuments.value = false
}
}
}
// 选择文档
const selectDocument = async (doc: any) => {
selectedFile.value = doc.id
selectedDocument.value = doc
previewUrl.value = ''
previewPage.value = 1
previewTotalPages.value = 1
// 尝试从多个字段获取文件URL
const fileUrl = doc.file_url || doc.fileUrl || doc.url || doc.FileURL
if (fileUrl) {
previewUrl.value = fileUrl
} else if (selectedKnowledge.value && doc.status === 'parsed') {
// 获取文档预览
loadingPreview.value = true
try {
const { getDocumentPreview } = await import('./knowledge/useKnowledge')
const result = await getDocumentPreview(selectedKnowledge.value.id, doc.id)
if (result.success && result.data?.content) {
previewUrl.value = result.data.content
previewTotalPages.value = result.data.total_pages || 1
previewPage.value = result.data.current_page || 1
}
} catch (error) {
console.error('Failed to get preview:', error)
} finally {
loadingPreview.value = false
}
}
}
// 翻页
const changePreviewPage = async (page: number) => {
if (!selectedKnowledge.value || !selectedFile.value || page < 1 || page > previewTotalPages.value) return
loadingPreview.value = true
previewPage.value = page
try {
const { getDocumentPreview } = await import('./knowledge/useKnowledge')
const result = await getDocumentPreview(selectedKnowledge.value.id, selectedFile.value, page)
if (result.success && result.data?.content) {
previewUrl.value = result.data.content
previewTotalPages.value = result.data.total_pages || 1
previewPage.value = result.data.current_page || page
}
} catch (error) {
console.error('Failed to get preview:', error)
} finally {
loadingPreview.value = false
}
}
// 触发文件上传
const triggerFileUpload = () => {
fileInput.value?.click()
}
// 处理文件选择
const handleFileSelect = async (event: Event) => {
const target = event.target as HTMLInputElement
const file = target.files?.[0]
if (!file || !selectedKnowledge.value) return
uploading.value = true
try {
const { uploadDocument } = await import('./knowledge/useKnowledge')
const result = await uploadDocument(selectedKnowledge.value.id, file)
if (result.success) {
ElMessage.success('File uploaded successfully')
// 后端返回 result.url 在顶层result.document 里有 file_url
const fileUrl = result.url || result.document?.file_url
// 添加到文档列表
const newDoc = result.document || {
id: result.id,
name: file.name,
file_size: file.size,
status: 'parsing',
chunk_count: 0,
uploaded_at: new Date().toISOString(),
file_url: fileUrl
}
// 如果返回了file_url添加到列表开头
if (fileUrl) {
previewUrl.value = fileUrl
// 设置选中的文档信息
selectedFile.value = result.id
selectedDocument.value = newDoc
// 添加到文档列表
knowledgeDocuments.value = [newDoc, ...knowledgeDocuments.value]
} else {
// 刷新文档列表
await changeFileFilter(fileFilter.value)
}
} else {
ElMessage.error(result.message || 'Failed to upload file')
}
} catch (error) {
ElMessage.error('Failed to upload file')
} finally {
uploading.value = false
// 清空 input 以便可以再次选择相同文件
target.value = ''
}
}
// 删除文档
const deleteDocument = async (docId: string) => {
if (!selectedKnowledge.value) return
try {
const { deleteDocument: apiDeleteDocument } = await import('./knowledge/useKnowledge')
const result = await apiDeleteDocument(selectedKnowledge.value.id, docId)
if (result.success) {
ElMessage.success('Document deleted')
// 如果删除的是当前选中的文件,清除选中状态
if (selectedFile.value === docId) {
selectedFile.value = null
selectedDocument.value = null
previewUrl.value = ''
}
// 刷新文档列表
await changeFileFilter(fileFilter.value)
} else {
ElMessage.error(result.message || 'Failed to delete document')
}
} catch (error) {
ElMessage.error('Failed to delete document')
}
}
</script>
<template>
@@ -515,7 +733,7 @@ const enterKnowledge = (kb: any) => {
</div>
<el-form label-position="top" class="kb-form">
<el-form-item label="Storage Type">
<el-select placeholder="Select storage type" class="w-full">
<el-select v-model="storageConfig.type" placeholder="Select storage type" class="w-full" popper-class="dark-select-dropdown">
<el-option label="Local" value="local" />
<el-option label="MinIO" value="minio" />
<el-option label="S3" value="s3" />
@@ -580,8 +798,8 @@ const enterKnowledge = (kb: any) => {
<!-- 文件上传弹窗 -->
<el-dialog
v-model="showFileUploadDialog"
:title="selectedKnowledge?.name || 'Knowledge Base'"
width="95vw"
title=" "
width="calc(100vw - 40px)"
top="20px"
:close-on-click-modal="false"
class="kb-dialog file-upload-dialog"
@@ -593,9 +811,16 @@ const enterKnowledge = (kb: any) => {
<i class="fa-solid fa-arrow-left"></i>
</button>
<h2 class="file-title">{{ selectedKnowledge?.name || 'Knowledge Base' }}</h2>
<button class="btn-primary">
<input
type="file"
ref="fileInput"
style="display: none"
accept=".pdf,.doc,.docx,.txt,.md"
@change="handleFileSelect"
/>
<button class="btn-primary" @click="triggerFileUpload">
<i class="fa-solid fa-upload"></i>
上传文档
Upload
</button>
</div>
@@ -604,30 +829,30 @@ const enterKnowledge = (kb: any) => {
<div
class="file-tab"
:class="{ active: fileFilter === 'all' }"
@click="fileFilter = 'all'"
@click="changeFileFilter('all')"
>
全部
All
</div>
<div
class="file-tab"
:class="{ active: fileFilter === 'parsed' }"
@click="fileFilter = 'parsed'"
@click="changeFileFilter('parsed')"
>
已解析
Parsed
</div>
<div
class="file-tab"
:class="{ active: fileFilter === 'parsing' }"
@click="fileFilter = 'parsing'"
@click="changeFileFilter('parsing')"
>
解析中
Parsing
</div>
<div
class="file-tab"
:class="{ active: fileFilter === 'failed' }"
@click="fileFilter = 'failed'"
@click="changeFileFilter('failed')"
>
解析失败
Failed
</div>
</div>
@@ -635,74 +860,98 @@ const enterKnowledge = (kb: any) => {
<div class="file-main">
<!-- 左侧文件列表 -->
<div class="file-list">
<!-- 模拟文件项 -->
<!-- 加载状态 -->
<div v-if="loadingDocuments" class="loading-state">
<i class="fa-solid fa-spinner fa-spin"></i>
<span>Loading...</span>
</div>
<!-- 无数据提示 -->
<div v-else-if="knowledgeDocuments.length === 0" class="empty-state">
<i class="fa-solid fa-folder-open"></i>
<span>No documents yet</span>
</div>
<!-- 文档列表 -->
<div
v-else
class="file-item"
:class="{ selected: selectedFile === i }"
v-for="i in 8"
:key="i"
@click="selectedFile = i"
:class="{ selected: selectedFile === doc.id }"
v-for="doc in knowledgeDocuments"
:key="doc.id"
@click="selectDocument(doc)"
>
<div class="file-item-icon">
<i class="fa-solid fa-file-pdf"></i>
</div>
<div class="file-item-info">
<div class="file-item-name">产品手册_v2.0.pdf</div>
<div class="file-item-name">{{ doc.name }}</div>
<div class="file-item-meta">
<span>2.4 MB</span>
<span>2024-01-15</span>
<span>{{ formatFileSize(doc.file_size) }}</span>
<span>{{ formatDate(doc.uploaded_at) }}</span>
</div>
</div>
<div class="file-item-status success">
<i class="fa-solid fa-check-circle"></i>
<div class="file-item-status" :class="doc.status">
<span v-if="doc.status === 'failed'" class="status-text">Failed</span>
<i v-else :class="getStatusIcon(doc.status)"></i>
</div>
</div>
</div>
<!-- 右侧预览面板 -->
<div class="file-preview">
<!-- 无选中文件时显示提示 -->
<div v-if="!selectedFile" class="preview-empty">
<i class="fa-solid fa-file-lines"></i>
<span>Select a document to preview</span>
</div>
<!-- 有选中文件时显示预览 -->
<template v-else>
<div class="preview-header">
<div class="preview-header-left">
<i class="fa-solid fa-file-pdf"></i>
<span class="preview-filename">产品手册_v2.0.pdf</span>
<span class="preview-filename">{{ selectedDocument?.name || 'Document' }}</span>
</div>
<div class="preview-header-info">
<span class="info-tag">2.4 MB</span>
<span class="info-tag">2024-01-15 10:30</span>
<span class="info-tag success">解析成功</span>
<span class="info-tag">156 个切片</span>
<span class="info-tag">{{ formatFileSize(selectedDocument?.file_size) }}</span>
<span class="info-tag">{{ formatDate(selectedDocument?.uploaded_at) }}</span>
<span class="info-tag" :class="selectedDocument?.status">{{ selectedDocument?.status === 'parsed' ? 'Parsed' : selectedDocument?.status === 'parsing' ? 'Parsing' : selectedDocument?.status === 'failed' ? 'Failed' : 'Pending' }}</span>
<span class="info-tag">{{ selectedDocument?.chunk_count || 0 }} chunks</span>
</div>
</div>
<div class="preview-content">
<!-- PDF文件预览 -->
<div class="pdf-preview">
<iframe
src="https://www.w3.org/WAI/ER/tests/xhtml/testfiles/resources/pdf/dummy.pdf"
class="pdf-iframe"
></iframe>
</div>
<div class="preview-pagination">
<button class="page-btn" disabled>
<i class="fa-solid fa-chevron-left"></i>
</button>
<span class="page-info">1 / 3</span>
<button class="page-btn">
<i class="fa-solid fa-chevron-right"></i>
</button>
<!-- 加载中状态 -->
<div v-if="loadingPreview" class="preview-loading">
<i class="fa-solid fa-spinner fa-spin"></i>
<span>Loading preview...</span>
</div>
<!-- 有预览URL时显示PDF -->
<embed
v-else-if="previewUrl"
type="application/pdf"
:src="previewUrl"
class="pdf-embed"
/>
<!-- 无预览时显示提示 -->
<div v-else class="preview-no-file">
<i class="fa-solid fa-file-pdf"></i>
<span>Document preview not available</span>
</div>
</div>
</div>
</template>
<div class="preview-actions">
<button class="action-btn delete">
<button class="action-btn delete" @click="deleteDocument(selectedFile)">
<i class="fa-solid fa-trash"></i>
删除
Delete
</button>
<button class="action-btn reparse">
<i class="fa-solid fa-rotate"></i>
重新解析
Reparse
</button>
<button class="action-btn confirm">
<i class="fa-solid fa-check"></i>
确认
Confirm
</button>
</div>
</div>

View File

@@ -1,90 +1,86 @@
<script setup lang="ts">
import { ref, nextTick, onMounted, computed } from 'vue'
import { ref, onMounted, computed } from 'vue'
import { useModelSettings } from './settings/useModelSettings'
import { useSkillServers } from './skill/useSkillServers'
import { useFileTree } from './skill/useFileTree'
import { useSkillChat } from './skill/useSkillChat'
import { useWorkflow } from './skill/useWorkflow'
import type { FileItem } from './skill/types'
interface MCPServer {
id: number
name: string
status: 'running' | 'stopped' | 'error'
type: string
port: number
createdAt: string
description: string
}
// MCP Server management
const {
mcpServers,
editingServer,
isEditing,
searchQuery,
filterStatus,
editForm,
openEdit,
saveEdit,
cancelEdit,
toggleStatus,
deleteServer,
filteredServers,
statusClass,
} = useSkillServers()
const mcpServers = ref<MCPServer[]>([
{ id: 1, name: 'linear-demo', status: 'running', type: 'Linear', port: 3001, createdAt: '2025-04-10', description: 'Linear API integration for project management' },
{ id: 2, name: 'google-maps', status: 'running', type: 'Google Maps', port: 3002, createdAt: '2025-04-08', description: 'Google Maps API for location services' },
{ id: 3, name: 'explorer-mcp', status: 'error', type: 'File System', port: 3003, createdAt: '2025-04-05', description: 'File system explorer and editor' },
{ id: 4, name: 'postgres-mcp', status: 'running', type: 'PostgreSQL', port: 3004, createdAt: '2025-04-12', description: 'PostgreSQL database operations' },
{ id: 5, name: 'github-mcp', status: 'stopped', type: 'GitHub', port: 3005, createdAt: '2025-04-11', description: 'GitHub API integration' },
])
// File tree management
const {
fileTree,
rootFiles,
inlineNewFolderName,
inlineCreatingFolderId,
inlineNewFileName,
inlineCreatingFileId,
isEditingFile,
isEditingFileClosing,
editingFile,
editingFileName,
editingFileContent,
editingFileType,
hoveringItem,
selectedFolder,
toggleFolder,
showInlineNewFolderInput,
confirmInlineCreateFolder,
cancelInlineCreateFolder,
findFolderById,
showInlineNewFileInput,
confirmInlineCreateFile,
cancelInlineCreateFile,
getFileIconByName,
handleFileUpload,
findFile,
selectFolder,
deleteFile,
deleteFolder,
openFile,
saveFileEdit,
cancelFileEdit,
} = useFileTree()
const editingServer = ref<MCPServer | null>(null)
const isEditing = ref(false)
const searchQuery = ref('')
const filterStatus = ref<string>('all')
// Chat functionality
const {
isChatOpen,
chatInput,
chatMessages,
toggleChat,
sendMessage,
} = useSkillChat()
const editForm = ref({
name: '',
type: '',
port: 3000,
description: '',
})
const openEdit = (server: MCPServer) => {
editingServer.value = server
editForm.value = {
name: server.name,
type: server.type,
port: server.port,
description: server.description,
}
isEditing.value = true
}
const saveEdit = () => {
if (editingServer.value) {
const index = mcpServers.value.findIndex(s => s.id === editingServer.value!.id)
if (index !== -1) {
mcpServers.value[index] = {
...mcpServers.value[index],
...editForm.value,
}
}
}
isEditing.value = false
}
const cancelEdit = () => {
isEditing.value = false
editingServer.value = null
}
const toggleStatus = (server: MCPServer) => {
if (server.status === 'running') {
server.status = 'stopped'
} else if (server.status === 'stopped') {
server.status = 'running'
}
}
const deleteServer = (id: number) => {
mcpServers.value = mcpServers.value.filter(s => s.id !== id)
}
const filteredServers = () => {
return mcpServers.value.filter(server => {
const matchSearch = server.name.toLowerCase().includes(searchQuery.value.toLowerCase()) ||
server.type.toLowerCase().includes(searchQuery.value.toLowerCase())
const matchStatus = filterStatus.value === 'all' || server.status === filterStatus.value
return matchSearch && matchStatus
})
}
// Workflow visualization
const {
workflowData,
isGeneratingGraph,
isGraphGenerated,
toggleNode,
renderWorkflow,
generateGraph,
} = useWorkflow()
// Skill
const isCreating = ref(false)
const createStep = ref(1) // 1= 2=
const createStep = ref(1)
const newSkillForm = ref({
name: '',
type: 'API',
@@ -98,7 +94,6 @@ const newSkillForm = ref({
// Model Settings
const { models, fetchModels } = useModelSettings()
// chat
const availableModels = computed(() => {
return models.value
.filter((m: any) => m.model_type === 'chat')
@@ -116,7 +111,6 @@ onMounted(() => {
const goToStep2 = () => {
if (!newSkillForm.value.name.trim()) return
createStep.value = 2
//
isGraphGenerated.value = false
isGeneratingGraph.value = false
}
@@ -125,462 +119,6 @@ const goBackToStep1 = () => {
createStep.value = 1
}
// - VS Code
interface FileItem {
id: string
name: string
icon: string
content: string
type: 'file'
}
interface FolderItem {
id: string
name: string
icon: string
type: 'folder'
expanded: boolean
children: (FileItem | FolderItem)[]
}
//
const skillMdContent = `# Skill Configuration
## Overview
This skill is designed to process user queries and generate appropriate responses.
## Data Sources
- user_data.csv
- config.json
## Scripts
- process.py
- transform.js
## Dependencies
- Python 3.8+
- Node.js 16+
`
const fileTree = ref<FolderItem[]>([
{
id: 'folder-1',
name: 'Data',
icon: 'fa-folder',
type: 'folder',
expanded: true,
children: [
{ id: 'file-1', name: 'user_data.csv', icon: 'fa-file', content: 'id,name,email,age\n1,John Doe,john@example.com,28\n2,Jane Smith,jane@example.com,32', type: 'file' },
{ id: 'file-2', name: 'config.json', icon: 'fa-code', content: '{\n "appName": "MyApp",\n "version": "1.0.0",\n "debug": true\n}', type: 'file' },
]
},
{
id: 'folder-2',
name: 'Script',
icon: 'fa-folder',
type: 'folder',
expanded: true,
children: [
{ id: 'file-3', name: 'process.py', icon: 'fa-code', content: 'import pandas as pd\n\ndef process_data(data):\n """Process the input data"""\n return data.apply(lambda x: x.strip())\n\nif __name__ == "__main__":\n df = pd.read_csv("user_data.csv")\n processed = process_data(df)\n processed.to_csv("output.csv", index=False)', type: 'file' },
{ id: 'file-4', name: 'transform.js', icon: 'fa-code', content: 'function transformData(input) {\n return input.map(item => ({\n ...item,\n processed: true,\n timestamp: Date.now()\n }));\n}\n\nmodule.exports = { transformData };', type: 'file' },
]
},
{
id: 'folder-3',
name: 'Reference',
icon: 'fa-folder',
type: 'folder',
expanded: true,
children: [
{ id: 'file-5', name: 'api_docs.md', icon: 'fa-code', content: '# API Documentation\n\n## Endpoints\n\n### GET /api/users\nReturns list of users.\n\n### POST /api/users\nCreate a new user.', type: 'file' },
]
},
])
//
const rootFiles = ref<FileItem[]>([
{ id: 'file-root-1', name: 'SKILL.md', icon: 'fa-file', content: skillMdContent, type: 'file' },
])
// /
const toggleFolder = (folder: FolderItem) => {
folder.expanded = !folder.expanded
}
//
const isCreatingFolder = ref(false)
const newFolderName = ref('')
const newFolderParentId = ref<string | null>(null)
const showNewFolderInput = (parentId: string | null = null) => {
newFolderName.value = ''
newFolderParentId.value = parentId
isCreatingFolder.value = true
}
const createFolder = () => {
if (!newFolderName.value.trim()) {
isCreatingFolder.value = false
return
}
const newFolder: FileItem = {
id: 'folder-' + Date.now(),
name: newFolderName.value,
icon: 'fa-folder',
type: 'folder',
expanded: true,
children: []
}
if (newFolderParentId.value) {
//
const parentFolder = findFolderById(fileTree.value, newFolderParentId.value)
if (parentFolder && parentFolder.children) {
parentFolder.children.push(newFolder)
}
} else {
//
fileTree.value.push(newFolder)
}
isCreatingFolder.value = false
}
//
const findFolderById = (folders: FileItem[], id: string): FileItem | null => {
for (const folder of folders) {
if (folder.id === id) return folder
if (folder.children) {
const found = findFolderById(folder.children, id)
if (found) return found
}
}
return null
}
//
const isCreatingFile = ref(false)
const newFileName = ref('')
const newFileParentFolder = ref<string | null>(null)
const showNewFileInput = (folderId: string | null = null) => {
newFileName.value = ''
newFileParentFolder.value = folderId
isCreatingFile.value = true
}
const createFile = () => {
if (!newFileName.value.trim()) {
isCreatingFile.value = false
return
}
const newFile: FileItem = {
id: 'file-' + Date.now(),
name: newFileName.value,
icon: getFileIconByName(newFileName.value),
content: '',
type: 'file'
}
if (newFileParentFolder.value) {
//
const folder = fileTree.value.find(f => f.id === newFileParentFolder.value)
if (folder && folder.children) {
folder.children.push(newFile)
}
} else {
//
rootFiles.value.push(newFile)
}
isCreatingFile.value = false
}
// - 使
const getFileIconByName = (filename: string): string => {
const ext = filename.split('.').pop()?.toLowerCase()
const codeExtensions = ['js', 'ts', 'py', 'html', 'css', 'json', 'xml', 'yaml', 'yml', 'java', 'c', 'cpp', 'go', 'rs', 'php', 'rb', 'swift', 'kt', 'sql', 'sh', 'bash', 'md']
if (codeExtensions.includes(ext || '')) {
return 'fa-code'
}
return 'fa-file'
}
//
const handleFileUpload = (event: Event, folderId: string | null = null) => {
const input = event.target as HTMLInputElement
if (!input.files?.length) return
Array.from(input.files).forEach(file => {
const reader = new FileReader()
reader.onload = (e) => {
const newFile: FileItem = {
id: 'file-' + Date.now() + Math.random(),
name: file.name,
icon: getFileIconByName(file.name),
content: e.target?.result as string || '',
type: 'file'
}
if (folderId) {
const folder = fileTree.value.find(f => f.id === folderId)
if (folder && folder.children) {
folder.children.push(newFile)
}
} else {
rootFiles.value.push(newFile)
}
}
reader.readAsText(file)
})
input.value = ''
}
//
const findFile = (items: (FileItem | FolderItem)[], fileId: string): FileItem | null => {
for (const item of items) {
if (item.type === 'file' && item.id === fileId) {
return item
}
if (item.type === 'folder' && item.children) {
const found = findFile(item.children, fileId)
if (found) return found
}
}
return null
}
//
const isEditingFile = ref(false)
const isEditingFileClosing = ref(false)
// AI
const isChatOpen = ref(false)
//
const isGeneratingGraph = ref(false)
const isGraphGenerated = ref(false)
const generateGraph = () => {
isGeneratingGraph.value = true
//
setTimeout(() => {
isGeneratingGraph.value = false
isGraphGenerated.value = true
// -
if (workflowData.value.length > 0) {
workflowData.value[0].expanded = true
if (workflowData.value[0].children) {
workflowData.value[0].children.forEach(child => {
child.expanded = false
if (child.children) {
child.children.forEach(gc => gc.expanded = false)
}
})
}
}
}, 2500)
}
const toggleChat = () => {
isChatOpen.value = !isChatOpen.value
}
const editingFile = ref<FileItem | null>(null)
const editingFileName = ref('')
const editingFileContent = ref('')
const editingFileType = ref('')
// /
//
const hoveringItem = ref<string | null>(null)
// /
const selectedFolder = ref<string | null>(null)
//
const selectFolder = (folderId: string) => {
if (selectedFolder.value === folderId) {
selectedFolder.value = null //
} else {
selectedFolder.value = folderId
}
}
//
const deleteFile = (file: FileItem, parentId: string | null) => {
if (!parentId) {
//
rootFiles.value = rootFiles.value.filter(f => f.id !== file.id)
} else {
//
const folder = fileTree.value.find(f => f.id === parentId)
if (folder && folder.children) {
folder.children = folder.children.filter(c => (c as FileItem).id !== file.id)
}
}
}
//
const deleteFolder = (folder: FolderItem) => {
fileTree.value = fileTree.value.filter(f => f.id !== folder.id)
}
//
const openFile = (file: FileItem) => {
editingFile.value = file
editingFileName.value = file.name
editingFileContent.value = file.content
editingFileType.value = 'file'
isEditingFile.value = true
}
const saveFileEdit = () => {
if (!editingFile.value) return
//
editingFile.value.content = editingFileContent.value
isEditingFileClosing.value = true
setTimeout(() => {
isEditingFile.value = false
isEditingFileClosing.value = false
editingFile.value = null
}, 300)
}
const cancelFileEdit = () => {
isEditingFileClosing.value = true
setTimeout(() => {
isEditingFile.value = false
isEditingFileClosing.value = false
editingFile.value = null
}, 300)
}
//
const chatInput = ref('')
const chatMessages = ref<{ id: number; text: string }[]>([])
const sendMessage = () => {
if (!chatInput.value.trim()) return
chatMessages.value.push({
id: Date.now(),
text: chatInput.value
})
const userInput = chatInput.value
chatInput.value = ''
// AI
setTimeout(() => {
chatMessages.value.push({
id: Date.now(),
text: `I've updated the configuration based on your request: "${userInput}". The changes have been applied to your skill settings.`
})
}, 1000)
}
// -
// -
interface WorkflowNode {
id: string
name: string
description: string
icon: string
color: string
bgColor: string
borderColor: string
status: 'pending' | 'processing' | 'completed'
expanded: boolean
children?: WorkflowNode[]
}
const workflowData = ref<WorkflowNode[]>([
{
id: 'output',
name: '最终输出',
description: '生成最终结果返回给用户',
icon: 'fa-check-circle',
color: 'text-green-400',
bgColor: 'from-green-500/20 to-green-500/5',
borderColor: 'border-green-500/30',
status: 'pending',
expanded: false,
children: [
{
id: 'generate',
name: '方案生成',
description: '基于分析结果构建解决方案',
icon: 'fa-lightbulb',
color: 'text-primary-orange',
bgColor: 'from-primary-orange/20 to-primary-orange/5',
borderColor: 'border-primary-orange/30',
status: 'pending',
expanded: false,
children: [
{
id: 'analyze',
name: '问题分析',
description: '拆解问题、提取关键信息、理解用户意图',
icon: 'fa-microscope',
color: 'text-purple-400',
bgColor: 'from-purple-500/20 to-purple-500/5',
borderColor: 'border-purple-500/30',
status: 'pending',
expanded: false,
children: [
{
id: 'question',
name: '用户问题',
description: '输入原始需求和问题描述',
icon: 'fa-question',
color: 'text-primary-cyan',
bgColor: 'from-primary-cyan/20 to-primary-cyan/5',
borderColor: 'border-primary-cyan/30',
status: 'pending',
expanded: false,
}
]
},
{
id: 'data',
name: '数据获取',
description: '从知识库、向量数据库获取相关信息',
icon: 'fa-database',
color: 'text-green-400',
bgColor: 'from-green-500/20 to-green-500/5',
borderColor: 'border-green-500/30',
status: 'pending',
expanded: false,
}
]
},
{
id: 'evaluate',
name: '方案评估',
description: '验证方案可行性和效果',
icon: 'fa-clipboard-check',
color: 'text-yellow-400',
bgColor: 'from-yellow-500/20 to-yellow-500/5',
borderColor: 'border-yellow-500/30',
status: 'pending',
expanded: false,
}
]
}
])
// /
const toggleNode = (node: WorkflowNode) => {
node.expanded = !node.expanded
}
//
const renderWorkflow = (nodes: WorkflowNode[], level: number = 0) => {
return nodes
}
const openCreate = () => {
newSkillForm.value = {
name: '',
@@ -613,15 +151,6 @@ const saveNewSkill = () => {
})
isCreating.value = false
}
const statusClass = (status: string) => {
switch (status) {
case 'running': return 'bg-primary-success'
case 'stopped': return 'bg-gray-500'
case 'error': return 'bg-primary-danger'
default: return 'bg-gray-500'
}
}
</script>
<style scoped>
@@ -1053,10 +582,10 @@ const statusClass = (status: string) => {
<span class="font-semibold text-white text-sm">Explorer</span>
</button>
<div class="flex items-center gap-1">
<button @click="showNewFolderInput" class="p-1.5 text-gray-500 hover:text-white hover:bg-dark-700 rounded transition-colors" title="New Folder">
<button @click="showInlineNewFolderInput(null)" class="p-1.5 text-gray-500 hover:text-white hover:bg-dark-700 rounded transition-colors" title="New Folder">
<i class="fa-solid fa-folder-plus text-sm"></i>
</button>
<button @click="showNewFileInput(null)" class="p-1.5 text-gray-500 hover:text-white hover:bg-dark-700 rounded transition-colors" title="New File">
<button @click="showInlineNewFileInput(null)" class="p-1.5 text-gray-500 hover:text-white hover:bg-dark-700 rounded transition-colors" title="New File">
<i class="fa-solid fa-file-circle-plus text-sm"></i>
</button>
<label class="p-1.5 text-gray-500 hover:text-white hover:bg-dark-700 rounded transition-colors cursor-pointer" title="Upload File">
@@ -1090,6 +619,50 @@ const statusClass = (status: string) => {
<i class="fa-solid fa-trash text-xs"></i>
</button>
</div>
<!-- 根目录新建文件夹/文件按钮 -->
<div v-if="inlineCreatingFolderId === 'root'" class="flex items-center gap-2 px-2 py-1">
<i class="fa-solid fa-folder text-yellow-400 text-sm"></i>
<input
v-model="inlineNewFolderName"
type="text"
placeholder="Folder name"
class="flex-1 bg-dark-700 border border-dark-500 rounded px-2 py-0.5 text-sm text-white placeholder-gray-500 focus:outline-none focus:border-primary-cyan"
@keyup.enter="confirmInlineCreateFolder"
@blur="confirmInlineCreateFolder"
@keyup.escape="cancelInlineCreateFolder"
autofocus
>
</div>
<div v-else-if="inlineCreatingFileId === 'root'" class="flex items-center gap-2 px-2 py-1">
<i class="fa-solid fa-file text-gray-400 text-sm"></i>
<input
v-model="inlineNewFileName"
type="text"
placeholder="File name"
class="flex-1 bg-dark-700 border border-dark-500 rounded px-2 py-0.5 text-sm text-white placeholder-gray-500 focus:outline-none focus:border-primary-cyan"
@keyup.enter="confirmInlineCreateFile"
@blur="confirmInlineCreateFile"
@keyup.escape="cancelInlineCreateFile"
autofocus
>
</div>
<div v-else class="flex items-center gap-2 px-2 py-1">
<button
@click="showInlineNewFolderInput(null)"
class="flex items-center gap-1 text-xs text-gray-500 hover:text-white transition-colors"
>
<i class="fa-solid fa-folder-plus"></i>
<span>New Folder</span>
</button>
<button
@click="showInlineNewFileInput(null)"
class="flex items-center gap-1 text-xs text-gray-500 hover:text-white transition-colors"
>
<i class="fa-solid fa-file-circle-plus"></i>
<span>New File</span>
</button>
</div>
</div>
<!-- 文件夹列表 -->
@@ -1130,7 +703,7 @@ const statusClass = (status: string) => {
</div>
<!-- 文件夹内的文件 -->
<div v-if="folder.expanded" class="ml-4">
<div v-if="folder.expanded || selectedFolder === folder.id" class="ml-4">
<div
v-for="child in folder.children"
:key="child.id"
@@ -1150,25 +723,54 @@ const statusClass = (status: string) => {
<i class="fa-solid fa-trash text-xs"></i>
</button>
</div>
<!-- 文件夹内新建文件夹/文件/上传按钮 -->
<div class="flex items-center gap-2 px-2 py-1">
<!-- 文件夹内新建文件夹/文件按钮或内联输入框 -->
<div v-if="inlineCreatingFolderId === folder.id" class="flex items-center gap-2 px-2 py-1">
<i class="fa-solid fa-folder text-yellow-400 text-xs"></i>
<input
v-model="inlineNewFolderName"
type="text"
placeholder="Folder name"
class="flex-1 bg-dark-700 border border-dark-500 rounded px-2 py-0.5 text-xs text-white placeholder-gray-500 focus:outline-none focus:border-primary-cyan"
@keyup.enter="confirmInlineCreateFolder"
@blur="confirmInlineCreateFolder"
@keyup.escape="cancelInlineCreateFolder"
autofocus
>
</div>
<div v-else-if="inlineCreatingFileId === folder.id" class="flex items-center gap-2 px-2 py-1">
<i class="fa-solid fa-file text-gray-400 text-xs"></i>
<input
v-model="inlineNewFileName"
type="text"
placeholder="File name"
class="flex-1 bg-dark-700 border border-dark-500 rounded px-2 py-0.5 text-xs text-white placeholder-gray-500 focus:outline-none focus:border-primary-cyan"
@keyup.enter="confirmInlineCreateFile"
@blur="confirmInlineCreateFile"
@keyup.escape="cancelInlineCreateFile"
autofocus
>
</div>
<div v-else class="flex items-center gap-2 px-2 py-1">
<button
@click.stop="showNewFolderInput(folder.id)"
class="flex items-center gap-1 text-xs text-gray-500 hover:text-white transition-colors opacity-0 group-hover:opacity-100"
@click.stop="showInlineNewFolderInput(folder.id)"
class="flex items-center gap-1 text-xs text-gray-500 hover:text-white transition-colors"
:class="selectedFolder === folder.id ? '' : 'opacity-0 group-hover:opacity-100'"
title="New Folder"
>
<i class="fa-solid fa-folder-plus"></i>
<span>New Folder</span>
</button>
<button
@click.stop="showNewFileInput(folder.id)"
class="flex items-center gap-1 text-xs text-gray-500 hover:text-white transition-colors opacity-0 group-hover:opacity-100"
@click.stop="showInlineNewFileInput(folder.id)"
class="flex items-center gap-1 text-xs text-gray-500 hover:text-white transition-colors"
:class="selectedFolder === folder.id ? '' : 'opacity-0 group-hover:opacity-100'"
>
<i class="fa-solid fa-plus"></i>
<span>New File</span>
</button>
<label
class="flex items-center gap-1 text-xs text-gray-500 hover:text-white transition-colors cursor-pointer opacity-0 group-hover:opacity-100"
class="flex items-center gap-1 text-xs text-gray-500 hover:text-white transition-colors cursor-pointer"
:class="selectedFolder === folder.id ? '' : 'opacity-0 group-hover:opacity-100'"
title="Upload File"
>
<i class="fa-solid fa-upload"></i>
@@ -1180,38 +782,6 @@ const statusClass = (status: string) => {
</div>
</div>
<!-- 新建文件夹输入框 -->
<div v-if="isCreatingFolder" class="p-2 border-t border-dark-600 bg-dark-800">
<div class="flex items-center gap-2">
<i class="fa-solid fa-folder text-yellow-400 text-sm"></i>
<input
v-model="newFolderName"
type="text"
placeholder="Folder name..."
class="flex-1 bg-dark-700 border border-dark-500 rounded px-2 py-1 text-sm text-white placeholder-gray-500 focus:outline-none focus:border-primary-orange"
@keyup.enter="createFolder"
@blur="createFolder"
autofocus
>
</div>
</div>
<!-- 新建文件输入框 -->
<div v-if="isCreatingFile" class="p-2 border-t border-dark-600 bg-dark-800">
<div class="flex items-center gap-2">
<i class="fa-solid fa-file text-gray-400 text-sm"></i>
<input
v-model="newFileName"
type="text"
placeholder="File name..."
class="flex-1 bg-dark-700 border border-dark-500 rounded px-2 py-1 text-sm text-white placeholder-gray-500 focus:outline-none focus:border-primary-orange"
@keyup.enter="createFile"
@blur="createFile"
autofocus
>
</div>
</div>
<!-- 底部操作 -->
<div class="p-3 border-t border-dark-600">
<button

9
web/src/views/Team.vue Normal file
View File

@@ -0,0 +1,9 @@
<script setup lang="ts">
</script>
<template>
<div class="p-6">
<h1 class="text-2xl font-semibold text-white mb-4">Team</h1>
<p class="text-gray-400">Team management page - Coming soon</p>
</div>
</template>

View File

@@ -95,6 +95,27 @@
border: 1px solid #374151;
}
/* 文件上传弹窗特殊样式 - 覆盖 kb-dialog */
.file-upload-dialog.kb-dialog :deep(.el-dialog) {
background-color: #1f2937;
border-radius: 12px;
border: 1px solid #374151;
}
.file-upload-dialog.kb-dialog :deep(.el-dialog__header) {
padding: 16px 24px;
border-bottom: 1px solid #374151;
}
.file-upload-dialog.kb-dialog :deep(.el-dialog__title) {
display: none;
}
.file-upload-dialog.kb-dialog :deep(.el-dialog__body) {
padding: 0;
background-color: #0f1419;
}
.kb-dialog :deep(.el-dialog__header) {
padding: 20px 24px;
border-bottom: 1px solid #374151;
@@ -575,23 +596,27 @@
/* 文件上传弹窗布局 */
.file-upload-dialog .el-dialog {
position: absolute !important;
top: 8px !important;
left: 50% !important;
transform: translateX(-50%) !important;
margin-bottom: 8px !important;
max-height: calc(100vh - 16px);
margin: auto !important;
top: 20px !important;
bottom: 20px !important;
height: auto !important;
max-height: calc(100vh - 40px);
overflow: hidden;
}
.file-upload-dialog .el-dialog__body {
padding: 0;
height: calc(100vh - 80px);
max-height: calc(100vh - 80px);
overflow: hidden;
}
.file-upload-layout {
display: flex;
flex-direction: column;
height: 85vh;
height: 100%;
background-color: #0f1419;
overflow: hidden;
}
/* 顶部导航 */
@@ -669,6 +694,26 @@
padding: 12px;
}
.loading-state,
.empty-state {
display: flex;
flex-direction: column;
align-items: center;
justify-content: center;
padding: 40px 20px;
color: #6b7280;
gap: 12px;
}
.loading-state i,
.empty-state i {
font-size: 32px;
}
.empty-state i {
color: #4b5563;
}
.file-item {
display: flex;
align-items: center;
@@ -744,10 +789,41 @@
color: #ef4444;
}
.file-item-status .status-text {
font-size: 12px;
color: #ef4444;
}
.file-item-status.parsing i {
color: #f59e0b;
}
.file-item-delete {
flex-shrink: 0;
width: 24px;
height: 24px;
border-radius: 4px;
border: none;
background: transparent;
color: #6b7280;
cursor: pointer;
display: flex;
align-items: center;
justify-content: center;
opacity: 0;
transition: all 0.2s;
margin-left: 4px;
}
.file-item:hover .file-item-delete {
opacity: 1;
}
.file-item-delete:hover {
background-color: rgba(239, 68, 68, 0.15);
color: #ef4444;
}
/* 右侧预览面板 */
.file-preview {
flex: 1;
@@ -756,6 +832,47 @@
background-color: #121218;
}
.preview-empty {
flex: 1;
display: flex;
flex-direction: column;
align-items: center;
justify-content: center;
color: #6b7280;
gap: 12px;
}
.preview-empty i {
font-size: 48px;
color: #4b5563;
}
.preview-empty span {
font-size: 14px;
}
.preview-loading,
.preview-no-file {
flex: 1;
display: flex;
flex-direction: column;
align-items: center;
justify-content: center;
color: #6b7280;
gap: 12px;
}
.preview-loading i,
.preview-no-file i {
font-size: 48px;
color: #4b5563;
}
.preview-loading span,
.preview-no-file span {
font-size: 14px;
}
.preview-header {
display: flex;
align-items: center;
@@ -817,7 +934,8 @@
background-color: #1a1a1a;
}
.pdf-iframe {
.pdf-iframe,
.pdf-embed {
width: 100%;
height: 100%;
border: none;

View File

@@ -0,0 +1,200 @@
// Knowledge Base API
const API_BASE = 'http://localhost:8082'
export interface KnowledgeBase {
id: string
name: string
description: string
llm_model_id: string
embedding_model_id: string
parsing_config: {
engine: string
docling_url?: string
enable_pdf: boolean
pandoc: boolean
}
status: string
document_count: number
chunk_count: number
created_at: string
updated_at: string
}
export interface KnowledgeDocument {
id: string
knowledge_base_id: string
name: string
file_key?: string
file_url?: string
file_size: number
status: string
chunk_count: number
uploaded_at: string
}
// 获取知识库列表
export const fetchKnowledgeBases = async (): Promise<KnowledgeBase[]> => {
try {
const response = await fetch(`${API_BASE}/api/knowledge/list`)
if (!response.ok) {
console.error('Server error:', response.status)
return []
}
const text = await response.text()
if (!text) {
return []
}
try {
const data = JSON.parse(text)
if (data.success) {
return data.data || []
}
console.error('API error:', data.message)
return []
} catch (parseError) {
console.error('Failed to parse JSON:', text)
return []
}
} catch (error) {
console.error('Failed to fetch knowledge bases:', error)
return []
}
}
// 创建知识库
export const createKnowledgeBase = async (params: {
name: string
description?: string
llm_model_id: string
embedding_model_id: string
parsing_config: {
engine: string
docling_url?: string
enable_pdf?: boolean
pandoc?: boolean
}
}): Promise<{ success: boolean; id?: string; message?: string }> => {
try {
const response = await fetch(`${API_BASE}/api/knowledge/create`, {
method: 'POST',
headers: {
'Content-Type': 'application/json',
},
body: JSON.stringify(params),
})
// Check if response is OK
if (!response.ok) {
const errorText = await response.text()
return { success: false, message: `Server error: ${response.status} - ${errorText}` }
}
const text = await response.text()
if (!text) {
return { success: false, message: 'Empty response from server' }
}
try {
const data = JSON.parse(text)
return data
} catch (parseError) {
console.error('Failed to parse JSON:', text)
return { success: false, message: `Invalid JSON response: ${text}` }
}
} catch (error) {
console.error('Failed to create knowledge base:', error)
return { success: false, message: `Network error: ${error}` }
}
}
// 删除知识库
export const deleteKnowledgeBase = async (id: string): Promise<{ success: boolean; message?: string }> => {
try {
const response = await fetch(`${API_BASE}/api/knowledge/${id}`, {
method: 'DELETE',
})
const data = await response.json()
return data
} catch (error) {
console.error('Failed to delete knowledge base:', error)
return { success: false, message: 'Failed to delete knowledge base' }
}
}
// 获取知识库下的文档列表
export const fetchKnowledgeDocuments = async (kbId: string, status?: string): Promise<KnowledgeDocument[]> => {
try {
const query = status && status !== 'all' ? `?status=${status}` : ''
const response = await fetch(`${API_BASE}/api/knowledge/${kbId}/documents${query}`)
const data = await response.json()
if (data.success) {
return data.data
}
throw new Error(data.message || 'Failed to fetch documents')
} catch (error) {
console.error('Failed to fetch documents:', error)
return []
}
}
// 上传文档到知识库
export const uploadDocument = async (kbId: string, file: File): Promise<{ success: boolean; id?: string; url?: string; document?: KnowledgeDocument; message?: string }> => {
try {
const formData = new FormData()
formData.append('file', file)
const response = await fetch(`${API_BASE}/api/knowledge/${kbId}/documents`, {
method: 'POST',
body: formData,
})
const data = await response.json()
return data
} catch (error) {
console.error('Failed to upload document:', error)
return { success: false, message: 'Failed to upload document' }
}
}
// 删除知识库文档
export const deleteDocument = async (kbId: string, docId: string): Promise<{ success: boolean; message?: string }> => {
try {
const response = await fetch(`${API_BASE}/api/knowledge/${kbId}/documents/${docId}`, {
method: 'DELETE',
})
const data = await response.json()
return data
} catch (error) {
console.error('Failed to delete document:', error)
return { success: false, message: 'Failed to delete document' }
}
}
// 重新解析文档
export const reparseDocument = async (kbId: string, docId: string): Promise<{ success: boolean; message?: string }> => {
try {
const response = await fetch(`${API_BASE}/api/knowledge/${kbId}/documents/${docId}/reparse`, {
method: 'POST',
})
const data = await response.json()
return data
} catch (error) {
console.error('Failed to reparse document:', error)
return { success: false, message: 'Failed to reparse document' }
}
}
// 获取文档预览内容
export const getDocumentPreview = async (kbId: string, docId: string, page: number = 1): Promise<{ success: boolean; data?: { total_pages: number; current_page: number; content: string }; message?: string }> => {
try {
const response = await fetch(`${API_BASE}/api/knowledge/${kbId}/documents/${docId}/preview?page=${page}`)
const data = await response.json()
return data
} catch (error) {
console.error('Failed to get document preview:', error)
return { success: false, message: 'Failed to get document preview' }
}
}

View File

@@ -0,0 +1,46 @@
// Skill page types
export interface MCPServer {
id: number
name: string
status: 'running' | 'stopped' | 'error'
type: string
port: number
createdAt: string
description: string
}
export interface FileItem {
id: string
name: string
icon?: string
content?: string
type: 'file'
}
export interface FolderItem {
id: string
name: string
icon?: string
type: 'folder'
expanded?: boolean
children: (FileItem | FolderItem)[]
}
export interface WorkflowNode {
id: string
name: string
description: string
icon: string
color: string
bgColor: string
borderColor: string
status: 'pending' | 'processing' | 'completed'
expanded: boolean
children?: WorkflowNode[]
}
export interface ChatMessage {
id: number
text: string
}

View File

@@ -0,0 +1,340 @@
import { ref } from 'vue'
import type { FileItem, FolderItem } from './types'
// 默认的 SKILL.md 内容
const DEFAULT_SKILL_CONTENT = `# Skill Configuration
## Overview
This skill is configured to process data and generate outputs.
## Dependencies
- Python 3.8+
- Node.js 16+
`
export function useFileTree() {
// 文件树数据
const fileTree = ref<FolderItem[]>([
{
id: 'folder-1',
name: 'Data',
icon: 'fa-folder',
type: 'folder',
expanded: true,
children: [
{ id: 'file-1', name: 'user_data.csv', icon: 'fa-file', content: 'id,name,email,age\n1,John Doe,john@example.com,28\n2,Jane Smith,jane@example.com,32', type: 'file' },
{ id: 'file-2', name: 'config.json', icon: 'fa-code', content: '{\n "appName": "MyApp",\n "version": "1.0.0",\n "debug": true\n}', type: 'file' },
]
},
{
id: 'folder-2',
name: 'Script',
icon: 'fa-folder',
type: 'folder',
expanded: true,
children: [
{ id: 'file-3', name: 'process.py', icon: 'fa-code', content: 'import pandas as pd\n\ndef process_data(data):\n """Process the input data"""\n return data.apply(lambda x: x.strip())\n\nif __name__ == "__main__":\n df = pd.read_csv("user_data.csv")\n processed = process_data(df)\n processed.to_csv("output.csv", index=False)', type: 'file' },
{ id: 'file-4', name: 'transform.js', icon: 'fa-code', content: 'function transformData(input) {\n return input.map(item => ({\n ...item,\n processed: true,\n timestamp: Date.now()\n }));\n}\n\nmodule.exports = { transformData };', type: 'file' },
]
},
{
id: 'folder-3',
name: 'Reference',
icon: 'fa-folder',
type: 'folder',
expanded: true,
children: [
{ id: 'file-5', name: 'api_docs.md', icon: 'fa-code', content: '# API Documentation\n\n## Endpoints\n\n### GET /api/users\nReturns list of users.\n\n### POST /api/users\nCreate a new user.', type: 'file' },
]
},
])
// 根目录文件
const rootFiles = ref<FileItem[]>([
{ id: 'file-root-1', name: 'SKILL.md', icon: 'fa-file', content: DEFAULT_SKILL_CONTENT, type: 'file' },
])
// 展开/收起文件夹
const toggleFolder = (folder: FolderItem) => {
folder.expanded = !folder.expanded
}
// 新建文件夹 - 内联输入模式
const inlineNewFolderName = ref('')
const inlineCreatingFolderId = ref<string | null>(null)
const showInlineNewFolderInput = (parentId: string | null = null) => {
inlineNewFolderName.value = ''
inlineCreatingFolderId.value = parentId === null ? 'root' : parentId
}
const confirmInlineCreateFolder = () => {
if (!inlineNewFolderName.value.trim()) {
inlineCreatingFolderId.value = null
return
}
const newFolder: FolderItem = {
id: 'folder-' + Date.now(),
name: inlineNewFolderName.value,
icon: 'fa-folder',
type: 'folder',
expanded: true,
children: []
}
if (inlineCreatingFolderId.value && inlineCreatingFolderId.value !== 'root') {
const parentFolder = findFolderById(fileTree.value, inlineCreatingFolderId.value)
if (parentFolder && parentFolder.children) {
parentFolder.children.push(newFolder)
// 选中新创建的文件夹
selectedFolder.value = newFolder.id
parentFolder.expanded = true
}
} else {
fileTree.value.push(newFolder)
// 选中新创建的文件夹
selectedFolder.value = newFolder.id
}
inlineCreatingFolderId.value = null
inlineNewFolderName.value = ''
}
const cancelInlineCreateFolder = () => {
inlineCreatingFolderId.value = null
inlineNewFolderName.value = ''
// 恢复选中之前选中的文件夹
if (selectedFolder.value) {
const folder = findFolderById(fileTree.value, selectedFolder.value)
if (folder) {
folder.expanded = true
}
}
}
// 递归查找文件夹
const findFolderById = (folders: (FileItem | FolderItem)[], id: string): FolderItem | null => {
for (const folder of folders) {
if ('children' in folder && folder.id === id) return folder as FolderItem
if ('children' in folder && folder.children) {
const found = findFolderById(folder.children, id)
if (found) return found
}
}
return null
}
// 新建文件 - 内联输入模式
const inlineNewFileName = ref('')
const inlineCreatingFileId = ref<string | null>(null)
const showInlineNewFileInput = (folderId: string | null = null) => {
inlineNewFileName.value = ''
inlineCreatingFileId.value = folderId === null ? 'root' : folderId
}
const confirmInlineCreateFile = () => {
if (!inlineNewFileName.value.trim()) {
inlineCreatingFileId.value = null
return
}
const newFile: FileItem = {
id: 'file-' + Date.now(),
name: inlineNewFileName.value,
icon: getFileIconByName(inlineNewFileName.value),
content: '',
type: 'file'
}
if (inlineCreatingFileId.value && inlineCreatingFileId.value !== 'root') {
const folder = findFolderById(fileTree.value, inlineCreatingFileId.value)
if (folder && folder.children) {
folder.children.push(newFile)
// 选中父文件夹
selectedFolder.value = folder.id
folder.expanded = true
}
} else {
rootFiles.value.push(newFile)
}
inlineCreatingFileId.value = null
inlineNewFileName.value = ''
}
const cancelInlineCreateFile = () => {
inlineCreatingFileId.value = null
inlineNewFileName.value = ''
// 恢复选中之前选中的文件夹
if (selectedFolder.value) {
const folder = findFolderById(fileTree.value, selectedFolder.value)
if (folder) {
folder.expanded = true
}
}
}
// 根据文件名获取图标
const getFileIconByName = (filename: string): string => {
const ext = filename.split('.').pop()?.toLowerCase()
const codeExtensions = ['js', 'ts', 'py', 'html', 'css', 'json', 'xml', 'yaml', 'yml', 'java', 'c', 'cpp', 'go', 'rs', 'php', 'rb', 'swift', 'kt', 'sql', 'sh', 'bash', 'md']
if (codeExtensions.includes(ext || '')) {
return 'fa-code'
}
return 'fa-file'
}
// 上传文件
const handleFileUpload = (event: Event, folderId: string | null = null) => {
const input = event.target as HTMLInputElement
if (!input.files?.length) return
Array.from(input.files).forEach(file => {
const reader = new FileReader()
reader.onload = (e) => {
const newFile: FileItem = {
id: 'file-' + Date.now() + Math.random(),
name: file.name,
icon: getFileIconByName(file.name),
content: e.target?.result as string || '',
type: 'file'
}
if (folderId) {
const folder = findFolderById(fileTree.value, folderId)
if (folder && folder.children) {
folder.children.push(newFile)
}
} else {
rootFiles.value.push(newFile)
}
}
reader.readAsText(file)
})
input.value = ''
}
// 递归查找文件
const findFile = (items: (FileItem | FolderItem)[], fileId: string): FileItem | null => {
for (const item of items) {
if (item.type === 'file' && item.id === fileId) {
return item
}
if (item.type === 'folder' && item.children) {
const found = findFile(item.children, fileId)
if (found) return found
}
}
return null
}
// 文件编辑相关状态
const isEditingFile = ref(false)
const isEditingFileClosing = ref(false)
const editingFile = ref<FileItem | null>(null)
const editingFileName = ref('')
const editingFileContent = ref('')
const editingFileType = ref('')
// 选中文件/文件夹状态
const hoveringItem = ref<string | null>(null)
const selectedFolder = ref<string | null>(null)
// 选中文件夹
const selectFolder = (folderId: string) => {
if (selectedFolder.value === folderId) {
selectedFolder.value = null
} else {
selectedFolder.value = folderId
const folder = findFolderById(fileTree.value, folderId)
if (folder) {
folder.expanded = true
}
}
}
// 删除文件
const deleteFile = (file: FileItem, parentId: string | null) => {
if (!parentId) {
rootFiles.value = rootFiles.value.filter(f => f.id !== file.id)
} else {
const folder = fileTree.value.find(f => f.id === parentId)
if (folder && folder.children) {
folder.children = folder.children.filter(c => (c as FileItem).id !== file.id)
}
}
}
// 删除文件夹
const deleteFolder = (folder: FolderItem) => {
fileTree.value = fileTree.value.filter(f => f.id !== folder.id)
}
// 打开文件编辑器
const openFile = (file: FileItem) => {
editingFile.value = file
editingFileName.value = file.name
editingFileContent.value = file.content || ''
editingFileType.value = 'file'
isEditingFile.value = true
}
const saveFileEdit = () => {
if (!editingFile.value) return
editingFile.value.content = editingFileContent.value
isEditingFileClosing.value = true
setTimeout(() => {
isEditingFile.value = false
isEditingFileClosing.value = false
editingFile.value = null
}, 300)
}
const cancelFileEdit = () => {
isEditingFileClosing.value = true
setTimeout(() => {
isEditingFile.value = false
isEditingFileClosing.value = false
editingFile.value = null
}, 300)
}
return {
// State
fileTree,
rootFiles,
inlineNewFolderName,
inlineCreatingFolderId,
inlineNewFileName,
inlineCreatingFileId,
isEditingFile,
isEditingFileClosing,
editingFile,
editingFileName,
editingFileContent,
editingFileType,
hoveringItem,
selectedFolder,
// Methods
toggleFolder,
showInlineNewFolderInput,
confirmInlineCreateFolder,
cancelInlineCreateFolder,
findFolderById,
showInlineNewFileInput,
confirmInlineCreateFile,
cancelInlineCreateFile,
getFileIconByName,
handleFileUpload,
findFile,
selectFolder,
deleteFile,
deleteFolder,
openFile,
saveFileEdit,
cancelFileEdit,
}
}

View File

@@ -0,0 +1,46 @@
import { ref } from 'vue'
import type { ChatMessage } from './types'
export function useSkillChat() {
// AI 对话面板
const isChatOpen = ref(false)
// 对话相关
const chatInput = ref('')
const chatMessages = ref<ChatMessage[]>([])
const toggleChat = () => {
isChatOpen.value = !isChatOpen.value
}
const sendMessage = () => {
if (!chatInput.value.trim()) return
chatMessages.value.push({
id: Date.now(),
text: chatInput.value
})
const userInput = chatInput.value
chatInput.value = ''
// 模拟 AI 回复
setTimeout(() => {
chatMessages.value.push({
id: Date.now(),
text: `I've updated the configuration based on your request: "${userInput}". The changes have been applied to your skill settings.`
})
}, 1000)
}
return {
// State
isChatOpen,
chatInput,
chatMessages,
// Methods
toggleChat,
sendMessage,
}
}

View File

@@ -0,0 +1,114 @@
import { ref } from 'vue'
import type { MCPServer } from './types'
export function useSkillServers() {
// MCP Server 列表
const mcpServers = ref<MCPServer[]>([
{ id: 1, name: 'linear-demo', status: 'running', type: 'Linear', port: 3001, createdAt: '2025-04-10', description: 'Linear API integration for project management' },
{ id: 2, name: 'google-maps', status: 'running', type: 'Google Maps', port: 3002, createdAt: '2025-04-08', description: 'Google Maps API for location services' },
{ id: 3, name: 'explorer-mcp', status: 'error', type: 'File System', port: 3003, createdAt: '2025-04-05', description: 'File system explorer and editor' },
{ id: 4, name: 'postgres-mcp', status: 'running', type: 'PostgreSQL', port: 3004, createdAt: '2025-04-12', description: 'PostgreSQL database operations' },
{ id: 5, name: 'github-mcp', status: 'stopped', type: 'GitHub', port: 3005, createdAt: '2025-04-11', description: 'GitHub API integration' },
])
// 编辑状态
const editingServer = ref<MCPServer | null>(null)
const isEditing = ref(false)
// 搜索和筛选
const searchQuery = ref('')
const filterStatus = ref<string>('all')
// 编辑表单
const editForm = ref({
name: '',
type: '',
port: 3000,
description: '',
})
// 打开编辑弹窗
const openEdit = (server: MCPServer) => {
editingServer.value = server
editForm.value = {
name: server.name,
type: server.type,
port: server.port,
description: server.description,
}
isEditing.value = true
}
// 保存编辑
const saveEdit = () => {
if (editingServer.value) {
const index = mcpServers.value.findIndex(s => s.id === editingServer.value!.id)
if (index !== -1) {
mcpServers.value[index] = {
...mcpServers.value[index],
...editForm.value,
}
}
}
isEditing.value = false
}
// 取消编辑
const cancelEdit = () => {
isEditing.value = false
editingServer.value = null
}
// 切换状态
const toggleStatus = (server: MCPServer) => {
if (server.status === 'running') {
server.status = 'stopped'
} else if (server.status === 'stopped') {
server.status = 'running'
}
}
// 删除服务器
const deleteServer = (id: number) => {
mcpServers.value = mcpServers.value.filter(s => s.id !== id)
}
// 筛选服务器
const filteredServers = () => {
return mcpServers.value.filter(server => {
const matchSearch = server.name.toLowerCase().includes(searchQuery.value.toLowerCase()) ||
server.type.toLowerCase().includes(searchQuery.value.toLowerCase())
const matchStatus = filterStatus.value === 'all' || server.status === filterStatus.value
return matchSearch && matchStatus
})
}
// 状态样式
const statusClass = (status: string) => {
switch (status) {
case 'running': return 'bg-primary-success'
case 'stopped': return 'bg-gray-500'
case 'error': return 'bg-primary-danger'
default: return 'bg-gray-500'
}
}
return {
// State
mcpServers,
editingServer,
isEditing,
searchQuery,
filterStatus,
editForm,
// Methods
openEdit,
saveEdit,
cancelEdit,
toggleStatus,
deleteServer,
filteredServers,
statusClass,
}
}

View File

@@ -0,0 +1,136 @@
import { ref } from 'vue'
interface WorkflowNode {
id: string
name: string
description: string
icon: string
color: string
bgColor: string
borderColor: string
status: 'pending' | 'processing' | 'completed'
expanded: boolean
children?: WorkflowNode[]
}
export function useWorkflow() {
// 流程生成相关
const isGeneratingGraph = ref(false)
const isGraphGenerated = ref(false)
const workflowData = ref<WorkflowNode[]>([
{
id: 'output',
name: '最终输出',
description: '生成最终结果返回给用户',
icon: 'fa-check-circle',
color: 'text-green-400',
bgColor: 'from-green-500/20 to-green-500/5',
borderColor: 'border-green-500/30',
status: 'pending',
expanded: false,
children: [
{
id: 'generate',
name: '方案生成',
description: '基于分析结果构建解决方案',
icon: 'fa-lightbulb',
color: 'text-primary-orange',
bgColor: 'from-primary-orange/20 to-primary-orange/5',
borderColor: 'border-primary-orange/30',
status: 'pending',
expanded: false,
children: [
{
id: 'analyze',
name: '问题分析',
description: '拆解问题、提取关键信息、理解用户意图',
icon: 'fa-microscope',
color: 'text-purple-400',
bgColor: 'from-purple-500/20 to-purple-500/5',
borderColor: 'border-purple-500/30',
status: 'pending',
expanded: false,
children: [
{
id: 'question',
name: '用户问题',
description: '输入原始需求和问题描述',
icon: 'fa-question',
color: 'text-primary-cyan',
bgColor: 'from-primary-cyan/20 to-primary-cyan/5',
borderColor: 'border-primary-cyan/30',
status: 'pending',
expanded: false,
}
]
},
{
id: 'data',
name: '数据获取',
description: '从知识库、向量数据库获取相关信息',
icon: 'fa-database',
color: 'text-green-400',
bgColor: 'from-green-500/20 to-green-500/5',
borderColor: 'border-green-500/30',
status: 'pending',
expanded: false,
}
]
},
{
id: 'evaluate',
name: '方案评估',
description: '验证方案可行性和效果',
icon: 'fa-clipboard-check',
color: 'text-yellow-400',
bgColor: 'from-yellow-500/20 to-yellow-500/5',
borderColor: 'border-yellow-500/30',
status: 'pending',
expanded: false,
}
]
}
])
// 展开/收起节点
const toggleNode = (node: WorkflowNode) => {
node.expanded = !node.expanded
}
// 递归渲染节点
const renderWorkflow = (nodes: WorkflowNode[], level: number = 0) => {
return nodes
}
const generateGraph = () => {
isGeneratingGraph.value = true
setTimeout(() => {
isGeneratingGraph.value = false
isGraphGenerated.value = true
if (workflowData.value.length > 0) {
workflowData.value[0].expanded = true
if (workflowData.value[0].children) {
workflowData.value[0].children.forEach(child => {
child.expanded = false
if (child.children) {
child.children.forEach(gc => gc.expanded = false)
}
})
}
}
}, 2500)
}
return {
// State
workflowData,
isGeneratingGraph,
isGraphGenerated,
// Methods
toggleNode,
renderWorkflow,
generateGraph,
}
}

View File

@@ -12,8 +12,8 @@
"noEmit": true,
"jsx": "preserve",
"strict": true,
"noUnusedLocals": true,
"noUnusedParameters": true,
"noUnusedLocals": false,
"noUnusedParameters": false,
"noFallthroughCasesInSwitch": true,
"baseUrl": ".",
"paths": {