From 51e07b82f52d02fe4e582de41e3d759235642408 Mon Sep 17 00:00:00 2001
From: nai-degen <44111-khanon@users.noreply.gitgud.io>
Date: Sat, 8 Apr 2023 08:34:46 -0500
Subject: [PATCH] removes unused files from failed rename

---
 src/auth.ts   |  16 --------
 src/kobold.ts |   6 ---
 src/openai.ts | 108 --------------------------------------------------
 3 files changed, 130 deletions(-)
 delete mode 100644 src/auth.ts
 delete mode 100644 src/kobold.ts
 delete mode 100644 src/openai.ts

diff --git a/src/auth.ts b/src/auth.ts
deleted file mode 100644
index f746241..0000000
--- a/src/auth.ts
+++ /dev/null
@@ -1,16 +0,0 @@
-import type { Request, Response, NextFunction } from "express";
-
-const PROXY_KEY = process.env.PROXY_KEY;
-
-export const auth = (req: Request, res: Response, next: NextFunction) => {
-  if (!PROXY_KEY) {
-    next();
-    return;
-  }
-  if (req.headers.authorization === `Bearer ${PROXY_KEY}`) {
-    delete req.headers.authorization;
-    next();
-  } else {
-    res.status(401).json({ error: "Unauthorized" });
-  }
-};
diff --git a/src/kobold.ts b/src/kobold.ts
deleted file mode 100644
index afd971a..0000000
--- a/src/kobold.ts
+++ /dev/null
@@ -1,6 +0,0 @@
-import { Request, Response, NextFunction } from "express";
-
-export const kobold = (req: Request, res: Response, next: NextFunction) => {
-  // TODO: Implement kobold
-  res.status(501).json({ error: "Not implemented" });
-};
diff --git a/src/openai.ts b/src/openai.ts
deleted file mode 100644
index bda297c..0000000
--- a/src/openai.ts
+++ /dev/null
@@ -1,108 +0,0 @@
-import { Request, Response, NextFunction, Router } from "express";
-import * as http from "http";
-import { createProxyMiddleware } from "http-proxy-middleware";
-import { logger } from "./logger";
-import { keys } from "./keys";
-
-/**
- * Modifies the request body to add a randomly selected API key.
- */
-const rewriteRequest = (proxyReq: http.ClientRequest, req: Request) => {
-  const key = keys.get(req.body?.model || "gpt-3.5")!;
-  req.key = key;
-
-  proxyReq.setHeader("Authorization", `Bearer ${key}`);
-  if (req.body?.stream) {
-    req.body.stream = false;
-    const updatedBody = JSON.stringify(req.body);
-    proxyReq.setHeader("Content-Length", Buffer.byteLength(updatedBody));
-    proxyReq.write(updatedBody);
-    proxyReq.end();
-  }
-};
-
-// TODO: extract this since Kobold will use it too
-const handleResponse = (
-  proxyRes: http.IncomingMessage,
-  req: Request,
-  res: Response
-) => {
-  const statusCode = proxyRes.statusCode || 500;
-
-  if (statusCode >= 400) {
-    // Consume body and then decide what to do
-    let body = "";
-    proxyRes.on("data", (chunk) => {
-      body += chunk;
-    });
-    proxyRes.on("end", () => {
-      let errorPayload: any = {
-        error: "Proxy couldn't parse error from OpenAI",
-      };
-      const canTryAgain = keys.anyAvailable()
-        ? "You can try again to get a different key."
-        : "There are no more keys available.";
-      try {
-        errorPayload = JSON.parse(body);
-      } catch (err) {
-        logger.error({ error: err }, errorPayload.error);
-        res.status(statusCode).json(errorPayload);
-        return;
-      }
-
-      if (statusCode === 401) {
-        // Key is invalid or was revoked
-        logger.warn(
-          `OpenAI key is invalid or revoked. Keyhash ${req.key?.hash}`
-        );
-        keys.disable(req.key!);
-        const message = `The OpenAI key is invalid or revoked. ${canTryAgain}`;
-        errorPayload.proxy_note = message;
-      } else if (statusCode === 429) {
-        // Rate limit exceeded
-        if (errorPayload.error?.type === "insufficient_quota") {
-          logger.warn(`OpenAI key is exhausted. Keyhash ${req.key?.hash}`);
-          keys.disable(req.key!);
-          const message = `The OpenAI key is exhausted. ${canTryAgain}`;
-          errorPayload.proxy_note = message;
-        } else {
-          logger.warn(
-            { errorCode: errorPayload.error?.type },
-            `OpenAI rate limit exceeded or model overloaded. Keyhash ${req.key?.hash}`
-          );
-        }
-      }
-
-      res.status(statusCode).json(errorPayload);
-    });
-  } else {
-    proxyRes.pipe(res);
-  }
-};
-
-const openaiProxy = createProxyMiddleware({
-  target: "https://api.openai.com",
-  changeOrigin: true,
-  onProxyReq: rewriteRequest,
-  onProxyRes: handleResponse,
-  selfHandleResponse: true,
-  pathRewrite: {
-    "^/proxy/openai": "",
-  },
-  logProvider: () => ({
-    debug: logger.debug.bind(logger),
-    info: logger.info.bind(logger),
-    warn: logger.warn.bind(logger),
-    error: logger.error.bind(logger),
-    log: logger.info.bind(logger),
-  }),
-});
-
-const openaiRouter = Router();
-openaiRouter.post("/v1/chat/completions", openaiProxy);
-openaiRouter.use((req, res) => {
-  logger.warn(`Blocked openai proxy request: ${req.method} ${req.path}`);
-  res.status(404).json({ error: "Not found" });
-});
-
-export const openai = openaiRouter;