File size: 4,839 Bytes
44d3414 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 |
const express = require('express');
const { v4: uuidv4 } = require('uuid');
const { stringToHex, chunkToUtf8String, getRandomIDPro } = require('./utils.js');
const app = express();
// Middleware configuration
app.use(express.json());
app.use(express.urlencoded({ extended: true }));
app.post('/v1/chat/completions', async (req, res) => {
// Models starting with o1 do not support streaming output
if (req.body.model.startsWith('o1-') && req.body.stream) {
return res.status(400).json({
error: 'Model not supported stream',
});
}
let currentKeyIndex = 0;
try {
const { model, messages, stream = false } = req.body;
let authToken = req.headers.authorization?.replace('Bearer ', '');
// Handle comma-separated keys
const keys = authToken.split(',').map((key) => key.trim());
if (keys.length > 0) {
// make sure currentKeyIndex won't cross the line
if (currentKeyIndex >= keys.length) {
currentKeyIndex = 0;
}
// Use the current index to get the key
authToken = keys[currentKeyIndex];
}
if (authToken && authToken.includes('%3A%3A')) {
authToken = authToken.split('%3A%3A')[1];
}
if (!messages || !Array.isArray(messages) || messages.length === 0 || !authToken) {
return res.status(400).json({
error: 'Invalid request. Messages should be a non-empty array and authorization is required',
});
}
const hexData = await stringToHex(messages, model);
// get checksum，req header Medium passes first, the environment variable ranks second, and is randomly generated last
const checksum =
req.headers['x-cursor-checksum'] ??
process.env['x-cursor-checksum'] ??
`zo${getRandomIDPro({ dictType: 'max', size: 6 })}${getRandomIDPro({ dictType: 'max', size: 64 })}/${getRandomIDPro({ dictType: 'max', size: 64 })}`;
const response = await fetch('https://api2.cursor.sh/aiserver.v1.AiService/StreamChat', {
method: 'POST',
headers: {
'Content-Type': 'application/connect+proto',
authorization: `Bearer ${authToken}`,
'connect-accept-encoding': 'gzip,br',
'connect-protocol-version': '1',
'user-agent': 'connect-es/1.4.0',
'x-amzn-trace-id': `Root=${uuidv4()}`,
'x-cursor-checksum': checksum,
'x-cursor-client-version': '0.42.3',
'x-cursor-timezone': 'Asia/Shanghai',
'x-ghost-mode': 'false',
'x-request-id': uuidv4(),
Host: 'api2.cursor.sh',
},
body: hexData,
});
if (stream) {
res.setHeader('Content-Type', 'text/event-stream');
res.setHeader('Cache-Control', 'no-cache');
res.setHeader('Connection', 'keep-alive');
const responseId = `chatcmpl-${uuidv4()}`;
// Use encapsulated function processing chunk
for await (const chunk of response.body) {
const text = await chunkToUtf8String(chunk);
if (text.length > 0) {
res.write(
`data: ${JSON.stringify({
id: responseId,
object: 'chat.completion.chunk',
created: Math.floor(Date.now() / 1000),
model,
choices: [
{
index: 0,
delta: {
content: text,
},
},
],
})}\n\n`,
);
}
}
res.write('data: [DONE]\n\n');
return res.end();
} else {
let text = '';
// Encapsulated functions are also used in non-streaming mode
for await (const chunk of response.body) {
text += await chunkToUtf8String(chunk);
}
// Further processing of the parsed string
text = text.replace(/^.*<\|END_USER\|>/s, '');
text = text.replace(/^\n[a-zA-Z]?/, '').trim();
// console.log(text)
return res.json({
id: `chatcmpl-${uuidv4()}`,
object: 'chat.completion',
created: Math.floor(Date.now() / 1000),
model,
choices: [
{
index: 0,
message: {
role: 'assistant',
content: text,
},
finish_reason: 'stop',
},
],
usage: {
prompt_tokens: 0,
completion_tokens: 0,
total_tokens: 0,
},
});
}
} catch (error) {
console.error('Error:', error);
if (!res.headersSent) {
if (req.body.stream) {
res.write(`data: ${JSON.stringify({ error: 'Internal server error' })}\n\n`);
return res.end();
} else {
return res.status(500).json({ error: 'Internal server error' });
}
}
}
});
// Start the server
const PORT = process.env.PORT || 3000;
app.listen(PORT, () => {
console.log(`The server runs on the port ${PORT}`);
});
|