hackagent 0.3.1__py3-none-any.whl

hackagent/models/agent.py

@@ -0,0 +1,223 @@
+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 ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.organization_minimal import OrganizationMinimal
+    from ..models.user_profile_minimal import UserProfileMinimal
+
+
+T = TypeVar("T", bound="Agent")
+
+
+@_attrs_define
+class Agent:
+    """Serializes Agent model instances to JSON and validates data for creating
+    or updating Agent instances.
+
+    This serializer provides a comprehensive representation of an Agent,
+    including its type, endpoint, and nested details for related 'organization'
+    and 'owner' for read operations, while allowing 'organization' and 'owner' IDs
+    for write operations.
+
+    Attributes:
+        organization_detail (OrganizationMinimalSerializer): Read-only nested
+            serializer for the agent's organization. Displays minimal details.
+        owner_detail (UserProfileMinimalSerializer): Read-only nested serializer
+            for the agent's owner's user profile. Displays minimal details.
+            Can be null if the agent has no owner or the owner has no profile.
+        agent_type (CharField): The type of the agent as a string
+                          (e.g., LITELLM, OPENAI_SDK, GOOGLE_ADK).
+
+    Meta:
+        model (Agent): The model class that this serializer works with.
+        fields (tuple): The fields to include in the serialized output.
+            Includes standard Agent fields like 'endpoint', 'type',
+            and the read-only nested details.
+        read_only_fields (tuple): Fields that are read-only and cannot be
+            set during create/update operations through this serializer.
+            This includes 'id', 'created_at', 'updated_at', and the
+            nested detail fields.
+
+        Attributes:
+            id (UUID):
+            name (str):
+            endpoint (str): The primary API endpoint URL for interacting with the agent.
+            organization (UUID):
+            organization_detail (OrganizationMinimal):
+            owner (Union[None, int]):
+            owner_detail (Union['UserProfileMinimal', None]):
+            created_at (datetime.datetime):
+            updated_at (datetime.datetime):
+            agent_type (Union[Unset, str]): The specific SDK, ADK, or API type the agent is built upon (e.g., OpenAI SDK,
+                Generic ADK).
+            description (Union[Unset, str]):
+            metadata (Union[Unset, Any]): Optional JSON data providing specific details and configuration. Structure depends
+                heavily on Agent Type. Examples:
+                - For GENERIC_ADK: {'adk_app_name': 'my_adk_app', 'protocol_version': '1.0'}
+                - For OPENAI_SDK: {'model': 'gpt-4-turbo', 'api_key_secret_name': 'MY_OPENAI_KEY', 'instructions': 'You are a
+                helpful assistant.'}
+                - For GOOGLE_ADK: {'project_id': 'my-gcp-project', 'location': 'us-central1'}
+                - General applicable: {'version': '1.2.0', 'custom_headers': {'X-Custom-Header': 'value'}}
+    """
+
+    id: UUID
+    name: str
+    endpoint: str
+    organization: UUID
+    organization_detail: "OrganizationMinimal"
+    owner: Union[None, int]
+    owner_detail: Union["UserProfileMinimal", None]
+    created_at: datetime.datetime
+    updated_at: datetime.datetime
+    agent_type: Union[Unset, str] = UNSET
+    description: Union[Unset, str] = UNSET
+    metadata: Union[Unset, Any] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        from ..models.user_profile_minimal import UserProfileMinimal
+
+        id = str(self.id)
+
+        name = self.name
+
+        endpoint = self.endpoint
+
+        organization = str(self.organization)
+
+        organization_detail = self.organization_detail.to_dict()
+
+        owner: Union[None, int]
+        owner = self.owner
+
+        owner_detail: Union[None, dict[str, Any]]
+        if isinstance(self.owner_detail, UserProfileMinimal):
+            owner_detail = self.owner_detail.to_dict()
+        else:
+            owner_detail = self.owner_detail
+
+        created_at = self.created_at.isoformat()
+
+        updated_at = self.updated_at.isoformat()
+
+        agent_type = self.agent_type
+
+        description = self.description
+
+        metadata = self.metadata
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "id": id,
+                "name": name,
+                "endpoint": endpoint,
+                "organization": organization,
+                "organization_detail": organization_detail,
+                "owner": owner,
+                "owner_detail": owner_detail,
+                "created_at": created_at,
+                "updated_at": updated_at,
+            }
+        )
+        if agent_type is not UNSET:
+            field_dict["agent_type"] = agent_type
+        if description is not UNSET:
+            field_dict["description"] = description
+        if metadata is not UNSET:
+            field_dict["metadata"] = metadata
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.organization_minimal import OrganizationMinimal
+        from ..models.user_profile_minimal import UserProfileMinimal
+
+        d = dict(src_dict)
+        id = UUID(d.pop("id"))
+
+        name = d.pop("name")
+
+        endpoint = d.pop("endpoint")
+
+        organization = UUID(d.pop("organization"))
+
+        organization_detail = OrganizationMinimal.from_dict(
+            d.pop("organization_detail")
+        )
+
+        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_detail(data: object) -> Union["UserProfileMinimal", None]:
+            if data is None:
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                owner_detail_type_1 = UserProfileMinimal.from_dict(data)
+
+                return owner_detail_type_1
+            except:  # noqa: E722
+                pass
+            return cast(Union["UserProfileMinimal", None], data)
+
+        owner_detail = _parse_owner_detail(d.pop("owner_detail"))
+
+        created_at = isoparse(d.pop("created_at"))
+
+        updated_at = isoparse(d.pop("updated_at"))
+
+        agent_type = d.pop("agent_type", UNSET)
+
+        description = d.pop("description", UNSET)
+
+        metadata = d.pop("metadata", UNSET)
+
+        agent = cls(
+            id=id,
+            name=name,
+            endpoint=endpoint,
+            organization=organization,
+            organization_detail=organization_detail,
+            owner=owner,
+            owner_detail=owner_detail,
+            created_at=created_at,
+            updated_at=updated_at,
+            agent_type=agent_type,
+            description=description,
+            metadata=metadata,
+        )
+
+        agent.additional_properties = d
+        return agent
+
+    @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/agent_request.py

@@ -0,0 +1,129 @@
+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 ..types import UNSET, Unset
+
+T = TypeVar("T", bound="AgentRequest")
+
+
+@_attrs_define
+class AgentRequest:
+    """Serializes Agent model instances to JSON and validates data for creating
+    or updating Agent instances.
+
+    This serializer provides a comprehensive representation of an Agent,
+    including its type, endpoint, and nested details for related 'organization'
+    and 'owner' for read operations, while allowing 'organization' and 'owner' IDs
+    for write operations.
+
+    Attributes:
+        organization_detail (OrganizationMinimalSerializer): Read-only nested
+            serializer for the agent's organization. Displays minimal details.
+        owner_detail (UserProfileMinimalSerializer): Read-only nested serializer
+            for the agent's owner's user profile. Displays minimal details.
+            Can be null if the agent has no owner or the owner has no profile.
+        agent_type (CharField): The type of the agent as a string
+                          (e.g., LITELLM, OPENAI_SDK, GOOGLE_ADK).
+
+    Meta:
+        model (Agent): The model class that this serializer works with.
+        fields (tuple): The fields to include in the serialized output.
+            Includes standard Agent fields like 'endpoint', 'type',
+            and the read-only nested details.
+        read_only_fields (tuple): Fields that are read-only and cannot be
+            set during create/update operations through this serializer.
+            This includes 'id', 'created_at', 'updated_at', and the
+            nested detail fields.
+
+        Attributes:
+            name (str):
+            endpoint (str): The primary API endpoint URL for interacting with the agent.
+            agent_type (Union[Unset, str]): The specific SDK, ADK, or API type the agent is built upon (e.g., OpenAI SDK,
+                Generic ADK).
+            description (Union[Unset, str]):
+            metadata (Union[Unset, Any]): Optional JSON data providing specific details and configuration. Structure depends
+                heavily on Agent Type. Examples:
+                - For GENERIC_ADK: {'adk_app_name': 'my_adk_app', 'protocol_version': '1.0'}
+                - For OPENAI_SDK: {'model': 'gpt-4-turbo', 'api_key_secret_name': 'MY_OPENAI_KEY', 'instructions': 'You are a
+                helpful assistant.'}
+                - For GOOGLE_ADK: {'project_id': 'my-gcp-project', 'location': 'us-central1'}
+                - General applicable: {'version': '1.2.0', 'custom_headers': {'X-Custom-Header': 'value'}}
+    """
+
+    name: str
+    endpoint: str
+    agent_type: Union[Unset, str] = UNSET
+    description: Union[Unset, str] = UNSET
+    metadata: Union[Unset, Any] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        name = self.name
+
+        endpoint = self.endpoint
+
+        agent_type = self.agent_type
+
+        description = self.description
+
+        metadata = self.metadata
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "name": name,
+                "endpoint": endpoint,
+            }
+        )
+        if agent_type is not UNSET:
+            field_dict["agent_type"] = agent_type
+        if description is not UNSET:
+            field_dict["description"] = description
+        if metadata is not UNSET:
+            field_dict["metadata"] = metadata
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        name = d.pop("name")
+
+        endpoint = d.pop("endpoint")
+
+        agent_type = d.pop("agent_type", UNSET)
+
+        description = d.pop("description", UNSET)
+
+        metadata = d.pop("metadata", UNSET)
+
+        agent_request = cls(
+            name=name,
+            endpoint=endpoint,
+            agent_type=agent_type,
+            description=description,
+            metadata=metadata,
+        )
+
+        agent_request.additional_properties = d
+        return agent_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/api_token_log.py

@@ -0,0 +1,184 @@
+import datetime
+from collections.abc import Mapping
+from typing import Any, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+from dateutil.parser import isoparse
+
+T = TypeVar("T", bound="APITokenLog")
+
+
+@_attrs_define
+class APITokenLog:
+    """Serializer for APITokenLog model, providing read-only access to log entries.
+
+    Attributes:
+        id (int):
+        timestamp (datetime.datetime):
+        api_key_prefix (Union[None, str]):
+        user_username (Union[None, str]):
+        organization_name (Union[None, str]):
+        model_id_used (str): Identifier of the AI model used.
+        api_endpoint (str): Internal endpoint name, e.g., 'generator' or 'judge'.
+        input_tokens (int):
+        output_tokens (int):
+        credits_deducted (str):
+        request_payload_preview (Union[None, str]): First ~256 chars of request payload
+        response_payload_preview (Union[None, str]): First ~256 chars of response payload
+    """
+
+    id: int
+    timestamp: datetime.datetime
+    api_key_prefix: Union[None, str]
+    user_username: Union[None, str]
+    organization_name: Union[None, str]
+    model_id_used: str
+    api_endpoint: str
+    input_tokens: int
+    output_tokens: int
+    credits_deducted: str
+    request_payload_preview: Union[None, str]
+    response_payload_preview: Union[None, str]
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        id = self.id
+
+        timestamp = self.timestamp.isoformat()
+
+        api_key_prefix: Union[None, str]
+        api_key_prefix = self.api_key_prefix
+
+        user_username: Union[None, str]
+        user_username = self.user_username
+
+        organization_name: Union[None, str]
+        organization_name = self.organization_name
+
+        model_id_used = self.model_id_used
+
+        api_endpoint = self.api_endpoint
+
+        input_tokens = self.input_tokens
+
+        output_tokens = self.output_tokens
+
+        credits_deducted = self.credits_deducted
+
+        request_payload_preview: Union[None, str]
+        request_payload_preview = self.request_payload_preview
+
+        response_payload_preview: Union[None, str]
+        response_payload_preview = self.response_payload_preview
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "id": id,
+                "timestamp": timestamp,
+                "api_key_prefix": api_key_prefix,
+                "user_username": user_username,
+                "organization_name": organization_name,
+                "model_id_used": model_id_used,
+                "api_endpoint": api_endpoint,
+                "input_tokens": input_tokens,
+                "output_tokens": output_tokens,
+                "credits_deducted": credits_deducted,
+                "request_payload_preview": request_payload_preview,
+                "response_payload_preview": response_payload_preview,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        id = d.pop("id")
+
+        timestamp = isoparse(d.pop("timestamp"))
+
+        def _parse_api_key_prefix(data: object) -> Union[None, str]:
+            if data is None:
+                return data
+            return cast(Union[None, str], data)
+
+        api_key_prefix = _parse_api_key_prefix(d.pop("api_key_prefix"))
+
+        def _parse_user_username(data: object) -> Union[None, str]:
+            if data is None:
+                return data
+            return cast(Union[None, str], data)
+
+        user_username = _parse_user_username(d.pop("user_username"))
+
+        def _parse_organization_name(data: object) -> Union[None, str]:
+            if data is None:
+                return data
+            return cast(Union[None, str], data)
+
+        organization_name = _parse_organization_name(d.pop("organization_name"))
+
+        model_id_used = d.pop("model_id_used")
+
+        api_endpoint = d.pop("api_endpoint")
+
+        input_tokens = d.pop("input_tokens")
+
+        output_tokens = d.pop("output_tokens")
+
+        credits_deducted = d.pop("credits_deducted")
+
+        def _parse_request_payload_preview(data: object) -> Union[None, str]:
+            if data is None:
+                return data
+            return cast(Union[None, str], data)
+
+        request_payload_preview = _parse_request_payload_preview(
+            d.pop("request_payload_preview")
+        )
+
+        def _parse_response_payload_preview(data: object) -> Union[None, str]:
+            if data is None:
+                return data
+            return cast(Union[None, str], data)
+
+        response_payload_preview = _parse_response_payload_preview(
+            d.pop("response_payload_preview")
+        )
+
+        api_token_log = cls(
+            id=id,
+            timestamp=timestamp,
+            api_key_prefix=api_key_prefix,
+            user_username=user_username,
+            organization_name=organization_name,
+            model_id_used=model_id_used,
+            api_endpoint=api_endpoint,
+            input_tokens=input_tokens,
+            output_tokens=output_tokens,
+            credits_deducted=credits_deducted,
+            request_payload_preview=request_payload_preview,
+            response_payload_preview=response_payload_preview,
+        )
+
+        api_token_log.additional_properties = d
+        return api_token_log
+
+    @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/attack.py

@@ -0,0 +1,154 @@
+import datetime
+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 dateutil.parser import isoparse
+
+T = TypeVar("T", bound="Attack")
+
+
+@_attrs_define
+class Attack:
+    """Serializer for the Attack model, which represents an Attack configuration.
+
+    Handles the conversion of Attack configuration instances to JSON (and vice-versa)
+    for API requests and responses. It includes read-only fields for related
+    object names (like agent_name, owner_username) for convenience in API outputs.
+
+        Attributes:
+            id (UUID):
+            type_ (str): A string identifier for the type of attack being configured (e.g., 'PREFIX_GENERATION',
+                'PROMPT_INJECTION').
+            agent (UUID):
+            agent_name (str):
+            owner (Union[None, int]):
+            owner_username (str):
+            organization (UUID):
+            organization_name (str):
+            configuration (Any): JSON containing client-provided configuration for an attack using this definition.
+            created_at (datetime.datetime):
+            updated_at (datetime.datetime):
+    """
+
+    id: UUID
+    type_: str
+    agent: UUID
+    agent_name: str
+    owner: Union[None, int]
+    owner_username: str
+    organization: UUID
+    organization_name: str
+    configuration: Any
+    created_at: datetime.datetime
+    updated_at: datetime.datetime
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        id = str(self.id)
+
+        type_ = self.type_
+
+        agent = str(self.agent)
+
+        agent_name = self.agent_name
+
+        owner: Union[None, int]
+        owner = self.owner
+
+        owner_username = self.owner_username
+
+        organization = str(self.organization)
+
+        organization_name = self.organization_name
+
+        configuration = self.configuration
+
+        created_at = self.created_at.isoformat()
+
+        updated_at = self.updated_at.isoformat()
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "id": id,
+                "type": type_,
+                "agent": agent,
+                "agent_name": agent_name,
+                "owner": owner,
+                "owner_username": owner_username,
+                "organization": organization,
+                "organization_name": organization_name,
+                "configuration": configuration,
+                "created_at": created_at,
+                "updated_at": updated_at,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        id = UUID(d.pop("id"))
+
+        type_ = d.pop("type")
+
+        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"))
+
+        owner_username = d.pop("owner_username")
+
+        organization = UUID(d.pop("organization"))
+
+        organization_name = d.pop("organization_name")
+
+        configuration = d.pop("configuration")
+
+        created_at = isoparse(d.pop("created_at"))
+
+        updated_at = isoparse(d.pop("updated_at"))
+
+        attack = cls(
+            id=id,
+            type_=type_,
+            agent=agent,
+            agent_name=agent_name,
+            owner=owner,
+            owner_username=owner_username,
+            organization=organization,
+            organization_name=organization_name,
+            configuration=configuration,
+            created_at=created_at,
+            updated_at=updated_at,
+        )
+
+        attack.additional_properties = d
+        return attack
+
+    @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