Update handler.py

handler.py

@@ -65,26 +65,21 @@ class EndpointHandler:
         # 4. Return coordinates and their corresponding sensitivity scores
         return [{"coord": idx, "score": score} for idx, score in zip(indices, distances)]
 
-    def __call__(self, data):
-        """
-        Args:
-            data (:obj:`dict`):
-                - "inputs": the DNA sequence
-                - "method": "embed" (default) or "sensitivity"
-        """
-        inputs = data.get("inputs", "")
-        method = data.get("method", "embed")
-        logger.info("inputs: ")
-        logger.info(inputs)
-        logger.info(method)
+def __call__(self, data):
+    inputs = data.get("inputs", "")
+    method = data.get("method", "embed")
+    
+    # Extract parameters with sensible defaults
+    # This allows your LangGraph agent to "tune" the resolution
+    window_size = data.get("window_size", 50)
+    step = data.get("step", 100)
 
-        if not inputs:
-            return {"error": "No input sequence provided"}
+    if not inputs:
+        return {"error": "No input sequence provided"}
 
-        if method == "sensitivity":
-            # Returns the map of high-leverage coordinates
-            return self.map_sensitivity(inputs)
-        else:
-            # Standard embedding behavior
-            embedding = self._get_embedding(inputs)
-            return embedding.cpu().tolist()[0]
+    if method == "sensitivity":
+        # Pass the new parameters into the mapper
+        return self.map_sensitivity(inputs, window_size=window_size, step=step)
+    else:
+        embedding = self._get_embedding(inputs)
+        return embedding.cpu().tolist()[0]