omnigate-ai 1.0.1 → 1.0.3

package/README.md

@@ -32,15 +32,66 @@ 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.
 
+## 🤖 Routing Your Custom AI Apps (Python & Node.js)
+
+Are you building your own AI apps? You can easily route them through OmniGate AI to track their token usage too! Just change the base URL in your official OpenAI SDK to point to your local proxy.
+
+**Python Example:**
+```python
+from openai import OpenAI
+
+client = OpenAI(
+    base_url="http://localhost:8080/v1", # Point to your proxy
+    api_key="your-gateway-secret-key"    # Use your local secret
+)
+```
+
+**Node.js Example:**
+```javascript
+import OpenAI from 'openai';
+
+const openai = new OpenAI({
+    baseURL: "http://localhost:8080/v1", // Point to your proxy
+    apiKey: "your-gateway-secret-key"    // Use your local secret
+});
+```
+
 ## ✨ Features
 
 - **🔒 100% Local Privacy:** Your real API keys are saved locally. Cloud-based AI tools never see them.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "omnigate-ai",
-  "version": "1.0.1",
+  "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/app.js

@@ -151,6 +151,40 @@ document.addEventListener('DOMContentLoaded', () => {
         console.error("Stream parse error:", err);
       }
     };
+
+    evtSource.addEventListener('clear_telemetry', () => {
+      // Reset state
+      totalTokens = 0;
+      totalInputTokens = 0;
+      totalOutputTokens = 0;
+      totalCost = 0.0;
+      totalRequests = 0;
+      requestTimestamps.length = 0;
+      
+      // Clear DOM
+      tbody.innerHTML = '';
+      elTotalTokens.textContent = '0';
+      elInputTokens.textContent = '0';
+      elOutputTokens.textContent = '0';
+      elTotalCost.textContent = '$0.000';
+      elAvgCost.textContent = '$0.000';
+      elTotalRequests.textContent = '0';
+      
+      updateGauges();
+    });
+  }
+
+  // ==========================================
+  // Reset Data Button
+  // ==========================================
+  const btnResetData = document.getElementById('btn-reset-data');
+  if (btnResetData) {
+    btnResetData.addEventListener('click', () => {
+      if (confirm('Are you sure you want to clear all telemetry history?')) {
+        fetch('/api/telemetry', { method: 'DELETE' })
+          .catch(err => console.error('Failed to clear telemetry:', err));
+      }
+    });
   }
 
   // ==========================================

package/public/index.html

@@ -27,9 +27,14 @@
           <span style="font-size:0.7rem;color:var(--text-dim);font-weight:400;display:block;margin-top:0.1rem;">Privacy-First API Gateway &middot; Token Cost Tracker</span>
         </div>
       </div>
-      <div class="status-indicator">
-        <span class="pulse-dot"></span>
-        <span>Gateway Connected</span>
+      <div class="header-actions" style="display: flex; gap: 1rem; align-items: center;">
+        <button id="btn-reset-data" class="btn-secondary" style="padding: 0.4rem 0.8rem; font-size: 0.8rem; border-color: rgba(255,8,68,0.5); color: #ffb199; background: rgba(255,8,68,0.1); border-radius: 50px;">
+          ⟲ Reset Data
+        </button>
+        <div class="status-indicator">
+          <span class="pulse-dot"></span>
+          <span>Gateway Connected</span>
+        </div>
       </div>
     </header>
 
@@ -202,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

@@ -84,6 +84,12 @@ app.get('/api/telemetry', (req, res) => {
   res.json(telemetryHistory);
 });
 
+app.delete('/api/telemetry', (req, res) => {
+  telemetryHistory.length = 0;
+  telemetryEmitter.emit('clear_telemetry', { status: 'cleared' });
+  res.json({ status: 'success', message: 'Telemetry history cleared' });
+});
+
 app.get('/api/telemetry/stream', (req, res) => {
   res.setHeader('Content-Type', 'text/event-stream');
   res.setHeader('Cache-Control', 'no-cache');
@@ -150,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);
@@ -564,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'
     }