ros/claude-code-router
1
0
Fork 0
mirror of https://github.com/musistudio/claude-code-router.git synced 2026-01-30 06:12:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

translate comment to english

Browse Source
This commit is contained in:
musistudio
2025-12-30 18:53:16 +08:00
parent 21d35e3d4c
commit 3cb1275e0c
18 changed files with 252 additions and 254 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
packages/server/src/agents/index.ts
View File

@@ -5,25 +5,25 @@ export class AgentsManager {
private agents: Map<string, IAgent> = new Map();
/**
* 注册一个agent
* @param agent 要注册的agent实例
* @param isDefault 是否设为默认agent
* Register an agent
* @param agent The agent instance to register
* @param isDefault Whether to set as default agent
*/
registerAgent(agent: IAgent): void {
this.agents.set(agent.name, agent);
}
/**
* 根据名称查找agent
* @param name agent名称
* @returns 找到的agent实例，未找到返回undefined
* Find agent by name
* @param name Agent name
* @returns Found agent instance, undefined if not found
*/
getAgent(name: string): IAgent | undefined {
return this.agents.get(name);
}
/**
* 获取所有已注册的agents
* @returns 所有agent实例的数组
* Get all registered agents
* @returns Array of all agent instances
*/
getAllAgents(): IAgent[] {
return Array.from(this.agents.values());
@@ -31,8 +31,8 @@ export class AgentsManager {
/**
* 获取所有agent的工具
* @returns 工具数组
* Get all agent tools
* @returns Array of tools
*/
getAllTools(): any[] {
const allTools: any[] = [];

26
packages/server/src/index.ts
View File

@@ -94,13 +94,13 @@ async function getServer(options: RunOptions = {}) {
let loggerConfig: any;
// 如果外部传入了 logger 配置，使用外部的
// Use external logger configuration if provided
if (options.logger !== undefined) {
loggerConfig = options.logger;
} else {
// 如果没有传入，并且 config.LOG !== false，则启用 logger
// Enable logger if not provided and config.LOG !== false
if (config.LOG !== false) {
// 将 config.LOG 设为 true（如果它还未设置）
// Set config.LOG to true (if not already set)
if (config.LOG === undefined) {
config.LOG = true;
}
@@ -166,7 +166,7 @@ async function getServer(options: RunOptions = {}) {
for (const agent of agentsManager.getAllAgents()) {
if (agent.shouldHandle(req, config)) {
// 设置agent标识
// Set agent identifier
useAgents.push(agent.name)
// change request body
@@ -209,10 +209,10 @@ async function getServer(options: RunOptions = {}) {
let currentToolId = ''
const toolMessages: any[] = []
const assistantMessages: any[] = []
// 存储Anthropic格式的消息体，区分文本和工具类型
// Store Anthropic format message body, distinguishing text and tool types
return done(null, rewriteStream(eventStream, async (data, controller) => {
try {
// 检测工具调用开始
// Detect tool call start
if (data.event === 'content_block_start' && data?.data?.content_block?.name) {
const agent = req.agents.find((name: string) => agentsManager.getAgent(name)?.tools.get(data.data.content_block.name))
if (agent) {
@@ -224,13 +224,13 @@ async function getServer(options: RunOptions = {}) {
}
}
// 收集工具参数
// Collect tool arguments
if (currentToolIndex > -1 && data.data.index === currentToolIndex && data.data?.delta?.type === 'input_json_delta') {
currentToolArgs += data.data?.delta?.partial_json;
return undefined;
}
// 工具调用完成，处理agent调用
// Tool call completed, handle agent invocation
if (currentToolIndex > -1 && data.data.index === currentToolIndex && data.data.type === 'content_block_stop') {
try {
const args = JSON5.parse(currentToolArgs);
@@ -293,7 +293,7 @@ async function getServer(options: RunOptions = {}) {
continue
}
// 检查流是否仍然可写
// Check if stream is still writable
if (!controller.desiredSize) {
break;
}
@@ -301,7 +301,7 @@ async function getServer(options: RunOptions = {}) {
controller.enqueue(eventData)
}catch (readError: any) {
if (readError.name === 'AbortError' || readError.code === 'ERR_STREAM_PREMATURE_CLOSE') {
abortController.abort(); // 中止所有相关操作
abortController.abort(); // Abort all related operations
break;
}
throw readError;
@@ -314,13 +314,13 @@ async function getServer(options: RunOptions = {}) {
}catch (error: any) {
console.error('Unexpected error in stream processing:', error);
// 处理流提前关闭的错误
// Handle premature stream closure error
if (error.code === 'ERR_STREAM_PREMATURE_CLOSE') {
abortController.abort();
return undefined;
}
// 其他错误仍然抛出
// Re-throw other errors
throw error;
}
}).pipeThrough(new SSESerializerTransform()))
@@ -405,7 +405,7 @@ export type { RunOptions };
export type { IAgent, ITool } from "./agents/type";
export { initDir, initConfig, readConfigFile, writeConfigFile, backupConfigFile } from "./utils";
// 如果是直接运行此文件，则启动服务
// Start service if this file is run directly
if (require.main === module) {
run().catch((error) => {
console.error('Failed to start server:', error);

6
packages/server/src/utils/SSEParser.transform.ts
View File

@@ -8,7 +8,7 @@ export class SSEParserTransform extends TransformStream<string, any> {
this.buffer += chunk;
const lines = this.buffer.split('\n');
// 保留最后一行（可能不完整）
// Keep last line (may be incomplete)
this.buffer = lines.pop() || '';
for (const line of lines) {
@@ -19,14 +19,14 @@ export class SSEParserTransform extends TransformStream<string, any> {
}
},
flush: (controller) => {
// 处理缓冲区中剩余的内容
// Process remaining content in buffer
if (this.buffer.trim()) {
const events: any[] = [];
this.processLine(this.buffer.trim(), events);
events.forEach(event => controller.enqueue(event));
}
// 推送最后一个事件（如果有）
// Push last event (if any)
if (Object.keys(this.currentEvent).length > 0) {
controller.enqueue(this.currentEvent);
}

2
packages/server/src/utils/rewriteStream.ts
View File

@@ -1,5 +1,5 @@
/**rewriteStream
* 读取源readablestream，返回一个新的readablestream，由processor对源数据进行处理后将返回的新值推送到新的stream，如果没有返回值则不推送
* Read source readablestream and return a new readablestream, processor processes source data and pushes returned new value to new stream, no push if no return value
* @param stream
* @param processor
*/