Ayeantics commited on
Commit
7f84064
·
verified ·
1 Parent(s): 2889205

Update server.js

Browse files
Files changed (1) hide show
  1. server.js +12 -6
server.js CHANGED
@@ -1,21 +1,27 @@
1
  const express = require('express');
2
  const proxy = require('express-http-proxy');
3
  const app = express();
 
4
  const targetUrl = 'https://api.openai.com';
5
- const openaiKey = process.env.OPENAI_KEY
6
  const port = 7860;
7
  const baseUrl = getExternalUrl(process.env.SPACE_ID);
8
 
9
- app.use('/api', proxy(targetUrl, {
 
 
 
 
 
10
  proxyReqOptDecorator: (proxyReqOpts, srcReq) => {
11
  // Modify the request headers if necessary
12
- proxyReqOpts.headers['Authorization'] = 'Bearer '+openaiKey;
13
  return proxyReqOpts;
14
  },
15
  }));
16
 
17
  app.get("/", (req, res) => {
18
- res.send(`This is your OpenAI Reverse Proxy URL: ${baseUrl}`);
19
  });
20
 
21
  function getExternalUrl(spaceId) {
@@ -28,5 +34,5 @@ function getExternalUrl(spaceId) {
28
  }
29
 
30
  app.listen(port, () => {
31
- console.log(`Reverse proxy server running on ${baseUrl}`);
32
- });
 
1
  const express = require('express');
2
  const proxy = require('express-http-proxy');
3
  const app = express();
4
+
5
  const targetUrl = 'https://api.openai.com';
6
+ const openaiKey = process.env.OPENAI_KEY;
7
  const port = 7860;
8
  const baseUrl = getExternalUrl(process.env.SPACE_ID);
9
 
10
+ let requestCount = 0; // Counter to keep track of the number of proxied requests
11
+
12
+ app.use('/api', (req, res, next) => {
13
+ requestCount++; // Increment request count for every request passed to /api
14
+ next();
15
+ }, proxy(targetUrl, {
16
  proxyReqOptDecorator: (proxyReqOpts, srcReq) => {
17
  // Modify the request headers if necessary
18
+ proxyReqOpts.headers['Authorization'] = 'Bearer ' + openaiKey;
19
  return proxyReqOpts;
20
  },
21
  }));
22
 
23
  app.get("/", (req, res) => {
24
+ res.send(`This is your OpenAI Reverse Proxy URL: ${baseUrl}. Requests proxied so far: ${requestCount}`);
25
  });
26
 
27
  function getExternalUrl(spaceId) {
 
34
  }
35
 
36
  app.listen(port, () => {
37
+ console.log(`Reverse proxy server running on ${baseUrl}. Requests proxied so far: ${requestCount}`);
38
+ });