omnigate-ai 1.0.2 → 1.0.3

package/README.md

@@ -32,12 +32,39 @@ You will be greeted by a beautiful, glassmorphic UI where you can:
 
 ## 🛠️ Integrating with VS Code / AI Agents
 
-To route your AI coding assistant through OmniGate AI:
+### Continue Extension (VS Code)
+
+Replace your `~/.continue/config.yaml` with the following:
+
+```yaml
+name: OmniGate Local Config
+version: 1.0.0
+schema: v1
+
+models:
+  - name: DeepSeek (via OmniGate)
+    provider: openai
+    model: deepseek-v4-pro
+    apiBase: http://localhost:8080/v1/
+    apiKey: your-gateway-secret-key
+    roles:
+      - chat
+      - edit
+      - apply
+    requestOptions:
+      headers:
+        X-Gateway-Key: your-gateway-secret-key
+        X-Gateway-User-ID: your-name
+        X-Gateway-Project-ID: your-project-name
+```
+
+> **Important:** Continue v2.0+ requires `name` (not `title`), `roles`, and the top-level `schema: v1` field. Without these, you'll see a "config error" in the VS Code status bar.
+
+### Roo Code / Cline / Other Agents
 
-1. Open your AI Tool's settings (e.g., Roo Code or Continue).
-2. Set the **Provider** to `OpenAI Compatible`.
-3. Set the **Base URL** to `http://localhost:8080/v1`.
-4. Set the **API Key** to your **Gateway Secret Key** (the one you created in the dashboard, *not* your real OpenAI key).
+1. Set the **Provider** to `OpenAI Compatible`.
+2. Set the **Base URL** to `http://localhost:8080/v1`.
+3. Set the **API Key** to your **Gateway Secret Key** (the one you created in the dashboard, *not* your real OpenAI key).
 
 Now, start coding! OmniGate AI will silently intercept the requests, inject your real API key, forward it to the provider, and track your tokens on the dashboard.
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "omnigate-ai",
-  "version": "1.0.2",
+  "version": "1.0.3",
   "description": "Privacy-first, self-hosted LLM API Gateway and Proxy for tracking token usage.",
   "main": "server.js",
   "type": "module",

package/public/index.html

@@ -207,17 +207,25 @@
               <p>Add this model block to your <code>~/.continue/config.yaml</code> config file:</p>
               <div class="code-container">
                 <button class="copy-btn" onclick="copySnippet('continue-code', this)">Copy</button>
-                <pre id="continue-code">models:
+                <pre id="continue-code">name: OmniGate Local Config
+version: 1.0.0
+schema: v1
+
+models:
   - name: DeepSeek (via OmniGate)
     provider: openai
     model: deepseek-v4-pro
     apiBase: http://localhost:8080/v1/
-    apiKey: stub-agency-key-for-local-testing
+    apiKey: your-gateway-secret-key
+    roles:
+      - chat
+      - edit
+      - apply
     requestOptions:
       headers:
-        X-Gateway-Key: stub-agency-key-for-local-testing
-        X-Gateway-User-ID: developer-erfan
-        X-Gateway-Project-ID: vscode-project-omega</pre>
+        X-Gateway-Key: your-gateway-secret-key
+        X-Gateway-User-ID: developer-name
+        X-Gateway-Project-ID: vscode-project-name</pre>
               </div>
             </div>
 

package/server.js

@@ -156,7 +156,14 @@ app.use((req, res, next) => {
     return next();
   }
 
-  const incomingGatewayKey = req.headers['x-gateway-key'] || '';
+  let incomingGatewayKey = req.headers['x-gateway-key'] || '';
+  if (!incomingGatewayKey && req.headers.authorization) {
+    const authHeader = req.headers.authorization;
+    if (authHeader.startsWith('Bearer ')) {
+      incomingGatewayKey = authHeader.slice(7);
+    }
+  }
+
   const expectedKey = activeConfig.agencyGatewayKey || '';
   
   const incomingBuffer = Buffer.from(incomingGatewayKey);
@@ -570,11 +577,38 @@ app.post('/v1/messages', async (req, res) => {
   }
 });
 
+// 4. OpenAI Models Proxy Handler
+app.get('/v1/models', (req, res) => {
+  res.json({
+    object: 'list',
+    data: [
+      {
+        id: 'deepseek-v4-pro',
+        object: 'model',
+        created: Date.now(),
+        owned_by: 'OmniGate'
+      },
+      {
+        id: 'gpt-4o',
+        object: 'model',
+        created: Date.now(),
+        owned_by: 'OmniGate'
+      },
+      {
+        id: 'gpt-3.5-turbo',
+        object: 'model',
+        created: Date.now(),
+        owned_by: 'OmniGate'
+      }
+    ]
+  });
+});
+
 // Catch-all route to reject unsupported endpoints
 app.use((req, res) => {
   res.status(404).json({
     error: {
-      message: `The endpoint ${req.method} ${req.path} is not supported by OmniGate AI gateway. Only /v1/chat/completions (OpenAI) and /v1/messages (Anthropic) are valid proxy routes.`,
+      message: `The endpoint ${req.method} ${req.path} is not supported by OmniGate AI gateway. Only /v1/chat/completions (OpenAI), /v1/models, and /v1/messages (Anthropic) are valid proxy routes.`,
       type: 'gateway_route_error',
       code: 'unsupported_route'
     }