hackagent 0.1.0__tar.gz → 0.2.0__tar.gz

{hackagent-0.1.0 → hackagent-0.2.0}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.3
 Name: hackagent
-Version: 0.1.0
-Summary: Python client for interacting with the HackAgent API.
+Version: 0.2.0
+Summary: HackAgent is an open-source security toolkit to detect vulnerabilities of your AI Agents.
 Author: Nicola Franco
 Author-email: nicola@vista-labs.ai
 Requires-Python: >=3.10,<4.0
@@ -10,8 +10,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Requires-Dist: google-adk (>=0.5.0,<0.6.0)
-Requires-Dist: litellm (>=1.69.1,<2.0.0)
+Requires-Dist: litellm (>=1.69.2,<2.0.0)
 Requires-Dist: pandas (>=2.2.3,<3.0.0)
 Requires-Dist: pydantic (>=2.0,<3.0)
 Requires-Dist: python-dotenv (>=1.1.0,<2.0.0)

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/agent.py

@@ -52,17 +52,21 @@ class HackAgent:
         name: str = None,
         agent_type: AgentTypeEnum = AgentTypeEnum.UNKNOWN,
         base_url: Optional[str] = None,
-        token: Optional[str] = None,
+        api_key: Optional[str] = None,
         predefined_prompts: Optional[Dict[str, Tuple[str, str]]] = None,
         raise_on_unexpected_status: bool = False,
         timeout: Optional[float] = None,
         env_file_path: Optional[str] = None,
     ):
-        display_hackagent_splash()  # Display the splash screen on init
+        display_hackagent_splash()
+
+        resolved_auth_token = self._resolve_api_token(
+            direct_api_key_param=api_key, env_file_path=env_file_path
+        )
 
         self.client = AuthenticatedClient(
             base_url=base_url,
-            token=self._resolve_api_token(token, env_file_path),
+            token=resolved_auth_token,
             prefix="Api-Key",
             raise_on_unexpected_status=raise_on_unexpected_status,
             timeout=timeout,
@@ -85,33 +89,38 @@ class HackAgent:
         }
 
     def _resolve_api_token(
-        self, token: Optional[str], env_file_path: Optional[str]
+        self, direct_api_key_param: Optional[str], env_file_path: Optional[str]
     ) -> str:
-        """Resolves the API token from direct input or environment variables."""
-        api_token_resolved = token
-        if api_token_resolved is None:
-            logger.debug(
-                "API token not provided directly, attempting to load from environment."
-            )
-            dotenv_to_load = env_file_path or find_dotenv(usecwd=True)
+        """Resolves the API token from the direct api_key parameter or environment variables."""
+        if direct_api_key_param is not None:
+            logger.debug("Using API token provided directly via 'api_key' parameter.")
+            return direct_api_key_param
+
+        # If direct_api_key_param is None, attempt to load from environment.
+        logger.debug(
+            "API token not provided via 'api_key' parameter, attempting to load from environment."
+        )
+        dotenv_to_load = env_file_path or find_dotenv(usecwd=True)
 
-            if dotenv_to_load:
-                logger.debug(f"Loading .env file from: {dotenv_to_load}")
-                load_dotenv(dotenv_to_load)
-            else:
-                logger.debug("No .env file found to load.")
+        if dotenv_to_load:
+            logger.debug(f"Loading .env file from: {dotenv_to_load}")
+            load_dotenv(dotenv_to_load)
+        else:
+            logger.debug("No .env file found to load.")
 
-            api_token_resolved = os.getenv("HACKAGENT_API_TOKEN")
+        api_token_resolved = os.getenv("HACKAGENT_API_KEY")
 
         if not api_token_resolved:
             error_message = (
-                "API token not provided and not found in HACKAGENT_API_TOKEN "
-                "environment variable (after attempting to load .env)."
+                "API token not provided via 'api_key' parameter, "
+                "and not found in HACKAGENT_API_KEY environment variable "
+                "(after attempting to load .env)."
             )
             raise ValueError(error_message)
+        logger.debug("Using API token from HACKAGENT_API_KEY environment variable.")
         return api_token_resolved
 
-    async def hack(
+    def hack(
         self,
         attack_config: Dict[str, Any],
         run_config_override: Optional[Dict[str, Any]] = None,
@@ -163,7 +172,7 @@ class HackAgent:
                 f"Using Victim Backend Agent ID: {backend_agent.id} for '{backend_agent.name}'"
             )
 
-            return await strategy.execute(
+            return strategy.execute(
                 attack_config=attack_config,
                 run_config_override=run_config_override,
                 fail_on_run_error=fail_on_run_error,

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/agent/agent_destroy.py

@@ -14,7 +14,9 @@ def _get_kwargs(
 ) -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
         "method": "delete",
-        "url": f"/api/agent/{id}",
+        "url": "/api/agent/{id}".format(
+            id=id,
+        ),
     }
 
     return _kwargs

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/agent/agent_partial_update.py

@@ -20,7 +20,9 @@ def _get_kwargs(
 
     _kwargs: dict[str, Any] = {
         "method": "patch",
-        "url": f"/api/agent/{id}",
+        "url": "/api/agent/{id}".format(
+            id=id,
+        ),
     }
 
     _body = body.to_dict()

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/agent/agent_retrieve.py

@@ -15,7 +15,9 @@ def _get_kwargs(
 ) -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
         "method": "get",
-        "url": f"/api/agent/{id}",
+        "url": "/api/agent/{id}".format(
+            id=id,
+        ),
     }
 
     return _kwargs

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/agent/agent_update.py

@@ -20,7 +20,9 @@ def _get_kwargs(
 
     _kwargs: dict[str, Any] = {
         "method": "put",
-        "url": f"/api/agent/{id}",
+        "url": "/api/agent/{id}".format(
+            id=id,
+        ),
     }
 
     _body = body.to_dict()

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/attack/attack_destroy.py

@@ -14,7 +14,9 @@ def _get_kwargs(
 ) -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
         "method": "delete",
-        "url": f"/api/attack/{id}",
+        "url": "/api/attack/{id}".format(
+            id=id,
+        ),
     }
 
     return _kwargs

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/attack/attack_partial_update.py

@@ -20,7 +20,9 @@ def _get_kwargs(
 
     _kwargs: dict[str, Any] = {
         "method": "patch",
-        "url": f"/api/attack/{id}",
+        "url": "/api/attack/{id}".format(
+            id=id,
+        ),
     }
 
     _body = body.to_dict()

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/attack/attack_retrieve.py

@@ -15,7 +15,9 @@ def _get_kwargs(
 ) -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
         "method": "get",
-        "url": f"/api/attack/{id}",
+        "url": "/api/attack/{id}".format(
+            id=id,
+        ),
     }
 
     return _kwargs

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/attack/attack_update.py

@@ -20,7 +20,9 @@ def _get_kwargs(
 
     _kwargs: dict[str, Any] = {
         "method": "put",
-        "url": f"/api/attack/{id}",
+        "url": "/api/attack/{id}".format(
+            id=id,
+        ),
     }
 
     _body = body.to_dict()

hackagent-0.2.0/hackagent/api/generator/generator_create.py

@@ -0,0 +1,99 @@
+from http import HTTPStatus
+from typing import Any, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...types import Response
+
+
+def _get_kwargs() -> dict[str, Any]:
+    _kwargs: dict[str, Any] = {
+        "method": "post",
+        "url": "/api/generator",
+    }
+
+    return _kwargs
+
+
+def _parse_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Optional[Any]:
+    if response.status_code == 200:
+        return None
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Response[Any]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    *,
+    client: AuthenticatedClient,
+) -> Response[Any]:
+    r"""Proxies POST requests to the configured OpenRouter generator model.
+    Requires a valid User API Key for access.
+    The client should send a POST request with a JSON body in the same format
+    as expected by LiteLLM or OpenRouter's /chat/completions endpoint,
+    including a \"model\" field.
+    Note: The \"model\" field provided by the client in the request body will be
+    overridden by the server-configured generator model ID for the actual call to OpenRouter.
+    e.g., {\"model\": \"client_specified_model_name\", \"messages\": [{\"role\": \"user\", \"content\":
+    \"Hello!\"}], \"stream\": False}
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Any]
+    """
+
+    kwargs = _get_kwargs()
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio_detailed(
+    *,
+    client: AuthenticatedClient,
+) -> Response[Any]:
+    r"""Proxies POST requests to the configured OpenRouter generator model.
+    Requires a valid User API Key for access.
+    The client should send a POST request with a JSON body in the same format
+    as expected by LiteLLM or OpenRouter's /chat/completions endpoint,
+    including a \"model\" field.
+    Note: The \"model\" field provided by the client in the request body will be
+    overridden by the server-configured generator model ID for the actual call to OpenRouter.
+    e.g., {\"model\": \"client_specified_model_name\", \"messages\": [{\"role\": \"user\", \"content\":
+    \"Hello!\"}], \"stream\": False}
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Any]
+    """
+
+    kwargs = _get_kwargs()
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)

hackagent-0.2.0/hackagent/api/judge/judge_create.py

@@ -0,0 +1,99 @@
+from http import HTTPStatus
+from typing import Any, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...types import Response
+
+
+def _get_kwargs() -> dict[str, Any]:
+    _kwargs: dict[str, Any] = {
+        "method": "post",
+        "url": "/api/judge",
+    }
+
+    return _kwargs
+
+
+def _parse_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Optional[Any]:
+    if response.status_code == 200:
+        return None
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Response[Any]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    *,
+    client: AuthenticatedClient,
+) -> Response[Any]:
+    r"""Proxies POST requests to the configured OpenRouter judge model.
+    Requires a valid User API Key for access.
+    The client should send a POST request with a JSON body in the same format
+    as expected by LiteLLM or OpenRouter's /chat/completions endpoint,
+    including a \"model\" field.
+    Note: The \"model\" field provided by the client in the request body will be
+    overridden by the server-configured judge model ID for the actual call to OpenRouter.
+    e.g., {\"model\": \"client_specified_model_name\", \"messages\": [{\"role\": \"user\", \"content\":
+    \"Is this good?\"}], \"stream\": False}
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Any]
+    """
+
+    kwargs = _get_kwargs()
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio_detailed(
+    *,
+    client: AuthenticatedClient,
+) -> Response[Any]:
+    r"""Proxies POST requests to the configured OpenRouter judge model.
+    Requires a valid User API Key for access.
+    The client should send a POST request with a JSON body in the same format
+    as expected by LiteLLM or OpenRouter's /chat/completions endpoint,
+    including a \"model\" field.
+    Note: The \"model\" field provided by the client in the request body will be
+    overridden by the server-configured judge model ID for the actual call to OpenRouter.
+    e.g., {\"model\": \"client_specified_model_name\", \"messages\": [{\"role\": \"user\", \"content\":
+    \"Is this good?\"}], \"stream\": False}
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Any]
+    """
+
+    kwargs = _get_kwargs()
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/key/key_destroy.py

@@ -13,7 +13,9 @@ def _get_kwargs(
 ) -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
         "method": "delete",
-        "url": f"/api/key/{prefix}",
+        "url": "/api/key/{prefix}".format(
+            prefix=prefix,
+        ),
     }
 
     return _kwargs

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/key/key_retrieve.py

@@ -14,7 +14,9 @@ def _get_kwargs(
 ) -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
         "method": "get",
-        "url": f"/api/key/{prefix}",
+        "url": "/api/key/{prefix}".format(
+            prefix=prefix,
+        ),
     }
 
     return _kwargs

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/prompt/prompt_destroy.py

@@ -14,7 +14,9 @@ def _get_kwargs(
 ) -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
         "method": "delete",
-        "url": f"/api/prompt/{id}",
+        "url": "/api/prompt/{id}".format(
+            id=id,
+        ),
     }
 
     return _kwargs

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/prompt/prompt_partial_update.py

@@ -20,7 +20,9 @@ def _get_kwargs(
 
     _kwargs: dict[str, Any] = {
         "method": "patch",
-        "url": f"/api/prompt/{id}",
+        "url": "/api/prompt/{id}".format(
+            id=id,
+        ),
     }
 
     _body = body.to_dict()

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/prompt/prompt_retrieve.py

@@ -15,7 +15,9 @@ def _get_kwargs(
 ) -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
         "method": "get",
-        "url": f"/api/prompt/{id}",
+        "url": "/api/prompt/{id}".format(
+            id=id,
+        ),
     }
 
     return _kwargs

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/prompt/prompt_update.py

@@ -20,7 +20,9 @@ def _get_kwargs(
 
     _kwargs: dict[str, Any] = {
         "method": "put",
-        "url": f"/api/prompt/{id}",
+        "url": "/api/prompt/{id}".format(
+            id=id,
+        ),
     }
 
     _body = body.to_dict()

hackagent-0.2.0/hackagent/api/result/__init__.py

@@ -0,0 +1 @@
+"""Contains endpoint functions for accessing the API"""

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/result/result_destroy.py

@@ -14,7 +14,9 @@ def _get_kwargs(
 ) -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
         "method": "delete",
-        "url": f"/api/result/{id}",
+        "url": "/api/result/{id}".format(
+            id=id,
+        ),
     }
 
     return _kwargs

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/result/result_partial_update.py

@@ -20,7 +20,9 @@ def _get_kwargs(
 
     _kwargs: dict[str, Any] = {
         "method": "patch",
-        "url": f"/api/result/{id}",
+        "url": "/api/result/{id}".format(
+            id=id,
+        ),
     }
 
     _body = body.to_dict()

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/result/result_retrieve.py

@@ -15,7 +15,9 @@ def _get_kwargs(
 ) -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
         "method": "get",
-        "url": f"/api/result/{id}",
+        "url": "/api/result/{id}".format(
+            id=id,
+        ),
     }
 
     return _kwargs

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/result/result_trace_create.py

@@ -20,7 +20,9 @@ def _get_kwargs(
 
     _kwargs: dict[str, Any] = {
         "method": "post",
-        "url": f"/api/result/{id}/trace",
+        "url": "/api/result/{id}/trace".format(
+            id=id,
+        ),
     }
 
     _body = body.to_dict()
@@ -63,7 +65,7 @@ def sync_detailed(
     body: TraceRequest,
 ) -> Response[Trace]:
     """Creates a new Trace associated with this Result.
-    The result instance is fetched using the 'pk' from the URL.
+    The result instance is fetched using the 'id' (the lookup_field) from the URL.
 
     Args:
         id (UUID):
@@ -96,7 +98,7 @@ def sync(
     body: TraceRequest,
 ) -> Optional[Trace]:
     """Creates a new Trace associated with this Result.
-    The result instance is fetched using the 'pk' from the URL.
+    The result instance is fetched using the 'id' (the lookup_field) from the URL.
 
     Args:
         id (UUID):
@@ -124,7 +126,7 @@ async def asyncio_detailed(
     body: TraceRequest,
 ) -> Response[Trace]:
     """Creates a new Trace associated with this Result.
-    The result instance is fetched using the 'pk' from the URL.
+    The result instance is fetched using the 'id' (the lookup_field) from the URL.
 
     Args:
         id (UUID):
@@ -155,7 +157,7 @@ async def asyncio(
     body: TraceRequest,
 ) -> Optional[Trace]:
     """Creates a new Trace associated with this Result.
-    The result instance is fetched using the 'pk' from the URL.
+    The result instance is fetched using the 'id' (the lookup_field) from the URL.
 
     Args:
         id (UUID):

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/result/result_update.py

@@ -20,7 +20,9 @@ def _get_kwargs(
 
     _kwargs: dict[str, Any] = {
         "method": "put",
-        "url": f"/api/result/{id}",
+        "url": "/api/result/{id}".format(
+            id=id,
+        ),
     }
 
     _body = body.to_dict()

hackagent-0.2.0/hackagent/api/run/__init__.py

@@ -0,0 +1 @@
+"""Contains endpoint functions for accessing the API"""

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/run/run_destroy.py

@@ -14,7 +14,9 @@ def _get_kwargs(
 ) -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
         "method": "delete",
-        "url": f"/api/run/{id}",
+        "url": "/api/run/{id}".format(
+            id=id,
+        ),
     }
 
     return _kwargs

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/run/run_partial_update.py

@@ -20,7 +20,9 @@ def _get_kwargs(
 
     _kwargs: dict[str, Any] = {
         "method": "patch",
-        "url": f"/api/run/{id}",
+        "url": "/api/run/{id}".format(
+            id=id,
+        ),
     }
 
     _body = body.to_dict()

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/run/run_result_create.py

@@ -20,7 +20,9 @@ def _get_kwargs(
 
     _kwargs: dict[str, Any] = {
         "method": "post",
-        "url": f"/api/run/{id}/result",
+        "url": "/api/run/{id}/result".format(
+            id=id,
+        ),
     }
 
     _body = body.to_dict()

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/run/run_retrieve.py

@@ -15,7 +15,9 @@ def _get_kwargs(
 ) -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
         "method": "get",
-        "url": f"/api/run/{id}",
+        "url": "/api/run/{id}".format(
+            id=id,
+        ),
     }
 
     return _kwargs

{hackagent-0.1.0 → hackagent-0.2.0}/hackagent/api/run/run_update.py

@@ -20,7 +20,9 @@ def _get_kwargs(
 
     _kwargs: dict[str, Any] = {
         "method": "put",
-        "url": f"/api/run/{id}",
+        "url": "/api/run/{id}".format(
+            id=id,
+        ),
     }
 
     _body = body.to_dict()