ragbits-guardrails 0.5.0__py3-none-any.whl → 1.4.0.dev202601310254__py3-none-any.whl

ragbits/guardrails/openai_moderation.py

@@ -1,5 +1,3 @@
-import base64
-
 from openai import AsyncOpenAI
 
 from ragbits.core.prompt import Prompt
@@ -27,18 +25,13 @@ class OpenAIModerationGuardrail(Guardrail):
         """
         if isinstance(input_to_verify, Prompt):
             inputs = [{"type": "text", "text": input_to_verify.rendered_user_prompt}]
+
             if input_to_verify.rendered_system_prompt is not None:
                 inputs.append({"type": "text", "text": input_to_verify.rendered_system_prompt})
-            if images := input_to_verify.images:
-                inputs.extend(
-                    [
-                        {
-                            "type": "image_url",
-                            "image_url": {"url": f"data:image/jpeg;base64,{base64.b64encode(im).decode('utf-8')}"},  # type: ignore
-                        }
-                        for im in images
-                    ]
-                )
+
+            if attachments := input_to_verify.attachments:
+                messages = [input_to_verify.create_message_with_attachment(attachment) for attachment in attachments]
+                inputs.extend(messages)
         else:
             inputs = [{"type": "text", "text": input_to_verify}]
         response = await self._openai_client.moderations.create(model=self._moderation_model, input=inputs)  # type: ignore

ragbits_guardrails-1.4.0.dev202601310254.dist-info/METADATA

@@ -0,0 +1,65 @@
+Metadata-Version: 2.4
+Name: ragbits-guardrails
+Version: 1.4.0.dev202601310254
+Summary: Guardrails module for Ragbits components
+Project-URL: Homepage, https://github.com/deepsense-ai/ragbits
+Project-URL: Bug Reports, https://github.com/deepsense-ai/ragbits/issues
+Project-URL: Documentation, https://ragbits.deepsense.ai/
+Project-URL: Source, https://github.com/deepsense-ai/ragbits
+Author-email: "deepsense.ai" <ragbits@deepsense.ai>
+License-Expression: MIT
+Keywords: Evaluation,GenAI,Generative AI,LLMs,Large Language Models,RAG,Retrieval Augmented Generation
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Natural Language :: English
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.10
+Requires-Dist: ragbits-core==1.4.0.dev202601310254
+Provides-Extra: openai
+Requires-Dist: openai<2.0.0,>=1.91.0; extra == 'openai'
+Description-Content-Type: text/markdown
+
+# Ragbits Guardrails
+
+Ragbits Guardrails is a Python package that contains utilities for ensuring the safety and relevance of responses generated by Ragbits components.
+
+## Installation
+
+You can install the latest version of Ragbits Guardrails using pip:
+
+```bash
+pip install ragbits-guardrails
+```
+
+## Quickstart
+Example of using the OpenAI Moderation Guardrail to verify a message:
+
+```python
+import asyncio
+from ragbits.guardrails.base import GuardrailManager, GuardrailVerificationResult
+from ragbits.guardrails.openai_moderation import OpenAIModerationGuardrail
+
+
+async def verify_message(message: str) -> list[GuardrailVerificationResult]:
+    manager = GuardrailManager([OpenAIModerationGuardrail()])
+    return await manager.verify(message)
+
+
+if __name__ == '__main__':
+    print(asyncio.run(verify_message("Test message")))
+```
+
+## Documentation
+* [How-To Guides - Guardrails](https://ragbits.deepsense.ai/how-to/use_guardrails/)
+<!--
+TODO:
+* Add link to API Reference once classes from the Guardrails package are added to the API Reference.
+-->

ragbits_guardrails-1.4.0.dev202601310254.dist-info/RECORD

@@ -0,0 +1,7 @@
+ragbits/guardrails/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ragbits/guardrails/base.py,sha256=L8S4DcZnLNgs0hwEy4JQpmy7_1JFJbg-z4SKTBNvniY,1309
+ragbits/guardrails/openai_moderation.py,sha256=B4-M7BjGiZl0gcVksYDquq3BNtuJajSJ3SPMiqmmeWs,1761
+ragbits/guardrails/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ragbits_guardrails-1.4.0.dev202601310254.dist-info/METADATA,sha256=ervjg0BSm1jYwL4ACQLAo6QSComWABnV_WMwIbA3r7Q,2426
+ragbits_guardrails-1.4.0.dev202601310254.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+ragbits_guardrails-1.4.0.dev202601310254.dist-info/RECORD,,

{ragbits_guardrails-0.5.0.dist-info → ragbits_guardrails-1.4.0.dev202601310254.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.26.3
+Generator: hatchling 1.28.0
 Root-Is-Purelib: true
 Tag: py3-none-any

ragbits_guardrails-0.5.0.dist-info/METADATA

@@ -1,30 +0,0 @@
-Metadata-Version: 2.3
-Name: ragbits-guardrails
-Version: 0.5.0
-Summary: Guardrails module for Ragbits components
-Project-URL: Homepage, https://github.com/deepsense-ai/ragbits
-Project-URL: Bug Reports, https://github.com/deepsense-ai/ragbits/issues
-Project-URL: Documentation, https://ragbits.deepsense.ai/
-Project-URL: Source, https://github.com/deepsense-ai/ragbits
-Author-email: "deepsense.ai" <ragbits@deepsense.ai>
-License: MIT
-Keywords: Evaluation,GenAI,Generative AI,LLMs,Large Language Models,RAG,Retrieval Augmented Generation
-Classifier: Development Status :: 4 - Beta
-Classifier: Environment :: Console
-Classifier: Intended Audience :: Science/Research
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Natural Language :: English
-Classifier: Operating System :: OS Independent
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
-Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
-Classifier: Topic :: Software Development :: Libraries :: Python Modules
-Requires-Python: >=3.10
-Requires-Dist: ragbits-core==0.5.0
-Provides-Extra: openai
-Requires-Dist: openai~=1.51.0; extra == 'openai'
-Description-Content-Type: text/markdown
-
-# Ragbits Guardrails

ragbits_guardrails-0.5.0.dist-info/RECORD

@@ -1,6 +0,0 @@
-ragbits/guardrails/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ragbits/guardrails/base.py,sha256=L8S4DcZnLNgs0hwEy4JQpmy7_1JFJbg-z4SKTBNvniY,1309
-ragbits/guardrails/openai_moderation.py,sha256=sA_kXDBf-a8M1E6TG0fof3fQDOfOxLa4YbfZBJQCb6o,1973
-ragbits_guardrails-0.5.0.dist-info/METADATA,sha256=zx3FC4wyZrqo40m0qd4IzFf3PIrQUSS2NuQCte50TV0,1342
-ragbits_guardrails-0.5.0.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-ragbits_guardrails-0.5.0.dist-info/RECORD,,