fix input

tdagent/tools/query_abuse_ip_db.py

@@ -121,7 +121,7 @@ def parse_response(response: Dict[str, Any]) -> Tuple[Optional[IPCheckResult], O
     return result, None
 
 
-def query_abuseipdb(ip_address: str, api_key: str, days: int) -> str:
+def query_abuseipdb(ip_address: str, days: int = 30) -> str:
     """
     Main function to query AbuseIPDB and format the response for Gradio
 
@@ -133,9 +133,11 @@ def query_abuseipdb(ip_address: str, api_key: str, days: int) -> str:
     Returns:
         A tuple containing (summary_text, reports_text)
     """
-    if not ip_address or not api_key:
+    if not ip_address:
         return "Please provide both an IP address and API key"
 
+    api_key = os.environ["ABUSEIPDB_KEY"]
+
     response = check_ip(ip_address, api_key, str(days))
     result, error = parse_response(response)
 
@@ -147,7 +149,7 @@ def query_abuseipdb(ip_address: str, api_key: str, days: int) -> str:
 
 gr_query_abuseipdb = gr.Interface(
     fn=query_abuseipdb,
-    inputs=["text", os.environ["ABUSEIPDB_KEY"]],
+    inputs=["text"],
     outputs="string",
     title="AbuseIPDB IP Checker",
     description="Check if an IP address has been reported for abusive behavior using AbuseIP DB API",