gemini-webapi 1.12.0__tar.gz → 1.12.1__tar.gz

{gemini_webapi-1.12.0 → gemini_webapi-1.12.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-webapi
-Version: 1.12.0
+Version: 1.12.1
 Summary: ✨ An elegant async Python wrapper for Google Gemini web app
 Author: UZQueen
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE

{gemini_webapi-1.12.0 → gemini_webapi-1.12.1}/src/gemini_webapi/types/candidate.py

@@ -1,4 +1,5 @@
-from pydantic import BaseModel
+import html
+from pydantic import BaseModel, field_validator
 
 from .image import Image, WebImage, GeneratedImage
 
@@ -33,6 +34,17 @@ class Candidate(BaseModel):
     def __repr__(self):
         return f"Candidate(rcid='{self.rcid}', text='{len(self.text) <= 20 and self.text or self.text[:20] + '...'}', images={self.images})"
 
+    @field_validator("text", "thoughts")
+    @classmethod
+    def decode_html(cls, value: str) -> str:
+        """
+        Auto unescape HTML entities in text/thoughts if any.
+        """
+
+        if value:
+            value = html.unescape(value)
+        return value
+
     @property
     def images(self) -> list[Image]:
         return self.web_images + self.generated_images

{gemini_webapi-1.12.0 → gemini_webapi-1.12.1}/src/gemini_webapi.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-webapi
-Version: 1.12.0
+Version: 1.12.1
 Summary: ✨ An elegant async Python wrapper for Google Gemini web app
 Author: UZQueen
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE

{gemini_webapi-1.12.0 → gemini_webapi-1.12.1}/src/gemini_webapi.egg-info/SOURCES.txt

@@ -31,5 +31,6 @@ src/gemini_webapi/utils/logger.py
 src/gemini_webapi/utils/rotate_1psidts.py
 src/gemini_webapi/utils/upload_file.py
 tests/test_client_features.py
+tests/test_html_entity_decode.py
 tests/test_rotate_cookies.py
 tests/test_save_image.py

gemini_webapi-1.12.1/tests/test_html_entity_decode.py

@@ -0,0 +1,48 @@
+import unittest
+from gemini_webapi.types.candidate import Candidate
+
+
+class TestHtmlEntityDecode(unittest.TestCase):
+    def test_html_entity_decoding(self):
+        # Test HTML entity decoding functionality
+        html_encoded_text = (
+            "This is a code snippet: <code>print('Hello, World!')</code>"
+        )
+        expected_decoded_text = (
+            "This is a code snippet: <code>print('Hello, World!')</code>"
+        )
+
+        # Create Candidate instance which should automatically decode HTML entities
+        candidate = Candidate(
+            rcid="test_rcid",
+            text=html_encoded_text,
+            thoughts="Testing &lt;b&gt;HTML&lt;/b&gt; entity decoding",
+        )
+
+        # Verify that text property is correctly decoded
+        self.assertEqual(candidate.text, expected_decoded_text)
+
+        # Verify that thoughts property is correctly decoded
+        self.assertEqual(candidate.thoughts, "Testing <b>HTML</b> entity decoding")
+
+    def test_non_html_text(self):
+        # Test plain text without any HTML entities
+        plain_text = "This is regular text with no HTML entities"
+
+        candidate = Candidate(rcid="test_rcid", text=plain_text)
+
+        # Verify the text remains unchanged
+        self.assertEqual(candidate.text, plain_text)
+
+    def test_complex_html_entities(self):
+        # Test more complex combinations of HTML entities
+        complex_html = "&lt;div&gt;This has &amp;amp; character\n and &quot;quotes&quot;&lt;/div&gt;"
+        expected_decoded = '<div>This has &amp; character\n and "quotes"</div>'
+
+        candidate = Candidate(rcid="test_rcid", text=complex_html)
+
+        self.assertEqual(candidate.text, expected_decoded)
+
+
+if __name__ == "__main__":
+    unittest.main()