@kya-os/agentshield-nextjs 0.1.47 → 0.2.0

package/dist/index.mjs

@@ -1990,10 +1990,12 @@ function createEnhancedAgentShieldMiddleware(config = {}) {
 
 // src/api-client.ts
 var DEFAULT_BASE_URL = "https://kya.vouched.id";
+var EDGE_DETECT_URL = "https://detect.kya-os.ai";
 var DEFAULT_TIMEOUT = 5e3;
 var AgentShieldClient = class {
   apiKey;
   baseUrl;
+  useEdge;
   timeout;
   debug;
   constructor(config) {
@@ -2001,7 +2003,8 @@ var AgentShieldClient = class {
       throw new Error("AgentShield API key is required");
     }
     this.apiKey = config.apiKey;
-    this.baseUrl = config.baseUrl || DEFAULT_BASE_URL;
+    this.useEdge = config.useEdge !== false;
+    this.baseUrl = config.baseUrl || (this.useEdge ? EDGE_DETECT_URL : DEFAULT_BASE_URL);
     this.timeout = config.timeout || DEFAULT_TIMEOUT;
     this.debug = config.debug || false;
   }
@@ -2014,7 +2017,8 @@ var AgentShieldClient = class {
       const controller = new AbortController();
       const timeoutId = setTimeout(() => controller.abort(), this.timeout);
       try {
-        const response = await fetch(`${this.baseUrl}/api/v1/enforce`, {
+        const endpoint = this.useEdge ? `${this.baseUrl}/__detect/enforce` : `${this.baseUrl}/api/v1/enforce`;
+        const response = await fetch(endpoint, {
           method: "POST",
           headers: {
             "Content-Type": "application/json",
@@ -2101,6 +2105,8 @@ function getAgentShieldClient(config) {
     clientInstance = new AgentShieldClient({
       apiKey,
       baseUrl: config?.baseUrl || process.env.AGENTSHIELD_API_URL,
+      // Default to edge detection unless explicitly disabled
+      useEdge: config?.useEdge ?? process.env.AGENTSHIELD_USE_EDGE !== "false",
       timeout: config?.timeout,
       debug: config?.debug || process.env.AGENTSHIELD_DEBUG === "true"
     });
@@ -2167,6 +2173,7 @@ function withAgentShield(config = {}) {
       client = getAgentShieldClient({
         apiKey: config.apiKey,
         baseUrl: config.apiUrl,
+        useEdge: config.useEdge,
         timeout: config.timeout,
         debug: config.debug
       });