fix some bugs
This commit is contained in:
53
src/index.ts
53
src/index.ts
@@ -16,6 +16,7 @@ import createWriteStream from "pino-rotating-file-stream";
|
||||
import { HOME_DIR } from "./constants";
|
||||
import { configureLogging } from "./utils/log";
|
||||
import { sessionUsageCache } from "./utils/cache";
|
||||
import Stream from "node:stream";
|
||||
|
||||
async function initializeClaudeConfig() {
|
||||
const homeDir = homedir();
|
||||
@@ -133,32 +134,50 @@ async function run(options: RunOptions = {}) {
|
||||
router(req, reply, config);
|
||||
}
|
||||
});
|
||||
server.addHook("onSend", async (req, reply, payload) => {
|
||||
server.addHook("onSend", (req, reply, payload, done) => {
|
||||
if (req.sessionId && req.url.startsWith("/v1/messages")) {
|
||||
if (payload instanceof ReadableStream) {
|
||||
const [originalStream, clonedStream] = payload.tee();
|
||||
const reader1 = clonedStream.getReader();
|
||||
while (true) {
|
||||
const { done, value } = await reader1.read();
|
||||
if (done) break;
|
||||
// Process the value if needed
|
||||
const dataStr = new TextDecoder().decode(value);
|
||||
if (!dataStr.startsWith("event: message_delta")) {
|
||||
continue;
|
||||
const read = async (stream: ReadableStream) => {
|
||||
const reader = stream.getReader();
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
// Process the value if needed
|
||||
const dataStr = new TextDecoder().decode(value);
|
||||
if (!dataStr.startsWith("event: message_delta")) {
|
||||
continue;
|
||||
}
|
||||
const str = dataStr.slice(27);
|
||||
try {
|
||||
const message = JSON.parse(str);
|
||||
sessionUsageCache.put(req.sessionId, message.usage);
|
||||
} catch {}
|
||||
}
|
||||
const str = dataStr.slice(27);
|
||||
try {
|
||||
const message = JSON.parse(str);
|
||||
sessionUsageCache.put(req.sessionId, message.usage);
|
||||
} catch {}
|
||||
}
|
||||
|
||||
return originalStream;
|
||||
read(clonedStream);
|
||||
done(originalStream)
|
||||
} else {
|
||||
req.log.debug({payload}, 'onSend Hook')
|
||||
sessionUsageCache.put(req.sessionId, payload.usage);
|
||||
if (payload instanceof Buffer || payload instanceof Response) {
|
||||
done(null, payload);
|
||||
} else if(typeof payload === "object") {
|
||||
done(null, JSON.stringify(payload));
|
||||
} else {
|
||||
done(null, payload);
|
||||
}
|
||||
}
|
||||
} else {
|
||||
if(payload instanceof Buffer || payload instanceof Response || payload === null || payload instanceof ReadableStream || payload instanceof Stream) {
|
||||
done(null, payload);
|
||||
} else if(typeof payload === "object") {
|
||||
req.log.debug({payload}, 'onSend Hook')
|
||||
done(null, JSON.stringify(payload));
|
||||
} else {
|
||||
done(null, payload);
|
||||
}
|
||||
}
|
||||
return payload;
|
||||
});
|
||||
server.start();
|
||||
}
|
||||
|
||||
@@ -16,18 +16,15 @@ export async function executeCodeCommand(args: string[] = []) {
|
||||
ANTHROPIC_BASE_URL: `http://127.0.0.1:${config.PORT || 3456}`,
|
||||
API_TIMEOUT_MS: String(config.API_TIMEOUT_MS ?? 600000), // Default to 10 minutes if not set
|
||||
};
|
||||
|
||||
const settingsFlag: Record<string, any> = {
|
||||
env,
|
||||
};
|
||||
let settingsFlag: Record<string, any> | undefined;
|
||||
if (config?.StatusLine?.enabled) {
|
||||
settingsFlag.statusLine = {
|
||||
type: "command",
|
||||
command: "ccr statusline",
|
||||
padding: 0,
|
||||
};
|
||||
}
|
||||
if (Object.keys(settingsFlag).length > 0 && process.platform !== 'win32') {
|
||||
settingsFlag = {
|
||||
statusLine: {
|
||||
type: "command",
|
||||
command: "ccr statusline",
|
||||
padding: 0,
|
||||
}
|
||||
}
|
||||
args.push(`--settings=${JSON.stringify(settingsFlag)}`);
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user