LiuVaayne
2e31a5bbcb
* feat: add api server
This reverts commit c76aa03566.
* update yarn.lock
* fix: correct import paths in ToolSettings component
Update import paths for PreprocessSettings and WebSearchSettings to reference correct locations in DocProcessSettings and WebSearchSettings directories.
* feat(settings): add API server settings link and route
* fix(auth): improve authorization handling and error responses
* feat(chat): enhance model validation and logging for chat completions
feat(models): improve logging for model retrieval and filtering
feat(utils): add model ID validation and support for OpenAI providers
* feat(api-server): refactor config loading and remove unused ToolSettings component
* refactor(ApiServerService): simplify config retrieval and improve error handling in ApiServerSettings
* fix(mcp): remove unnecessary await in listTools return statement
* refactor(ApiServerSettings): replace window.message with window.toast for notifications
---------
Co-authored-by: kangfenmao <kangfenmao@qq.com>
129 lines
3.1 KiB
TypeScript
129 lines
3.1 KiB
TypeScript
import { loggerService } from '@main/services/LoggerService'
|
|
import cors from 'cors'
|
|
import express from 'express'
|
|
import { v4 as uuidv4 } from 'uuid'
|
|
|
|
import { authMiddleware } from './middleware/auth'
|
|
import { errorHandler } from './middleware/error'
|
|
import { setupOpenAPIDocumentation } from './middleware/openapi'
|
|
import { chatRoutes } from './routes/chat'
|
|
import { mcpRoutes } from './routes/mcp'
|
|
import { modelsRoutes } from './routes/models'
|
|
|
|
const logger = loggerService.withContext('ApiServer')
|
|
|
|
const app = express()
|
|
|
|
// Global middleware
|
|
app.use((req, res, next) => {
|
|
const start = Date.now()
|
|
res.on('finish', () => {
|
|
const duration = Date.now() - start
|
|
logger.info(`${req.method} ${req.path} - ${res.statusCode} - ${duration}ms`)
|
|
})
|
|
next()
|
|
})
|
|
|
|
app.use((_req, res, next) => {
|
|
res.setHeader('X-Request-ID', uuidv4())
|
|
next()
|
|
})
|
|
|
|
app.use(
|
|
cors({
|
|
origin: '*',
|
|
allowedHeaders: ['Content-Type', 'Authorization'],
|
|
methods: ['GET', 'POST', 'PUT', 'DELETE', 'OPTIONS']
|
|
})
|
|
)
|
|
|
|
/**
|
|
* @swagger
|
|
* /health:
|
|
* get:
|
|
* summary: Health check endpoint
|
|
* description: Check server status (no authentication required)
|
|
* tags: [Health]
|
|
* security: []
|
|
* responses:
|
|
* 200:
|
|
* description: Server is healthy
|
|
* content:
|
|
* application/json:
|
|
* schema:
|
|
* type: object
|
|
* properties:
|
|
* status:
|
|
* type: string
|
|
* example: ok
|
|
* timestamp:
|
|
* type: string
|
|
* format: date-time
|
|
* version:
|
|
* type: string
|
|
* example: 1.0.0
|
|
*/
|
|
app.get('/health', (_req, res) => {
|
|
res.json({
|
|
status: 'ok',
|
|
timestamp: new Date().toISOString(),
|
|
version: process.env.npm_package_version || '1.0.0'
|
|
})
|
|
})
|
|
|
|
/**
|
|
* @swagger
|
|
* /:
|
|
* get:
|
|
* summary: API information
|
|
* description: Get basic API information and available endpoints
|
|
* tags: [General]
|
|
* security: []
|
|
* responses:
|
|
* 200:
|
|
* description: API information
|
|
* content:
|
|
* application/json:
|
|
* schema:
|
|
* type: object
|
|
* properties:
|
|
* name:
|
|
* type: string
|
|
* example: Cherry Studio API
|
|
* version:
|
|
* type: string
|
|
* example: 1.0.0
|
|
* endpoints:
|
|
* type: object
|
|
*/
|
|
app.get('/', (_req, res) => {
|
|
res.json({
|
|
name: 'Cherry Studio API',
|
|
version: '1.0.0',
|
|
endpoints: {
|
|
health: 'GET /health',
|
|
models: 'GET /v1/models',
|
|
chat: 'POST /v1/chat/completions',
|
|
mcp: 'GET /v1/mcps'
|
|
}
|
|
})
|
|
})
|
|
|
|
// API v1 routes with auth
|
|
const apiRouter = express.Router()
|
|
apiRouter.use(authMiddleware)
|
|
apiRouter.use(express.json())
|
|
// Mount routes
|
|
apiRouter.use('/chat', chatRoutes)
|
|
apiRouter.use('/mcps', mcpRoutes)
|
|
apiRouter.use('/models', modelsRoutes)
|
|
app.use('/v1', apiRouter)
|
|
|
|
// Setup OpenAPI documentation
|
|
setupOpenAPIDocumentation(app)
|
|
|
|
// Error handling (must be last)
|
|
app.use(errorHandler)
|
|
|
|
export { app }
|