adds prompt counter

This commit is contained in:
nai-degen
2023-04-08 11:13:55 -05:00
committed by nai-degen
parent c74c527f46
commit 11308b2baa
3 changed files with 15 additions and 1 deletions
+3
View File
@@ -59,6 +59,9 @@ export const handleResponse = (
res.status(statusCode).json(errorPayload);
});
} else {
// Increment key's usage count
keys.incrementPrompt(req.key?.hash);
Object.keys(proxyRes.headers).forEach((key) => {
res.setHeader(key, proxyRes.headers[key] as string);
});