From 54a2b66fdfdb6dd3ca2508e838529655f33193e6 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Thu, 15 Feb 2024 21:45:13 -0800
Subject: [PATCH] feat(google_text_moderation.py): initial commit for google's
 text moderation endpoint

---
 enterprise/hooks/google_text_moderation.py | 53 ++++++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100644 enterprise/hooks/google_text_moderation.py

diff --git a/enterprise/hooks/google_text_moderation.py b/enterprise/hooks/google_text_moderation.py
new file mode 100644
index 000000000..b79a32805
--- /dev/null
+++ b/enterprise/hooks/google_text_moderation.py
@@ -0,0 +1,53 @@
+# +-----------------------------------------------+
+#
+#            Google Text Moderation
+#   https://cloud.google.com/natural-language/docs/moderating-text
+#
+# +-----------------------------------------------+
+#  Thank you users! We ❤️ you! - Krrish & Ishaan
+
+
+from typing import Optional, Literal, Union
+import litellm, traceback, sys, uuid
+from litellm.caching import DualCache
+from litellm.proxy._types import UserAPIKeyAuth
+from litellm.integrations.custom_logger import CustomLogger
+from fastapi import HTTPException
+from litellm._logging import verbose_proxy_logger
+from litellm.utils import (
+    ModelResponse,
+    EmbeddingResponse,
+    ImageResponse,
+    StreamingChoices,
+)
+from datetime import datetime
+import aiohttp, asyncio
+
+
+class _ENTERPRISE_GoogleTextModeration(CustomLogger):
+    user_api_key_cache = None
+
+    # Class variables or attributes
+    def __init__(self, mock_testing: bool = False):
+        pass
+
+    def print_verbose(self, print_statement):
+        try:
+            verbose_proxy_logger.debug(print_statement)
+            if litellm.set_verbose:
+                print(print_statement)  # noqa
+        except:
+            pass
+
+    async def async_pre_call_hook(
+        self,
+        user_api_key_dict: UserAPIKeyAuth,
+        cache: DualCache,
+        data: dict,
+        call_type: str,
+    ):
+        """
+        - Calls Google's Text Moderation API
+        - Rejects request if it fails safety check
+        """
+        pass