fixes kobold endpoint not actually being rate-limited
This commit is contained in:
parent
6b47c1cdc5
commit
f0afc8c420
|
@ -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" });
|
||||
|
|
Loading…
Reference in New Issue