fixes kobold endpoint not actually being rate-limited

2023-04-10 06:42:12 -05:00
parent 6b47c1cdc5
commit f0afc8c420
1 changed files with 2 additions and 1 deletions
@@ -13,6 +13,7 @@ import {
  handleInternalError,
  incrementKeyUsage,
 } from "./common";
+import { ipLimiter } from "./rate-limit";
 import {
  addKey,
  disableStream,
@@ -123,7 +124,7 @@ const koboldOaiProxy = createProxyMiddleware({
 const koboldRouter = Router();
 koboldRouter.get("/api/v1/model", handleModelRequest);
 koboldRouter.get("/api/v1/config/soft_prompts_list", handleSoftPromptsRequest);
-koboldRouter.post("/api/v1/generate", koboldOaiProxy);
+koboldRouter.post("/api/v1/generate", ipLimiter, koboldOaiProxy);
 koboldRouter.use((req, res) => {
  logger.warn(`Unhandled kobold request: ${req.method} ${req.path}`);
  res.status(404).json({ error: "Not found" });