@dropout-ai/runtime 0.1.0 → 0.2.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dropout-ai/runtime",
-  "version": "0.1.0",
+  "version": "0.2.01",
   "description": "Invisible Node.js runtime for capturing AI interactions.",
   "main": "src/index.js",
   "scripts": {

package/src/index.js

@@ -1,11 +1,24 @@
+const originalFetch = global.fetch;
+
+// 4. Internal telemetry sender (Neutral naming)
+async function safeSendTelemetry(payload) {
+  try {
+    await originalFetch('http://localhost:4000/capture', {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify(payload)
+    });
+  } catch (e) {
+    // Fail silently
+  }
+}
+
 (function () {
   // 1. Guard global.fetch existence and idempotency
   if (typeof global.fetch !== 'function' || global.fetch.__dropout_patched__) {
     return;
   }
 
-  const originalFetch = global.fetch;
-
   // 2. Default Configuration
   let config = {
     version: 1,
@@ -27,30 +40,50 @@
     }
   }, 0);
 
-  // 4. Internal telemetry sender (Neutral naming)
-  async function safeSendTelemetry(payload) {
+
+  // 5. Detection Helpers
+  function matchesKnownProvider(url) {
+    return (
+      url &&
+      (
+        url.includes('openai.com/v1') ||
+        url.includes('api.anthropic.com') ||
+        url.includes('generativelanguage.googleapis.com') ||
+        url.includes('api-inference.huggingface.co')
+      )
+    );
+  }
+
+  function matchesGenericLLMPayload(init) {
     try {
-      await originalFetch('http://localhost:4000/capture', {
-        method: 'POST',
-        headers: { 'Content-Type': 'application/json' },
-        body: JSON.stringify(payload)
-      });
+      if (!init || typeof init.body !== 'string') return false;
+
+      const body = JSON.parse(init.body);
+
+      // Common LLM request shapes
+      return (
+        body.messages ||
+        body.prompt ||
+        body.input
+      );
     } catch (e) {
-      // Fail silently
+      return false;
     }
   }
 
-  // 5. The Monkey Patch
+  // 6. The Monkey Patch
   global.fetch = async function (input, init) {
     const start = Date.now();
     const response = await originalFetch(input, init);
 
     try {
-      // 6. Detection Logic (Dumb detection + Config Check)
+      // 7. Detection Logic (Dumb detection + Config Check)
       const url = typeof input === 'string' ? input : (input && input.url);
-      const isAIProvider = config.enabledProviders.some(p => url && url.includes(`${p}.com/v1`));
+      const isLikelyAI =
+        matchesKnownProvider(url) ||
+        matchesGenericLLMPayload(init);
 
-      if (url && isAIProvider) {
+      if (url && isLikelyAI) {
         const latency = Date.now() - start;
         const contentType = response.headers.get('content-type') || '';
         const isStream = contentType.includes('text/event-stream');
@@ -80,7 +113,8 @@
         }
 
         const payload = {
-          provider: 'openai',
+          provider: url.includes('openai.com') ? 'openai' : 'unknown',
+          confidence: matchesKnownProvider(url) ? 'high' : 'heuristic',
           url,
           model,
           prompt,
@@ -99,3 +133,45 @@
 
   global.fetch.__dropout_patched__ = true;
 })();
+
+// 11. Manual capture (combo mode)
+async function capture(target, options = {}) {
+  const start = Date.now();
+
+  try {
+    const result =
+      typeof target === 'function'
+        ? await target()
+        : await Promise.resolve(target);
+
+    const payload = {
+      provider: options.provider || 'manual',
+      model: options.model,
+      prompt: options.prompt,
+      output: typeof result === 'string'
+        ? result
+        : JSON.stringify(result).slice(0, 20000),
+      latency_ms: Date.now() - start,
+      timestamp: Math.floor(Date.now() / 1000),
+      mode: 'manual'
+    };
+
+    setTimeout(() => safeSendTelemetry(payload), 0);
+    return result;
+  } catch (error) {
+    const payload = {
+      provider: options.provider || 'manual',
+      error: error?.message || String(error),
+      latency_ms: Date.now() - start,
+      timestamp: Math.floor(Date.now() / 1000),
+      mode: 'manual'
+    };
+
+    setTimeout(() => safeSendTelemetry(payload), 0);
+    throw error;
+  }
+}
+
+module.exports = {
+  capture
+};