mirror of
https://github.com/CherryHQ/cherry-studio.git
synced 2025-12-26 20:12:38 +08:00
58dbb514e0
- Added `aisdk-stream-protocel.md` to document text and data stream protocols. - Created `ClaudeCodeService` for invoking and streaming responses from the Claude Code CLI. - Introduced built-in tools for Claude Code, including Bash, Edit, and WebFetch. - Developed transformation functions to convert Claude Code messages to AI SDK format. - Enhanced OCR utility with delayed loading of the Sharp module. - Updated agent types and session message structures to accommodate new features. - Modified API tests to reflect changes in session creation and message streaming. - Upgraded `uuid` package to version 13.0.0 for improved UUID generation.
213 lines
6.6 KiB
TypeScript
213 lines
6.6 KiB
TypeScript
import express from 'express'
|
|
|
|
import { agentHandlers, messageHandlers, sessionHandlers } from './handlers'
|
|
import { checkAgentExists, handleValidationErrors } from './middleware'
|
|
import {
|
|
validateAgent,
|
|
validateAgentId,
|
|
validateAgentUpdate,
|
|
validatePagination,
|
|
validateSession,
|
|
validateSessionId,
|
|
validateSessionMessage,
|
|
validateSessionUpdate
|
|
} from './validators'
|
|
|
|
// Create main agents router
|
|
const agentsRouter = express.Router()
|
|
|
|
/**
|
|
* @swagger
|
|
* components:
|
|
* schemas:
|
|
* AgentEntity:
|
|
* type: object
|
|
* properties:
|
|
* id:
|
|
* type: string
|
|
* description: Unique agent identifier
|
|
* name:
|
|
* type: string
|
|
* description: Agent name
|
|
* description:
|
|
* type: string
|
|
* description: Agent description
|
|
* avatar:
|
|
* type: string
|
|
* description: Agent avatar URL
|
|
* instructions:
|
|
* type: string
|
|
* description: System prompt/instructions
|
|
* model:
|
|
* type: string
|
|
* description: Main model ID
|
|
* plan_model:
|
|
* type: string
|
|
* description: Optional planning model ID
|
|
* small_model:
|
|
* type: string
|
|
* description: Optional small/fast model ID
|
|
* built_in_tools:
|
|
* type: array
|
|
* items:
|
|
* type: string
|
|
* description: Built-in tool IDs
|
|
* mcps:
|
|
* type: array
|
|
* items:
|
|
* type: string
|
|
* description: MCP tool IDs
|
|
* knowledges:
|
|
* type: array
|
|
* items:
|
|
* type: string
|
|
* description: Knowledge base IDs
|
|
* configuration:
|
|
* type: object
|
|
* description: Extensible settings
|
|
* accessible_paths:
|
|
* type: array
|
|
* items:
|
|
* type: string
|
|
* description: Accessible directory paths
|
|
* permission_mode:
|
|
* type: string
|
|
* enum: [readOnly, acceptEdits, bypassPermissions]
|
|
* description: Permission mode
|
|
* max_steps:
|
|
* type: integer
|
|
* description: Maximum steps the agent can take
|
|
* created_at:
|
|
* type: string
|
|
* format: date-time
|
|
* updated_at:
|
|
* type: string
|
|
* format: date-time
|
|
* required:
|
|
* - id
|
|
* - name
|
|
* - model
|
|
* - created_at
|
|
* - updated_at
|
|
* CreateAgentRequest:
|
|
* type: object
|
|
* properties:
|
|
* name:
|
|
* type: string
|
|
* description: Agent name
|
|
* description:
|
|
* type: string
|
|
* description: Agent description
|
|
* avatar:
|
|
* type: string
|
|
* description: Agent avatar URL
|
|
* instructions:
|
|
* type: string
|
|
* description: System prompt/instructions
|
|
* model:
|
|
* type: string
|
|
* description: Main model ID
|
|
* plan_model:
|
|
* type: string
|
|
* description: Optional planning model ID
|
|
* small_model:
|
|
* type: string
|
|
* description: Optional small/fast model ID
|
|
* built_in_tools:
|
|
* type: array
|
|
* items:
|
|
* type: string
|
|
* description: Built-in tool IDs
|
|
* mcps:
|
|
* type: array
|
|
* items:
|
|
* type: string
|
|
* description: MCP tool IDs
|
|
* knowledges:
|
|
* type: array
|
|
* items:
|
|
* type: string
|
|
* description: Knowledge base IDs
|
|
* configuration:
|
|
* type: object
|
|
* description: Extensible settings
|
|
* accessible_paths:
|
|
* type: array
|
|
* items:
|
|
* type: string
|
|
* description: Accessible directory paths
|
|
* permission_mode:
|
|
* type: string
|
|
* enum: [readOnly, acceptEdits, bypassPermissions]
|
|
* description: Permission mode
|
|
* max_steps:
|
|
* type: integer
|
|
* description: Maximum steps the agent can take
|
|
* required:
|
|
* - name
|
|
* - model
|
|
*/
|
|
|
|
// Agent CRUD routes
|
|
agentsRouter.post('/', validateAgent, handleValidationErrors, agentHandlers.createAgent)
|
|
agentsRouter.get('/', validatePagination, handleValidationErrors, agentHandlers.listAgents)
|
|
agentsRouter.get('/:agentId', validateAgentId, handleValidationErrors, agentHandlers.getAgent)
|
|
agentsRouter.put('/:agentId', validateAgentId, validateAgentUpdate, handleValidationErrors, agentHandlers.updateAgent)
|
|
agentsRouter.patch('/:agentId', validateAgentId, validateAgentUpdate, handleValidationErrors, agentHandlers.patchAgent)
|
|
agentsRouter.delete('/:agentId', validateAgentId, handleValidationErrors, agentHandlers.deleteAgent)
|
|
|
|
// Create sessions router with agent context
|
|
const createSessionsRouter = (): express.Router => {
|
|
const sessionsRouter = express.Router({ mergeParams: true })
|
|
|
|
// Session CRUD routes (nested under agent)
|
|
sessionsRouter.post('/', validateSession, handleValidationErrors, sessionHandlers.createSession)
|
|
sessionsRouter.get('/', validatePagination, handleValidationErrors, sessionHandlers.listSessions)
|
|
sessionsRouter.get('/:sessionId', validateSessionId, handleValidationErrors, sessionHandlers.getSession)
|
|
sessionsRouter.put(
|
|
'/:sessionId',
|
|
validateSessionId,
|
|
validateSessionUpdate,
|
|
handleValidationErrors,
|
|
sessionHandlers.updateSession
|
|
)
|
|
sessionsRouter.patch(
|
|
'/:sessionId',
|
|
validateSessionId,
|
|
validateSessionUpdate,
|
|
handleValidationErrors,
|
|
sessionHandlers.patchSession
|
|
)
|
|
sessionsRouter.delete('/:sessionId', validateSessionId, handleValidationErrors, sessionHandlers.deleteSession)
|
|
|
|
return sessionsRouter
|
|
}
|
|
|
|
// Create messages router with agent and session context
|
|
const createMessagesRouter = (): express.Router => {
|
|
const messagesRouter = express.Router({ mergeParams: true })
|
|
|
|
// Message CRUD routes (nested under agent/session)
|
|
messagesRouter.post('/', validateSessionMessage, handleValidationErrors, messageHandlers.createMessageStream)
|
|
return messagesRouter
|
|
}
|
|
|
|
// Mount nested resources with clear hierarchy
|
|
const sessionsRouter = createSessionsRouter()
|
|
const messagesRouter = createMessagesRouter()
|
|
|
|
// Mount sessions under specific agent
|
|
agentsRouter.use('/:agentId/sessions', validateAgentId, checkAgentExists, handleValidationErrors, sessionsRouter)
|
|
|
|
// Mount messages under specific agent/session
|
|
agentsRouter.use(
|
|
'/:agentId/sessions/:sessionId/messages',
|
|
validateAgentId,
|
|
validateSessionId,
|
|
handleValidationErrors,
|
|
messagesRouter
|
|
)
|
|
|
|
// Export main router and convenience router
|
|
export const agentsRoutes = agentsRouter
|