ModelKit

ModelKit

ModelKit 是一个强大的AI模型管理平台，支持多种AI服务提供商，提供统一的模型管理、配置验证和API接口服务。

🚀 功能特性

• 多模型提供商支持: 支持 OpenAI、Ollama、DeepSeek、SiliconFlow、Moonshot、Azure OpenAI、百智云、腾讯混元、百炼、火山引擎、Gemini、智谱等主流AI服务商
• 模型类型管理: 支持聊天模型、嵌入模型、重排序模型、视觉模型、代码模型、函数调用等多种模型类型
• 配置验证: 提供模型配置的实时验证功能，确保API配置正确性
• 统一API接口: 提供标准化的RESTful API，简化AI模型集成
• 现代化Web界面: 基于React 19和Material-UI构建的响应式用户界面
• 国际化支持: 内置中英文多语言支持
• 可复用组件: 提供开箱即用的ModelModal组件，支持在其他项目中快速集成

🏗️ 技术架构

后端 (Go)

• 框架: Echo v4 (HTTP框架)
• 语言: Go 1.24.0+
• 架构: 分层架构 (Handler -> UseCase -> Domain)
• 依赖管理: Go Modules

前端 (React)

• 框架: React 19 + TypeScript
• UI库: Material-UI v6 + CT-MUI
• 构建工具: Vite 6
• 状态管理: Redux Toolkit
• 路由: React Router v7
• 编辑器: TipTap (富文本编辑器)

📁 项目结构

ModelKit/
├── consts/           # 常量定义
├── domain/           # 领域模型和接口
├── errcode/          # 错误码和国际化
├── handler/          # HTTP处理器
├── pkg/              # 公共包
├── usecase/          # 业务用例
├── ui/               # 前端应用
│   ├── src/          # 源代码
│   │   ├── components/     # 可复用组件
│   │   │   └── Card/       # 卡片组件
│   │   ├── constant/       # 常量定义
│   │   ├── api/            # API接口
│   │   └── services/       # 服务层
│   ├── public/       # 静态资源
│   └── package.json  # 前端依赖
├── utils/            # 工具函数
├── go.mod            # Go模块文件
└── Makefile          # 构建脚本

🛠️ 安装部署

环境要求

• Go 1.24.0+
• Node.js 18+
• pnpm 10.12.1+

前端部署(开发中)

1. 进入前端目录

cd ui

1. 安装依赖

pnpm install

1. 开发模式运行

pnpm dev

1. 构建生产版本

pnpm build

🧩 组件使用指南

ModelModal 组件

ModelModal 是一个功能完整的AI模型配置组件，支持添加、编辑和测试各种AI模型。该组件可以在其他React项目中复用。

安装依赖

# 必需依赖
npm install @mui/material @emotion/react @emotion/styled
npm install react-hook-form @mui/icons-material

# 可选依赖（用于图标）
npm install @mui/lab

基本使用

import { ModelModal } from './components/ModelModal'
import { ModelService } from './services/ModelService'

function App() {
  const [open, setOpen] = useState(false)
  const [modelData, setModelData] = useState(null)
  
  const modelService = new ModelService()
  
  const handleClose = () => {
    setOpen(false)
    setModelData(null)
  }
  
  const handleRefresh = () => {
    // 刷新模型列表
    console.log('刷新模型列表')
  }
  
  return (
    <div>
      <button onClick={() => setOpen(true)}>添加模型</button>
      
      <ModelModal
        open={open}
        data={modelData}
        type="chat"
        onClose={handleClose}
        refresh={handleRefresh}
        modelService={modelService}
      />
    </div>
  )
}

组件属性

实现ModelService接口

interface ModelService {
  createModel: (data: CreateModelData) => Promise<{ ModelName: string }>
  getModelNameList: (data: GetModelNameData) => Promise<{ models: { model: string }[] }>
  testModel: (data: CheckModelData) => Promise<{ error: string }>
  updateModel: (data: UpdateModelData) => Promise<void>
}

class CustomModelService implements ModelService {
  async createModel(data: CreateModelData) {
    // 实现创建模型的逻辑
    const response = await fetch('/api/models', {
      method: 'POST',
      headers: { 'Content-Type': 'application/json' },
      body: JSON.stringify(data)
    })
    return response.json()
  }
  
  async getModelNameList(data: GetModelNameData) {
    // 实现获取模型列表的逻辑
    const response = await fetch('/api/models/list', {
      method: 'POST',
      headers: { 'Content-Type': 'application/json' },
      body: JSON.stringify(data)
    })
    return response.json()
  }
  
  async testModel(data: CheckModelData) {
    // 实现测试模型的逻辑
    const response = await fetch('/api/models/test', {
      method: 'POST',
      headers: { 'Content-Type': 'application/json' },
      body: JSON.stringify(data)
    })
    return response.json()
  }
  
  async updateModel(data: UpdateModelData) {
    // 实现更新模型的逻辑
    await fetch(`/api/models/${data.id}`, {
      method: 'PUT',
      headers: { 'Content-Type': 'application/json' },
      body: JSON.stringify(data)
    })
  }
}

支持的数据类型

// 模型提供商
type ModelProvider = 'BaiZhiCloud' | 'DeepSeek' | 'Hunyuan' | 'BaiLian' | 
                    'Volcengine' | 'OpenAI' | 'Ollama' | 'SiliconFlow' | 
                    'Moonshot' | 'AzureOpenAI' | 'Gemini' | 'ZhiPu' | 'Other'

// 模型类型
type ModelType = 'chat' | 'embedding' | 'rerank'

// 表单数据
interface AddModelForm {
  provider: keyof typeof ModelProvider
  model: string
  base_url: string
  api_version: string
  api_key: string
  api_header_key: string
  api_header_value: string
  type: ModelType
}

自定义样式

组件使用Material-UI主题系统，可以通过主题配置自定义样式：

import { createTheme, ThemeProvider } from '@mui/material/styles'

const theme = createTheme({
  palette: {
    primary: {
      main: '#1976d2',
    },
    secondary: {
      main: '#dc004e',
    },
  },
})

function App() {
  return (
    <ThemeProvider theme={theme}>
      <ModelModal {...props} />
    </ThemeProvider>
  )
}

完整示例项目

查看 examples/model-modal-demo 目录获取完整的集成示例。

📚 API文档

获取模型列表

GET /api/v1/modelkit/models

查询参数:

• provider: 模型提供商 (必需)
• base_url: 基础URL (必需)
• api_key: API密钥 (可选)
• api_header: API头部 (可选)
• type: 模型类型 (必需)

响应示例:

{
  "success": true,
  "message": "获取模型列表成功",
  "data": {
    "models": [
      {
        "id": "gpt-4",
        "object": "model",
        "owned_by": "OpenAI",
        "model_type": "chat"
      }
    ]
  }
}

检查模型配置

POST /api/v1/modelkit/check

请求体:

{
  "provider": "OpenAI",
  "model": "gpt-4",
  "base_url": "https://api.openai.com",
  "api_key": "sk-xxx",
  "type": "chat"
}

响应示例:

{
  "success": true,
  "message": "模型配置验证成功",
  "data": {
    "error": "",
    "content": "配置验证通过"
  }
}

🔧 配置说明

支持的模型提供商

• OpenAI: GPT系列模型
• Ollama: 本地部署模型
• DeepSeek: DeepSeek系列模型
• SiliconFlow: SiliconFlow模型
• Moonshot: Moonshot系列模型
• Azure OpenAI: Azure OpenAI服务
• 百智云: 通义千问等模型
• 腾讯混元: 混元系列模型
• 百炼: 百炼系列模型
• 火山引擎: 火山引擎模型
• Gemini: Google Gemini模型
• 智谱: 智谱AI模型

支持的模型类型

• chat: 聊天模型
• embedding: 嵌入模型
• rerank: 重排序模型
• vision: 视觉模型
• coder: 代码模型
• functioncall: 函数调用模型

📝 许可证

本项目采用 MIT 许可证。

🤝 贡献指南

欢迎提交 Issue 和 Pull Request！

1. Fork 本仓库
2. 创建特性分支 (git checkout -b feature/AmazingFeature)
3. 提交更改 (git commit -m 'Add some AmazingFeature')
4. 推送到分支 (git push origin feature/AmazingFeature)
5. 打开 Pull Request

📞 联系我们

• 项目主页: https://github.com/chaitin/ModelKit
• 问题反馈: Issues

🙏 致谢

感谢所有为这个项目做出贡献的开发者和用户！