hackagent 0.1.0__py3-none-any.whl

hackagent/models/run.py

@@ -0,0 +1,233 @@
+import datetime
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar, Union, cast
+from uuid import UUID
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+from dateutil.parser import isoparse
+
+from ..models.status_enum import StatusEnum
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.result import Result
+
+
+T = TypeVar("T", bound="Run")
+
+
+@_attrs_define
+class Run:
+    """Serializer for the Run model, used for both input and output.
+
+    Attributes:
+        id (UUID):
+        agent (UUID):
+        agent_name (str):
+        owner (Union[None, int]):
+        owner_username (Union[None, str]):
+        organization (UUID):
+        organization_name (str):
+        timestamp (datetime.datetime):
+        is_client_executed (bool): Indicates if the run was initiated via an Attack by a client application.
+        results (list['Result']):
+        attack (Union[None, UUID, Unset]): The Attack this run is an instance of, if applicable.
+        run_config (Union[Unset, Any]): JSON containing specific settings for this run. If linked to an Attack, this
+            might be a copy or subset of its configuration.
+        status (Union[Unset, StatusEnum]): * `PENDING` - Pending
+            * `RUNNING` - Running
+            * `COMPLETED` - Completed
+            * `FAILED` - Failed
+            * `CANCELLED` - Cancelled
+        run_notes (Union[Unset, str]):
+    """
+
+    id: UUID
+    agent: UUID
+    agent_name: str
+    owner: Union[None, int]
+    owner_username: Union[None, str]
+    organization: UUID
+    organization_name: str
+    timestamp: datetime.datetime
+    is_client_executed: bool
+    results: list["Result"]
+    attack: Union[None, UUID, Unset] = UNSET
+    run_config: Union[Unset, Any] = UNSET
+    status: Union[Unset, StatusEnum] = UNSET
+    run_notes: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        id = str(self.id)
+
+        agent = str(self.agent)
+
+        agent_name = self.agent_name
+
+        owner: Union[None, int]
+        owner = self.owner
+
+        owner_username: Union[None, str]
+        owner_username = self.owner_username
+
+        organization = str(self.organization)
+
+        organization_name = self.organization_name
+
+        timestamp = self.timestamp.isoformat()
+
+        is_client_executed = self.is_client_executed
+
+        results = []
+        for results_item_data in self.results:
+            results_item = results_item_data.to_dict()
+            results.append(results_item)
+
+        attack: Union[None, Unset, str]
+        if isinstance(self.attack, Unset):
+            attack = UNSET
+        elif isinstance(self.attack, UUID):
+            attack = str(self.attack)
+        else:
+            attack = self.attack
+
+        run_config = self.run_config
+
+        status: Union[Unset, str] = UNSET
+        if not isinstance(self.status, Unset):
+            status = self.status.value
+
+        run_notes = self.run_notes
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "id": id,
+                "agent": agent,
+                "agent_name": agent_name,
+                "owner": owner,
+                "owner_username": owner_username,
+                "organization": organization,
+                "organization_name": organization_name,
+                "timestamp": timestamp,
+                "is_client_executed": is_client_executed,
+                "results": results,
+            }
+        )
+        if attack is not UNSET:
+            field_dict["attack"] = attack
+        if run_config is not UNSET:
+            field_dict["run_config"] = run_config
+        if status is not UNSET:
+            field_dict["status"] = status
+        if run_notes is not UNSET:
+            field_dict["run_notes"] = run_notes
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.result import Result
+
+        d = dict(src_dict)
+        id = UUID(d.pop("id"))
+
+        agent = UUID(d.pop("agent"))
+
+        agent_name = d.pop("agent_name")
+
+        def _parse_owner(data: object) -> Union[None, int]:
+            if data is None:
+                return data
+            return cast(Union[None, int], data)
+
+        owner = _parse_owner(d.pop("owner"))
+
+        def _parse_owner_username(data: object) -> Union[None, str]:
+            if data is None:
+                return data
+            return cast(Union[None, str], data)
+
+        owner_username = _parse_owner_username(d.pop("owner_username"))
+
+        organization = UUID(d.pop("organization"))
+
+        organization_name = d.pop("organization_name")
+
+        timestamp = isoparse(d.pop("timestamp"))
+
+        is_client_executed = d.pop("is_client_executed")
+
+        results = []
+        _results = d.pop("results")
+        for results_item_data in _results:
+            results_item = Result.from_dict(results_item_data)
+
+            results.append(results_item)
+
+        def _parse_attack(data: object) -> Union[None, UUID, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, str):
+                    raise TypeError()
+                attack_type_0 = UUID(data)
+
+                return attack_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[None, UUID, Unset], data)
+
+        attack = _parse_attack(d.pop("attack", UNSET))
+
+        run_config = d.pop("run_config", UNSET)
+
+        _status = d.pop("status", UNSET)
+        status: Union[Unset, StatusEnum]
+        if isinstance(_status, Unset):
+            status = UNSET
+        else:
+            status = StatusEnum(_status)
+
+        run_notes = d.pop("run_notes", UNSET)
+
+        run = cls(
+            id=id,
+            agent=agent,
+            agent_name=agent_name,
+            owner=owner,
+            owner_username=owner_username,
+            organization=organization,
+            organization_name=organization_name,
+            timestamp=timestamp,
+            is_client_executed=is_client_executed,
+            results=results,
+            attack=attack,
+            run_config=run_config,
+            status=status,
+            run_notes=run_notes,
+        )
+
+        run.additional_properties = d
+        return run
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

hackagent/models/run_list_status.py

@@ -0,0 +1,12 @@
+from enum import Enum
+
+
+class RunListStatus(str, Enum):
+    CANCELLED = "CANCELLED"
+    COMPLETED = "COMPLETED"
+    FAILED = "FAILED"
+    PENDING = "PENDING"
+    RUNNING = "RUNNING"
+
+    def __str__(self) -> str:
+        return str(self.value)

hackagent/models/run_request.py

@@ -0,0 +1,133 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar, Union, cast
+from uuid import UUID
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.status_enum import StatusEnum
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="RunRequest")
+
+
+@_attrs_define
+class RunRequest:
+    """Serializer for the Run model, used for both input and output.
+
+    Attributes:
+        agent (UUID):
+        attack (Union[None, UUID, Unset]): The Attack this run is an instance of, if applicable.
+        run_config (Union[Unset, Any]): JSON containing specific settings for this run. If linked to an Attack, this
+            might be a copy or subset of its configuration.
+        status (Union[Unset, StatusEnum]): * `PENDING` - Pending
+            * `RUNNING` - Running
+            * `COMPLETED` - Completed
+            * `FAILED` - Failed
+            * `CANCELLED` - Cancelled
+        run_notes (Union[Unset, str]):
+    """
+
+    agent: UUID
+    attack: Union[None, UUID, Unset] = UNSET
+    run_config: Union[Unset, Any] = UNSET
+    status: Union[Unset, StatusEnum] = UNSET
+    run_notes: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        agent = str(self.agent)
+
+        attack: Union[None, Unset, str]
+        if isinstance(self.attack, Unset):
+            attack = UNSET
+        elif isinstance(self.attack, UUID):
+            attack = str(self.attack)
+        else:
+            attack = self.attack
+
+        run_config = self.run_config
+
+        status: Union[Unset, str] = UNSET
+        if not isinstance(self.status, Unset):
+            status = self.status.value
+
+        run_notes = self.run_notes
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "agent": agent,
+            }
+        )
+        if attack is not UNSET:
+            field_dict["attack"] = attack
+        if run_config is not UNSET:
+            field_dict["run_config"] = run_config
+        if status is not UNSET:
+            field_dict["status"] = status
+        if run_notes is not UNSET:
+            field_dict["run_notes"] = run_notes
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        agent = UUID(d.pop("agent"))
+
+        def _parse_attack(data: object) -> Union[None, UUID, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, str):
+                    raise TypeError()
+                attack_type_0 = UUID(data)
+
+                return attack_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[None, UUID, Unset], data)
+
+        attack = _parse_attack(d.pop("attack", UNSET))
+
+        run_config = d.pop("run_config", UNSET)
+
+        _status = d.pop("status", UNSET)
+        status: Union[Unset, StatusEnum]
+        if isinstance(_status, Unset):
+            status = UNSET
+        else:
+            status = StatusEnum(_status)
+
+        run_notes = d.pop("run_notes", UNSET)
+
+        run_request = cls(
+            agent=agent,
+            attack=attack,
+            run_config=run_config,
+            status=status,
+            run_notes=run_notes,
+        )
+
+        run_request.additional_properties = d
+        return run_request
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

hackagent/models/status_enum.py

@@ -0,0 +1,12 @@
+from enum import Enum
+
+
+class StatusEnum(str, Enum):
+    CANCELLED = "CANCELLED"
+    COMPLETED = "COMPLETED"
+    FAILED = "FAILED"
+    PENDING = "PENDING"
+    RUNNING = "RUNNING"
+
+    def __str__(self) -> str:
+        return str(self.value)

hackagent/models/step_type_enum.py

@@ -0,0 +1,14 @@
+from enum import Enum
+
+
+class StepTypeEnum(str, Enum):
+    A2A_COMM = "A2A_COMM"
+    AGENT_RESPONSE_CHUNK = "AGENT_RESPONSE_CHUNK"
+    AGENT_THOUGHT = "AGENT_THOUGHT"
+    MCP_STEP = "MCP_STEP"
+    OTHER = "OTHER"
+    TOOL_CALL = "TOOL_CALL"
+    TOOL_RESPONSE = "TOOL_RESPONSE"
+
+    def __str__(self) -> str:
+        return str(self.value)

hackagent/models/trace.py

@@ -0,0 +1,121 @@
+import datetime
+from collections.abc import Mapping
+from typing import Any, TypeVar, Union
+from uuid import UUID
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+from dateutil.parser import isoparse
+
+from ..models.step_type_enum import StepTypeEnum
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="Trace")
+
+
+@_attrs_define
+class Trace:
+    """Serializer for the Trace model.
+
+    Attributes:
+        id (int):
+        result (UUID):
+        sequence (int):
+        timestamp (datetime.datetime):
+        step_type (Union[Unset, StepTypeEnum]): * `TOOL_CALL` - Tool Call
+            * `TOOL_RESPONSE` - Tool Response
+            * `AGENT_THOUGHT` - Agent Thought/Reasoning
+            * `AGENT_RESPONSE_CHUNK` - Agent Response Chunk
+            * `OTHER` - Other
+            * `MCP_STEP` - Multi-Context Prompting Step
+            * `A2A_COMM` - Agent-to-Agent Communication
+        content (Union[Unset, Any]):
+    """
+
+    id: int
+    result: UUID
+    sequence: int
+    timestamp: datetime.datetime
+    step_type: Union[Unset, StepTypeEnum] = UNSET
+    content: Union[Unset, Any] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        id = self.id
+
+        result = str(self.result)
+
+        sequence = self.sequence
+
+        timestamp = self.timestamp.isoformat()
+
+        step_type: Union[Unset, str] = UNSET
+        if not isinstance(self.step_type, Unset):
+            step_type = self.step_type.value
+
+        content = self.content
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "id": id,
+                "result": result,
+                "sequence": sequence,
+                "timestamp": timestamp,
+            }
+        )
+        if step_type is not UNSET:
+            field_dict["step_type"] = step_type
+        if content is not UNSET:
+            field_dict["content"] = content
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        id = d.pop("id")
+
+        result = UUID(d.pop("result"))
+
+        sequence = d.pop("sequence")
+
+        timestamp = isoparse(d.pop("timestamp"))
+
+        _step_type = d.pop("step_type", UNSET)
+        step_type: Union[Unset, StepTypeEnum]
+        if isinstance(_step_type, Unset):
+            step_type = UNSET
+        else:
+            step_type = StepTypeEnum(_step_type)
+
+        content = d.pop("content", UNSET)
+
+        trace = cls(
+            id=id,
+            result=result,
+            sequence=sequence,
+            timestamp=timestamp,
+            step_type=step_type,
+            content=content,
+        )
+
+        trace.additional_properties = d
+        return trace
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

hackagent/models/trace_request.py

@@ -0,0 +1,94 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.step_type_enum import StepTypeEnum
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="TraceRequest")
+
+
+@_attrs_define
+class TraceRequest:
+    """Serializer for the Trace model.
+
+    Attributes:
+        sequence (int):
+        step_type (Union[Unset, StepTypeEnum]): * `TOOL_CALL` - Tool Call
+            * `TOOL_RESPONSE` - Tool Response
+            * `AGENT_THOUGHT` - Agent Thought/Reasoning
+            * `AGENT_RESPONSE_CHUNK` - Agent Response Chunk
+            * `OTHER` - Other
+            * `MCP_STEP` - Multi-Context Prompting Step
+            * `A2A_COMM` - Agent-to-Agent Communication
+        content (Union[Unset, Any]):
+    """
+
+    sequence: int
+    step_type: Union[Unset, StepTypeEnum] = UNSET
+    content: Union[Unset, Any] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        sequence = self.sequence
+
+        step_type: Union[Unset, str] = UNSET
+        if not isinstance(self.step_type, Unset):
+            step_type = self.step_type.value
+
+        content = self.content
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "sequence": sequence,
+            }
+        )
+        if step_type is not UNSET:
+            field_dict["step_type"] = step_type
+        if content is not UNSET:
+            field_dict["content"] = content
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        sequence = d.pop("sequence")
+
+        _step_type = d.pop("step_type", UNSET)
+        step_type: Union[Unset, StepTypeEnum]
+        if isinstance(_step_type, Unset):
+            step_type = UNSET
+        else:
+            step_type = StepTypeEnum(_step_type)
+
+        content = d.pop("content", UNSET)
+
+        trace_request = cls(
+            sequence=sequence,
+            step_type=step_type,
+            content=content,
+        )
+
+        trace_request.additional_properties = d
+        return trace_request
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties