Update server.js

server.js

@@ -2,7 +2,7 @@ const express = require('express');
 const proxy = require('express-http-proxy');
 const app = express();
 const bodyParser = require('body-parser');
-const targetUrl = 'https://openrouter.ai/';
+const targetUrl = 'https://openrouter.ai';
 const openaiKey = process.env.OPENAI_KEY;
 const proxyKey = process.env.PROXY_KEY; // Your secret proxy key
 const port = 7860;
@@ -10,8 +10,34 @@ const baseUrl = getExternalUrl(process.env.SPACE_ID);
 
 app.use(bodyParser.json({ limit: '50mb' }));
 
+// Middleware to log request details
+app.use((req, res, next) => {
+  console.log(`Incoming Request - Method: ${req.method}, URL: ${req.url}`);
+  console.log('Headers:', req.headers);
+  console.log('Body:', req.body);
+  next();
+});
+
+
+// Middleware to authenticate requests with the proxy key and check the model
+function authenticateProxyKeyAndModel(req, res, next) {
+  const providedKey = req.headers['auro']; // Assuming the key is sent in the 'x-proxy-key' header
+  const requestedModel = req.body.model;
+
+  // List of allowed models
+  const allowedModels = ['gryphe/mythomist-7b', 'gryphe/mythomax-l2-13b'];
+
+  if (providedKey && providedKey === proxyKey && allowedModels.includes(requestedModel)) {
+    // If the provided key matches the expected key and the requested model is allowed, allow the request to proceed
+    next();
+  } else {
+    // If the key is missing or incorrect, or the model is not allowed, reject the request with an error response
+    res.status(401).json({ error: 'Unauthorized or invalid model' });
+  }
+}
+
 
-app.use('/api', proxy(targetUrl, {
+app.use('/api', authenticateProxyKeyAndModel, proxy(targetUrl, {
   proxyReqOptDecorator: (proxyReqOpts, srcReq) => {
     // Modify the request headers if necessary