pangea-sdk 5.2.0__tar.gz → 5.2.0b1__tar.gz

{pangea_sdk-5.2.0 → pangea_sdk-5.2.0b1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pangea-sdk
-Version: 5.2.0
+Version: 5.2.0b1
 Summary: Pangea API SDK
 Home-page: https://pangea.cloud/docs/sdk/python/
 License: MIT
@@ -17,11 +17,11 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Software Development
 Classifier: Topic :: Software Development :: Libraries
-Requires-Dist: aiohttp (>=3.11.10,<4.0.0)
-Requires-Dist: cryptography (>=43.0.3,<44.0.0)
-Requires-Dist: deprecated (>=1.2.15,<2.0.0)
+Requires-Dist: aiohttp (>=3.10.10,<4.0.0)
+Requires-Dist: cryptography (>=43.0.1,<44.0.0)
+Requires-Dist: deprecated (>=1.2.14,<2.0.0)
 Requires-Dist: google-crc32c (>=1.5.0,<2.0.0)
-Requires-Dist: pydantic (>=2.10.3,<3.0.0)
+Requires-Dist: pydantic (>=2.9.2,<3.0.0)
 Requires-Dist: python-dateutil (>=2.9.0,<3.0.0)
 Requires-Dist: requests (>=2.31.0,<3.0.0)
 Requires-Dist: requests-toolbelt (>=1.0.0,<2.0.0)
@@ -64,10 +64,9 @@ $ poetry add pangea-sdk
 #### Beta releases
 
 Pre-release versions may be available with the `b` (beta) denotation in the
-version number. These releases serve to preview Beta and Early Access services
-and APIs. Per Semantic Versioning, they are considered unstable and do not carry
-the same compatibility guarantees as stable releases.
-[Beta changelog](https://github.com/pangeacyber/pangea-python/blob/beta/CHANGELOG.md).
+version number. These releases serve to preview beta services and APIs. Per
+Semantic Versioning, they are considered unstable and do not carry the same
+compatibility guarantees as stable releases. [Beta changelog](https://github.com/pangeacyber/pangea-python/blob/beta/CHANGELOG.md).
 
 Via pip:
 

{pangea_sdk-5.2.0 → pangea_sdk-5.2.0b1}/README.md

@@ -33,10 +33,9 @@ $ poetry add pangea-sdk
 #### Beta releases
 
 Pre-release versions may be available with the `b` (beta) denotation in the
-version number. These releases serve to preview Beta and Early Access services
-and APIs. Per Semantic Versioning, they are considered unstable and do not carry
-the same compatibility guarantees as stable releases.
-[Beta changelog](https://github.com/pangeacyber/pangea-python/blob/beta/CHANGELOG.md).
+version number. These releases serve to preview beta services and APIs. Per
+Semantic Versioning, they are considered unstable and do not carry the same
+compatibility guarantees as stable releases. [Beta changelog](https://github.com/pangeacyber/pangea-python/blob/beta/CHANGELOG.md).
 
 Via pip:
 

{pangea_sdk-5.2.0 → pangea_sdk-5.2.0b1}/pangea/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "5.2.0"
+__version__ = "5.2.0beta1"
 
 from pangea.asyncio.request import PangeaRequestAsync
 from pangea.config import PangeaConfig

{pangea_sdk-5.2.0 → pangea_sdk-5.2.0b1}/pangea/asyncio/request.py

@@ -5,11 +5,12 @@ from __future__ import annotations
 import asyncio
 import json
 import time
-from typing import Any, Dict, List, Optional, Sequence, Tuple, Type, Union
+from typing import Dict, List, Optional, Sequence, Tuple, Type, Union, cast
 
 import aiohttp
 from aiohttp import FormData
 from pydantic import BaseModel
+from pydantic_core import to_jsonable_python
 from typing_extensions import Any, TypeVar
 
 import pangea.exceptions as pe
@@ -55,17 +56,20 @@ class PangeaRequestAsync(PangeaRequestBase):
         if data is None:
             data = {}
 
+        # Normalize.
+        data = cast(dict[str, Any], to_jsonable_python(data))
+
         if url is None:
             url = self._url(endpoint)
 
         # Set config ID if available
-        if self.config_id and data.get("config_id", None) is None:  # type: ignore[union-attr]
-            data["config_id"] = self.config_id  # type: ignore[index]
+        if self.config_id and data.get("config_id", None) is None:
+            data["config_id"] = self.config_id
 
         self.logger.debug(
             json.dumps({"service": self.service, "action": "post", "url": url, "data": data}, default=default_encoder)
         )
-        transfer_method = data.get("transfer_method", None)  # type: ignore[union-attr]
+        transfer_method = data.get("transfer_method", None)
 
         if files and type(data) is dict and (transfer_method == TransferMethod.POST_URL.value):
             requests_response = await self._full_post_presigned_url(
@@ -182,20 +186,7 @@ class PangeaRequestAsync(PangeaRequestBase):
         if resp.status < 200 or resp.status >= 300:
             raise pe.PresignedUploadError(f"presigned PUT failure: {resp.status}", await resp.text())
 
-    async def download_file(self, url: str, filename: str | None = None) -> AttachedFile:
-        """
-        Download file
-
-        Download a file from the specified URL and save it with the given
-        filename.
-
-        Args:
-            url: URL of the file to download
-            filename: Name to save the downloaded file as. If not provided, the
-              filename will be determined from the Content-Disposition header or
-              the URL.
-        """
-
+    async def download_file(self, url: str, filename: Optional[str] = None) -> AttachedFile:
         self.logger.debug(
             json.dumps(
                 {

{pangea_sdk-5.2.0 → pangea_sdk-5.2.0b1}/pangea/asyncio/services/audit.py

@@ -174,16 +174,14 @@ class AuditAsync(ServiceBaseAsync, AuditBase):
         verbose: Optional[bool] = None,
     ) -> PangeaResponse[LogResult]:
         """
-        Log an event
+        Log an entry
 
         Create a log entry in the Secure Audit Log.
-
         Args:
             event (dict[str, Any]): event to be logged
             verify (bool, optional): True to verify logs consistency after response.
             sign_local (bool, optional): True to sign event with local key.
             verbose (bool, optional): True to get a more verbose response.
-
         Raises:
             AuditException: If an audit based api exception happens
             PangeaAPIException: If an API Error happens
@@ -194,7 +192,13 @@ class AuditAsync(ServiceBaseAsync, AuditBase):
                 Available response fields can be found in our [API documentation](https://pangea.cloud/docs/api/audit#log-an-entry).
 
         Examples:
-            response = await audit.log_event({"message": "hello world"}, verbose=True)
+            try:
+                log_response = audit.log({"message"="Hello world"}, verbose=False)
+                print(f"Response. Hash: {log_response.result.hash}")
+            except pe.PangeaAPIException as e:
+                print(f"Request Error: {e.response.summary}")
+                for err in e.errors:
+                    print(f"\\t{err.detail} \\n")
         """
 
         input = self._get_log_request(event, sign_local=sign_local, verify=verify, verbose=verbose)
@@ -628,9 +632,9 @@ class AuditAsync(ServiceBaseAsync, AuditBase):
             if pub_root is not None:
                 self.pub_roots[tree_size] = pub_root
 
-        await self._fix_consistency_proofs(tree_sizes)
+        await self.fix_consistency_proofs(tree_sizes)
 
-    async def _fix_consistency_proofs(self, tree_sizes: Iterable[int]) -> None:
+    async def fix_consistency_proofs(self, tree_sizes: Iterable[int]):
         # on very rare occasions, the consistency proof in Arweave may be wrong
         # override it with the proof from pangea (not the root hash, just the proof)
         for tree_size in tree_sizes:

{pangea_sdk-5.2.0 → pangea_sdk-5.2.0b1}/pangea/asyncio/services/authz.py

@@ -1,12 +1,9 @@
 # Copyright 2022 Pangea Cyber Corporation
 # Author: Pangea Cyber Corporation
 
-from __future__ import annotations
-
 from typing import Any, Dict, List, Optional
 
 from pangea.asyncio.services.base import ServiceBaseAsync
-from pangea.config import PangeaConfig
 from pangea.response import PangeaResponse
 from pangea.services.authz import (
     CheckRequest,
@@ -39,8 +36,8 @@ class AuthZAsync(ServiceBaseAsync):
 
     Examples:
         import os
-        from pangea.asyncio.services import AuthZAsync
         from pangea.config import PangeaConfig
+        from pangea.services import AuthZ
 
         PANGEA_TOKEN = os.getenv("PANGEA_AUTHZ_TOKEN")
 
@@ -52,25 +49,7 @@ class AuthZAsync(ServiceBaseAsync):
 
     service_name = "authz"
 
-    def __init__(
-        self, token: str, config: PangeaConfig | None = None, logger_name: str = "pangea", config_id: str | None = None
-    ) -> None:
-        """
-        AuthZ client
-
-        Initializes a new AuthZ client.
-
-        Args:
-            token: Pangea API token.
-            config: Configuration.
-            logger_name: Logger name.
-            config_id: Configuration ID.
-
-        Examples:
-             config = PangeaConfig(domain="aws.us.pangea.cloud")
-             authz = AuthZAsync(token="pangea_token", config=config)
-        """
-
+    def __init__(self, token: str, config=None, logger_name="pangea", config_id: Optional[str] = None):
         super().__init__(token, config, logger_name, config_id=config_id)
 
     async def tuple_create(self, tuples: List[Tuple]) -> PangeaResponse[TupleCreateResult]:

{pangea_sdk-5.2.0 → pangea_sdk-5.2.0b1}/pangea/asyncio/services/base.py

@@ -1,11 +1,8 @@
 # Copyright 2022 Pangea Cyber Corporation
 # Author: Pangea Cyber Corporation
-from __future__ import annotations
 
 from typing import Dict, Optional, Type, Union
 
-from typing_extensions import override
-
 from pangea.asyncio.request import PangeaRequestAsync
 from pangea.exceptions import AcceptedRequestException
 from pangea.response import AttachedFile, PangeaResponse, PangeaResponseResult
@@ -26,7 +23,6 @@ class ServiceBaseAsync(ServiceBase):
 
         return self._request
 
-    @override
     async def poll_result(  # type: ignore[override]
         self,
         exception: Optional[AcceptedRequestException] = None,
@@ -40,8 +36,7 @@ class ServiceBaseAsync(ServiceBase):
         Returns request's result that has been accepted by the server
 
         Args:
-            exception: Exception that was previously raised by the SDK on a call
-              that is being processed.
+            exception (AcceptedRequestException): Exception raise by SDK on the call that is been processed.
 
         Returns:
             PangeaResponse
@@ -63,21 +58,7 @@ class ServiceBaseAsync(ServiceBase):
         else:
             raise AttributeError("Need to set exception, response or request_id")
 
-    @override
-    async def download_file(self, url: str, filename: str | None = None) -> AttachedFile:  # type: ignore[override]
-        """
-        Download file
-
-        Download a file from the specified URL and save it with the given
-        filename.
-
-        Args:
-            url: URL of the file to download
-            filename: Name to save the downloaded file as. If not provided, the
-              filename will be determined from the Content-Disposition header or
-              the URL.
-        """
-
+    async def download_file(self, url: str, filename: Optional[str] = None) -> AttachedFile:  # type: ignore[override]
         return await self.request.download_file(url=url, filename=filename)
 
     async def close(self):

pangea_sdk-5.2.0b1/pangea/asyncio/services/data_guard.py

@@ -0,0 +1,65 @@
+from __future__ import annotations
+
+from pangea.asyncio.services.base import ServiceBaseAsync
+from pangea.response import PangeaResponse, PangeaResponseResult
+from pangea.services.data_guard import TextGuardResult
+
+
+class DataGuard(ServiceBaseAsync):
+    """Data Guard service client.
+
+    Provides methods to interact with Pangea's Data Guard service.
+    """
+
+    service_name = "data-guard"
+
+    async def guard_text(
+        self,
+        text: str,
+        *,
+        recipe: str = "pangea_prompt_guard",
+        debug: bool = False,
+    ) -> PangeaResponse[TextGuardResult]:
+        """
+        Text guard (Beta)
+
+        Guard text.
+
+        How to install a [Beta release](https://pangea.cloud/docs/sdk/python/#beta-releases).
+
+        OperationId: data_guard_post_v1_text_guard
+
+        Args:
+            text: Text.
+            recipe: Recipe.
+            debug: Debug.
+
+        Examples:
+            response = await data_guard.guard_text("text")
+        """
+
+        return await self.request.post(
+            "v1/text/guard", TextGuardResult, data={"text": text, "recipe": recipe, "debug": debug}
+        )
+
+    async def guard_file(
+        self,
+        file_url: str,
+    ) -> PangeaResponse[PangeaResponseResult]:
+        """
+        File guard (Beta)
+
+        Guard a file URL.
+
+        How to install a [Beta release](https://pangea.cloud/docs/sdk/python/#beta-releases).
+
+        OperationId: data_guard_post_v1_file_guard
+
+        Args:
+            file_url: File URL.
+
+        Examples:
+            response = await data_guard.guard_file("https://example.org/file.txt")
+        """
+
+        return await self.request.post("v1/file/guard", PangeaResponseResult, data={"file_url": file_url})

{pangea_sdk-5.2.0 → pangea_sdk-5.2.0b1}/pangea/asyncio/services/intel.py

@@ -790,7 +790,6 @@ class UserIntelAsync(ServiceBaseAsync):
         verbose: Optional[bool] = None,
         raw: Optional[bool] = None,
         provider: Optional[str] = None,
-        cursor: Optional[str] = None,
     ) -> PangeaResponse[m.UserBreachedResult]:
         """
         Look up breached users
@@ -809,7 +808,6 @@ class UserIntelAsync(ServiceBaseAsync):
             verbose (bool, optional): Echo the API parameters in the response
             raw (bool, optional): Include raw data from this provider
             provider (str, optional): Use reputation data from this provider: "crowdstrike"
-            cursor (str, optional): A token given in the raw response from SpyCloud. Post this back to paginate results
 
         Raises:
             PangeaAPIException: If an API Error happens
@@ -837,7 +835,6 @@ class UserIntelAsync(ServiceBaseAsync):
             end=end,
             verbose=verbose,
             raw=raw,
-            cursor=cursor,
         )
         return await self.request.post(
             "v1/user/breached", m.UserBreachedResult, data=input.model_dump(exclude_none=True)

pangea_sdk-5.2.0b1/pangea/asyncio/services/prompt_guard.py

@@ -0,0 +1,39 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from pangea.asyncio.services.base import ServiceBaseAsync
+from pangea.services.prompt_guard import GuardResult, Message
+
+if TYPE_CHECKING:
+    from collections.abc import Iterable
+
+    from pangea.response import PangeaResponse
+
+
+class PromptGuard(ServiceBaseAsync):
+    """Prompt Guard service client.
+
+    Provides methods to interact with Pangea's Prompt Guard service.
+    """
+
+    service_name = "prompt-guard"
+
+    async def guard(self, messages: Iterable[Message]) -> PangeaResponse[GuardResult]:
+        """
+        Guard (Beta)
+
+        Guard messages.
+
+        How to install a [Beta release](https://pangea.cloud/docs/sdk/python/#beta-releases).
+
+        OperationId: prompt_guard_post_v1_guard
+
+        Args:
+            messages: Messages..
+
+        Examples:
+            response = await prompt_guard.guard([Message(role="user", content="hello world")])
+        """
+
+        return await self.request.post("v1/guard", GuardResult, data={"messages": messages})

{pangea_sdk-5.2.0 → pangea_sdk-5.2.0b1}/pangea/asyncio/services/sanitize.py

@@ -1,13 +1,10 @@
 # Copyright 2022 Pangea Cyber Corporation
 # Author: Pangea Cyber Corporation
-from __future__ import annotations
-
 import io
 from typing import List, Optional, Tuple
 
 import pangea.services.sanitize as m
 from pangea.asyncio.services.base import ServiceBaseAsync
-from pangea.config import PangeaConfig
 from pangea.response import PangeaResponse, TransferMethod
 from pangea.utils import FileUploadParams, get_file_upload_params
 
@@ -19,38 +16,17 @@ class SanitizeAsync(ServiceBaseAsync):
         import os
 
         # Pangea SDK
-        from pangea.asyncio.services import SanitizeAsync
         from pangea.config import PangeaConfig
+        from pangea.asyncio.services import Sanitize
 
         PANGEA_SANITIZE_TOKEN = os.getenv("PANGEA_SANITIZE_TOKEN")
         config = PangeaConfig(domain="pangea.cloud")
 
-        sanitize = SanitizeAsync(token=PANGEA_SANITIZE_TOKEN, config=config)
+        sanitize = Sanitize(token=PANGEA_SANITIZE_TOKEN, config=config)
     """
 
     service_name = "sanitize"
 
-    def __init__(
-        self, token: str, config: PangeaConfig | None = None, logger_name: str = "pangea", config_id: str | None = None
-    ) -> None:
-        """
-        Sanitize client
-
-        Initializes a new Sanitize client.
-
-        Args:
-            token: Pangea API token.
-            config: Configuration.
-            logger_name: Logger name.
-            config_id: Configuration ID.
-
-        Examples:
-             config = PangeaConfig(domain="aws.us.pangea.cloud")
-             authz = SanitizeAsync(token="pangea_token", config=config)
-        """
-
-        super().__init__(token, config, logger_name, config_id=config_id)
-
     async def sanitize(
         self,
         transfer_method: TransferMethod = TransferMethod.POST_URL,

{pangea_sdk-5.2.0 → pangea_sdk-5.2.0b1}/pangea/asyncio/services/share.py

@@ -7,38 +7,16 @@ from typing import Dict, List, Optional, Tuple, Union
 
 import pangea.services.share.share as m
 from pangea.asyncio.services.base import ServiceBaseAsync
-from pangea.config import PangeaConfig
 from pangea.response import PangeaResponse, TransferMethod
 from pangea.services.share.file_format import FileFormat
 from pangea.utils import get_file_size, get_file_upload_params
 
 
 class ShareAsync(ServiceBaseAsync):
-    """Secure Share service client."""
+    """Share service client."""
 
     service_name = "share"
 
-    def __init__(
-        self, token: str, config: PangeaConfig | None = None, logger_name: str = "pangea", config_id: str | None = None
-    ) -> None:
-        """
-        Secure Share client
-
-        Initializes a new Secure Share client.
-
-        Args:
-            token: Pangea API token.
-            config: Configuration.
-            logger_name: Logger name.
-            config_id: Configuration ID.
-
-        Examples:
-             config = PangeaConfig(domain="aws.us.pangea.cloud")
-             authz = ShareAsync(token="pangea_token", config=config)
-        """
-
-        super().__init__(token, config, logger_name, config_id=config_id)
-
     async def buckets(self) -> PangeaResponse[m.BucketsResult]:
         """
         Buckets

{pangea_sdk-5.2.0 → pangea_sdk-5.2.0b1}/pangea/deep_verify.py

@@ -263,14 +263,8 @@ def main():
         audit = init_audit(args.token, args.domain)
         errors = deep_verify(audit, args.file)
 
-        print("\n\nWarnings:")
-        val = errors["not_persisted"]
-        print(f"\tnot_persisted: {val}")
-
-        print("\nTotal errors:")
+        print("\n\nTotal errors:")
         for key, val in errors.items():
-            if key == "not_persisted":
-                continue
             print(f"\t{key.title()}: {val}")
         print()
 

{pangea_sdk-5.2.0 → pangea_sdk-5.2.0b1}/pangea/dump_audit.py

@@ -63,12 +63,11 @@ def dump_before(audit: Audit, output: io.TextIOWrapper, start: datetime) -> int:
     cnt = 0
     if search_res.result and search_res.result.count > 0:
         leaf_index = search_res.result.events[0].leaf_index
-        if leaf_index is not None:
-            for row in reversed(search_res.result.events):
-                if row.leaf_index != leaf_index:
-                    break
-                dump_event(output, row, search_res)
-                cnt += 1
+        for row in reversed(search_res.result.events):
+            if row.leaf_index != leaf_index:
+                break
+            dump_event(output, row, search_res)
+            cnt += 1
     print(f"Dumping before... {cnt} events")
     return cnt
 
@@ -90,7 +89,7 @@ def dump_after(audit: Audit, output: io.TextIOWrapper, start: datetime, last_eve
     cnt = 0
     if search_res.result and search_res.result.count > 0:
         leaf_index = search_res.result.events[0].leaf_index
-        if leaf_index is not None and leaf_index == last_leaf_index:
+        if leaf_index == last_leaf_index:
             start_idx: int = 1 if last_event_hash == search_res.result.events[0].hash else 0
             for row in search_res.result.events[start_idx:]:
                 if row.leaf_index != leaf_index:
@@ -125,7 +124,7 @@ def dump_page(
     msg = f"Dumping... {search_res.result.count} events"
 
     if search_res.result.count <= 1:
-        return end, 0, True, "", 0
+        return end, 0  # type: ignore[return-value]
 
     offset = 0
     result_id = search_res.result.id

{pangea_sdk-5.2.0 → pangea_sdk-5.2.0b1}/pangea/request.py

@@ -6,12 +6,13 @@ import copy
 import json
 import logging
 import time
-from typing import TYPE_CHECKING, Any, Dict, List, Optional, Sequence, Tuple, Type, Union
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Sequence, Tuple, Type, Union, cast
 
 import requests
 from pydantic import BaseModel
+from pydantic_core import to_jsonable_python
 from requests.adapters import HTTPAdapter, Retry
-from requests_toolbelt import MultipartDecoder  # type: ignore[import-untyped]
+from requests_toolbelt import MultipartDecoder  # type: ignore
 from typing_extensions import TypeVar
 
 import pangea
@@ -232,6 +233,9 @@ class PangeaRequest(PangeaRequestBase):
         if data is None:
             data = {}
 
+        # Normalize.
+        data = cast(dict[str, Any], to_jsonable_python(data))
+
         if url is None:
             url = self._url(endpoint)
 
@@ -402,20 +406,7 @@ class PangeaRequest(PangeaRequestBase):
 
         return self._check_response(pangea_response)
 
-    def download_file(self, url: str, filename: str | None = None) -> AttachedFile:
-        """
-        Download file
-
-        Download a file from the specified URL and save it with the given
-        filename.
-
-        Args:
-            url: URL of the file to download
-            filename: Name to save the downloaded file as. If not provided, the
-              filename will be determined from the Content-Disposition header or
-              the URL.
-        """
-
+    def download_file(self, url: str, filename: Optional[str] = None) -> AttachedFile:
         self.logger.debug(
             json.dumps(
                 {

{pangea_sdk-5.2.0 → pangea_sdk-5.2.0b1}/pangea/services/__init__.py

@@ -1,9 +1,11 @@
 from .audit.audit import Audit
 from .authn.authn import AuthN
 from .authz import AuthZ
+from .data_guard import DataGuard
 from .embargo import Embargo
 from .file_scan import FileScan
 from .intel import DomainIntel, FileIntel, IpIntel, UrlIntel, UserIntel
+from .prompt_guard import PromptGuard
 from .redact import Redact
 from .sanitize import Sanitize
 from .share.share import Share

{pangea_sdk-5.2.0 → pangea_sdk-5.2.0b1}/pangea/services/audit/audit.py

@@ -492,7 +492,7 @@ class Audit(ServiceBase, AuditBase):
         verbose: Optional[bool] = None,
     ) -> PangeaResponse[LogResult]:
         """
-        Log an event
+        Log an entry
 
         Create a log entry in the Secure Audit Log.
 
@@ -501,7 +501,6 @@ class Audit(ServiceBase, AuditBase):
             verify (bool, optional): True to verify logs consistency after response.
             sign_local (bool, optional): True to sign event with local key.
             verbose (bool, optional): True to get a more verbose response.
-
         Raises:
             AuditException: If an audit based api exception happens
             PangeaAPIException: If an API Error happens
@@ -512,7 +511,13 @@ class Audit(ServiceBase, AuditBase):
                 Available response fields can be found in our [API documentation](https://pangea.cloud/docs/api/audit#/v1/log).
 
         Examples:
-            response = audit.log_event({"message": "hello world"}, verbose=True)
+            try:
+                log_response = audit.log({"message": "hello world"}, verbose=True)
+                print(f"Response. Hash: {log_response.result.hash}")
+            except pe.PangeaAPIException as e:
+                print(f"Request Error: {e.response.summary}")
+                for err in e.errors:
+                    print(f"\\t{err.detail} \\n")
         """
 
         input = self._get_log_request(event, sign_local=sign_local, verify=verify, verbose=verbose)
@@ -956,9 +961,9 @@ class Audit(ServiceBase, AuditBase):
             if pub_root is not None:
                 self.pub_roots[tree_size] = pub_root
 
-        self._fix_consistency_proofs(tree_sizes)
+        self.fix_consistency_proofs(tree_sizes)
 
-    def _fix_consistency_proofs(self, tree_sizes: Iterable[int]) -> None:
+    def fix_consistency_proofs(self, tree_sizes: Iterable[int]):
         # on very rare occasions, the consistency proof in Arweave may be wrong
         # override it with the proof from pangea (not the root hash, just the proof)
         for tree_size in tree_sizes: