contentgrid-extension-helpers 0.0.1__py3-none-any.whl → 0.0.2__py3-none-any.whl

contentgrid_extension_helpers/__init__.py

@@ -0,0 +1,3 @@
+from .exceptions import LLMDenyException, ExtensionHelperException, IllegalActivityError, MissingInputError, InjectionError, MalformedInputError, NotRelatedError, SensitiveInformationError, SecurityError # noqa: F401
+from .middleware.exception_middleware import catch_exceptions_middleware # noqa: F401
+from .structured_output.model_deny import DenyReason, ModelDeny # noqa: F401

contentgrid_extension_helpers/authentication/__init__.py

@@ -1,2 +1,2 @@
-from .oidc import create_current_user_dependency, decode_and_verify_token, create_oauth2_scheme, get_jwks_client
-from .user import ContentGridUser
+from .oidc import create_current_user_dependency, decode_and_verify_token, create_oauth2_scheme, get_oauth_jwks_client, get_oidc_jwks_client # noqa: F401
+from .user import ContentGridUser # noqa: F401

contentgrid_extension_helpers/exceptions.py

@@ -0,0 +1,48 @@
+
+class ExtensionHelperException(Exception):
+    """Base class for all custom exceptions in the extension helpers library."""
+    def __init__(self, *args: object) -> None:
+        super().__init__(*args)
+
+class LLMDenyException(ExtensionHelperException):
+    """
+    Raised when the LLM denies a request based on its rules.
+    This is a base class for more specific denial reasons.
+    """
+    def __init__(self, *args: object) -> None:
+        super().__init__(*args)
+
+class NotRelatedError(LLMDenyException):
+    """Raised when the input is unrelated to the required extraction."""
+    def __init__(self, *args: object) -> None:
+        super().__init__(*args)
+
+class MissingInputError(LLMDenyException):
+    """Raised when required input is missing or empty."""
+    def __init__(self, *args: object) -> None:
+        super().__init__(*args)
+
+class MalformedInputError(LLMDenyException):
+    """Raised when input is present but malformed or contains only strange characters."""
+    def __init__(self, *args: object) -> None:
+        super().__init__(*args)
+
+class IllegalActivityError(LLMDenyException):
+    """Raised when the input describes illegal activities."""
+    def __init__(self, *args: object) -> None:
+        super().__init__(*args)
+
+class SensitiveInformationError(LLMDenyException):
+    """Raised when the input contains sensitive information."""
+    def __init__(self, *args: object) -> None:
+        super().__init__(*args)
+
+class SecurityError(LLMDenyException):
+    """Raised when the input attempts to compromise system security."""
+    def __init__(self, *args: object) -> None:
+        super().__init__(*args)
+
+class InjectionError(SecurityError):
+    """Raised when the input is a prompt injection attack."""
+    def __init__(self, *args: object) -> None:
+        super().__init__(*args)

contentgrid_extension_helpers/logging/__init__.py

@@ -1 +1 @@
-from .json_logging import setup_json_logging, XenitJsonFormatter
+from .json_logging import setup_json_logging, XenitJsonFormatter # noqa: F401

contentgrid_extension_helpers/logging/json_logging.py

@@ -2,7 +2,7 @@ import datetime
 import json
 import logging
 import time
-from typing import List, Optional, Any, Dict
+from typing import Optional, Any, Dict
 
 class XenitJsonFormatter(logging.Formatter):
     def __init__(
@@ -54,7 +54,7 @@ class XenitJsonFormatter(logging.Formatter):
 
 def setup_json_logging(
     component: str = "cg-extension", 
-    additional_keys: dict[str, str] = [], 
+    additional_keys: dict[str,str] = {}, 
     log_level: int = logging.DEBUG
 ) -> logging.Logger:
     """

contentgrid_extension_helpers/middleware/exception_middleware.py

@@ -0,0 +1,124 @@
+from fastapi import Request, status
+import logging
+from contentgrid_extension_helpers.problem_response import ProblemResponse
+from contentgrid_hal_client.exceptions import (
+    NotFound,
+    Unauthorized,
+    BadRequest,
+    IncorrectAttributeType,
+    NonExistantAttribute,
+    MissingRequiredAttribute,
+    MissingHALTemplate,
+)
+from requests.exceptions import HTTPError
+from contentgrid_extension_helpers.exceptions import LLMDenyException
+
+async def catch_exceptions_middleware(request: Request, call_next, problem_base_url: str = "https://docs.contentgrid.com"):
+    """
+    Catches exceptions and returns ProblemResponse objects.
+
+    Args:
+        request: The incoming request.
+        call_next: The next middleware or route handler.
+        problem_base_url: The base URL for problem type URIs.
+    """
+
+    def format_type(type_name: str) -> str:
+        """Formats problem type URIs."""
+        return f"{problem_base_url}/{type_name}"
+    
+    cors_headers = {
+        "Access-Control-Allow-Origin": "*",
+        "Access-Control-Allow-Methods": "*",
+        "Access-Control-Allow-Headers": "*, Authorization",
+    }
+    
+    try:
+        return await call_next(request)
+    except LLMDenyException as e:
+        logging.error(f"LLM denied request: {e}")
+        return ProblemResponse(
+            title="Request Denied",
+            problem_type=format_type("request-denied"),
+            detail=str(e),
+            status=status.HTTP_400_BAD_REQUEST,
+            headers=cors_headers,
+        )
+    except NotFound:
+        return ProblemResponse(
+            title="Not found",
+            problem_type=format_type("not-found"),
+            detail="Resource not found",
+            status=status.HTTP_404_NOT_FOUND,
+            headers=cors_headers,
+        )
+    except Unauthorized:
+        return ProblemResponse(
+            title="Unauthorized",
+            problem_type=format_type("unauthorized"),
+            detail="user is not authorized for requested resource",
+            status=status.HTTP_401_UNAUTHORIZED,
+            headers=cors_headers,
+        )
+    except BadRequest:
+        return ProblemResponse(
+            title="Bad Request",
+            problem_type=format_type("bad-request"),
+            detail="The request was malformed or invalid.",
+            status=status.HTTP_400_BAD_REQUEST,
+            headers=cors_headers,
+        )
+    except IncorrectAttributeType as e:
+        return ProblemResponse(
+            title="Incorrect Attribute Type",
+            problem_type=format_type("incorrect-attribute-type"),
+            detail=str(e),
+            status=status.HTTP_400_BAD_REQUEST,
+            headers=cors_headers,
+        )
+    except NonExistantAttribute as e:
+        return ProblemResponse(
+            title="Non-Existent Attribute",
+            problem_type=format_type("non-existent-attribute"),
+            detail=str(e),
+            status=status.HTTP_404_NOT_FOUND,
+            headers=cors_headers,
+        )
+    except MissingRequiredAttribute as e:
+        return ProblemResponse(
+            title="Missing Required Attribute",
+            problem_type=format_type("missing-required-attribute"),
+            detail=str(e),
+            status=status.HTTP_400_BAD_REQUEST,
+            headers=cors_headers,
+        )
+    except MissingHALTemplate as e:
+        return ProblemResponse(
+            title="Missing HAL Template",
+            problem_type=format_type("missing-hal-template"),
+            detail=str(e),
+            status=status.HTTP_404_NOT_FOUND,
+            headers=cors_headers,
+        )
+    except HTTPError as e:
+        logging.exception(f"HTTP Error: {str(e)}", exc_info=True)
+        return ProblemResponse(
+            title="HTTP Error",
+            problem_type=format_type("http-error"),
+            detail=f"An HTTP error occurred: {str(e)}",
+            status=e.response.status_code if e.response else status.HTTP_500_INTERNAL_SERVER_ERROR,
+            headers=cors_headers,
+        )
+    except Exception as e:
+        logging.exception(
+            f"Untyped exception caught in backend request...: {str(e)}",
+            exc_info=True,
+            stack_info=True,
+        )
+        return ProblemResponse(
+            title="Internal server error",
+            problem_type=format_type("unknown"),
+            detail=f"An unexpected error occurred: {str(e)}",
+            status=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            headers=cors_headers,
+        )

contentgrid_extension_helpers/problem_response.py

@@ -0,0 +1,45 @@
+from typing import Any, Dict, Mapping
+from fastapi.responses import JSONResponse
+
+
+def get_problem_dict(title: str | None, problem_type: str | None, status: int | None, detail: str | None, instance: str | None, **kwargs) -> Dict[str, Any]:
+    result: Dict[str, Any] = dict()
+    if title is not None:
+        result["title"] = title
+    if problem_type is not None:
+        result["type"] = problem_type
+    if status is not None:
+        result["status"] = status
+    if detail is not None:
+        result["detail"] = detail
+    if instance is not None:
+        result["instance"] = instance
+
+    for key, value in kwargs.items():
+        if key == "type":
+            raise ValueError(f"{key} is a reserved property name!")
+        result[key] = value
+
+    return result
+
+
+class ProblemResponse(JSONResponse):
+    def __init__(
+            self,
+            title: str | None = None,
+            problem_type: str | None = "about:blank",  # not 'type' because it is a built-in function
+            status: int = 400,
+            detail: str | None = None,
+            instance: str | None = None,
+            headers: Mapping[str, str] | None = None,
+            media_type: str | None = "application/problem+json",
+            background: Any | None = None,
+            **kwargs
+    ):
+        super().__init__(
+            get_problem_dict(title, problem_type, status, detail, instance, **kwargs),
+            status_code=status,
+            headers=headers,
+            media_type=media_type,
+            background=background
+        )

contentgrid_extension_helpers/structured_output/model_deny.py

@@ -0,0 +1,45 @@
+from enum import Enum
+from pydantic import BaseModel, Field
+
+from contentgrid_extension_helpers import exceptions
+
+
+class DenyReason(Enum):
+    NOT_RELATED = "not_related"
+    MISSING_INPUT = "missing_input"
+    MALFORMED_INPUT = "malformed_input"
+    ILLEGAL_ACTIVITY = "illegal_activity"
+    SENSITIVE_INFORMATION = "sensitive_information"
+    SECURITY_ERROR = "security_error"
+    INJECTION_ATTEMPT = "injection_attempt"
+
+    def to_exception(self) -> type[exceptions.LLMDenyException]:
+        """Maps a DenyReason to its corresponding exception class."""
+        match self:
+            case DenyReason.NOT_RELATED:
+                return exceptions.NotRelatedError
+            case DenyReason.MISSING_INPUT:
+                return exceptions.MissingInputError
+            case DenyReason.MALFORMED_INPUT:
+                return exceptions.MalformedInputError
+            case DenyReason.ILLEGAL_ACTIVITY:
+                return exceptions.IllegalActivityError
+            case DenyReason.SENSITIVE_INFORMATION:
+                return exceptions.SensitiveInformationError
+            case DenyReason.SECURITY_ERROR:
+                return exceptions.SecurityError
+            case DenyReason.INJECTION_ATTEMPT:
+                return exceptions.InjectionError
+
+
+class ModelDeny(BaseModel):
+    deny_type: DenyReason = Field(description="Type of denial")
+    reason: str = Field(
+        description="Reason for denying the request. Should be a short sentence that can be shown to the user. Not longer than two lines"
+    )
+
+    def __init__(self, **data):
+        super().__init__(**data)
+        # Raise the corresponding exception immediately
+        exception_class = self.deny_type.to_exception()
+        raise exception_class(self.reason)

{contentgrid_extension_helpers-0.0.1.dist-info → contentgrid_extension_helpers-0.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: contentgrid-extension-helpers
-Version: 0.0.1
+Version: 0.0.2
 Summary: Helper functions for contentgrid extensions.
 Author-email: Ranec Belpaire <ranec.belpaire@xenit.eu>
 License: Copyright 2024 Xenit Solutions
@@ -26,6 +26,7 @@ Requires-Dist: requests<3,>=2.20.0
 Requires-Dist: uri-template<2
 Requires-Dist: fastapi>=0.111
 Requires-Dist: PyJWT>2
+Requires-Dist: contentgrid_hal_client>=0.0.10
 
 ### ContentGrid-Extension-Helpers
 

contentgrid_extension_helpers-0.0.2.dist-info/RECORD

@@ -0,0 +1,15 @@
+contentgrid_extension_helpers/__init__.py,sha256=Mw64JI29DW5HOzmL1UA8grdyrOqAQeHlllR_zxjagy8,380
+contentgrid_extension_helpers/exceptions.py,sha256=GJOIwDC_51pyhiPaq6wRrkmS3X1aU_PQoEE4sPTgopQ,1812
+contentgrid_extension_helpers/problem_response.py,sha256=v2z_hx92pHWyqMZfnJDi6TaGVIeoGh2AXVr1F9VA00w,1542
+contentgrid_extension_helpers/authentication/__init__.py,sha256=XnupVcladC7H4LudetZ46da9IP69pWKhtRJvikhmIKM,200
+contentgrid_extension_helpers/authentication/oidc.py,sha256=14XEmp_WWDVygb3oBKB9S29UlgJV8wnm_1lw36U4xxc,9820
+contentgrid_extension_helpers/authentication/user.py,sha256=pr3DVZKchLxpJXU6k2uUNLquwr9LjWfr_sArdAjUjZU,185
+contentgrid_extension_helpers/logging/__init__.py,sha256=15tz-g0fLdBjJto8kWcjCYhfXCJg0qrwSXVuXRH9i3Q,77
+contentgrid_extension_helpers/logging/json_logging.py,sha256=NrzoBfEUAwQT7mmCHK0GoFJv5t9rrgjomDTXzZ-vgFI,3112
+contentgrid_extension_helpers/middleware/exception_middleware.py,sha256=RXbdTzdtfufrGBfViTIQZw691jgXE3rmogKF28Vlhzs,4462
+contentgrid_extension_helpers/structured_output/model_deny.py,sha256=n2sEls0kyhvL8hHUKeo3_JQ_ZssyknIbIoDAOUvMVxc,1748
+contentgrid_extension_helpers-0.0.2.dist-info/LICENSE,sha256=tk6n-p8lEmzLJg-O4052CkMgfUtt1q2Zoh1QLAyL7S8,555
+contentgrid_extension_helpers-0.0.2.dist-info/METADATA,sha256=VIMjWfLhujv6cT5eJvnUozPrKy1G4yVwo8XsNPpmRGM,1393
+contentgrid_extension_helpers-0.0.2.dist-info/WHEEL,sha256=beeZ86-EfXScwlR_HKu4SllMC9wUEj_8Z_4FJ3egI2w,91
+contentgrid_extension_helpers-0.0.2.dist-info/top_level.txt,sha256=yJGGofrNVsl5psVGO0vLFHO1610ob88GtB9zpvS8iIk,30
+contentgrid_extension_helpers-0.0.2.dist-info/RECORD,,

{contentgrid_extension_helpers-0.0.1.dist-info → contentgrid_extension_helpers-0.0.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.8.2)
+Generator: setuptools (76.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

contentgrid_extension_helpers-0.0.1.dist-info/RECORD

@@ -1,11 +0,0 @@
-contentgrid_extension_helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-contentgrid_extension_helpers/authentication/__init__.py,sha256=b6Bp8jCg2-T7bcTGid7pCk5PfNYwH-52UFUX2Op8ArQ,146
-contentgrid_extension_helpers/authentication/oidc.py,sha256=14XEmp_WWDVygb3oBKB9S29UlgJV8wnm_1lw36U4xxc,9820
-contentgrid_extension_helpers/authentication/user.py,sha256=pr3DVZKchLxpJXU6k2uUNLquwr9LjWfr_sArdAjUjZU,185
-contentgrid_extension_helpers/logging/__init__.py,sha256=u6edUMrsULUinBNN94yJywZyINz4azhIAXO2LMbmtaY,64
-contentgrid_extension_helpers/logging/json_logging.py,sha256=QRoDHZjtMXM-JGPR6JWqJnCljwSQ912wjsZ38XMA7Tw,3119
-contentgrid_extension_helpers-0.0.1.dist-info/LICENSE,sha256=tk6n-p8lEmzLJg-O4052CkMgfUtt1q2Zoh1QLAyL7S8,555
-contentgrid_extension_helpers-0.0.1.dist-info/METADATA,sha256=eP5YG_IsyyRtfonDnTiwnLlKPuQHZ7JWkB2s4aFaO2k,1347
-contentgrid_extension_helpers-0.0.1.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
-contentgrid_extension_helpers-0.0.1.dist-info/top_level.txt,sha256=yJGGofrNVsl5psVGO0vLFHO1610ob88GtB9zpvS8iIk,30
-contentgrid_extension_helpers-0.0.1.dist-info/RECORD,,