Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2026-01-07 05:39:05 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Refactor logging levels in transform.ts and adjust JSON body parser configuration in app.ts

Browse Source
This commit is contained in:
Vaayne 2025-09-22 16:32:32 +08:00
parent 634c478e18
commit 2385fba695
3 changed files with 9 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/main/apiServer/app.ts
View File

@ -15,6 +15,11 @@ import { modelsRoutes } from './routes/models'
const logger = loggerService.withContext('ApiServer')
const app = express()
app.use(
express.json({
limit: '50mb'
})
)
// Global middleware
app.use((req, res, next) => {
@ -111,7 +116,6 @@ app.get('/', (_req, res) => {
// Provider-specific API routes with auth (must be before /v1 to avoid conflicts)
const providerRouter = express.Router({ mergeParams: true })
providerRouter.use(authMiddleware)
providerRouter.use(express.json())
// Mount provider-specific messages route
providerRouter.use('/v1/messages', messagesProviderRoutes)
app.use('/:provider', providerRouter)
@ -119,7 +123,6 @@ app.use('/:provider', providerRouter)
// 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)

2
src/main/apiServer/utils/index.ts
View File

@ -7,7 +7,7 @@ const logger = loggerService.withContext('ApiServerUtils')
// Cache configuration
const PROVIDERS_CACHE_KEY = 'api-server:providers'
const PROVIDERS_CACHE_TTL = 5 * 60 * 1000 // 5 minutes
const PROVIDERS_CACHE_TTL = 1 * 60 * 1000 // 1 minutes
export async function getAvailableProviders(): Promise<Provider[]> {
try {

11
src/main/services/agents/services/claudecode/transform.ts
View File

@ -34,7 +34,7 @@ const generateMessageId = (): string => `msg_${uuidv4().replace(/-/g, '')}`
// Main transform function
export function transformSDKMessageToStreamParts(sdkMessage: SDKMessage): AgentStreamPart[] {
const chunks: AgentStreamPart[] = []
logger.debug('Transforming SDKMessage to stream parts', sdkMessage)
logger.silly('Transforming SDKMessage to stream parts', sdkMessage)
switch (sdkMessage.type) {
case 'assistant':
case 'user':
@ -105,7 +105,6 @@ function handleUserOrAssistantMessage(message: Extract<SDKMessage, { type: 'assi
}
} else if (Array.isArray(message.message.content)) {
for (const block of message.message.content) {
logger.debug('Handling user or assistant message:', { block })
switch (block.type) {
case 'text':
chunks.push(...generateTextChunks(messageId, block.text, message))
@ -127,8 +126,7 @@ function handleUserOrAssistantMessage(message: Extract<SDKMessage, { type: 'assi
})
break
case 'tool_result': {
logger.debug('Handling tool result:', { block })
logger.debug('contentblockState', { content: contentBlockState })
logger.silly('Handling tool result:', { block, content: contentBlockState })
const hasToolCall = contentBlockState.has(block.tool_use_id)
const toolCall = contentBlockState.get(block.tool_use_id) as toolCallBlock
chunks.push({
@ -161,7 +159,7 @@ function handleStreamEvent(message: Extract<SDKMessage, { type: 'stream_event' }
const chunks: AgentStreamPart[] = []
const event = message.event
const blockKey = `${message.uuid ?? message.session_id ?? 'session'}:${event.type}`
logger.debug('Handling stream event:', { event })
logger.silly('Handling stream event:', { event })
switch (event.type) {
case 'message_start':
// No specific UI chunk needed for message start in this protocol
@ -291,9 +289,6 @@ function handleStreamEvent(message: Extract<SDKMessage, { type: 'stream_event' }
// Handle system messages
function handleSystemMessage(message: Extract<SDKMessage, { type: 'system' }>): AgentStreamPart[] {
const chunks: AgentStreamPart[] = []
logger.debug('Received system message', {
subtype: message.subtype
})
switch (message.subtype) {
case 'init': {
chunks.push({